From bd6f16c7a6695606d7ebb5baa56eefe65b603cf2 Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期二, 11 六月 2024 16:48:01 +0800 Subject: [PATCH] # --- src/core/warehouse.jsx | 72 +++++++++++++++++++++++++----------- 1 files changed, 50 insertions(+), 22 deletions(-) diff --git a/src/core/warehouse.jsx b/src/core/warehouse.jsx index 23eda47..b37c8ae 100644 --- a/src/core/warehouse.jsx +++ b/src/core/warehouse.jsx @@ -1,6 +1,7 @@ import { useState, useEffect, useRef, useMemo } from 'react'; import { useFrame } from '@react-three/fiber'; import * as THREE from 'three'; +import WebSocketClient from './websocket' import Tunnel from '../components/tunnel'; import Area from '../components/area'; import Shelf from '../components/shelf'; @@ -12,32 +13,59 @@ import shelfData from '@/assets/data/shelf'; import agvRealDataList from '@/assets/data/agv'; import { getBoxData } from '../api/box'; +import { getAgvData } from '../api/agv'; 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([]); 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; + } + } + } + const timer = setInterval(() => { getBoxData().then(res => { + // res.push({ + // locNo: "A100301001", + // row: 3, + // bay: 1, + // lev: 1, + // locSts: "STOCK", + // }) + // console.log(res); setBoxData(res); }) - - const agvRealData = agvRealDataList[index]; - if (agvRealData) { - setAgvData(agvRealData); - } - index++; - }, INTERVAL_TIME * 1.1); + }, INTERVAL_TIME); return () => { - clearInterval(timer); + if (timer) { + clearInterval(timer); + } + + if (websocket) { + websocket.onMessage = (data) => { } + websocket.close(); + } } }, []) @@ -53,17 +81,17 @@ return shelfData.map((data, index) => <Shelf key={index} {...data} />) }, []); - const shelfEl1 = useMemo(() => { - return shelfData.slice(0, 1).map((data, index) => <Shelf key={index} {...data} />) - }, []); - const boxEl = useMemo(() => { 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]); useFrame((state, delta) => { @@ -76,8 +104,8 @@ {tunnelEl} {areaEl} {shelfEl} - {/* {shelfEl1} */} - {agvEl} + {agv1El} + {agv2El} {boxEl} </group> </> -- Gitblit v1.9.1