若依 3.0

RuoYi

2020-07-19 cee572f237cb293e4a9070ef458356bc6daa2105

 ruoyi-quartz/src/main/java/com/ruoyi/quartz/util/AbstractQuartzJob.java

ÎļþÃû´Ó ruoyi/src/main/java/com/ruoyi/common/utils/job/AbstractQuartzJob.java ÐÞ¸Ä
@@ -1,107 +1,107 @@
package com.ruoyi.common.utils.job;

import java.util.Date;
import org.quartz.Job;
import org.quartz.JobExecutionContext;
import org.quartz.JobExecutionException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.ruoyi.common.constant.Constants;
import com.ruoyi.common.constant.ScheduleConstants;
import com.ruoyi.common.utils.ExceptionUtil;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.common.utils.bean.BeanUtils;
import com.ruoyi.common.utils.spring.SpringUtils;
import com.ruoyi.project.monitor.domain.SysJob;
import com.ruoyi.project.monitor.domain.SysJobLog;
import com.ruoyi.project.monitor.service.ISysJobLogService;

/**
 * æŠ½è±¡quartz调用
 *
 * @author ruoyi
 */
public abstract class AbstractQuartzJob implements Job
{
    private static final Logger log = LoggerFactory.getLogger(AbstractQuartzJob.class);

    /**
     * çº¿ç¨‹æœ¬åœ°å˜é‡
     */
    private static ThreadLocal<Date> threadLocal = new ThreadLocal<>();

    @Override
    public void execute(JobExecutionContext context) throws JobExecutionException
    {
        SysJob sysJob = new SysJob();
        BeanUtils.copyBeanProp(sysJob, context.getMergedJobDataMap().get(ScheduleConstants.TASK_PROPERTIES));
        try
        {
            before(context, sysJob);
            if (sysJob != null)
            {
                doExecute(context, sysJob);
            }
            after(context, sysJob, null);
        }
        catch (Exception e)
        {
            log.error("任务执行异常  - ï¼š", e);
            after(context, sysJob, e);
        }
    }

    /**
     * æ‰§è¡Œå‰
     *
     * @param context å·¥ä½œæ‰§è¡Œä¸Šä¸‹æ–‡å¯¹è±¡
     * @param sysJob ç³»ç»Ÿè®¡åˆ’任务
     */
    protected void before(JobExecutionContext context, SysJob sysJob)
    {
        threadLocal.set(new Date());
    }

    /**
     * æ‰§è¡ŒåŽ
     *
     * @param context å·¥ä½œæ‰§è¡Œä¸Šä¸‹æ–‡å¯¹è±¡
     * @param sysScheduleJob ç³»ç»Ÿè®¡åˆ’任务
     */
    protected void after(JobExecutionContext context, SysJob sysJob, Exception e)
    {
        Date startTime = threadLocal.get();
        threadLocal.remove();

        final SysJobLog sysJobLog = new SysJobLog();
        sysJobLog.setJobName(sysJob.getJobName());
        sysJobLog.setJobGroup(sysJob.getJobGroup());
        sysJobLog.setInvokeTarget(sysJob.getInvokeTarget());
        sysJobLog.setStartTime(startTime);
        sysJobLog.setStopTime(new Date());
        long runMs = sysJobLog.getStopTime().getTime() - sysJobLog.getStartTime().getTime();
        sysJobLog.setJobMessage(sysJobLog.getJobName() + " æ€»å…±è€—时：" + runMs + "毫秒");
        if (e != null)
        {
            sysJobLog.setStatus(Constants.FAIL);
            String errorMsg = StringUtils.substring(ExceptionUtil.getExceptionMessage(e), 0, 2000);
            sysJobLog.setExceptionInfo(errorMsg);
        }
        else
        {
            sysJobLog.setStatus(Constants.SUCCESS);
        }

        // å†™å…¥æ•°æ®åº“当中
        SpringUtils.getBean(ISysJobLogService.class).addJobLog(sysJobLog);
    }

    /**
     * æ‰§è¡Œæ–¹æ³•ï¼Œç”±å­ç±»é‡è½½
     *
     * @param context å·¥ä½œæ‰§è¡Œä¸Šä¸‹æ–‡å¯¹è±¡
     * @param sysJob ç³»ç»Ÿè®¡åˆ’任务
     * @throws Exception æ‰§è¡Œè¿‡ç¨‹ä¸­çš„异常
     */
    protected abstract void doExecute(JobExecutionContext context, SysJob sysJob) throws Exception;
}
package com.ruoyi.quartz.util;



