#

luxiaotao1123

2024-06-17 619d7232bb601224ed4a442f01ef9768f9e96f40

 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';
@@ -19,30 +20,52 @@

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);
            })

            getAgvData().then(res => {
                // setAgvData(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();
            }
        }
    }, [])

@@ -58,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) => {
@@ -81,8 +104,8 @@
                {tunnelEl}
                {areaEl}
                {shelfEl}
                {/* {shelfEl1} */}
                {agvEl}
                {agv1El}
                {agv2El}
                {boxEl}
            </group>
        </>