package com.zy.acs.manager.core;
|
|
import com.alibaba.fastjson.JSON;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.zy.acs.common.utils.Utils;
|
import com.zy.acs.manager.common.domain.param.HandlerPublishParam;
|
import com.zy.acs.manager.common.exception.BusinessException;
|
import com.zy.acs.manager.core.service.AvoidWaveCalculator;
|
import com.zy.acs.manager.core.service.MainService;
|
import com.zy.acs.manager.core.service.ThreadPoolRegulator;
|
import com.zy.acs.manager.core.service.TrafficService;
|
import com.zy.acs.manager.core.service.astart.MapDataDispatcher;
|
import com.zy.acs.manager.manager.entity.Agv;
|
import com.zy.acs.manager.manager.entity.AgvDetail;
|
import com.zy.acs.manager.manager.entity.Loc;
|
import com.zy.acs.manager.manager.entity.Task;
|
import com.zy.acs.manager.manager.enums.ActionStsType;
|
import com.zy.acs.manager.manager.enums.LocStsType;
|
import com.zy.acs.manager.manager.enums.TaskStsType;
|
import com.zy.acs.manager.manager.service.*;
|
import com.zy.acs.manager.system.controller.BaseController;
|
import com.zy.acs.framework.common.Cools;
|
import com.zy.acs.framework.common.R;
|
import com.zy.acs.framework.common.SnowflakeIdWorker;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.jdbc.core.JdbcTemplate;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.web.bind.annotation.*;
|
|
import java.util.Date;
|
import java.util.List;
|
import java.util.concurrent.CompletableFuture;
|
import java.util.concurrent.ExecutionException;
|
|
/**
|
* Created by vincent on 8/1/2024
|
*/
|
@Slf4j
|
@RestController
|
@RequestMapping("/handler")
|
public class HandlerController extends BaseController {
|
|
public static final String APP_KEY = "xltys1995";
|
|
@Autowired
|
private AgvService agvService;
|
@Autowired
|
private AgvDetailService agvDetailService;
|
@Autowired
|
private TaskService taskService;
|
@Autowired
|
private MainService mainService;
|
@Autowired
|
private SnowflakeIdWorker snowflakeIdWorker;
|
@Autowired
|
private CodeService codeService;
|
@Autowired
|
private JdbcTemplate jdbcTemplate;
|
@Autowired
|
private LocService locService;
|
@Autowired
|
private TrafficService trafficService;
|
@Autowired
|
private ThreadPoolRegulator threadPoolRegulator;
|
@Autowired
|
private MapDataDispatcher mapDataDispatcher;
|
@Autowired
|
private AvoidWaveCalculator avoidWaveCalculator;
|
|
@RequestMapping(value = "/publish", method = {RequestMethod.GET, RequestMethod.POST})
|
@Transactional
|
public R publish(@RequestHeader String appKey,
|
@RequestBody HandlerPublishParam param) {
|
if (Cools.isEmpty(param.getAgvNo(), param.getTaskType(), appKey)) {
|
return R.error();
|
}
|
if (!APP_KEY.equals(appKey)) {
|
return R.error();
|
}
|
|
Date now = new Date();
|
Agv agv = agvService.selectByUuid(param.getAgvNo());
|
AgvDetail agvDetail = agvDetailService.selectByAgvId(agv.getId());
|
|
// check
|
if (taskService.count(new LambdaQueryWrapper<Task>()
|
.eq(Task::getAgvId, agv.getId())
|
.and(i -> {
|
i.eq(Task::getTaskSts, TaskStsType.WAITING.val())
|
.or()
|
.eq(Task::getTaskSts, TaskStsType.ASSIGN.val())
|
.or().eq(Task::getTaskSts, TaskStsType.PROGRESS.val());
|
})) > 0) {
|
log.info(agv.getUuid() + "号AGV不可用，已经存在进行中的任务...");
|
return R.error();
|
}
|
if (!agvService.judgeEnable(agv.getId())) {
|
log.info(agv.getUuid() + "号AGV不可用，任务无法计算...");
|
return R.error();
|
}
|
|
// generate
|
switch (param.getTaskType()) {
|
case MOVE:
|
if (Cools.isEmpty(param.getEndCode())) {
|
return R.error();
|
}
|
if (null == codeService.selectByData(param.getEndCode())) {
|
return R.error();
|
}
|
if (!mainService.buildMinorTask(agv, agvDetail, param.getTaskType(), param.getEndCode())) {
|
return R.error();
|
}
|
break;
|
case TO_CHARGE:
|
case TO_STANDBY:
|
if (!mainService.buildMinorTask(agv, agvDetail, param.getTaskType(), null)) {
|
return R.error();
|
}
|
break;
|
case LOC_TO_LOC:
|
if (Cools.isEmpty(param.getStartLocNo(), param.getEndLocNo())) {
|
return R.error();
|
}
|
Loc oriLoc = locService.selecatByLocNo(param.getStartLocNo());
|
if (null == oriLoc) {
|
return R.error();
|
}
|
Loc destLoc = locService.selecatByLocNo(param.getEndLocNo());
|
if (null == destLoc) {
|
return R.error();
|
}
|
|
// task
|
Task task = new Task();
|
task.setAgvId(agv.getId());
|
task.setUuid(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
|
List<Task> lastTasks = taskService.list(new LambdaQueryWrapper<Task>().orderByDesc(Task::getId));
|
task.setSeqNum(Utils.generateSeqNum(Cools.isEmpty(lastTasks)?null:lastTasks.get(0).getSeqNum()));
|
|
task.setOriLoc(oriLoc.getId());
|
task.setDestLoc(destLoc.getId());
|
|
task.setTaskType(param.getTaskType().val());
|
task.setTaskSts(TaskStsType.WAITING.val());
|
|
task.setPriority(999);
|
task.setIoTime(now);
|
task.setStartTime(now);
|
if (!taskService.save(task)) {
|
throw new BusinessException(task.getSeqNum() + "任务保存失败");
|
}
|
|
// loc
|
oriLoc = locService.getById(task.getOriLoc());
|
if (!oriLoc.getLocSts().equals(LocStsType.STOCK.val())) {
|
throw new BusinessException("oriLoc：" + task.getOriLoc$() + " 不是在库状态");
|
}
|
oriLoc.setLocSts(LocStsType.PAKOUT.val());
|
oriLoc.setUpdateTime(now);
|
if (!locService.updateById(oriLoc)) {
|
throw new BusinessException("oriLoc：" + task.getOriLoc$() + " 修改库位状态失败");
|
}
|
|
destLoc = locService.getById(task.getDestLoc());
|
if (!destLoc.getLocSts().equals(LocStsType.IDLE.val())) {
|
throw new BusinessException("destLoc：" + task.getDestLoc$() + " 不是空闲状态");
|
}
|
destLoc.setLocSts(LocStsType.PAKIN.val());
|
destLoc.setUpdateTime(now);
|
if (!locService.updateById(destLoc)) {
|
throw new BusinessException("destLoc：" + task.getDestLoc$() + " 修改库位状态失败");
|
}
|
|
break;
|
default:
|
break;
|
}
|
|
return R.ok();
|
}
|
|
@RequestMapping(value = "/reset", method = {RequestMethod.GET, RequestMethod.POST})
|
public R reset(@RequestHeader String appKey,
|
@RequestBody HandlerPublishParam param) {
|
if (Cools.isEmpty(appKey)) {
|
return R.error();
|
}
|
if (!APP_KEY.equals(appKey)) {
|
return R.error();
|
}
|
|
Agv agv = null;
|
if (!Cools.isEmpty(param.getAgvNo())) {
|
agv = agvService.selectByUuid(param.getAgvNo());
|
}
|
|
if (null != agv) {
|
mapDataDispatcher.modifyDynamicMatrix(null, null, agv.getUuid(), true);
|
avoidWaveCalculator.calcDynamicNodeByVehicle(agv, null);
|
|
jdbcTemplate.update("UPDATE man_task SET task_sts = " + TaskStsType.COMPLETE.val() + " where agv_id = " + agv.getId());
|
jdbcTemplate.update("UPDATE man_travel SET state = 'FINISH'" + " where agv_id = " + agv.getId());
|
jdbcTemplate.update("UPDATE man_segment SET state = 'FINISH'" + " where agv_id = " + agv.getId());
|
jdbcTemplate.update("UPDATE man_action SET action_sts = " + ActionStsType.FINISH.val() + " where agv_id = " + agv.getId());
|
} else {
|
List<Agv> agvList = agvService.list(new LambdaQueryWrapper<Agv>());
|
for (Agv one : agvList) {
|
mapDataDispatcher.modifyDynamicMatrix(null, null, one.getUuid(), true);
|
avoidWaveCalculator.calcDynamicNodeByVehicle(one, null);
|
}
|
|
jdbcTemplate.update("UPDATE man_task SET task_sts = " + TaskStsType.COMPLETE.val());
|
jdbcTemplate.update("UPDATE man_travel SET state = 'FINISH'");
|
jdbcTemplate.update("UPDATE man_segment SET state = 'FINISH'");
|
jdbcTemplate.update("UPDATE man_action SET action_sts = " + ActionStsType.FINISH.val());
|
}
|
|
return R.ok();
|
}
|
|
@RequestMapping(value = "/unlock", method = {RequestMethod.GET, RequestMethod.POST})
|
public R unlockPath(@RequestHeader String appKey,
|
@RequestBody HandlerPublishParam param) throws ExecutionException, InterruptedException {
|
if (Cools.isEmpty(param.getAgvNo(), appKey)) {
|
return R.error();
|
}
|
if (!APP_KEY.equals(appKey)) {
|
return R.error();
|
}
|
|
// block
|
// Future<R> future = threadPoolRegulator.getInstance().submit(() -> {
|
// mapDataDispatcher.modifyDynamicMatrix(null, null, param.getAgvNo(), true);
|
// return success();
|
// });
|
// System.out.println(JSON.toJSONString(future.get()));
|
|
// non-block
|
CompletableFuture<?> completableFuture = CompletableFuture.supplyAsync(() -> {
|
mapDataDispatcher.modifyDynamicMatrix(null, null, param.getAgvNo(), true);
|
return R.ok();
|
}, threadPoolRegulator.getInstance());
|
|
completableFuture.thenAccept(result -> {
|
System.out.println(JSON.toJSONString(result));
|
}).exceptionally(e -> {
|
log.error("unlockPath", e);
|
return null;
|
});
|
|
return R.ok();
|
}
|
|
}
|
