# DbSchedulerController.java

    package cloud.wconnect.workperformance.controller.database;

    import cloud.wconnect.commonlib.annotation.ApiId;
    import cloud.wconnect.commonlib.component.BeanMapper;
    import cloud.wconnect.commonlib.component.SessionContext;
    import cloud.wconnect.commonlib.cons.*;
    import cloud.wconnect.commonlib.exception.BadRequestException;
    import cloud.wconnect.commonlib.pojo.dto.PropertyDTO;
    import cloud.wconnect.commonlib.pojo.dto.result.ResultDTO;
    import cloud.wconnect.commonlib.pojo.dto.scheduler.JobParameterDTO;
    import cloud.wconnect.commonlib.redis.PropertyI18nRedisService;
    import cloud.wconnect.commonlib.util.CronRuleUtil;
    import cloud.wconnect.commonlib.util.I18nMessageUtils;
    import cloud.wconnect.commonlib.util.JsonUtils;
    import cloud.wconnect.workperformance.cons.database.DbCons;
    import cloud.wconnect.workperformance.cons.database.DbPmsReportDataRangeEnum;
    import cloud.wconnect.workperformance.entity.database.DbScheduler;
    import cloud.wconnect.workperformance.pojo.dto.database.notification.DbSchedulerDTO;
    import cloud.wconnect.workperformance.service.database.IDbSchedulerService;
    import com.fasterxml.jackson.databind.ObjectMapper;
    import com.google.common.collect.ImmutableMap;
    import com.google.common.collect.Lists;
    import jakarta.annotation.Resource;
    import jakarta.validation.Valid;
    import org.apache.commons.collections.CollectionUtils;
    import org.apache.commons.lang3.StringUtils;
    import org.springframework.web.bind.annotation.*;

    import java.text.MessageFormat;
    import java.util.Collections;
    import java.util.List;
    import java.util.Map;
    import java.util.Objects;

    /**
     * <p>
     * Database scheduler 前端控制器
     * </p>
     *
     * @author zc
     * @since 2025-12-02
     */
    @RestController
    @RequestMapping("/database/scheduler")
    public class DbSchedulerController {
        @Resource
        private IDbSchedulerService schedulerService;
        @Resource
        private PropertyI18nRedisService propertyI18nRedisService;
        @Resource
        private SessionContext sessionContext;
        @Resource
        private BeanMapper beanMapper;
        @Resource
        private ObjectMapper objectMapper;
        @Resource
        private I18nMessageUtils i18nMessageUtils;

        /**
         * 查询调度任务
         *
         * @param property 资产标识
         * @param category 调度任务类型
         * @return 调度任务
         * @author zc
         * @date 2025/12/02
         */
        @ApiId(1654)
        @GetMapping("/{category:\\d+}")
        public ResultDTO<List<DbSchedulerDTO>> list(String property, @PathVariable int category) {
            int propertyId = sessionContext.getPropertyId(property);
            PropertyDTO propertyDTO = propertyI18nRedisService.get(propertyId);
            // 校验支持的定时任务类型
            checkSchedulerCategory(propertyDTO.getPmsType(), category);
            // 查询调度任务
            List<DbScheduler> schedulerList = schedulerService.list(propertyId, category);
            if (CollectionUtils.isEmpty(schedulerList)) {
                return ResultDTO.success(Collections.emptyList());
            }
            return ResultDTO.success(beanMapper.mapList(schedulerList, DbSchedulerDTO.class));
        }

        /**
         * 保存调度任务
         *
         * @param property     资产标识
         * @param id           调度任务id
         * @param schedulerDTO 调度任务DTO
         * @return 调度任务 Job Id
         * @author zc
         * @date 2025/12/02
         */
        @ApiId(1655)
        @PostMapping("/{id}")
        public ResultDTO<Map<String, String>> save(@RequestParam String property, @PathVariable String id,
                                                   @RequestBody @Valid DbSchedulerDTO schedulerDTO) {
            int propertyId = sessionContext.getPropertyId(property);
            PropertyDTO propertyDTO = propertyI18nRedisService.get(propertyId);
            // 校验支持的定时任务类型
            checkSchedulerCategory(propertyDTO.getPmsType(), schedulerDTO.getCategory());
            // 检查 parameters json 格式是否合法
            checkJobParameters(schedulerDTO.getParameters());
            // 检查 scheduler json 格式是否合法
            CronRuleUtil.validateCronRuleForScheduler(objectMapper, Lists.newArrayList(SchedulerCons.EVERY_DAY),
                    schedulerDTO.getScheduler());
            // 设置 Job Parameter
            setJobParameter(schedulerDTO);

            schedulerDTO.setPropertyId(propertyDTO.getId());
            if (id.equals("0")) {
                // 校验分类下最大可配置调度任务的数量
                schedulerService.checkMaxNum(propertyDTO.getId(), schedulerDTO.getCategory(), 1);
                // 新增调度任务
                id = schedulerService.save(schedulerDTO, propertyDTO, sessionContext.getLoginUserId());
            } else {
                // 校验调度任务是否存在
                DbScheduler scheduler = schedulerService.checkExist(id);
                // 校验归属
                schedulerService.checkBelong(propertyDTO.getId(), scheduler);
                if (!scheduler.getCategory().equals(schedulerDTO.getCategory())) {
                    throw new BadRequestException(CommonErrorI18nConstant.SCHEDULER_CANNOT_CHANGE_CATEGORY,
                            BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                            "The Scheduler category cannot update, SchedulerJobId=" + scheduler.getId());
                }
                // 修改调度任务
                schedulerDTO.setId(id);
                id = schedulerService.update(propertyDTO, scheduler, schedulerDTO, sessionContext.getLoginUserId());
            }
            return ResultDTO.success(ImmutableMap.of("id", id));
        }

        /**
         * 设置 Job Parameter
         *
         * @param schedulerDTO 调度任务DTO
         * @author zc
         * @date 2025/12/02
         */
        private void setJobParameter(DbSchedulerDTO schedulerDTO) {
            JobParameterDTO jobParameterDTO = JsonUtils.convertStringToObject(objectMapper,
                    schedulerDTO.getParameters(), JobParameterDTO.class);
            jobParameterDTO.setDateRangeUnit(JobParameterDTO.DAY);
            jobParameterDTO.setDateRange(DbPmsReportDataRangeEnum.getDateRange(schedulerDTO.getCategory()));
            schedulerDTO.setParameters(JsonUtils.convertObjToString(objectMapper, jobParameterDTO));
        }

        /**
         * 校验定时任务 parameters 参数
         *
         * @param parameters 调度任务自定义参数
         * @author zc
         * @date 2025/12/02
         */
        private void checkJobParameters(String parameters) {
            if (StringUtils.isBlank(parameters)) {
                throw new BadRequestException(CommonErrorI18nConstant.CANNOT_EMPTY,
                        i18nMessageUtils.getErrorParamMapNeedTranslate(CommonFieldI18nConstant.SCHEDULER_EXEC_PARAMETERS),
                        BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                        "The parameters cannot be blank");
            }
            JobParameterDTO jobParameterDTO = JsonUtils.convertStringToObject(objectMapper, parameters, JobParameterDTO.class);
            // 校验阈值
            if (Objects.isNull(jobParameterDTO.getThreshold())) {
                throw new BadRequestException(CommonErrorI18nConstant.CANNOT_EMPTY,
                        i18nMessageUtils.getErrorParamMapNeedTranslate(CommonFieldI18nConstant.SCHEDULER_THRESHOLD),
                        BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                        "The threshold cannot be null");
            }
        }

        /**
         * 校验支持的定时任务类型
         *
         * @param propertyPmsType 属性PMS类型
         * @param schedulerType   定时任务类型
         * @author zc
         * @date 2025/12/02
         */
        private void checkSchedulerCategory(String propertyPmsType, int schedulerType) {
            List<Integer> effectiveSchedulerTypeList;
            if (SystemPlatformEnum.OPERA.getDesc().equals(propertyPmsType)) {
                effectiveSchedulerTypeList = DbCons.OPERA_NOTIFY_TYPE_LIST;
            } else if (SystemPlatformEnum.SHI_JI.getDesc().equals(propertyPmsType)) {
                effectiveSchedulerTypeList = DbCons.SHIJI_NOTIFY_TYPE_LIST;
            } else {
                effectiveSchedulerTypeList = DbCons.PEP_NOTIFY_TYPE_LIST;
            }
            if (!effectiveSchedulerTypeList.contains(schedulerType)) {
                throw new BadRequestException(CommonErrorI18nConstant.NOTIFY_TYPE_NOT_SUPPORTED,
                        BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                        MessageFormat.format("The currently selected scheduler type is not supported, " +
                                "schedulerCategory must be in {0}", effectiveSchedulerTypeList));
            }
        }
    }