From 33817dd27d0527bf7eff142bf74c09fceb628590 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@63.com>
Date: 星期五, 24 九月 2021 13:09:37 +0800
Subject: [PATCH] 1

---
 src/main/java/com/zy/asrs/controller/WorkController.java |   31 +++++++++++++++++++++++++------
 1 files changed, 25 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/zy/asrs/controller/WorkController.java b/src/main/java/com/zy/asrs/controller/WorkController.java
index ff968e0..074d0d8 100644
--- a/src/main/java/com/zy/asrs/controller/WorkController.java
+++ b/src/main/java/com/zy/asrs/controller/WorkController.java
@@ -12,10 +12,7 @@
 import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 /**
  * 宸ヤ綔娴佹帴鍙f帶鍒跺櫒
@@ -23,6 +20,10 @@
  */
 @RestController
 public class WorkController extends BaseController {
+
+    public static final ArrayList<Integer> AGV_EMPTY_AREA = new ArrayList<Integer>() {{
+       add(1061);add(1062);add(1063);add(1064);add(1065);add(1066);add(1067);add(1068);
+    }};
 
     @Autowired
     private WorkService workService;
@@ -44,7 +45,16 @@
     @RequestMapping("/available/empty/put/site")
     @ManagerAuth(memo = "鑾峰彇绌烘澘鍏ュ簱绔欑偣")
     public R availableEmptyPutSite(){
-        return R.ok().add(basDevpService.getAvailableEmptyInSite());
+        List<Integer> availableEmptyInSite = basDevpService.getAvailableEmptyInSite();
+        List<Integer> agvSite = new ArrayList<>();
+        List<Integer> availableEmptyInSite1 = basDevpService.getAgvAvailableInSite();
+        for (Integer integer : availableEmptyInSite1) {
+            if (AGV_EMPTY_AREA.contains(integer)) {
+                agvSite.add(integer);
+            }
+        }
+        availableEmptyInSite.addAll(agvSite);
+        return R.ok().add(availableEmptyInSite);
     }
 
     @RequestMapping("/available/take/site")
@@ -84,7 +94,16 @@
     @RequestMapping("/available/empty/take/site")
     @ManagerAuth(memo = "鑾峰彇绌烘澘鍑哄簱绔欑偣")
     public R availableEmptyTakeSite(){
-        return R.ok().add(basDevpService.getAvailableEmptyOutSite());
+        List<Integer> availableEmptyOutSite = basDevpService.getAvailableEmptyOutSite();
+        List<Integer> agvSite = new ArrayList<>();
+        List<Integer> availableEmptyInSite1 = basDevpService.getAgvAvailableInSite();
+        for (Integer integer : availableEmptyInSite1) {
+            if (AGV_EMPTY_AREA.contains(integer)) {
+                agvSite.add(integer);
+            }
+        }
+        availableEmptyOutSite.addAll(agvSite);
+        return R.ok().add(availableEmptyOutSite);
     }
 
     @RequestMapping("/full/store/put/start")

--
Gitblit v1.9.1