1.場景描述

老專案需要多機部署，專案中有幾十個定時任務，一旦多機部署，定時任務就會重複執行，固定ip與錯開時間方案都存在較大弊端，最終採用的方案是:AOP+排他鎖的方式，軟體老王已驗證通過，介紹下，有需要的朋友可以參考下。

2.解決方案

軟體老王基本方案是採用：AOP+排他鎖的方式。

（1）目前老專案有幾十個定時任務，採用AOP的方式，可以保證程式碼的無侵入（即使簡單的微侵入，例如增加幾行程式碼，測試驗證的工作量也會比較大的）。
（2）採用排他鎖的方式，保證批處理的高可用，不重複執行。

2.1 AOP程式設計

Aop的概念就不說了，就是面向切面程式設計，通俗點就是統一處理一類問題，比如日誌、請求鑑權等，剛開始不確定是否可行，系統中的批處理是使用spring註解的方式@Scheduled進行批處理，採用aop對註解@Scheduled進行程式設計，統一攔截批處理，程式碼如下：

/**
 * 軟體老王-AOP處理類
 */
@Aspect
@Component
public class ScheduledAspect {
    @Autowired
    ScheduleService scheduleService ;

    @Pointcut( "@annotation(org.springframework.scheduling.annotation.Scheduled)")
    public void scheduled() {
    }
    @Around("scheduled()")
    public Object scheduled(ProceedingJoinPoint pjd) {
        Object result = null;
        String taskName = pjd.getSignature().getName();
        try {
            if (scheduleService.isInvoke(taskName)){
                return result;
            }
            result = pjd.proceed();
            scheduleService.end(taskName);
        } catch (Throwable e) {
            throw new RuntimeException(e);
        }
        return result;
    }
}
 

說明：

（1）面向標籤程式設計

  @Pointcut( "@annotation(org.springframework.scheduling.annotation.Scheduled)")

這樣註解會攔截標籤@Scheduled。

（2）使用aop的環繞標籤 @Around("scheduled()")

@before標籤拿不到執行完成狀態，需要使用環繞標籤@@Around，在標籤中可以拿到執行完成後狀態，以便放開鎖。

 result = pjd.proceed();

（3）結合排他鎖使用

   @Autowired
    ScheduleService scheduleService ;
 

2.2 排他鎖

排他鎖，簡單來說就是通過資料庫總的標誌位+版版號進行的控制.

軟體老王的程式碼如下，：

/**
 * 軟體老王-排他鎖服務類
 */
@Service
public class ScheduleService {
    @Autowired
    ScheduleClusterMapper scheduleClusterMapper;

    public boolean isInvoke(String taskName) {
        boolean isValid = false;
        try {
            ScheduleCluster carIndexEntity = scheduleClusterMapper.selectByTaskName(taskName);
            int execute = carIndexEntity.getExecute();
            String ip = InetAddress.getLocalHost().getHostAddress();
            long currentTimeMillis = System.currentTimeMillis();
            long time = carIndexEntity.getUpdatedate().getTime();
            if (execute == 0) {
                isValid = start(taskName, carIndexEntity.getVersion(), ip);
            }
        } catch (UnknownHostException e) {
            e.printStackTrace();
        }
        return isValid;
    }
    //執行鎖機制，軟體老王
    public boolean start(String taskName, int version, String ip) {
        ScheduleCluster scheduleCluster = new ScheduleCluster();
        scheduleCluster.setVersion(version);
        scheduleCluster.setExecuteIp(ip);
        scheduleCluster.setUpdatedate(DateUtil.getCurrentTime());
        scheduleCluster.setTaskName(taskName);
        scheduleCluster.setExecute(1);
        int count = scheduleClusterMapper.updateByTaskName(scheduleCluster);
        if (count > 0) {
            return true;
        }
        return false;
    }
    //執行解鎖機制，軟體老王
    public void end(String taskName) {
        ScheduleCluster scheduleCluster = new ScheduleCluster();
        scheduleCluster.setUpdatedate(DateUtil.getCurrentTime());
        scheduleCluster.setTaskName(taskName);
        scheduleCluster.setExecute(0);
        scheduleClusterMapper.updateNormalByTaskName(scheduleCluster);
    }
}

說明：

大的原理是在where條件後帶上版本號，在update中更新version+1，這樣通過影響資料庫的影響條數，來判斷是否拿到鎖。

（1）主類中呼叫start方法，該方法是更新批處理狀態，軟體老王這裡設定了一個小點，在updateByTaskName的mybatis方法中，有個version+1的更新；

（2）end方法放在更新完成後，釋放鎖。

（3）其實還有一個點，可以考慮下，需要有個機制，比如出現異常情況，剛好批處理執行中，重啟服務了等，下次批處理執行前，假如鎖還未釋放，程式碼中增加釋放鎖的機制。

2.3 資料庫相關

（1）資料庫表設計

（2）mybatis相關方法

（1）第一個是start對應方法，執行鎖和version增加。

 <update id="updateByTaskName" parameterType="com.yutong.dmp.entity.ScheduleCluster">
    update t_schedule_cluster
    <set>
        <if test="executeIp != null">
            execute_ip = #{executeIp,jdbcType=VARCHAR},
        </if>
        <if test="version != null">
            version = #{version,jdbcType=INTEGER} + 1,
        </if>
        <if test="execute != null">
            execute = #{execute,jdbcType=INTEGER},
        </if>
        <if test="status != null">
            status = #{status,jdbcType=VARCHAR},
        </if>
        <if test="createby != null">
            createby = #{createby,jdbcType=VARCHAR},
        </if>
        <if test="createdate != null">
            createdate = #{createdate,jdbcType=TIMESTAMP},
        </if>
        <if test="updateby != null">
            updateby = #{updateby,jdbcType=VARCHAR},
        </if>
        <if test="updatedate != null">
            updatedate = #{updatedate,jdbcType=TIMESTAMP},
        </if>
    </set>
    where task_name = #{taskName,jdbcType=VARCHAR}
    and version = #{version,jdbcType=INTEGER}
    and status ='1'
  </update>

（2）第二個是釋放鎖，更改excute為0。

<update id="updateNormalByTaskName" parameterType="com.yutong.dmp.entity.ScheduleCluster">
    update t_schedule_cluster
    <set>
        <if test="executeIp != null">
            execute_ip = #{executeIp,jdbcType=VARCHAR},
        </if>
        <if test="version != null">
            version = #{version,jdbcType=INTEGER},
        </if>
        <if test="execute != null">
            execute = #{execute,jdbcType=INTEGER},
        </if>
        <if test="status != null">
            status = #{status,jdbcType=VARCHAR},
        </if>
        <if test="createby != null">
            createby = #{createby,jdbcType=VARCHAR},
        </if>
        <if test="createdate != null">
            createdate = #{createdate,jdbcType=TIMESTAMP},
        </if>
        <if test="updateby != null">
            updateby = #{updateby,jdbcType=VARCHAR},
        </if>
        <if test="updatedate != null">
            updatedate = #{updatedate,jdbcType=TIMESTAMP},
        </if>
    </set>
    where task_name = #{taskName,jdbcType=VARCHAR}
    and status ='1'
</update>

I’m 「軟體老王」，如果覺得還可以的話，關注下唄，後續更新秒知！歡迎討論區、同名公眾號留言交流