From 471d7b5bc4c8cf7ffeb258f208a66cca202d64de Mon Sep 17 00:00:00 2001
From: Junjie <540245094@qq.com>
Date: 星期一, 14 七月 2025 14:50:42 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/core/thread/fake/FakeNyShuttleThread.java |  233 ++++++++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 171 insertions(+), 62 deletions(-)

diff --git a/src/main/java/com/zy/core/thread/fake/FakeNyShuttleThread.java b/src/main/java/com/zy/core/thread/fake/FakeNyShuttleThread.java
index 53e7eee..e6fed97 100644
--- a/src/main/java/com/zy/core/thread/fake/FakeNyShuttleThread.java
+++ b/src/main/java/com/zy/core/thread/fake/FakeNyShuttleThread.java
@@ -20,28 +20,35 @@
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
-import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 
 @Slf4j
 @SuppressWarnings("all")
 public class FakeNyShuttleThread implements ThreadHandler {
 
-
     private RedisUtil redisUtil;
     private JSONObject fakeStatusDemo = JSONObject.parseObject("{\"mode\":1,\"extend\":{\"countQuantity\":400,\"suspendState\":0,\"minCellVoltage\":3279,\"chargeCycleTimes\":0,\"maxCellVoltage\":3281,\"surplusQuantity\":204,\"voltage\":5248},\"hasLift\":false,\"hasPallet\":false,\"batteryVoltage\":5248,\"runDirection\":\"2\",\"currentCode\":\"{\\\"x\\\":19,\\\"y\\\":11,\\\"z\\\":2}\",\"errorCode\":\"0\",\"hasCharge\":false,\"batteryPower\":\"51\",\"speed\":0,\"deviceStatus\":1}");
-    private HashMap<String, Socket> fakeServerMap = new HashMap();
-    private HashMap<String, JSONObject> fakeStatusMap = new HashMap();
-    private HashMap<String, JSONObject> fakeCommandMap = new HashMap();
+    private ServerSocket serverSocket;
+    private Integer gatewayPort;
+    private ConcurrentHashMap<String, Thread> fakeThreadMap = new ConcurrentHashMap();
+    private ConcurrentHashMap<String, Socket> fakeServerMap = new ConcurrentHashMap();
+    private ConcurrentHashMap<String, JSONObject> fakeStatusMap = new ConcurrentHashMap();
+    private ConcurrentHashMap<String, JSONObject> fakeCommandMap = new ConcurrentHashMap();
 
     private boolean fake = false;
 
-    public FakeNyShuttleThread(RedisUtil redisUtil) {
+    public FakeNyShuttleThread(RedisUtil redisUtil, Integer gatewayPort) {
         this.redisUtil = redisUtil;
+        this.gatewayPort = gatewayPort;
     }
 
     @Override
     public void run() {
         News.info("Fake Server is Started");
+
+        acceptorThread();
+        fakeCommandThread();
+
         while (true) {
             try {
                 initFakeDeviceServer();
@@ -51,7 +58,111 @@
         }
     }
 
-    private void initFakeDeviceServer() {
+    private void acceptorThread() {
+        Thread acceptorThread = new Thread(() -> {
+            log.info("{}:acceptorThread is start");
+            try {
+                while (true) {
+                    try {
+                        if (serverSocket == null) {
+                            serverSocket = new ServerSocket(gatewayPort);
+                        }
+                    }catch (Exception e){
+                        e.printStackTrace();
+                    }
+
+                    if(serverSocket == null){
+                        continue;
+                    }
+
+                    Socket fakeSocket = serverSocket.accept();
+                    InputStream inputStream = fakeSocket.getInputStream();
+                    if(inputStream == null){
+                        continue;
+                    }
+                    // 鑾峰彇杈撳叆娴�
+                    BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream));
+                    // 璇诲彇鏈嶅姟鍣ㄧ殑鍝嶅簲
+                    StringBuffer sb = new StringBuffer();
+                    char[] chars = new char[2048];//缂撳啿鍖�
+                    while (true) {
+                        reader.read(chars);
+                        String trim = new String(chars);
+                        sb.append(trim);
+                        if (trim.lastIndexOf("\r\n") != -1) {
+                            break;
+                        }
+                    }
+
+//                    log.info("acceptorThread is end:{}", sb.toString());
+                    JSONObject result = null;
+                    try {
+                        result = JSON.parseObject(sb.toString());
+                    }catch (Exception e) {
+                        log.error("error body:{}", sb.toString());
+                        e.printStackTrace();
+                    }
+                    if(result == null){
+                        continue;
+                    }
+
+                    Object msgType = result.get("msgType");
+                    if (msgType != null) {
+                        if ("fakeDeviceFirstConnect".equals(msgType)) {
+                            DeviceConfig device = JSON.parseObject(JSON.toJSONString(result.get("deviceConfig")), DeviceConfig.class);
+                            log.info("{}:device is start,devices:{}", device.getDeviceNo(),JSON.toJSONString(fakeServerMap));
+
+                            String fakeStatus = JSON.toJSONString(fakeStatusDemo);
+                            if (!Cools.isEmpty(device.getFakeInitStatus())) {
+                                fakeStatus = device.getFakeInitStatus();
+                            }
+
+                            fakeServerMap.put(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo(), fakeSocket);
+                            fakeStatusMap.put(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo(), JSON.parseObject(fakeStatus));
+
+                            HashMap<String, Object> map = new HashMap<>();
+                            map.put("deviceNo", device.getDeviceNo());
+                            map.put("status", "success");
+                            // 鑾峰彇杈撳嚭娴�
+                            OutputStreamWriter writer = new OutputStreamWriter(fakeSocket.getOutputStream());
+                            writer.write(JSON.toJSONString(map) + "\r\n");
+                            writer.flush();
+                        }
+                    }
+
+                    Thread.sleep(1000);
+                }
+            }catch (Exception e){
+                e.printStackTrace();
+            }
+        });
+        acceptorThread.start();
+    }
+
+    private void fakeCommandThread() {
+        Thread fakeCommandThread = new Thread(() -> {
+            while (true) {
+                try {
+                    FakeDeviceUtils fakeDeviceUtils = null;
+                    try {
+                        fakeDeviceUtils = SpringUtils.getBean(FakeDeviceUtils.class);
+                    }catch (Exception e){}
+                    if(fakeDeviceUtils == null){
+                        continue;
+                    }
+                    List<DeviceConfig> deviceConfigs = fakeDeviceUtils.getFakeDeviceConfig();
+                    for (DeviceConfig device : deviceConfigs) {
+                        excuteFakeCommand(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo());
+                    }
+                }catch (Exception e){
+                    e.printStackTrace();
+                }
+            }
+        });
+        fakeCommandThread.start();
+    }
+
+    private synchronized void initFakeDeviceServer() {
         FakeDeviceUtils fakeDeviceUtils = null;
         try {
             fakeDeviceUtils = SpringUtils.getBean(FakeDeviceUtils.class);
@@ -59,64 +170,53 @@
         if(fakeDeviceUtils == null){
             return;
         }
-        String fakeDeviceConfig = fakeDeviceUtils.getFakeDeviceConfig();
-        if(null != fakeDeviceConfig){
-            List<DeviceConfig> deviceConfigs = JSON.parseArray(fakeDeviceConfig, DeviceConfig.class);
-            for (DeviceConfig device : deviceConfigs) {
-                if (!device.getDeviceType().equals(String.valueOf(SlaveType.Shuttle))) {
-                    continue;
-                }
+        List<DeviceConfig> deviceConfigs = fakeDeviceUtils.getFakeDeviceConfig();
+        for (DeviceConfig device : deviceConfigs) {
+            if (!device.getDeviceType().equals(String.valueOf(SlaveType.Shuttle))) {
+                continue;
+            }
 
-                if (!device.getThreadImpl().equals("NyShuttleThread")) {
-                    continue;
-                }
+            if (!device.getThreadImpl().equals("NyShuttleThread")) {
+                continue;
+            }
 
-                if (fakeServerMap.containsKey(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo())) {
-                    continue;
-                }
+            if (fakeThreadMap.containsKey(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo())) {
+                continue;
+            }
 
-                Thread fakeThread = new Thread(() -> {
-                    try {
-                        ServerSocket serverSocket = new ServerSocket(device.getPort());
-                        Socket fakeSocket = serverSocket.accept();
-
-                        String fakeStatus = JSON.toJSONString(fakeStatusDemo);
-                        if (!Cools.isEmpty(device.getFakeInitStatus())) {
-                            fakeStatus = device.getFakeInitStatus();
-                        }
-
-                        fakeServerMap.put(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo(), fakeSocket);
-                        fakeStatusMap.put(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo(), JSON.parseObject(fakeStatus));
-                        while (true) {
-                            if(fakeSocket == null) {
-                                fakeServerMap.remove(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo());
-                                break;
-                            }
-
-                            handleClient(device);
-                        }
-                    }catch (Exception e){
-                        e.printStackTrace();
-                    }
-                });
-                fakeThread.start();
-
-                Thread fakeCommandThread = new Thread(() -> {
-                    while (true) {
-                        try {
-                            excuteFakeCommand(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo());
-                        }catch (Exception e){
-                            e.printStackTrace();
-                        }
-                    }
-                });
-                fakeCommandThread.start();
-
+            NyShuttleThread shuttleThread = (NyShuttleThread) SlaveConnection.get(SlaveType.Shuttle, device.getDeviceNo());
+            if (shuttleThread == null) {
                 // init
-                ThreadHandler thread = new NyShuttleThread(device, redisUtil);;
+                NyShuttleThread thread = new NyShuttleThread(device, redisUtil);;
                 new Thread(thread).start();
                 SlaveConnection.put(SlaveType.Shuttle, device.getDeviceNo(), thread);
+            }else {
+                Socket socket = shuttleThread.getSocket();
+                if(socket == null){
+                    continue;
+                }
             }
+
+            log.info("{}:device is run,devices:{}", device.getDeviceNo(),JSON.toJSONString(fakeServerMap));
+
+            Thread fakeThread = new Thread(() -> {
+                log.info("{}:device is start handle client", device.getDeviceNo());
+                try {
+                    while (true) {
+                        if(fakeServerMap.containsKey(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo())){
+                            handleClient(device);
+                        }
+                    }
+                }catch (Exception e){
+                    e.printStackTrace();
+                }
+            });
+            fakeThread.start();
+            fakeThreadMap.put(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo(), fakeThread);
+
+            try {
+                Thread.sleep(2000);
+            }catch (Exception e){}
         }
     }
 
@@ -261,6 +361,12 @@
     }
 
     private void removeFake(DeviceConfig device) {
+        Thread thread = fakeThreadMap.get(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo());
+        if (thread != null) {
+            thread.interrupt();
+        }
+
+        fakeThreadMap.remove(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo());
         fakeServerMap.remove(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo());
         fakeStatusMap.remove(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo());
         fakeCommandMap.remove(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo());
@@ -302,7 +408,7 @@
             fakeCommandMap.put(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo(), fakeCommand);
             response = genereateFakeCommandResponse(requestId, taskId, requestType);
         } else if (requestType.equals("readState")) {
-            response = genereateFakeStatusResponse(requestId, fakeStatus);
+            response = genereateFakeStatusResponse(requestId, fakeStatus, device.getDeviceNo());
         }
 
         fakeStatusMap.put(String.valueOf(SlaveType.Shuttle) + device.getDeviceNo(), fakeStatus);
@@ -314,11 +420,11 @@
         }
     }
 
-    public String genereateFakeStatusResponse(Integer taskId, JSONObject fakeStatus) {
+    public String genereateFakeStatusResponse(Integer taskId, JSONObject fakeStatus, Integer deviceNo) {
         JSONObject result = new JSONObject();
         JSONObject response = new JSONObject();
         result.put("msgType", "responseMsg");
-        result.put("robotId", 5001);
+        result.put("robotId", deviceNo);
         result.put("response", response);
 
         JSONObject header = new JSONObject();
@@ -335,7 +441,10 @@
         body.put("point", fakeStatus.getString("currentCode"));
         body.put("powerPercent", fakeStatus.getString("batteryPower"));
         body.put("voltage", fakeStatus.getInteger("batteryVoltage"));
-        body.put("errCode", new ArrayList<Integer>(){{add(fakeStatus.getInteger("errorCode"));add(0);}});
+        body.put("errCode", new ArrayList<Integer>() {{
+            add(fakeStatus.getInteger("errorCode"));
+            add(0);
+        }});
         body.put("liftPosition", fakeStatus.getBoolean("hasLift") == true ? 2 : 1);
         body.put("loadState", fakeStatus.getBoolean("hasPallet") == true ? 1 : 0);
         body.put("runDir", fakeStatus.getString("runDirection"));

--
Gitblit v1.9.1