#
vincentlu
8 天以前 efabc6ba991acfd01d38bb0bf4e8cfd772416617
zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/TaskServiceImpl.java
@@ -6,6 +6,9 @@
import com.zy.acs.framework.common.Cools;
import com.zy.acs.framework.common.SnowflakeIdWorker;
import com.zy.acs.framework.exception.CoolException;
import com.zy.acs.manager.common.domain.BaseParam;
import com.zy.acs.manager.common.domain.PageParam;
import com.zy.acs.manager.common.domain.PageResult;
import com.zy.acs.manager.common.exception.BusinessException;
import com.zy.acs.manager.core.domain.Lane;
import com.zy.acs.manager.core.service.LaneService;
@@ -25,6 +28,8 @@
public class TaskServiceImpl extends ServiceImpl<TaskMapper, Task> implements TaskService {
    @Autowired
    private BusService busService;
    @Autowired
    private CodeService codeService;
    @Autowired
    private LocService locService;
@@ -36,6 +41,13 @@
    private SnowflakeIdWorker snowflakeIdWorker;
    @Autowired
    private LaneService laneService;
    @Autowired
    private StaReserveService staReserveService;
    @Override
    public PageResult<Task> pageRel(PageParam<Task, BaseParam> pageParam) {
        return new PageResult<>(this.baseMapper.selectPageRel(pageParam, pageParam.checkoutMap()), pageParam.getTotal());
    }
    @Override
    public synchronized String generateSeqNum() {
@@ -86,7 +98,7 @@
        if (null == task) {
            return Boolean.FALSE;
        }
        this.maintainLocSts(task, Boolean.TRUE);
        this.maintainLocAndStaHandler(task, Boolean.TRUE);
        task.setTaskSts(TaskStsType.COMPLETE.val());
        task.setUpdateTime(new Date());
@@ -94,6 +106,7 @@
        if (!this.updateById(task)) {
            throw new CoolException(BaseRes.ERROR);
        }
        busService.checkoutComplete(task.getBusId());
        return Boolean.TRUE;
    }
@@ -104,7 +117,7 @@
        if (null == task) {
            return Boolean.FALSE;
        }
        this.maintainLocSts(task, Boolean.FALSE);
        this.maintainLocAndStaHandler(task, Boolean.FALSE);
        task.setTaskSts(TaskStsType.CANCEL.val());
        task.setUpdateTime(new Date());
@@ -112,6 +125,7 @@
        if (!this.updateById(task)) {
            throw new CoolException(BaseRes.ERROR);
        }
        busService.checkoutComplete(task.getBusId());
        return Boolean.TRUE;
    }
@@ -135,7 +149,7 @@
        if (null == codeId) {
            return null;
        }
        return laneService.search(codeService.getById(codeId).getData());
        return laneService.search(codeService.getCacheById(codeId).getData());
    }
    @Override
@@ -158,7 +172,7 @@
        if (null == codeId) {
            return null;
        }
        return laneService.search(codeService.getById(codeId).getData());
        return laneService.search(codeService.getCacheById(codeId).getData());
    }
    @Override
