udi-wms-java/src/main/java/com/glxp/api/task/SyncHeartTask.java

package com.glxp.api.task;

import cn.hutool.core.thread.ThreadUtil;
import com.glxp.api.dao.schedule.ScheduledDao;
import com.glxp.api.dao.system.SyncDataSetDao;
import com.glxp.api.entity.collect.IoCollectSet;
import com.glxp.api.entity.system.ScheduledEntity;
import com.glxp.api.entity.system.SyncDataSetEntity;
import com.glxp.api.req.system.ScheduledRequest;
import com.glxp.api.service.sync.HeartService;
import com.glxp.api.util.IntUtil;
import com.glxp.api.util.RedisUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.SchedulingConfigurer;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;
import org.springframework.scheduling.support.CronTrigger;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.concurrent.ExecutorService;

@Slf4j
@Component
@EnableScheduling
public class SyncHeartTask implements SchedulingConfigurer {

    @Resource
    protected ScheduledDao scheduledDao;
    @Resource
    SyncHeartService syncHeartService;
    @Resource
    RedisUtil redisUtil;
    @Resource
    HeartService heartService;
    @Resource
    private SyncDataSetDao syncDataSetDao;
    private volatile ExecutorService executor;

    private ExecutorService getExecutor() {
        if (null == executor) {
            synchronized (this) {
                executor = ThreadUtil.newExecutor(10, 100, Integer.MAX_VALUE);
            }
        }
        return executor;
    }

    @Override
    public void configureTasks(ScheduledTaskRegistrar scheduledTaskRegistrar) {

        scheduledTaskRegistrar.addTriggerTask(() -> process(),
                triggerContext -> {
                    ScheduledRequest scheduledRequest = new ScheduledRequest();
                    scheduledRequest.setCronName("heartTask");
                    ScheduledEntity scheduledEntity = scheduledDao.findScheduled(scheduledRequest);
                    String cron = scheduledEntity.getCron();
                    if (cron.isEmpty()) {
                        log.error("cron is null");
                    }
                    return new CronTrigger(cron).nextExecutionTime(triggerContext);
                });
    }


    private void process() {


        String lockKey = "syncProcessTask_lock";
        try {
            // 尝试获取分布式锁，设置20*60秒超时
            boolean locked = (Boolean) redisUtil.get(lockKey);
            if (locked) {
                log.info("延迟分标定时任务已在其他节点执行中");
                return;
            }
            redisUtil.set(lockKey, true);
            getExecutor().submit(() -> {
                syncHeartService.syncProcess();
                redisUtil.set(lockKey, false);
            });

        } finally {
            // 释放分布式锁
            redisUtil.set(lockKey, false);
            log.info("延迟分标定时任务执行结束，已释放锁");
        }


    }


}