WebSocket 集成

src/main/java/com/rymcu/vertical/config/ShiroConfig.java

@@ -51,6 +51,7 @@ public class ShiroConfig implements EnvironmentAware {
         filterChainDefinitionMap.put("/login", "anon");
 
         filterChainDefinitionMap.put("/api/**", "anon");
+        filterChainDefinitionMap.put("/ws/**", "anon");
         filterChainDefinitionMap.put("/**", "auth");
         //配置shiro默认登录界面地址，前后端分离中登录界面跳转应由前端路由控制，后台仅返回json数据
         shiroFilterFactoryBean.setLoginUrl("/login");

src/main/java/com/rymcu/vertical/config/WebSocketConfigurer.java

@@ -1,17 +0,0 @@
-package com.rymcu.vertical.config;
-
-import org.springframework.context.annotation.Bean;
-import org.springframework.context.annotation.Configuration;
-import org.springframework.web.socket.server.standard.ServerEndpointExporter;
-
-/**
- * WebSocket 配置类
- * @author ronger
- */
-@Configuration
-public class WebSocketConfigurer {
-    @Bean
-    public ServerEndpointExporter serverEndpointExporter() {
-        return new ServerEndpointExporter();
-    }
-}

src/main/java/com/rymcu/vertical/config/WebSocketStompConfig.java

@@ -0,0 +1,40 @@
+package com.rymcu.vertical.config;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.messaging.simp.config.MessageBrokerRegistry;
+import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
+import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
+import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
+
+/**
+ * @author ronger
+ */
+@Configuration
+@EnableWebSocketMessageBroker
+public class WebSocketStompConfig implements WebSocketMessageBrokerConfigurer {
+    /**
+     * 注册stomp端点
+     * @param registry
+     */
+    @Override
+    public void registerStompEndpoints(StompEndpointRegistry registry) {
+
+        // 允许使用socketJs方式访问 即可通过http://IP:PORT/ws来和服务端websocket连接
+        registry.addEndpoint("/ws").setAllowedOrigins("*").withSockJS();
+    }
+
+    /**
+     * 配置信息代理
+     * @param registry
+     */
+    @Override
+    public void configureMessageBroker(MessageBrokerRegistry registry) {
+
+        // 订阅Broker名称 user点对点 topic广播即群发
+        registry.enableSimpleBroker("/user","/public");
+        // 全局(客户端)使用的消息前缀
+        registry.setApplicationDestinationPrefixes("/app");
+        // 点对点使用的前缀 无需配置 默认/user
+        registry.setUserDestinationPrefix("/user");
+    }
+}

src/main/java/com/rymcu/vertical/dto/TokenUser.java

@@ -8,6 +8,8 @@ import lombok.Data;
 @Data
 public class TokenUser {
 
+    private Integer idUser;
+
     private String account;
 
     private String nickname;

src/main/java/com/rymcu/vertical/web/api/common/WebSocketController.java

@@ -0,0 +1,59 @@
+package com.rymcu.vertical.web.api.common;
+
+import com.alibaba.fastjson.JSONObject;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.messaging.handler.annotation.DestinationVariable;
+import org.springframework.messaging.handler.annotation.MessageMapping;
+import org.springframework.messaging.handler.annotation.SendTo;
+import org.springframework.messaging.simp.SimpMessagingTemplate;
+import org.springframework.messaging.simp.annotation.SendToUser;
+import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
+import org.springframework.stereotype.Controller;
+
+import java.util.HashMap;
+
+/**
+ * @author ronger
+ */
+@Controller
+public class WebSocketController {
+
+    @Autowired
+    private SimpMessagingTemplate template;
+
+    @MessageMapping("/sendMessage")
+    @SendTo("/public/greetings")
+    public void sendMessage(JSONObject message, StompHeaderAccessor headerAccessor){
+        this.template.convertAndSend("/public/greetings",message);
+    }
+
+    @MessageMapping("/message")
+    @SendToUser("/message")
+    public void message(JSONObject message){
+        String type = message.get("type").toString();
+        HashMap res = (HashMap) message.get("data");
+        HashMap mine = (HashMap) res.get("mine");
+        HashMap to = (HashMap) res.get("to");
+        System.out.println(to.get("type"));
+        boolean flag = to.get("type").equals("friend")?true:false;
+        String id = to.get("id").toString();
+        HashMap map = new HashMap();
+        map.put("id",mine.get("id"));
+        map.put("avatar",mine.get("avatar"));
+        map.put("formid",mine.get("id"));
+        map.put("username",mine.get("username"));
+        map.put("type",to.get("type"));
+        map.put("content",mine.get("content"));
+        map.put("mine",false);
+        map.put("cid",0);
+        map.put("timestamp","");
+        JSONObject json = new JSONObject();
+        json.put("type",type);
+        json.put("data",map);
+        if(flag){
+            this.template.convertAndSendToUser(id,"/message",json);
+        }else{
+            this.template.convertAndSendToUser(id,"/message",json);
+        }
+    }
+}

src/main/java/com/rymcu/vertical/web/api/common/WebSocketServer.java

@@ -1,130 +0,0 @@
-package com.rymcu.vertical.web.api.common;
-
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-import org.springframework.stereotype.Component;
-
-import javax.websocket.*;
-import javax.websocket.server.PathParam;
-import javax.websocket.server.ServerEndpoint;
-import java.io.IOException;
-import java.util.concurrent.CopyOnWriteArraySet;
-
-/**
- * @author ronger
- */
-@ServerEndpoint("/api/v1/websocket/{sid}")
-@Component
-public class WebSocketServer {
-    static final Logger log= LoggerFactory.getLogger(WebSocketServer.class);
-    /**
-     * 静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
-     * */
-    private static int onlineCount = 0;
-    /**
-     * concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。
-     */
-    private static CopyOnWriteArraySet<WebSocketServer> webSocketSet = new CopyOnWriteArraySet<WebSocketServer>();
-
-
-    /**
-     * 与某个客户端的连接会话，需要通过它来给客户端发送数据
-     */
-    private Session session;
-
-    /**
-     * 接收sid
-     * */
-    private String sid = "";
-    /**
-     * 连接建立成功调用的方法*/
-    @OnOpen
-    public void onOpen(Session session,@PathParam("sid") String sid) {
-        this.session = session;
-        webSocketSet.add(this);
-        addOnlineCount();
-        log.info("有新窗口开始监听:" + sid + ",当前在线人数为" + getOnlineCount());
-        this.sid=sid;
-        try {
-            sendMessage("连接成功");
-        } catch (IOException e) {
-            log.error("websocket IO异常");
-        }
-    }
-
-    /**
-     * 连接关闭调用的方法
-     */
-    @OnClose
-    public void onClose() {
-        webSocketSet.remove(this);
-        subOnlineCount();
-        log.info("有一连接关闭！当前在线人数为" + getOnlineCount());
-    }
-
-    /**
-     * 收到客户端消息后调用的方法
-     *
-     * @param message 客户端发送过来的消息*/
-    @OnMessage
-    public void onMessage(String message, Session session) {
-        log.info("收到来自窗口" + sid + "的信息:" + message);
-        //群发消息
-        for (WebSocketServer item : webSocketSet) {
-            try {
-                item.sendMessage(message);
-            } catch (IOException e) {
-                e.printStackTrace();
-            }
-        }
-    }
-
-    /**
-     *
-     * @param session
-     * @param error
-     */
-    @OnError
-    public void onError(Session session, Throwable error) {
-        log.error("发生错误");
-        error.printStackTrace();
-    }
-    /**
-     * 实现服务器主动推送
-     */
-    public void sendMessage(String message) throws IOException {
-        this.session.getBasicRemote().sendText(message);
-    }
-
-
-    /**
-     * 群发自定义消息
-     * */
-    public static void sendInfo(String message,@PathParam("sid") String sid) throws IOException {
-        log.info("推送消息到窗口"+sid+"，推送内容:"+message);
-        for (WebSocketServer item : webSocketSet) {
-            try {
-                //这里可以设定只推送给这个sid的，为null则全部推送
-                if(sid==null) {
-                    item.sendMessage(message);
-                }else if(item.sid.equals(sid)){
-                    item.sendMessage(message);
-                }
-            } catch (IOException e) {
-                continue;
-            }
-        }
-    }
-
-    public static synchronized int getOnlineCount() {
-        return onlineCount;
-    }
-
-    public static synchronized void addOnlineCount() {
-        WebSocketServer.onlineCount++;
-    }
-
-    public static synchronized void subOnlineCount() {
-        WebSocketServer.onlineCount--;
-    }
-}