import java.util.Date;

import org.quartz.Job;

import org.quartz.JobExecutionContext;

import org.quartz.JobExecutionException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import com.ruoyi.common.constant.Constants;

import com.ruoyi.common.constant.ScheduleConstants;

import com.ruoyi.common.utils.ExceptionUtil;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.common.utils.bean.BeanUtils;

import com.ruoyi.common.utils.spring.SpringUtils;

import com.ruoyi.quartz.domain.SysJob;

import com.ruoyi.quartz.domain.SysJobLog;

import com.ruoyi.quartz.service.ISysJobLogService;



/**

 * æŠ½è±¡quartz调用

 *

 * @author ruoyi

 */

public abstract class AbstractQuartzJob implements Job

{

    private static final Logger log = LoggerFactory.getLogger(AbstractQuartzJob.class);



    /**

     * çº¿ç¨‹æœ¬åœ°å˜é‡

     */

    private static ThreadLocal<Date> threadLocal = new ThreadLocal<>();



    @Override

    public void execute(JobExecutionContext context) throws JobExecutionException

    {

        SysJob sysJob = new SysJob();

        BeanUtils.copyBeanProp(sysJob, context.getMergedJobDataMap().get(ScheduleConstants.TASK_PROPERTIES));

        try

        {

            before(context, sysJob);

            if (sysJob != null)

            {

                doExecute(context, sysJob);

            }

            after(context, sysJob, null);

        }

        catch (Exception e)

        {

            log.error("任务执行异常  - ï¼š", e);

            after(context, sysJob, e);

        }

    }



    /**

     * æ‰§è¡Œå‰

     *

     * @param context å·¥ä½œæ‰§è¡Œä¸Šä¸‹æ–‡å¯¹è±¡

     * @param sysJob ç³»ç»Ÿè®¡åˆ’任务

     */

    protected void before(JobExecutionContext context, SysJob sysJob)

    {

        threadLocal.set(new Date());

    }



    /**

     * æ‰§è¡ŒåŽ

     *

     * @param context å·¥ä½œæ‰§è¡Œä¸Šä¸‹æ–‡å¯¹è±¡

     * @param sysScheduleJob ç³»ç»Ÿè®¡åˆ’任务

     */

    protected void after(JobExecutionContext context, SysJob sysJob, Exception e)

    {

        Date startTime = threadLocal.get();

        threadLocal.remove();



        final SysJobLog sysJobLog = new SysJobLog();

        sysJobLog.setJobName(sysJob.getJobName());

        sysJobLog.setJobGroup(sysJob.getJobGroup());

        sysJobLog.setInvokeTarget(sysJob.getInvokeTarget());

        sysJobLog.setStartTime(startTime);

        sysJobLog.setStopTime(new Date());

        long runMs = sysJobLog.getStopTime().getTime() - sysJobLog.getStartTime().getTime();

        sysJobLog.setJobMessage(sysJobLog.getJobName() + " æ€»å…±è€—时：" + runMs + "毫秒");

        if (e != null)

        {

            sysJobLog.setStatus(Constants.FAIL);

            String errorMsg = StringUtils.substring(ExceptionUtil.getExceptionMessage(e), 0, 2000);

            sysJobLog.setExceptionInfo(errorMsg);

        }

        else

        {

            sysJobLog.setStatus(Constants.SUCCESS);

        }



        // å†™å…¥æ•°æ®åº“当中

        SpringUtils.getBean(ISysJobLogService.class).addJobLog(sysJobLog);

    }



    /**

     * æ‰§è¡Œæ–¹æ³•ï¼Œç”±å­ç±»é‡è½½

     *

     * @param context å·¥ä½œæ‰§è¡Œä¸Šä¸‹æ–‡å¯¹è±¡

     * @param sysJob ç³»ç»Ÿè®¡åˆ’任务

     * @throws Exception æ‰§è¡Œè¿‡ç¨‹ä¸­çš„异常

     */

    protected abstract void doExecute(JobExecutionContext context, SysJob sysJob) throws Exception;

}