@@ -176,7 +190,7 @@
    }
    @Override
    public List<Task> findRunningTasksByAgv(Long agvId) {
    public List<Task> findTransportTasksByAgv(Long agvId) {
        if (null == agvId) {
            return new ArrayList<>();
        }
@@ -186,8 +200,107 @@
        return this.list(wrapper);
    }
    @Override
    public Integer findTransportTasksCountByAgv(Long agvId) {
        LambdaQueryWrapper<Task> wrapper = new LambdaQueryWrapper<Task>().eq(Task::getAgvId, agvId);
        wrapper.in(Task::getTaskSts, TaskStsType.WAITING.val(), TaskStsType.ASSIGN.val(), TaskStsType.PROGRESS.val());
        wrapper.notIn(Task::getTaskType, TaskTypeType.MOVE.val(), TaskTypeType.TO_CHARGE.val(), TaskTypeType.TO_STANDBY.val());
        return this.count(wrapper);
    }
    @Override
    public Task findLatestTask(Long agvId, TaskStsType taskSts) {
        LambdaQueryWrapper<Task> wrapper = new LambdaQueryWrapper<Task>()
                .orderByDesc(Task::getCreateTime)
                .last("limit 0, 1");
        if (null != agvId) {
            wrapper.eq(Task::getAgvId, agvId);
        }
        if (null != taskSts) {
            wrapper.eq(Task::getTaskSts, taskSts.val());
        }
        return this.list(wrapper).stream().findFirst().orElse(null);
    }
    @Override
    public void maintainLocAndSta(Task task) {
        if (null == task) {
            return;
        }
        if (!task.getTaskSts().equals(TaskStsType.COMPLETE.val())) {
            return;
        }
        Date now = new Date();
        // loc status
        Loc oriLoc = null;
        Loc destLoc = null;
        Sta oriSta = null;
        Sta destSta = null;
        switch (Objects.requireNonNull(TaskTypeType.get(task.getTaskTypeEl()))) {
            case LOC_TO_LOC:
                oriLoc = locService.getById(task.getOriLoc());
                if (oriLoc.getLocSts().equals(LocStsType.PAKOUT.val())) {
                    oriLoc.setLocSts(LocStsType.IDLE.val());
                    oriLoc.setUpdateTime(now);
                    if (!locService.updateById(oriLoc)) {
                        log.error("Loc [{}] 库位修改状态失败", task.getOriLoc$());
                    }
                }
                destLoc = locService.getById(task.getDestLoc());
                if (destLoc.getLocSts().equals(LocStsType.PAKIN.val())) {
                    destLoc.setLocSts(LocStsType.STOCK.val());
                    destLoc.setUpdateTime(now);
                    if (!locService.updateById(destLoc)) {
                        log.error("Loc [{}] 库位修改状态失败", task.getDestLoc$());
                    }
                }
                break;
            case LOC_TO_STA:
                oriLoc = locService.getById(task.getOriLoc());
                if (oriLoc.getLocSts().equals(LocStsType.PAKOUT.val())) {
                    oriLoc.setLocSts(LocStsType.IDLE.val());
                    oriLoc.setUpdateTime(now);
                    if (!locService.updateById(oriLoc)) {
                        log.error("Loc [{}] 库位修改状态失败", task.getOriLoc$());
                    }
                }
                destSta = staService.getById(task.getDestSta());
                staReserveService.confirmStaReserve(destSta, task, 1, StaReserveType.IN);
                break;
            case STA_TO_LOC:
                oriSta = staService.getById(task.getOriSta());
                staReserveService.confirmStaReserve(oriSta, task, 1, StaReserveType.OUT);
                destLoc = locService.getById(task.getDestLoc());
                if (destLoc.getLocSts().equals(LocStsType.PAKIN.val())) {
                    destLoc.setLocSts(LocStsType.STOCK.val());
                    destLoc.setUpdateTime(now);
                    if (!locService.updateById(destLoc)) {
                        log.error("Loc [{}] 库位修改状态失败", task.getDestLoc$());
                    }
                }
                break;
            case STA_TO_STA:
                oriSta = staService.getById(task.getOriSta());
                staReserveService.confirmStaReserve(oriSta, task, 1, StaReserveType.OUT);
                destSta = staService.getById(task.getDestSta());
                staReserveService.confirmStaReserve(destSta, task, 1, StaReserveType.IN);
                break;
            case TO_CHARGE:
            case TO_STANDBY:
            case MOVE:
                break;
            default:
                break;
        }
    }
    @Transactional
    public void maintainLocSts(Task task, Boolean complete) {
    public void maintainLocAndStaHandler(Task task, Boolean complete) {
        Loc oriLoc = null; Loc destLoc = null;
        Sta oriSta = null; Sta destSta = null;
        Date now = new Date();
@@ -200,13 +313,13 @@
                oriLoc.setLocSts(complete?LocStsType.IDLE.val():LocStsType.STOCK.val());
                oriLoc.setUpdateTime(now);
                if (!locService.updateById(oriLoc)) {
                    throw new BusinessException("Loc [" + task.getOriLoc$() + "] 库位修改状态失败 !!!");
                    throw new BusinessException("Loc [" + oriLoc.getLocNo() + "] 库位修改状态失败 !!!");
                }
                destLoc.setLocSts(complete?LocStsType.STOCK.val():LocStsType.IDLE.val());
                destLoc.setUpdateTime(now);
                if (!locService.updateById(destLoc)) {
                    throw new BusinessException("Loc [" + task.getDestLoc$() + "] 库位修改状态失败 !!!");
                    throw new BusinessException("Loc [" + destLoc.getLocNo() + "] 库位修改状态失败 !!!");
                }
                break;
            case LOC_TO_STA:
@@ -214,18 +327,45 @@
                oriLoc.setLocSts(complete?LocStsType.IDLE.val():LocStsType.STOCK.val());
                oriLoc.setUpdateTime(now);
                if (!locService.updateById(oriLoc)) {
                    throw new BusinessException("Loc [" + task.getOriLoc$() + "] 库位修改状态失败 !!!");
                    throw new BusinessException("Loc [" + oriLoc.getLocNo() + "] 库位修改状态失败 !!!");
                }
                destSta = staService.getById(task.getDestSta());
                if (complete) {
                    staReserveService.confirmStaReserve(destSta, task, 1, StaReserveType.IN);
                } else {
                    staReserveService.cancelStaReserve(destSta, task, 1, StaReserveType.IN);
                }
                break;
            case STA_TO_LOC:
                oriSta = staService.getById(task.getOriSta());
                if (complete) {
                    staReserveService.confirmStaReserve(oriSta, task, 1, StaReserveType.OUT);
                } else {
                    staReserveService.cancelStaReserve(oriSta, task, 1, StaReserveType.OUT);
                }
                destLoc = locService.getById(task.getDestLoc());
                destLoc.setLocSts(complete?LocStsType.STOCK.val():LocStsType.IDLE.val());
                destLoc.setUpdateTime(now);
                if (!locService.updateById(destLoc)) {
                    throw new BusinessException("Loc [" + task.getDestLoc$() + "] 库位修改状态失败 !!!");
                    throw new BusinessException("Loc [" + destLoc.getLocNo() + "] 库位修改状态失败 !!!");
                }
                break;
            case STA_TO_STA:
                oriSta = staService.getById(task.getOriSta());
                if (complete) {
                    staReserveService.confirmStaReserve(oriSta, task, 1, StaReserveType.OUT);
                } else {
                    staReserveService.cancelStaReserve(oriSta, task, 1, StaReserveType.OUT);
                }
                destSta = staService.getById(task.getDestSta());
                if (complete) {
                    staReserveService.confirmStaReserve(destSta, task, 1, StaReserveType.IN);
                } else {
                    staReserveService.cancelStaReserve(destSta, task, 1, StaReserveType.IN);
                }
                break;
            case TO_CHARGE:
            case TO_STANDBY: