| package com.zy.asrs.wcs.core.controller; | 
|   | 
| import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; | 
| import com.baomidou.mybatisplus.extension.plugins.pagination.Page; | 
| import com.zy.asrs.framework.common.Cools; | 
| import com.zy.asrs.framework.common.R; | 
| import com.zy.asrs.framework.common.SnowflakeIdWorker; | 
| import com.zy.asrs.wcs.common.annotation.OperationLog; | 
| import com.zy.asrs.wcs.common.domain.BaseParam; | 
| import com.zy.asrs.wcs.common.domain.KeyValVo; | 
| import com.zy.asrs.wcs.common.domain.PageParam; | 
| import com.zy.asrs.wcs.core.domain.param.ShuttleMoveLocParam; | 
| import com.zy.asrs.wcs.core.domain.param.ShuttleOperatorParam; | 
| import com.zy.asrs.wcs.core.entity.BasShuttle; | 
| import com.zy.asrs.wcs.core.entity.Motion; | 
| import com.zy.asrs.wcs.core.entity.Task; | 
| import com.zy.asrs.wcs.core.entity.TaskCtg; | 
| import com.zy.asrs.wcs.core.kernel.AnalyzeService; | 
| import com.zy.asrs.wcs.core.model.enums.DeviceCtgType; | 
| import com.zy.asrs.wcs.core.model.enums.TaskCtgType; | 
| import com.zy.asrs.wcs.core.model.enums.TaskStsType; | 
| import com.zy.asrs.wcs.core.service.BasShuttleService; | 
| import com.zy.asrs.wcs.core.service.MotionService; | 
| import com.zy.asrs.wcs.core.service.TaskCtgService; | 
| import com.zy.asrs.wcs.core.service.TaskService; | 
| import com.zy.asrs.wcs.core.utils.Utils; | 
| import com.zy.asrs.wcs.rcs.News; | 
| import com.zy.asrs.wcs.rcs.cache.SlaveConnection; | 
| import com.zy.asrs.wcs.rcs.entity.Device; | 
| import com.zy.asrs.wcs.rcs.model.enums.SlaveType; | 
| import com.zy.asrs.wcs.rcs.model.protocol.ShuttleProtocol; | 
| import com.zy.asrs.wcs.rcs.service.DeviceService; | 
| import com.zy.asrs.wcs.rcs.thread.ShuttleThread; | 
| import com.zy.asrs.wcs.system.controller.BaseController; | 
| import com.zy.asrs.wcs.utils.ExcelUtil; | 
| import org.springframework.beans.factory.annotation.Autowired; | 
| import org.springframework.security.access.prepost.PreAuthorize; | 
| import org.springframework.transaction.annotation.Transactional; | 
| import org.springframework.web.bind.annotation.*; | 
|   | 
| import javax.servlet.http.HttpServletResponse; | 
| import java.util.*; | 
|   | 
| @RestController | 
| @RequestMapping("/api") | 
| public class BasShuttleController extends BaseController { | 
|   | 
|     @Autowired | 
|     private BasShuttleService basShuttleService; | 
|     @Autowired | 
|     private TaskCtgService taskCtgService; | 
|     @Autowired | 
|     private SnowflakeIdWorker snowflakeIdWorker; | 
|     @Autowired | 
|     private TaskService taskService; | 
|     @Autowired | 
|     private AnalyzeService analyzeService; | 
|     @Autowired | 
|     private MotionService motionService; | 
|     @Autowired | 
|     private DeviceService deviceService; | 
|   | 
|     @PreAuthorize("hasAuthority('core:basShuttle:list')") | 
|     @PostMapping("/basShuttle/page") | 
|     public R page(@RequestBody Map<String, Object> map) { | 
|         BaseParam baseParam = buildParam(map, BaseParam.class); | 
|         PageParam<BasShuttle, BaseParam> pageParam = new PageParam<>(baseParam, BasShuttle.class); | 
|         return R.ok().add(basShuttleService.page(pageParam, pageParam.buildWrapper(true))); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('core:basShuttle:list')") | 
|     @PostMapping("/basShuttle/list") | 
|     public R list(@RequestBody Map<String, Object> map) { | 
|         return R.ok().add(basShuttleService.list()); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('core:basShuttle:list')") | 
|     @GetMapping("/basShuttle/{id}") | 
|     public R get(@PathVariable("id") Long id) { | 
|         return R.ok().add(basShuttleService.getById(id)); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('core:basShuttle:save')") | 
|     @OperationLog("添加四向车设备列表") | 
|     @PostMapping("/basShuttle/save") | 
|     public R save(@RequestBody BasShuttle basShuttle) { | 
|         if (!basShuttleService.save(basShuttle)) { | 
|             return R.error("添加失败"); | 
|         } | 
|         return R.ok("添加成功"); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('core:basShuttle:update')") | 
|     @OperationLog("修改四向车设备列表") | 
|     @PostMapping("/basShuttle/update") | 
|     public R update(@RequestBody BasShuttle basShuttle) { | 
|         if (!basShuttleService.updateById(basShuttle)) { | 
|             return R.error("修改失败"); | 
|         } | 
|         return R.ok("修改成功"); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('core:basShuttle:remove')") | 
|     @OperationLog("删除四向车设备列表") | 
|     @PostMapping("/basShuttle/remove/{ids}") | 
|     public R remove(@PathVariable Long[] ids) { | 
|         if (!basShuttleService.removeByIds(Arrays.asList(ids))) { | 
|             return R.error("删除失败"); | 
|         } | 
|         return R.ok("删除成功"); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('core:basShuttle:list')") | 
|     @PostMapping("/basShuttle/query") | 
|     public R query(@RequestParam(required = false) String condition) { | 
|         List<KeyValVo> vos = new ArrayList<>(); | 
|         LambdaQueryWrapper<BasShuttle> wrapper = new LambdaQueryWrapper<>(); | 
|         if (!Cools.isEmpty(condition)) { | 
|             wrapper.like(BasShuttle::getShuttleNo, condition); | 
|         } | 
|         basShuttleService.page(new Page<>(1, 30), wrapper).getRecords().forEach( | 
|                 item -> vos.add(new KeyValVo(item.getShuttleNo(), item.getShuttleNo())) | 
|         ); | 
|         return R.ok().add(vos); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('core:basShuttle:list')") | 
|     @PostMapping("/basShuttle/export") | 
|     public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception { | 
|         ExcelUtil.build(ExcelUtil.create(basShuttleService.list(), BasShuttle.class), response); | 
|     } | 
|   | 
|     //手动操作 | 
|     @PreAuthorize("hasAuthority('core:basShuttle:operator')") | 
|     @PostMapping("/basShuttle/operator/shuttle") | 
|     @Transactional | 
|     public synchronized R shuttleOperator(@RequestBody ShuttleOperatorParam param) { | 
|         if (Cools.isEmpty(param.getShuttleNo(), param.getShuttleTaskMode())) { | 
|             return R.error("参数为空"); | 
|         } | 
|   | 
|         Integer shuttleNo = param.getShuttleNo(); | 
|   | 
|         //获取手动任务类型 | 
|         TaskCtg taskCtg = taskCtgService.getOne(new LambdaQueryWrapper<TaskCtg>() | 
|                 .eq(TaskCtg::getFlag, String.valueOf(TaskCtgType.MANUAL)) | 
|                 .eq(TaskCtg::getStatus, 1)); | 
|         if (taskCtg == null) { | 
|             return R.error(); | 
|         } | 
|   | 
|         String sourceLoc = null;//源库位 | 
|         String targetLoc = null;//任务目标(借用字段) | 
|         String targetSite = param.getShuttleTaskMode();//任务类型(借用字段) | 
|         if (param.getShuttleTaskMode().equals("moveLoc")) { | 
|             //迁移任务 | 
|             targetLoc = param.getTargetLocNo(); | 
|         } else if (param.getShuttleTaskMode().equals("takeMove")) { | 
|             //取放货任务 | 
|             targetLoc = param.getTargetLocNo(); | 
|             sourceLoc = param.getSourceLocNo(); | 
|         } else if (param.getShuttleTaskMode().equals("reset")) { | 
|             //复位 | 
|             Device device = deviceService.getOne(new LambdaQueryWrapper<Device>() | 
|                     .eq(Device::getDeviceType, DeviceCtgType.SHUTTLE.val()) | 
|                     .eq(Device::getStatus, 1) | 
|                     .eq(Device::getDeviceNo, shuttleNo)); | 
|             if (device == null) { | 
|                 return R.error("设备不存在"); | 
|             } | 
|   | 
|             ShuttleThread shuttleThread = (ShuttleThread) SlaveConnection.get(SlaveType.Shuttle, device.getId().intValue()); | 
|             if (shuttleThread == null) { | 
|                 return R.error("设备离线"); | 
|             } | 
|   | 
|             shuttleThread.reset(null); | 
|             return R.ok(); | 
|         } else if (param.getShuttleTaskMode().equals("requestWaiting")) { | 
|             //申请确认 | 
|             Device device = deviceService.getOne(new LambdaQueryWrapper<Device>() | 
|                     .eq(Device::getDeviceType, DeviceCtgType.SHUTTLE.val()) | 
|                     .eq(Device::getStatus, 1) | 
|                     .eq(Device::getDeviceNo, shuttleNo)); | 
|             if (device == null) { | 
|                 return R.error("设备不存在"); | 
|             } | 
|   | 
|             ShuttleThread shuttleThread = (ShuttleThread) SlaveConnection.get(SlaveType.Shuttle, device.getId().intValue()); | 
|             if (shuttleThread == null) { | 
|                 return R.error("设备离线"); | 
|             } | 
|   | 
|             shuttleThread.requestWaiting(); | 
|             return R.ok(); | 
|         } | 
|   | 
|         Task task = new Task(); | 
|         task.setUuid(String.valueOf(snowflakeIdWorker.nextId())); | 
|         task.setTaskNo(String.valueOf(Utils.getTaskNo("MANUAL"))); | 
|         task.setTaskSts(TaskStsType.NEW_MANUAL.sts); | 
|         task.setTaskCtg(taskCtg.getId()); | 
|         task.setPriority(10); | 
|         task.setOriginSite(null); | 
|         task.setOriginLoc(sourceLoc); | 
|         task.setDestSite(null); | 
|         task.setDestLoc(targetLoc); | 
|         task.setIoTime(new Date()); | 
|         task.setStartTime(new Date()); | 
|         task.setStatus(1); | 
|         task.setMemo(targetSite); | 
|         task.setShuttleNo(param.getShuttleNo()); | 
|   | 
|         if (param.getShuttleTaskMode().equals("takeMove")) { | 
|             task.setRecordLoc("Y");//记录库存信息 | 
|         } | 
|         boolean result = taskService.save(task); | 
|         if (!result) { | 
|             return R.error(); | 
|         } | 
|   | 
|         task = taskService.getById(task.getId()); | 
|   | 
|         // generate motion list | 
|         List<Motion> motionList = analyzeService.generateShuttleManualMotion(task); | 
|         if (Cools.isEmpty(motionList)) { | 
|             News.error("保存{}号四向穿梭车手动任务失败!!!", shuttleNo); | 
|             return R.error(); | 
|         } | 
|         motionService.batchInsert(motionList, task.getUuid(), Integer.valueOf(task.getTaskNo())); | 
|   | 
|         task.setTaskSts(TaskStsType.ANALYZE_MANUAL.sts); | 
|         if (!taskService.updateById(task)) { | 
|             News.error("保存{}号四向穿梭车手动任务失败!!!", shuttleNo); | 
|             return R.error(); | 
|         } | 
|         return R.ok(); | 
|     } | 
|   | 
|     //跑库系统 | 
|     @PreAuthorize("hasAuthority('core:basShuttle:operator')") | 
|     @PostMapping("/basShuttle/moveLoc") | 
|     public synchronized R shuttleMoveLoc(@RequestBody ShuttleMoveLocParam param) { | 
|         Device device = deviceService.getOne(new LambdaQueryWrapper<Device>() | 
|                 .eq(Device::getDeviceType, DeviceCtgType.SHUTTLE.val()) | 
|                 .eq(Device::getStatus, 1) | 
|                 .eq(Device::getHostId, getHostId()) | 
|                 .eq(Device::getDeviceNo, param.getShuttleNo())); | 
|         if (device == null) { | 
|             return R.error(); | 
|         } | 
|   | 
|         ShuttleThread shuttleThread = (ShuttleThread) SlaveConnection.get(SlaveType.Shuttle, device.getId().intValue()); | 
|         if (shuttleThread == null) { | 
|             return R.error(); | 
|         } | 
|   | 
|         ShuttleProtocol shuttleProtocol = shuttleThread.getStatus(); | 
|         if (shuttleProtocol == null) { | 
|             return R.error(); | 
|         } | 
|   | 
|         if (!shuttleThread.isIdle()) { | 
|             return R.error("小车忙碌中"); | 
|         } | 
|   | 
|         shuttleThread.enableMoveLoc(param, true); | 
|         return R.ok(); | 
|     } | 
|   | 
|     //跑库系统关闭 | 
|     @PreAuthorize("hasAuthority('core:basShuttle:operator')") | 
|     @PostMapping("/basShuttle/moveLocClose") | 
|     public synchronized R shuttleMoveLocClose(@RequestBody ShuttleMoveLocParam param) { | 
|         Device device = deviceService.getOne(new LambdaQueryWrapper<Device>() | 
|                 .eq(Device::getDeviceType, DeviceCtgType.SHUTTLE.val()) | 
|                 .eq(Device::getStatus, 1) | 
|                 .eq(Device::getHostId, getHostId()) | 
|                 .eq(Device::getDeviceNo, param.getShuttleNo())); | 
|         if (device == null) { | 
|             return R.error(); | 
|         } | 
|   | 
|         ShuttleThread shuttleThread = (ShuttleThread) SlaveConnection.get(SlaveType.Shuttle, device.getId().intValue()); | 
|         if (shuttleThread == null) { | 
|             return R.error(); | 
|         } | 
|   | 
|         ShuttleProtocol shuttleProtocol = shuttleThread.getStatus(); | 
|         if (shuttleProtocol == null) { | 
|             return R.error(); | 
|         } | 
|   | 
|         shuttleThread.enableMoveLoc(null, false); | 
|         return R.ok(); | 
|     } | 
|   | 
| } |