From 9d71e24e7982f6915e5f832512925bee9b245f35 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期二, 20 八月 2024 16:13:26 +0800
Subject: [PATCH] #

---
 src/core/warehouse.jsx |   71 +++++++++++++++++++++++------------
 1 files changed, 47 insertions(+), 24 deletions(-)

diff --git a/src/core/warehouse.jsx b/src/core/warehouse.jsx
index c3117e7..c9b74c3 100644
--- a/src/core/warehouse.jsx
+++ b/src/core/warehouse.jsx
@@ -7,6 +7,7 @@
 import Shelf from '../components/shelf';
 import Box from '../components/box';
 import Agv from '../components/agv';
+import Path from '../components/path'
 
 import tunnelData from '@/assets/data/tunnel';
 import areaData from '@/assets/data/area';
@@ -14,19 +15,38 @@
 import agvRealDataList from '@/assets/data/agv';
 import { getBoxData } from '../api/box';
 import { getAgvData } from '../api/agv';
-import { INTERVAL_TIME, WEBSOCKET_BASE_URL, IP, PORT } from '@/config/setting'
+import { INTERVAL_TIME } from '@/config/setting'
 
 let index = 0;
 
 const Warehouse = (props) => {
 
-    const [agvData, setAgvData] = useState([
-        {}
-    ]);
-    const [boxData, setBoxData] = useState([])
+    const [agv1Data, setAgv1Data] = useState({});
+    const [agv2Data, setAgv2Data] = useState({});
+    const [boxData, setBoxData] = useState([]);
+    const [pathData, setPathData] = useState([]);
 
     useEffect(() => {
-
+        const websocket = new WebSocketClient('/ws/digitalTwin/websocket');
+        websocket.connect();
+        websocket.onMessage = (jsonStr) => {
+            const data = JSON.parse(jsonStr);
+            if (data.type === 'agv') {
+                switch (data.no) {
+                    case "1":
+                        setAgv1Data(data.obj);
+                        break;
+                    case "2":
+                        setAgv2Data(data.obj);
+                        break;
+                    default:
+                        break;
+                }
+            }
+            if (data.type === 'path') {
+                setPathData(data.obj);
+            }
+        }
 
         const timer = setInterval(() => {
             getBoxData().then(res => {
@@ -40,23 +60,17 @@
                 // console.log(res);
                 setBoxData(res);
             })
-
-            getAgvData().then(res => {
-                // console.log(JSON.stringify(res))
-                if (res && res.length > 0) {
-                    setAgvData(res);
-                }
-            })
-
-            const agvRealData = agvRealDataList[index];
-            if (agvRealData) {
-                // setAgvData(agvRealData);
-            }
-            index++;
         }, INTERVAL_TIME);
 
         return () => {
-            clearInterval(timer);
+            if (timer) {
+                clearInterval(timer);
+            }
+
+            if (websocket) {
+                websocket.onMessage = (data) => { }
+                websocket.close();
+            }
         }
     }, [])
 
@@ -76,10 +90,17 @@
         return boxData.map((data, idx) => <Box key={idx} {...data} />)
     }, [boxData]);
 
-    const agvEl = useMemo(() => {
-        return agvData.map((data, idx) => <Agv key={idx} {...data} />)
-    }, [agvData]);
+    const agv1El = useMemo(() => {
+        return <Agv key={1} {...agv1Data} />
+    }, [agv1Data]);
 
+    const agv2El = useMemo(() => {
+        return <Agv key={2} {...agv2Data} />
+    }, [agv2Data]);
+
+    const pathEl = useMemo(() => {
+        return pathData.map((data, idx) => <Path key={idx} list={data} />)
+    }, [pathData])
 
     useFrame((state, delta) => {
 
@@ -91,8 +112,10 @@
                 {tunnelEl}
                 {areaEl}
                 {shelfEl}
-                {agvEl}
+                {agv1El}
+                {agv2El}
                 {boxEl}
+                {pathEl}
             </group>
         </>
     )

--
Gitblit v1.9.1