test WebSocketHandlerDecorator

Author: netbuffer

src/main/java/cn/netbuffer/springboot/websocket/demo/config/SpringWebSocketConfig.java

@@ -2,6 +2,8 @@
 
 import cn.netbuffer.springboot.websocket.demo.interceptor.RbacHandshakeInterceptor;
 import cn.netbuffer.springboot.websocket.demo.websocket.handler.MessageHandler;
+import cn.netbuffer.springboot.websocket.demo.websocket.handler.PersistenceToDatabaseWebSocketHandlerDecorator;
+import cn.netbuffer.springboot.websocket.demo.websocket.handler.PersistenceToElasticsearchWebSocketHandlerDecorator;
 import cn.netbuffer.springboot.websocket.demo.websocket.handler.TextHandler;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.context.annotation.Bean;
@@ -32,7 +34,7 @@ public WebSocketHandler buildTextHandler() {
 
     @Bean
     public WebSocketHandler buildMessageHandler() {
-        return new MessageHandler();
+        return new PersistenceToElasticsearchWebSocketHandlerDecorator(new PersistenceToDatabaseWebSocketHandlerDecorator(new MessageHandler()));
     }
 
 }

src/main/java/cn/netbuffer/springboot/websocket/demo/interceptor/RbacHandshakeInterceptor.java

@@ -22,8 +22,8 @@ public boolean beforeHandshake(ServerHttpRequest serverHttpRequest, ServerHttpRe
         log.debug("beforeHandshake {} to {} websocket request", serverHttpRequest.getMethodValue(), serverHttpRequest.getURI());
         log.debug("serverHttpRequest.getHeaders()={}", serverHttpRequest.getHeaders());
 //        boolean result = handleByCookie(serverHttpRequest);
-//        boolean result = handleByQueryString(serverHttpRequest);
-        boolean result = handleByHttpSession(serverHttpRequest);
+        boolean result = handleByQueryString(serverHttpRequest);
+//        boolean result = handleByHttpSession(serverHttpRequest);
         if (!result) {
             serverHttpResponse.setStatusCode(HttpStatus.FORBIDDEN);
         }

src/main/java/cn/netbuffer/springboot/websocket/demo/websocket/handler/PersistenceToDatabaseWebSocketHandlerDecorator.java

@@ -0,0 +1,27 @@
+package cn.netbuffer.springboot.websocket.demo.websocket.handler;
+
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.web.socket.*;
+import org.springframework.web.socket.handler.WebSocketHandlerDecorator;
+
+@Slf4j
+public class PersistenceToDatabaseWebSocketHandlerDecorator extends WebSocketHandlerDecorator {
+
+    public PersistenceToDatabaseWebSocketHandlerDecorator(WebSocketHandler delegate) {
+        super(delegate);
+    }
+
+    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) {
+        try {
+            this.getDelegate().handleMessage(session, message);
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        if (message instanceof TextMessage) {
+            log.debug("save text message[{}] to database", message.getPayload());
+            //todo 这里可以基于任意的jdbc框架来写库
+        } else if (message instanceof BinaryMessage) {
+            log.debug("save binary message to database");
+        }
+    }
+}

src/main/java/cn/netbuffer/springboot/websocket/demo/websocket/handler/PersistenceToElasticsearchWebSocketHandlerDecorator.java

@@ -0,0 +1,27 @@
+package cn.netbuffer.springboot.websocket.demo.websocket.handler;
+
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.web.socket.*;
+import org.springframework.web.socket.handler.WebSocketHandlerDecorator;
+
+@Slf4j
+public class PersistenceToElasticsearchWebSocketHandlerDecorator extends WebSocketHandlerDecorator {
+
+    public PersistenceToElasticsearchWebSocketHandlerDecorator(WebSocketHandler delegate) {
+        super(delegate);
+    }
+
+    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) {
+        try {
+            this.getDelegate().handleMessage(session, message);
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        if (message instanceof TextMessage) {
+            log.debug("save text message[{}] to elasticsearch", message.getPayload());
+            //todo 这里可以基于任意的elasticsearch框架来写入搜索引擎
+        } else if (message instanceof BinaryMessage) {
+            log.debug("save binary message to elasticsearch");
+        }
+    }
+}