agv定时任务后分配站点

chen.llin

2 天以前 03b9be24e5a9bc5cf746beac1ac43330183c2c63

agv定时任务后分配站点

 src/main/java/com/zy/asrs/service/impl/MobileServiceImpl.java

@@ -257,106 +257,68 @@
            throw new CoolException("请等待出库完成,type:" + type);
        }

        // 检查站点是否有未完成的AGV任务
        // 规则：当某个站点有未完成的AGV任务时，不分配该站点；只从没有未完成任务的站点中选择
        // 将站点列表转换为字符串列表（Task表的sta_no是String类型）
        List<String> siteStrList = canInSites.stream()
                .map(String::valueOf)
                .collect(Collectors.toList());
        
        // 查询这些站点中有未完成AGV任务的站点（wrk_sts不在5和15之间表示未完成）
        List<Task> unfinishedTasks = taskService.selectList(new EntityWrapper<Task>()
                .in("sta_no", siteStrList)
                .eq("task_type", "agv") // 只查询AGV任务
                .last("AND wrk_sts NOT IN (5, 15)") // 排除已完成状态（5和15表示已完成）
        );
        
        // 获取有未完成任务的站点集合（这些站点将被排除，不参与分配）
        Set<String> sitesWithUnfinishedTasks = unfinishedTasks.stream()
                .map(Task::getStaNo)
                .collect(Collectors.toSet());
        
        // 从可用站点中排除有未完成任务的站点，只保留没有未完成任务的站点
        List<Integer> availableSites = canInSites.stream()
                .filter(site -> !sitesWithUnfinishedTasks.contains(String.valueOf(site)))
                .collect(Collectors.toList());
        
        // 如果所有站点都有未完成任务，则没有可用站点，不分配
        if (availableSites.isEmpty()) {
            String groupName = (whsType != null && whsType.equals(agvProperties.getWhsTypeMapping().getInboundArea())) 
                    ? "东侧" : "西侧";
            log.warn("{}所有站点（{}）都有未完成的AGV任务，无法分配站点，请等待任务完成", groupName, canInSites);
            throw new CoolException(groupName + "所有站点都有未完成的AGV任务，请等待任务完成后再试");
        }
        
        // 记录站点分配信息
        if (!sitesWithUnfinishedTasks.isEmpty()) {
            log.info("站点分配检查：总站点数={}，有未完成任务的站点={}（已排除），可用站点数={}，可用站点={}", 
                    canInSites.size(), sitesWithUnfinishedTasks, availableSites.size(), availableSites);
        } else {
            log.info("站点分配检查：所有站点（{}）都没有未完成任务，全部可用", canInSites);
        }

        // 寻找入库任务最少的站点（只从可用站点中选择，且必须in_enable="Y"能入 和 canining="Y"可入）
        // 寻找入库任务最少的站点（且必须in_enable="Y"能入 和 canining="Y"可入）
        // 注意：不在此处检查未完成的AGV任务，允许PDA持续申请下单
        // 未完成任务的检查将在发送AGV请求时进行（AgvHandler.callAgv）
        List<BasDevp> devList = basDevpMapper.selectList(new EntityWrapper<BasDevp>()
                .in("dev_no", availableSites)
                .in("dev_no", canInSites)
                .eq("in_enable", "Y") // in_enable是能入
                .eq("canining", "Y") // canining是可入
        );
        
        // 如果查询结果为空，说明没有可入的站点
        if (devList.isEmpty()) {
        // 选择站点（如果可入站点为空，则不在创建任务时分配站点，由定时任务分配）
        Integer endSite = null;
        if (!devList.isEmpty()) {
            // 入库任务数排序
            devList.sort(Comparator.comparing(BasDevp::getInQty));

            // 选择站点
            BasDevp basDevp;
            
            // 获取最少任务数
            int minInQty = devList.get(0).getInQty();
            
            // 筛选出任务数最少的站点列表
            List<BasDevp> minTaskSites = devList.stream()
                    .filter(dev -> dev.getInQty() == minInQty)
                    .collect(Collectors.toList());
            
            // 根据配置选择分配策略
            String strategy = agvProperties.getSiteAllocation().getStrategy();
            boolean enableRoundRobin = agvProperties.getSiteAllocation().isEnableRoundRobin();
            
            if (minTaskSites.size() > 1 && enableRoundRobin && "round-robin".equals(strategy)) {
                // 轮询分配：当多个站点任务数相同时，使用轮询
                String groupKey = (whsType != null && whsType.equals(agvProperties.getWhsTypeMapping().getInboundArea())) 
                        ? "east" : "west";
                AtomicInteger counter = siteRoundRobinCounters.computeIfAbsent(groupKey, k -> new AtomicInteger(0));
                int index = counter.getAndIncrement() % minTaskSites.size();
                basDevp = minTaskSites.get(index);
                log.info("使用轮询分配策略，站点组：{}，轮询索引：{}，选中站点：{}", groupKey, index, basDevp.getDevNo());
            } else if (minTaskSites.size() > 1 && enableRoundRobin && "random".equals(strategy)) {
                // 随机分配
                Random random = new Random();
                int index = random.nextInt(minTaskSites.size());
                basDevp = minTaskSites.get(index);
                log.info("使用随机分配策略，选中站点：{}", basDevp.getDevNo());
            } else {
                // 默认：选择第一个（任务最少的）
                basDevp = devList.get(0);
                if (minTaskSites.size() > 1) {
                    log.info("多个站点任务数相同（{}），但未启用轮询，选择第一个站点：{}", minInQty, basDevp.getDevNo());
                }
            }
            
            endSite = basDevp.getDevNo();

            // 入库暂存+1
            basDevpMapper.incrementInQty(endSite);
        } else {
            // 没有可入站点，记录日志但不阻止下单，站点分配将在定时任务中处理
            String groupName = (whsType != null && whsType.equals(agvProperties.getWhsTypeMapping().getInboundArea())) 
                    ? "东侧" : "西侧";
            log.warn("{}可用站点（{}）中没有可入站点（in_enable='Y'且canining='Y'），无法分配", groupName, availableSites);
            throw new CoolException(groupName + "可用站点中没有可入站点，请检查站点配置");
            log.warn("{}可用站点（{}）中没有可入站点（in_enable='Y'且canining='Y'），暂不分配站点，将在定时任务中分配", groupName, canInSites);
        }
        
        // 入库任务数排序
        devList.sort(Comparator.comparing(BasDevp::getInQty));

        // 选择站点
        BasDevp basDevp;
        int endSite;
        
        // 获取最少任务数
        int minInQty = devList.get(0).getInQty();
        
        // 筛选出任务数最少的站点列表
        List<BasDevp> minTaskSites = devList.stream()
                .filter(dev -> dev.getInQty() == minInQty)
                .collect(Collectors.toList());
        
        // 根据配置选择分配策略
        String strategy = agvProperties.getSiteAllocation().getStrategy();
        boolean enableRoundRobin = agvProperties.getSiteAllocation().isEnableRoundRobin();
        
        if (minTaskSites.size() > 1 && enableRoundRobin && "round-robin".equals(strategy)) {
            // 轮询分配：当多个站点任务数相同时，使用轮询
            String groupKey = (whsType != null && whsType.equals(agvProperties.getWhsTypeMapping().getInboundArea())) 
                    ? "east" : "west";
            AtomicInteger counter = siteRoundRobinCounters.computeIfAbsent(groupKey, k -> new AtomicInteger(0));
            int index = counter.getAndIncrement() % minTaskSites.size();
            basDevp = minTaskSites.get(index);
            log.info("使用轮询分配策略，站点组：{}，轮询索引：{}，选中站点：{}", groupKey, index, basDevp.getDevNo());
        } else if (minTaskSites.size() > 1 && enableRoundRobin && "random".equals(strategy)) {
            // 随机分配
            Random random = new Random();
            int index = random.nextInt(minTaskSites.size());
            basDevp = minTaskSites.get(index);
            log.info("使用随机分配策略，选中站点：{}", basDevp.getDevNo());
        } else {
            // 默认：选择第一个（任务最少的）
            basDevp = devList.get(0);
            if (minTaskSites.size() > 1) {
                log.info("多个站点任务数相同（{}），但未启用轮询，选择第一个站点：{}", minInQty, basDevp.getDevNo());
            }
        }
        
        endSite = basDevp.getDevNo();

        // 入库暂存+1
        basDevpMapper.incrementInQty(endSite);


        // 获取工作号
@@ -370,7 +332,7 @@
                .setIoType(ioType) // 入出库状态： 1.入库
                .setTaskType("agv")
                .setIoPri(10D)
                .setStaNo(String.valueOf(endSite))
                .setStaNo(endSite != null ? String.valueOf(endSite) : null) // 如果分配了站点则设置，否则为null，由定时任务分配
                .setSourceStaNo(sourceSite) // 设置源站点
                .setInvWh(robotGroup) // 根据whs_type设置机器人组
                .setFullPlt(ioType != 10 ? "N" : "Y")// 满板：Y

 src/main/java/com/zy/asrs/task/handler/AgvHandler.java

@@ -6,6 +6,8 @@
import com.zy.asrs.entity.Task;
import com.zy.asrs.entity.TaskLog;
import com.zy.asrs.entity.WrkMast;
import com.zy.asrs.entity.BasDevp;
import com.zy.asrs.mapper.BasDevpMapper;
import com.zy.asrs.mapper.BasStationMapper;
import com.zy.asrs.mapper.WrkMastMapper;
import com.zy.asrs.service.ApiLogService;
@@ -20,11 +22,9 @@
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;

/**
@@ -52,7 +52,16 @@
    private BasStationMapper basStationMapper;

    @Resource
    private BasDevpMapper basDevpMapper;

    @Resource
    private AgvProperties agvProperties;

    /**
     * 站点轮询计数器，用于平均分配站点
     * Key: 站点组标识（如 "east" 或 "west"），Value: 当前轮询索引
     */
    private final Map<String, AtomicInteger> siteRoundRobinCounters = new ConcurrentHashMap<>();

    /**
     * 呼叫agv搬运
@@ -64,6 +73,37 @@
        }

        for (Task task : taskList) {
            // 如果任务没有分配站点，先分配站点
            String staNo = task.getStaNo();
            if (staNo == null || staNo.isEmpty()) {
                Integer allocatedSite = allocateSiteForTask(task);
                if (allocatedSite == null) {
                    log.warn("任务ID：{}无法分配站点，跳过本次发送", task.getId());
                    continue; // 无法分配站点，跳过本次发送
                }
                staNo = String.valueOf(allocatedSite);
                task.setStaNo(staNo);
                taskService.updateById(task);
                log.info("任务ID：{}已分配站点：{}", task.getId(), staNo);
            }
            
            // 检查目标站点是否有未完成的AGV任务
            // 如果站点有未完成的任务，则跳过本次发送，等待下次
            if (staNo != null && !staNo.isEmpty()) {
                List<Task> unfinishedTasks = taskService.selectList(new EntityWrapper<Task>()
                        .eq("sta_no", staNo)
                        .eq("task_type", "agv")
                        .ne("id", task.getId()) // 排除当前任务本身
                        .last("AND wrk_sts NOT IN (5, 15)") // 排除已完成状态
                );
                
                if (!unfinishedTasks.isEmpty()) {
                    log.info("站点{}有{}个未完成的AGV任务，跳过本次发送，等待任务完成。当前任务ID：{}", 
                            staNo, unfinishedTasks.size(), task.getId());
                    continue; // 跳过本次发送，等待下次
                }
            }
            
            // 呼叫agv
            String response = "";
            boolean success = false;
@@ -197,6 +237,115 @@
    }

    /**
     * 为任务分配站点（定时任务中调用）
     * @param task 任务对象
     * @return 分配的站点编号，如果无法分配则返回null
     */
    private Integer allocateSiteForTask(Task task) {
        // 根据任务的invWh（机器人组）判断是东侧还是西侧
        String robotGroup = task.getInvWh();
        List<String> targetStations;
        String groupKey;
        
        if (robotGroup != null && robotGroup.equals(agvProperties.getRobotGroupEast())) {
            // 东侧站点
            targetStations = agvProperties.getEastStations();
            groupKey = "east";
        } else if (robotGroup != null && robotGroup.equals(agvProperties.getRobotGroupWest())) {
            // 西侧站点
            targetStations = agvProperties.getWestStations();
            groupKey = "west";
        } else {
            // 默认使用东侧
            targetStations = agvProperties.getEastStations();
            groupKey = "east";
            log.warn("任务ID：{}的机器人组{}未识别，使用默认东侧站点", task.getId(), robotGroup);
        }
        
        if (targetStations.isEmpty()) {
            log.warn("任务ID：{}没有可用的目标站点配置", task.getId());
            return null;
        }
        
        // 将站点字符串列表转换为整数列表
        List<Integer> siteIntList = targetStations.stream()
                .map(Integer::parseInt)
                .collect(Collectors.toList());
        
        // 判断能入站点（in_enable="Y"表示能入）
        List<Integer> sites = basDevpMapper.selectList(
                new EntityWrapper<BasDevp>()
                        .eq("in_enable", "Y")
                        .in("dev_no", siteIntList)
        ).stream().map(BasDevp::getDevNo).collect(Collectors.toList());
        
        if (sites.isEmpty()) {
            log.warn("任务ID：{}没有能入站点", task.getId());
            return null;
        }
        
        // 获取没有出库任务的站点
        List<Integer> canInSites = basDevpMapper.getCanInSites(sites);
        if (canInSites.isEmpty()) {
            log.warn("任务ID：{}没有可入站点（请等待出库完成）", task.getId());
            return null;
        }
        
        // 寻找入库任务最少的站点（且必须in_enable="Y"能入 和 canining="Y"可入）
        List<BasDevp> devList = basDevpMapper.selectList(new EntityWrapper<BasDevp>()
                .in("dev_no", canInSites)
                .eq("in_enable", "Y")
                .eq("canining", "Y")
        );
        
        if (devList.isEmpty()) {
            log.warn("任务ID：{}没有可入站点（in_enable='Y'且canining='Y'）", task.getId());
            return null;
        }
        
        // 入库任务数排序
        devList.sort(Comparator.comparing(BasDevp::getInQty));
        
        // 选择站点
        BasDevp basDevp;
        int minInQty = devList.get(0).getInQty();
        
        // 筛选出任务数最少的站点列表
        List<BasDevp> minTaskSites = devList.stream()
                .filter(dev -> dev.getInQty() == minInQty)
                .collect(Collectors.toList());
        
        // 根据配置选择分配策略
        String strategy = agvProperties.getSiteAllocation().getStrategy();
        boolean enableRoundRobin = agvProperties.getSiteAllocation().isEnableRoundRobin();
        
        if (minTaskSites.size() > 1 && enableRoundRobin && "round-robin".equals(strategy)) {
            // 轮询分配
            AtomicInteger counter = siteRoundRobinCounters.computeIfAbsent(groupKey, k -> new AtomicInteger(0));
            int index = counter.getAndIncrement() % minTaskSites.size();
            basDevp = minTaskSites.get(index);
            log.info("使用轮询分配策略，站点组：{}，轮询索引：{}，选中站点：{}", groupKey, index, basDevp.getDevNo());
        } else if (minTaskSites.size() > 1 && enableRoundRobin && "random".equals(strategy)) {
            // 随机分配
            Random random = new Random();
            int index = random.nextInt(minTaskSites.size());
            basDevp = minTaskSites.get(index);
            log.info("使用随机分配策略，选中站点：{}", basDevp.getDevNo());
        } else {
            // 默认：选择第一个（任务最少的）
            basDevp = devList.get(0);
        }
        
        Integer endSite = basDevp.getDevNo();
        
        // 入库暂存+1
        basDevpMapper.incrementInQty(endSite);
        
        log.info("任务ID：{}已分配站点：{}", task.getId(), endSite);
        return endSite;
    }

    /**
     * 根据站点编号判断机器人组
     * @param staNo 站点编号
     * @return 机器人组名称