优化代码

java110

2021-05-08 ea6e1675bbd411e30b21369698ed4b0587c21c0e

 service-job/src/main/java/com/java110/job/smo/impl/TaskInnerServiceSMOImpl.java

old mode 100644
new mode 100755
@@ -1,20 +1,38 @@
package com.java110.job.smo.impl;


import com.java110.job.dao.ITaskServiceDao;
import com.java110.core.smo.task.ITaskInnerServiceSMO;
import com.java110.dto.task.TaskDto;
import com.java110.utils.util.BeanConvertUtil;
import com.java110.core.base.smo.BaseServiceSMO;
import com.java110.dto.user.UserDto;
import com.java110.core.smo.user.IUserInnerServiceSMO;
import com.java110.intf.job.ITaskInnerServiceSMO;
import com.java110.intf.user.IUserInnerServiceSMO;
import com.java110.dto.PageDto;
import com.java110.dto.task.TaskDto;
import com.java110.dto.task.TaskTemplateDto;
import com.java110.dto.task.TaskTemplateSpecDto;
import com.java110.dto.taskAttr.TaskAttrDto;
import com.java110.dto.user.UserDto;
import com.java110.job.dao.ITaskAttrServiceDao;
import com.java110.job.dao.ITaskServiceDao;
import com.java110.job.quartz.TaskSystemJob;
import com.java110.utils.util.Assert;
import com.java110.utils.util.BeanConvertUtil;
import org.quartz.CronScheduleBuilder;
import org.quartz.CronTrigger;
import org.quartz.JobBuilder;
import org.quartz.JobDetail;
import org.quartz.JobKey;
import org.quartz.Scheduler;
import org.quartz.TriggerBuilder;
import org.quartz.TriggerKey;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @ClassName FloorInnerServiceSMOImpl
@@ -27,14 +45,28 @@
@RestController
public class TaskInnerServiceSMOImpl extends BaseServiceSMO implements ITaskInnerServiceSMO {

    private static final Logger logger = LoggerFactory.getLogger(TaskInnerServiceSMOImpl.class);

    private static final String defaultCronExpression = "0 * * * * ?";// 每分钟执行一次

    private static final String prefixJobName = "task_"; // job
    private static final String triggerNames = "taskToData_"; // job

    @Autowired
    private Scheduler scheduler;


    @Autowired
    private ITaskServiceDao taskServiceDaoImpl;

    @Autowired
    private ITaskAttrServiceDao taskAttrServiceDaoImpl;

    @Autowired
    private IUserInnerServiceSMO userInnerServiceSMOImpl;

