package com.mes.plannedAmount.service.impl;

import cn.hutool.json.JSONObject;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.mes.plannedAmount.entity.PlannedAmount;
import com.mes.plannedAmount.mapper.PlannedAmountMapper;
import com.mes.plannedAmount.service.PlannedAmountService;
import com.mes.websocket.WebSocketServer;
import org.springframework.stereotype.Service;

import java.util.List;

@Service
public class PlannedAmountImpl extends ServiceImpl<PlannedAmountMapper, PlannedAmount> implements PlannedAmountService {

    @Override
    public void notifyPlannedUpdate(PlannedAmount value) {
        JSONObject message = new JSONObject();
        message.set("type", "planned_update");
        message.set("data", value);

        List<WebSocketServer> servers = WebSocketServer.SESSIONMAP.get("value");
        if (servers != null) {
            for (WebSocketServer server : servers) {
                server.sendToWeb("",message.toString());
            }
        }
    }
}