|
@@ -1,7 +1,8 @@
|
|
|
package cn.com.ty.lift.push.websocket.config;
|
|
|
|
|
|
import cn.com.ty.lift.common.constants.ApiConstants;
|
|
|
-import cn.com.ty.lift.push.websocket.dao.entity.TimeyUser;
|
|
|
+import cn.com.ty.lift.common.constants.WebSocketConstants;
|
|
|
+import cn.com.ty.lift.common.model.TimeyUser;
|
|
|
import org.springframework.http.server.ServerHttpRequest;
|
|
|
import org.springframework.http.server.ServerHttpResponse;
|
|
|
import org.springframework.http.server.ServletServerHttpRequest;
|
|
@@ -12,36 +13,36 @@ import javax.servlet.http.HttpSession;
|
|
|
import java.util.Map;
|
|
|
|
|
|
public class WebSocketInterceptor implements HandshakeInterceptor {
|
|
|
-
|
|
|
- /**
|
|
|
- * 进入handler之前的拦截
|
|
|
- */
|
|
|
- @Override
|
|
|
- public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response,
|
|
|
- WebSocketHandler wsHandler, Map<String, Object> map) {
|
|
|
|
|
|
- if(request instanceof ServletServerHttpRequest) {
|
|
|
+ /**
|
|
|
+ * 进入handler之前的拦截
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response,
|
|
|
+ WebSocketHandler wsHandler, Map<String, Object> map) {
|
|
|
+
|
|
|
+ if (request instanceof ServletServerHttpRequest) {
|
|
|
ServletServerHttpRequest serverHttpRequest = (ServletServerHttpRequest) request;
|
|
|
- HttpSession session = serverHttpRequest.getServletRequest().getSession();
|
|
|
- TimeyUser timeyUser = (TimeyUser) session.getAttribute(ApiConstants.CURRENT_USER);
|
|
|
- if(timeyUser != null) {
|
|
|
- map.put(ApiConstants.CURRENT_USER_ID, timeyUser.getUserId());
|
|
|
+ HttpSession session = serverHttpRequest.getServletRequest().getSession();
|
|
|
+ TimeyUser timeyUser = (TimeyUser) session.getAttribute(WebSocketConstants.CURRENT_TIMEY_USER);
|
|
|
+ if (timeyUser != null) {
|
|
|
+ map.put(ApiConstants.CURRENT_USER_ID, timeyUser.getUserId());
|
|
|
}
|
|
|
- }
|
|
|
- return true;
|
|
|
- }
|
|
|
+ }
|
|
|
+ return true;
|
|
|
+ }
|
|
|
|
|
|
- @Override
|
|
|
- public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
|
|
|
- Exception exception) {
|
|
|
+ @Override
|
|
|
+ public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
|
|
|
+ Exception exception) {
|
|
|
ServletServerHttpRequest serverHttpRequest = (ServletServerHttpRequest) request;
|
|
|
- HttpSession session = serverHttpRequest.getServletRequest().getSession();
|
|
|
- TimeyUser timeyUser = (TimeyUser) session.getAttribute(ApiConstants.CURRENT_USER);
|
|
|
- if(timeyUser != null) {
|
|
|
- System.out.println("用户"+ timeyUser.getUserName() +"连接进入系统");
|
|
|
- } else {
|
|
|
- System.out.println("当前连接为空");
|
|
|
- }
|
|
|
- }
|
|
|
+ HttpSession session = serverHttpRequest.getServletRequest().getSession();
|
|
|
+ TimeyUser timeyUser = (TimeyUser) session.getAttribute(WebSocketConstants.CURRENT_TIMEY_USER);
|
|
|
+ if (timeyUser != null) {
|
|
|
+ System.out.println("用户" + timeyUser.getUserName() + "连接进入系统");
|
|
|
+ } else {
|
|
|
+ System.out.println("当前连接为空");
|
|
|
+ }
|
|
|
+ }
|
|
|
|
|
|
}
|