package cn.org.hentai.jtt1078.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.HandshakeInterceptor;
|
|
import java.util.Map;
|
|
public class CustomHandshakeInterceptor implements HandshakeInterceptor {
|
|
@Override
|
public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response,
|
WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
|
//从路径中获取userId
|
String path = request.getURI().getPath();
|
String[] segments = path.split("/");
|
if (segments.length > 2) {
|
String sim = segments[segments.length - 1];
|
attributes.put("sim", sim);
|
}
|
|
return true;
|
}
|
|
@Override
|
public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response,
|
WebSocketHandler wsHandler, Exception ex) {
|
}
|
}
|