增加 spring 定时任务用例

spring/spring-websocket/src/main/java/com/heibaiying/websocket/CustomHandshakeInterceptor.java

@@ -0,0 +1,29 @@
+package com.heibaiying.websocket;
+
+import org.springframework.http.server.ServerHttpRequest;
+import org.springframework.http.server.ServerHttpResponse;
+import org.springframework.web.socket.WebSocketHandler;
+import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;
+
+import java.net.InetSocketAddress;
+import java.util.Map;
+
+/**
+ * @author : heibaiying
+ * @description : 握手拦截器
+ */
+public class CustomHandshakeInterceptor extends HttpSessionHandshakeInterceptor {
+
+    /*@Override
+    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
+        InetSocketAddress remoteAddress = request.getRemoteAddress();
+        System.out.println(remoteAddress);
+        return true;
+    }*/
+
+    @Override
+    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
+        return super.beforeHandshake(request, response, wsHandler, attributes);
+    }
+}
+

spring/spring-websocket/src/main/java/com/heibaiying/websocket/CustomerHandler.java

@@ -0,0 +1,21 @@
+package com.heibaiying.websocket;
+
+import org.springframework.web.socket.TextMessage;
+import org.springframework.web.socket.WebSocketSession;
+import org.springframework.web.socket.handler.TextWebSocketHandler;
+
+import java.time.LocalDateTime;
+
+/**
+ * @author : heibaiying
+ * @description : 自定义消息处理器
+ */
+public class CustomerHandler extends TextWebSocketHandler {
+
+    @Override
+    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
+        String stringMessage = new String(message.asBytes());
+        System.out.println("服务端收到消息：" + stringMessage);
+        session.sendMessage(new TextMessage(stringMessage+LocalDateTime.now()));
+    }
+}