mul-agent-virtual-simulation-system/train-server/src/main/java/top/haijunit/train/utils/TrainWebSocketUtils.java

package top.haijunit.train.utils;

import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.experimental.UtilityClass;
import lombok.extern.slf4j.Slf4j;
import top.haijunit.common.utils.JsonUtils;
import top.haijunit.train.device.domain.CommandEnum;
import top.haijunit.websocket.utils.WebSocketUtils;

/**
 * @author zhanghaijun
 * @date 2023/11/26 21:31
 * @description [一句话描述该类的功能]
 */
@Slf4j
@UtilityClass
public class TrainWebSocketUtils {

    public static void sendMessage(CommandEnum command, Object message) {
        WebSocketUtils.sendMessage(JsonUtils.toJsonString(new SocketMessage(command, message)));
    }

    public static void sendMessage(String trainNo, int state) {
        JSONObject object = JSONUtil.createObj();
        object.set(CommandEnum.TRAIN_OPEN_DOOR.getNameStr(), trainNo);
        object.set(CommandEnum.TRAIN_OPEN_DOOR.getStatusStr(), state);
        sendMessage(CommandEnum.TRAIN_OPEN_DOOR, object);
    }


    @Data
    @AllArgsConstructor
    public static class SocketMessage {
        private final CommandEnum cmd;
        private final Object data;
    }
}