firsh

leutu

2024-05-20 a4e99a6a39284643fdf2b04dbe55212e5954973c

 src/main/java/com/skyline/electricity/controller/WebSocketServer.java

@@ -17,12 +17,12 @@
    private static Logger log;
    private static final AtomicInteger OnlineCount;
    private static CopyOnWriteArraySet<Session> SessionSet;
    

    @PostConstruct
    public void init() {
        System.out.println("websocket 加载");
       // System.out.println("websocket 加载");
    }
    

    @OnOpen
    public void onOpen(final Session session) {
        WebSocketServer.SessionSet.add(session);
@@ -30,26 +30,26 @@
        WebSocketServer.log.info("有连接加入，当前连接数为：{}", (Object)cnt);
        SendMessage(session, "连接成功");
    }
    

    @OnClose
    public void onClose(final Session session) {
        WebSocketServer.SessionSet.remove(session);
        final int cnt = WebSocketServer.OnlineCount.decrementAndGet();
        WebSocketServer.log.info("有连接关闭，当前连接数为：{}", (Object)cnt);
    }
    

    @OnMessage
    public void onMessage(final String message, final Session session) {
        WebSocketServer.log.info("来自客户端的消息：{}", (Object)message);
        SendMessage(session, "收到消息，消息内容：" + message);
    }
    

    @OnError
    public void onError(final Session session, final Throwable error) {
        WebSocketServer.log.error("发生错误：{}，Session ID： {}", (Object)error.getMessage(), (Object)session.getId());
        error.printStackTrace();
    }
    

    public static void SendMessage(final Session session, final String message) {
        try {
            session.getBasicRemote().sendText(String.format("%s", message));
@@ -59,7 +59,7 @@
            e.printStackTrace();
        }
    }
    

    public static void BroadCastInfo(final String message) throws IOException {

        for (final Session session : WebSocketServer.SessionSet) {
@@ -68,7 +68,7 @@
            }
        }
    }
    

    public static void SendMessage(final String message, final String sessionId) throws IOException {
        Session session = null;

@@ -85,7 +85,7 @@
            WebSocketServer.log.warn("没有找到你指定ID的会话：{}", (Object)sessionId);
        }
    }
    

    static {
        WebSocketServer.log = LoggerFactory.getLogger((Class)WebSocketServer.class);
        OnlineCount = new AtomicInteger(0);