用springboot写一个排班demo

好的，以下是一个使用 Spring Boot 框架编写的简单排班示例。这个示例使用了 Spring Data JPA 来访问数据库，数据库使用了 H2 内存数据库。你可以将代码复制到一个 Java 项目中，并添加必要的依赖来运行它。

这个示例的核心是一个名为 `Shift` 的实体类，它表示一个班次。`Shift` 类有三个属性：`id`、`name` 和 `startTime`。`id` 是主键，`name` 表示班次的名称，`startTime` 表示班次开始时间。

@Entity
public class Shift {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    private String name;

    private LocalTime startTime;

    // 构造函数、getter 和 setter 略
}

接下来是一个名为 `ShiftRepository` 的接口，它继承了 Spring Data JPA 的 `JpaRepository` 接口。`ShiftRepository` 接口用于访问数据库中的班次数据。

@Repository
public interface ShiftRepository extends JpaRepository<Shift, Long> {
}

然后是一个名为 `Schedule` 的实体类，它表示一个排班计划。`Schedule` 类有三个属性：`id`、`date` 和 `shift`。`id` 是主键，`date` 表示排班日期，`shift` 表示当天的班次。

@Entity
public class Schedule {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    private LocalDate date;

    @ManyToOne
    private Shift shift;

    // 构造函数、getter 和 setter 略
}

接下来是一个名为 `ScheduleRepository` 的接口，它继承了 Spring Data JPA 的 `JpaRepository` 接口。`ScheduleRepository` 接口用于访问数据库中的排班计划数据。

@Repository
public interface ScheduleRepository extends JpaRepository<Schedule, Long> {
    Schedule findByDate(LocalDate date);
}

然后是一个名为 `ScheduleService` 的服务类，它提供了一些方法来操作排班计划数据。`ScheduleService` 类使用了 `ShiftRepository` 和 `ScheduleRepository` 接口来访问数据库。

@Service
public class ScheduleService {
    private final ShiftRepository shiftRepository;
    private final ScheduleRepository scheduleRepository;

    public ScheduleService(ShiftRepository shiftRepository, ScheduleRepository scheduleRepository) {
        this.shiftRepository = shiftRepository;
        this.scheduleRepository = scheduleRepository;
    }

    public List<Shift> getAllShifts() {
        return shiftRepository.findAll();
    }

    public Shift getShiftById(Long id) {
        return shiftRepository.findById(id).orElse(null);
    }

    public void saveSchedule(LocalDate date, Long shiftId) {
        Shift shift = shiftRepository.findById(shiftId).orElse(null);
        if (shift == null) {
            throw new IllegalArgumentException("Invalid shift ID: " + shiftId);
        }

        Schedule schedule = new Schedule();
        schedule.setDate(date);
        schedule.setShift(shift);
        scheduleRepository.save(schedule);
    }

    public void deleteSchedule(LocalDate date) {
        Schedule schedule = scheduleRepository.findByDate(date);
        if (schedule != null) {
            scheduleRepository.delete(schedule);
        }
    }

    public List<Schedule> getAllSchedules() {
        return scheduleRepository.findAll();
    }
}

最后是一个名为 `ScheduleController` 的控制器类，它处理 HTTP 请求并调用 `ScheduleService` 类中的方法来操作排班计划数据。

@RestController
public class ScheduleController {
    private final ScheduleService scheduleService;

    public ScheduleController(ScheduleService scheduleService) {
        this.scheduleService = scheduleService;
    }

    @GetMapping("/shifts")
    public List<Shift> getAllShifts() {
        return scheduleService.getAllShifts();
    }

    @GetMapping("/shifts/{id}")
    public Shift getShiftById(@PathVariable Long id) {
        return scheduleService.getShiftById(id);
    }

    @PostMapping("/schedules")
    public void saveSchedule(@RequestBody Map<String, Object> payload) {
        LocalDate date = LocalDate.parse((String) payload.get("date"));
        Long shiftId = Long.parseLong((String) payload.get("shiftId"));
        scheduleService.saveSchedule(date, shiftId);
    }

    @DeleteMapping("/schedules/{date}")
    public void deleteSchedule(@PathVariable String date) {
        scheduleService.deleteSchedule(LocalDate.parse(date));
    }

    @GetMapping("/schedules")
    public List<Schedule> getAllSchedules() {
        return scheduleService.getAllSchedules();
    }
}

这个示例使用了 H2 内存数据库，因此你不需要安装任何数据库软件。你可以在 `src/main/resources/application.properties` 文件中设置数据库相关的配置：

spring.datasource.url=jdbc:h2:mem:testdb
spring.jpa.hibernate.ddl-auto=create
spring.h2.console.enabled=true
spring.h2.console.path=/h2-console

最后，你可以使用以下命令来编译和运行这个示例：

./mvnw spring-boot:run

这个示例提供了以下几个 HTTP 接口：

- `GET /shifts`：获取所有班次；
- `GET /shifts/{id}`：根据 ID 获取一个班次；
- `POST /schedules`：创建一个排班计划；
- `DELETE /schedules/{date}`：删除一个排班计划；
- `GET /schedules`：获取所有排班计划。

你可以使用 Postman 或类似的工具来测试这些接口。这个示例可以作为一个简单的排班系统的基础，你可以根据自己的需求进行扩展。