消息通知系统

src/main/java/com/ai/da/common/websocket/NotificationConnection.java

@@ -0,0 +1,43 @@
+package com.ai.da.common.websocket;
+
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+
+import javax.websocket.*;
+import javax.websocket.server.ServerEndpoint;
+import java.io.IOException;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+
+@ServerEndpoint(value = "/notification")
+@Component
+@Slf4j
+public class NotificationConnection {
+
+    static Map<String,Session> sessionMap = new ConcurrentHashMap<>();
+    //连接建立时执行的操作
+    @OnOpen
+    public void onOpen(Session session){
+        sessionMap.put(session.getId(),session);
+        log.info("websocket is open");
+    }
+    //收到了客户端消息执行的操作
+    @OnMessage
+    public void onMessage(String text){
+        log.info("收到了一条消息："+text);
+//        return "已收到你的消息";
+    }
+    //连接关闭的时候执行的操作
+    @OnClose
+    public void onClose(Session session){
+        sessionMap.remove(session.getId());
+        log.info("websocket is close");
+    }
+
+    public void sendMsg(String message) throws IOException {
+        for(String key:sessionMap.keySet()){
+            sessionMap.get(key).getBasicRemote().sendText(message);
+        }
+    }
+}

src/main/java/com/ai/da/common/websocket/config/WebSocketConfig.java

@@ -0,0 +1,18 @@
+package com.ai.da.common.websocket.config;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.socket.server.standard.ServerEndpointExporter;
+
+/**
+ * Configuration of WebSocket
+ *
+ * @author db1995
+ */
+@Configuration
+public class WebSocketConfig {
+    @Bean
+    public ServerEndpointExporter serverEndpointExporter() {
+        return new ServerEndpointExporter();
+    }
+}