huang
2025-04-15 ef714be504f98f6b9549b134148a18d416a9dcb0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
package com.mes.yield.service.impl;
 
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.mes.yield.entity.Yield;
import com.mes.yield.mapper.YieldMapper;
import com.mes.yield.service.YieldService;
import com.mes.tools.WebSocketServer;
import cn.hutool.json.JSONObject;
import org.springframework.stereotype.Service;
 
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
 
@Service
public class YieldServiceImpl extends ServiceImpl<YieldMapper, Yield> implements YieldService {
 
    @Override
    public void notifyYieldUpdate(Yield yield) {
        JSONObject message = new JSONObject();
        message.set("type", "yield_update");
        message.set("data", yield);
 
        ArrayList<WebSocketServer> servers = WebSocketServer.sessionMap.get("yield");
        if (servers != null) {
            for (WebSocketServer server : servers) {
                server.sendMessage(message.toString());
            }
        }
    }