Blame view

src/main/java/com/objecteye/websocket/MyHandshakeInterceptor.java 1.16 KB
c83b5b39   Liu Haoyu   项目创建, 集成spring-se...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
  package com.objecteye.websocket;
  
  import org.springframework.http.server.ServerHttpRequest;
  import org.springframework.http.server.ServerHttpResponse;
  import org.springframework.http.server.ServletServerHttpRequest;
  import org.springframework.stereotype.Component;
  import org.springframework.web.socket.WebSocketHandler;
  import org.springframework.web.socket.server.HandshakeInterceptor;
  
  import java.util.Map;
  
  /**
   * @Author: lr
   * @Date: 2019/9/5 14:32
   * @Version 1.0
   * @Message:
   */
  @Component
  public class MyHandshakeInterceptor implements HandshakeInterceptor {
      @Override
      public boolean beforeHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Map<String, Object> map) throws Exception {
          String deviceId = ((ServletServerHttpRequest) serverHttpRequest).getServletRequest().getParameter("deviceId");
          if (deviceId != null) {
              map.put("deviceId", deviceId);
          }
          return true;
      }
  
      @Override
      public void afterHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Exception e) {
      }
  }