    @Override
    public List<TaskDto> queryTasks(@RequestBody  TaskDto taskDto) {
    public List<TaskDto> queryTasks(@RequestBody TaskDto taskDto) {

        //校验是否传了 分页信息

@@ -63,7 +95,7 @@
    /**
     * 从用户列表中查询用户，将用户中的信息 刷新到 floor对象中
     *
     * @param task 小区定时任务信息
     * @param task  小区定时任务信息
     * @param users 用户列表
     */
    private void refreshTask(TaskDto task, List<UserDto> users) {
@@ -80,7 +112,7 @@
     * @param tasks 小区楼信息
     * @return 批量userIds 信息
     */
     private String[] getUserIds(List<TaskDto> tasks) {
    private String[] getUserIds(List<TaskDto> tasks) {
        List<String> userIds = new ArrayList<String>();
        for (TaskDto task : tasks) {
            userIds.add(task.getTaskId());
@@ -91,7 +123,155 @@

    @Override
    public int queryTasksCount(@RequestBody TaskDto taskDto) {
        return taskServiceDaoImpl.queryTasksCount(BeanConvertUtil.beanCovertMap(taskDto));    }
        return taskServiceDaoImpl.queryTasksCount(BeanConvertUtil.beanCovertMap(taskDto));
    }


    @Override
    public int queryTaskTemplateCount(@RequestBody TaskTemplateDto taskTemplateDto) {
        return taskServiceDaoImpl.queryTaskTemplateCount(BeanConvertUtil.beanCovertMap(taskTemplateDto));
    }


    @Override
    public List<TaskTemplateDto> queryTaskTemplate(@RequestBody TaskTemplateDto taskTemplateDto) {

        //校验是否传了 分页信息

        int page = taskTemplateDto.getPage();

        if (page != PageDto.DEFAULT_PAGE) {
            taskTemplateDto.setPage((page - 1) * taskTemplateDto.getRow());
        }

        List<TaskTemplateDto> taskTemplates = BeanConvertUtil.covertBeanList(taskServiceDaoImpl.getTaskTemplateInfo(BeanConvertUtil.beanCovertMap(taskTemplateDto)), TaskTemplateDto.class);

        return taskTemplates;
    }


    @Override
    public int queryTaskTemplateSpecCount(@RequestBody TaskTemplateSpecDto taskTemplateSpecDto) {
        return taskServiceDaoImpl.queryTaskTemplateSpecCount(BeanConvertUtil.beanCovertMap(taskTemplateSpecDto));
    }


    @Override
    public List<TaskTemplateSpecDto> queryTaskTemplateSpec(@RequestBody TaskTemplateSpecDto taskTemplateSpecDto) {

        //校验是否传了 分页信息

        int page = taskTemplateSpecDto.getPage();

        if (page != PageDto.DEFAULT_PAGE) {
            taskTemplateSpecDto.setPage((page - 1) * taskTemplateSpecDto.getRow());
        }

        List<TaskTemplateSpecDto> taskTemplates = BeanConvertUtil.covertBeanList(taskServiceDaoImpl.getTaskTemplateSpecInfo(BeanConvertUtil.beanCovertMap(taskTemplateSpecDto)), TaskTemplateSpecDto.class);

        return taskTemplates;
    }

    /**
     * 启动任务
     *
     * @param taskDto
     * @return
     */
    public int startTask(@RequestBody TaskDto taskDto) {
        List<TaskAttrDto> attrDtos = BeanConvertUtil.covertBeanList(taskAttrServiceDaoImpl.getTaskAttrInfo(BeanConvertUtil.beanCovertMap(taskDto)),
                TaskAttrDto.class);
        Map info = new HashMap();
        info.put("templateId", taskDto.getTemplateId());
        List<TaskTemplateDto> taskTemplateDtos = BeanConvertUtil.covertBeanList(taskServiceDaoImpl.getTaskTemplateInfo(info), TaskTemplateDto.class);

        Assert.listOnlyOne(taskTemplateDtos, "模板不存在或存在多个");

        taskDto.setTaskTemplateDto(taskTemplateDtos.get(0));
        taskDto.setTaskAttr(attrDtos);

        try {
            String cronExpression = taskDto.getTaskCron();// 如果没有配置则，每一分运行一次

            CronScheduleBuilder cronScheduleBuilder = CronScheduleBuilder.cronSchedule(cronExpression);

            String jobName = prefixJobName + taskDto.getTaskId();

            String triggerName = triggerNames + taskDto.getTaskId();

            //设置任务名称
            JobKey jobKey = new JobKey(jobName, TaskSystemJob.JOB_GROUP_NAME);
            JobDetail jobDetail = scheduler.getJobDetail(jobKey);

            if (jobDetail != null) {
                return 0;
            }

            String taskCfgName = taskDto.getTaskName();
            JobDetail warnJob = JobBuilder.newJob(TaskSystemJob.class).withIdentity(jobName, TaskSystemJob.JOB_GROUP_NAME).withDescription("任务启动").build();

            warnJob.getJobDataMap().put(TaskSystemJob.JOB_DATA_CONFIG_NAME, taskCfgName);

            warnJob.getJobDataMap().put(TaskSystemJob.JOB_DATA_TASK_ID, taskDto.getTaskId());
            warnJob.getJobDataMap().put(TaskSystemJob.JOB_DATA_TASK, taskDto);
            warnJob.getJobDataMap().put(TaskSystemJob.JOB_DATA_TASK_ATTR, taskDto);

            // 触发时间点
            CronTrigger warnTrigger = TriggerBuilder.newTrigger().withIdentity(triggerName, triggerName + "_group").withSchedule(cronScheduleBuilder).build();

            // 错过执行后，立即执行
            //warnTrigger(CronTrigger.MISFIRE_INSTRUCTION_FIRE_ONCE_NOW);
            //交由Scheduler安排触发
            scheduler.scheduleJob(warnJob, warnTrigger);
            Map paramIn = new HashMap();
            paramIn.put("taskId", taskDto.getTaskId());
            paramIn.put("state", "002");
            paramIn.put("statusCd", "0");
            taskServiceDaoImpl.updateTaskInfoInstance(paramIn);

        } catch (Exception e) {
            logger.error("启动侦听失败", e);
            return 0;
        }
        return 1;
    }

    /**
     * 停止任务
     *
     * @param taskDto
     * @return
     */
    public int stopTask(@RequestBody TaskDto taskDto) {

        try {
            String jobName = prefixJobName + taskDto.getTaskId();

            String triggerName = prefixJobName + taskDto.getTaskId();

            TriggerKey triggerKey = TriggerKey.triggerKey(jobName, TaskSystemJob.JOB_GROUP_NAME);
            // 停止触发器
            scheduler.pauseTrigger(triggerKey);
            // 移除触发器
            scheduler.unscheduleJob(triggerKey);

            JobKey jobKey = new JobKey(jobName, TaskSystemJob.JOB_GROUP_NAME);
            // 删除任务
            scheduler.deleteJob(jobKey);

            Map paramIn = new HashMap();
            paramIn.put("taskId", taskDto.getTaskId());
            paramIn.put("state", "001");
            paramIn.put("statusCd", "0");
            taskServiceDaoImpl.updateTaskInfoInstance(paramIn);

        } catch (Exception e) {
            logger.error("启动侦听失败", e);
            return 0;
        }
        return 1;
    }


    public ITaskServiceDao getTaskServiceDaoImpl() {
        return taskServiceDaoImpl;