服务站大屏

src/main/java/com/care/bigscreen/websocket/BigScreenWebSocketEndpoint.java

@@ -1,11 +1,16 @@
 package com.care.bigscreen.websocket;
 
 
+import cn.hutool.json.JSONUtil;
+import com.care.common.vo.UserLogindConvertVO;
 import com.care.util.ExUtil;
 import com.care.util.JsonUtil;
+import com.care.util.JwtUtils;
 import com.care.util.Result;
+import io.jsonwebtoken.Claims;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import org.springframework.util.StringUtils;
 
@@ -24,6 +29,10 @@ import java.util.concurrent.CopyOnWriteArraySet;
 @Component
 public class BigScreenWebSocketEndpoint {
     private static final Logger logger = LoggerFactory.getLogger(BigScreenWebSocketEndpoint.class);
+
+    @Autowired
+    JwtUtils jwtUtils;
+
     //用来存放每个客户端对应的BigScreenWebSocketEndpoint对象
     private static CopyOnWriteArraySet<BigScreenWebSocketEndpoint> currentWebSocketSession = new CopyOnWriteArraySet<>();
 
@@ -31,8 +40,8 @@ public class BigScreenWebSocketEndpoint {
     private Session session;
     //用于关联当前客户端和websocket会话的关系
 
-    //登录名
-    private String username;
+    //电话作为登陆用户名
+    private String phone;
 
     //服务站ID
     private String stationId;
@@ -52,8 +61,8 @@ public class BigScreenWebSocketEndpoint {
                 closeSession(session, "-1", "连接失败,token为空");
                 return;
             }
-            //建立websocket和自动外呼坐席的关联关系
-            //从token中解析出坐席的信息
+            //建立websocket和登录用户、站点ID的关联关系
+            //从token中解析出登录用户的信息
             String[] keyValues = requestParamString.split("&");
             //取出token 字符串
             String ts = null;
@@ -65,25 +74,22 @@ public class BigScreenWebSocketEndpoint {
                     stationId = t.replaceAll("stationId=", "");
                 }
             }
-//            if (StringUtils.isEmpty(ts) || StringUtils.isEmpty(stationId)) {
-//                closeSession(session, "-1", "连接失败,token或stationId为空");
-//                return;
-//            }
-//
-//            //连接对象初始化
-//            Token token = null;
-//            try {
-//                token = TokenUtil.parseTokenStr(ts);
-//            } catch (Exception e) {
-//                logger.error("websocket会话建立异常:"+ ExUtil.exToDetail(e));
-//            }
-//            if(token == null){
-//                closeSession(session, "-1", "连接失败,token为空");
-//                return;
-//            }
-//
-//            logger.info("websocket会话建立:登录用户={},服务站ID={}登录！", token.getPayload().getUsername(), this.stationId);
-//            this.username = token.getPayload().getUsername();
+            if (StringUtils.isEmpty(ts) || StringUtils.isEmpty(stationId)) {
+                closeSession(session, "-1", "连接失败,token或stationId为空");
+                return;
+            }
+
+            Claims claims = jwtUtils.tokenParse(ts);
+            if (claims != null) {
+                UserLogindConvertVO vo = JSONUtil.toBean(claims.getSubject(), UserLogindConvertVO.class);
+                if (vo != null) {
+                    this.phone = vo.getPhone();
+                } else{
+                    closeSession(session, "-1", "连接失败,token解析失败");
+                    return;
+                }
+            }
+            logger.info("websocket会话建立:服务站ID={},登录用户={}登录！", this.stationId,this.phone);
             this.session = session;
             currentWebSocketSession.add(this);
         } catch (Exception e) {
@@ -96,12 +102,12 @@ public class BigScreenWebSocketEndpoint {
     public void onClose(Session session) {
         //移除连接
         currentWebSocketSession.remove(this);
-        logger.info("客户端断开连接,客户端id={},坐席={},退出自动外呼", session.getId(), this.username);
+        logger.info("客户端断开连接,客户端id={},服务站ID={},登录用户={},退出websocket连接", session.getId(), this.stationId, this.phone);
     }
 
     @OnError
     public void onError(Session session, Throwable error) {
-        logger.error("客户端id={},服务站ID={},登录用户={},连接异常信息={}.", session.getId(), this.stationId, this.username, error.getCause());
+        logger.error("客户端id={},服务站ID={},登录用户={},连接异常信息={}.", session.getId(), this.stationId, this.phone, error.getCause());
     }
 
     @OnMessage
@@ -121,7 +127,7 @@ public class BigScreenWebSocketEndpoint {
                 try {
                     session.getBasicRemote().sendText(message);
                 } catch (IOException e) {
-                    logger.info("给服务站ID={},登录用户={} 发送数据失败，客户端连接已经断开.", endpoint.getStationId(), endpoint.getUsername());
+                    logger.info("给服务站ID={},登录用户={} 发送数据失败，客户端连接已经断开.", endpoint.getStationId(), endpoint.getPhone());
                 }
             }
         }
@@ -142,7 +148,7 @@ public class BigScreenWebSocketEndpoint {
                 try {
                     session.getBasicRemote().sendText(msg);
                 } catch (IOException e) {
-                    logger.info("给服务站ID={}，登录用户={} 发送数据失败，客户端连接已经断开.", endpoint.getStationId(), endpoint.getUsername());
+                    logger.info("给服务站ID={}，登录用户={} 发送数据失败，客户端连接已经断开.", endpoint.getStationId(), endpoint.getPhone());
                 }
             }
         }
@@ -178,8 +184,8 @@ public class BigScreenWebSocketEndpoint {
         return session;
     }
 
-    public String getUsername() {
-        return username;
+    public String getPhone() {
+        return phone;
     }
 
     public String getStationId() {