TaskInnerServiceSMOImpl.java 9.64 KB

Edit Raw Blame History Permalink

package com.java110.job.smo.impl;


import com.java110.core.base.smo.BaseServiceSMO;
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.task.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 com.java110.core.log.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
 * @Description 定时任务内部服务实现类
 * @Author wuxw
 * @Date 2019/4/24 9:20
 * @Version 1.0
 * add by wuxw 2019/4/24
 **/
@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) {

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

        int page = taskDto.getPage();

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

        List<TaskDto> tasks = BeanConvertUtil.covertBeanList(taskServiceDaoImpl.getTaskInfo(BeanConvertUtil.beanCovertMap(taskDto)), TaskDto.class);

        if (tasks == null || tasks.size() == 0) {
            return tasks;
        }

        String[] userIds = getUserIds(tasks);
        //根据 userId 查询用户信息
        List<UserDto> users = userInnerServiceSMOImpl.getUserInfo(userIds);

        for (TaskDto task : tasks) {
            refreshTask(task, users);
        }
        return tasks;
    }

    /**
     * 从用户列表中查询用户，将用户中的信息 刷新到 floor对象中
     *
     * @param task  项目定时任务信息
     * @param users 用户列表
     */
    private void refreshTask(TaskDto task, List<UserDto> users) {
        for (UserDto user : users) {
            if (task.getTaskId().equals(user.getUserId())) {
                BeanConvertUtil.covertBean(user, task);
            }
        }
    }

    /**
     * 获取批量userId
     *
     * @param tasks 项目楼信息
     * @return 批量userIds 信息
     */
    private String[] getUserIds(List<TaskDto> tasks) {
        List<String> userIds = new ArrayList<String>();
        for (TaskDto task : tasks) {
            userIds.add(task.getTaskId());
        }

        return userIds.toArray(new String[userIds.size()]);
    }

    @Override
    public int queryTasksCount(@RequestBody TaskDto 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;
    }

    public void setTaskServiceDaoImpl(ITaskServiceDao taskServiceDaoImpl) {
        this.taskServiceDaoImpl = taskServiceDaoImpl;
    }

    public IUserInnerServiceSMO getUserInnerServiceSMOImpl() {
        return userInnerServiceSMOImpl;
    }

    public void setUserInnerServiceSMOImpl(IUserInnerServiceSMO userInnerServiceSMOImpl) {
        this.userInnerServiceSMOImpl = userInnerServiceSMOImpl;
    }
}