#

vincentlu

2025-05-13 ebd2f4397a92c6a5096de1b86d59154363344720

 zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/KernelScheduler.java

@@ -2,23 +2,26 @@

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.zy.acs.manager.core.domain.AgvTaskDto;
import com.zy.acs.manager.core.service.MainLockWrapService;
import com.zy.acs.manager.core.service.MainService;
import com.zy.acs.manager.core.service.TrafficService;
import com.zy.acs.manager.manager.entity.Bus;
import com.zy.acs.manager.manager.entity.Segment;
import com.zy.acs.manager.manager.entity.Task;
import com.zy.acs.manager.manager.enums.ActionStsType;
import com.zy.acs.manager.manager.enums.BusStsType;
import com.zy.acs.manager.manager.enums.SegmentStateType;
import com.zy.acs.manager.manager.enums.TaskStsType;
import com.zy.acs.manager.manager.service.ActionService;
import com.zy.acs.manager.manager.service.BusService;
import com.zy.acs.manager.manager.service.SegmentService;
import com.zy.acs.manager.manager.service.TaskService;
import com.zy.acs.manager.manager.service.*;
import com.zy.acs.manager.system.service.ConfigService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.context.event.ApplicationReadyEvent;
import org.springframework.context.event.EventListener;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import javax.annotation.PreDestroy;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
@@ -33,9 +36,18 @@
@Component
public class KernelScheduler {

    public static int CORE_SCAN_FREQUENCY_MILLISECOND = 15;

    private static final int LOCK_TIMEOUT = 5;

    private final ReentrantLock lock = new ReentrantLock(Boolean.TRUE);

    private Thread trafficCalcThread;

    private Thread actionPublicThread;

    @Autowired
    private AgvService agvService;
    @Autowired
    private BusService busService;
    @Autowired
@@ -43,23 +55,32 @@
    @Autowired
    private MainService mainService;
    @Autowired
    private MainLockWrapService mainLockWrapService;
    @Autowired
    private ActionService actionService;
    @Autowired
    private SegmentService segmentService;
    @Autowired
    private TrafficService trafficService;
    @Autowired
    private ConfigService configService;
    @Autowired
    private LocService locService;
    @Autowired
    private AgvModelService agvModelService;

    @Scheduled(cron = "0/3 * * * * ? ")
    @Scheduled(cron = "0/1 * * * * ? ")
    private void startupBus() throws InterruptedException {
//        if (!configService.getVal("TaskAssignMode", Boolean.class)) { return; }
        if (!this.lock.tryLock(LOCK_TIMEOUT, TimeUnit.SECONDS)) { return; }
        List<Bus> busList = busService.selectBySts(BusStsType.RECEIVE);
        for (Bus bus : busList) {
            mainService.infuseAgvForTask(bus);
            mainService.allocateTask(bus);
        }
        this.lock.unlock();
    }

    @Scheduled(cron = "0/3 * * * * ? ")
    @Scheduled(cron = "0/1 * * * * ? ")
    private void calculateSeg() throws InterruptedException {
        if (!this.lock.tryLock(LOCK_TIMEOUT, TimeUnit.SECONDS)) { return; }
        List<Task> taskList = taskService.selectBySts(TaskStsType.WAITING);
@@ -77,28 +98,54 @@
        }

        for (AgvTaskDto dto : taskDtoList) {
            mainService.mergeMajorTask(dto.getAgvId(), dto.getTaskList());
            mainLockWrapService.buildMajorTask(dto.getAgvId(), dto.getTaskList());
        }
        this.lock.unlock();
    }

    @Scheduled(cron = "0/1 * * * * ? ")
    private void traffic() {
        List<Segment> segments = segmentService.list(new LambdaQueryWrapper<Segment>()
                .eq(Segment::getState, SegmentStateType.WAITING.toString())
        );
        for (Segment segment : segments) {
            trafficService.trigger(segment);
        }
    @EventListener(ApplicationReadyEvent.class)
    public void init() {
        try { Thread.sleep(1200); } catch (InterruptedException ignore) {}
        // traffic calculate
        this.trafficCalcThread = new Thread(() -> {
            while (!Thread.currentThread().isInterrupted()) {
                try {
                    Thread.sleep(CORE_SCAN_FREQUENCY_MILLISECOND);

                    List<Segment> segments = segmentService.list(new LambdaQueryWrapper<Segment>()
                            .eq(Segment::getState, SegmentStateType.WAITING.toString())
                    );
                    for (Segment segment : segments) {
                        long startTime = System.currentTimeMillis();
                        trafficService.trigger(segment);
//                        log.info("traffic calculation spend {} ms", System.currentTimeMillis() - startTime);
                    }

                } catch (Exception e) {
                    log.error("KernelScheduler.trafficCalcThread fail", e);
                }
            }
        });
        this.trafficCalcThread.start();
        // public action
        this.actionPublicThread = new Thread(() -> {
            while (!Thread.currentThread().isInterrupted()) {
                try {
                    Thread.sleep(CORE_SCAN_FREQUENCY_MILLISECOND);

                    List<String> actionGroupIds = actionService.selectGroupNo(ActionStsType.PREPARE);
                    for (String actionGroupId : actionGroupIds) {
                        mainService.publishAction(actionGroupId);
                    }

                } catch (Exception e) {
                    log.error("KernelScheduler.actionPublicThread fail", e);
                }
            }
        });
        this.actionPublicThread.start();
    }

    @Scheduled(cron = "0/1 * * * * ? ")
    private void publishAction(){
        List<String> actionGroupIds = actionService.selectPrepareGroup();
        for (String actionGroupId : actionGroupIds) {
            mainService.publishAction(actionGroupId);
        }
    }

    // patch ----------------------------------------------------------------------------------------------

@@ -133,4 +180,15 @@
        }
    }


    @PreDestroy
    public void shutDown(){
        if (this.trafficCalcThread != null) {
            this.trafficCalcThread .interrupt();
        }
        if (this.actionPublicThread != null) {
            this.actionPublicThread .interrupt();
        }
    }

}