nnkwrik 6 lat temu
rodzic
commit
deb45dc79c

+ 1 - 0
im-service/src/main/java/io/github/nnkwrik/imservice/redis/RedisClient.java

@@ -8,6 +8,7 @@ import java.util.List;
 
 /**
  * redis client
+ * 存储未读消息。key=ChatId,Value=List<WsMessage>
  *
  * @author nnkwrik
  * @date 18/12/06 15:08

+ 7 - 7
im-service/src/main/java/io/github/nnkwrik/imservice/websocket/ChatEndpoint.java

@@ -53,12 +53,12 @@ public class ChatEndpoint {
         String token = (String) config.getUserProperties().get(JWTUser.class.getName());
         JWTUser user = solveToken(token);
 
-//        if (user == null || !user.getOpenId().equals(openId)) {
-//            log.info("【websocket消息】token检验失败,拒绝连接, openId = [{}]", openId);
-//            rejectConnect(session);
-//            session.close();
-//            return;
-//        }
+        if (user == null || !user.getOpenId().equals(openId)) {
+            log.info("【websocket消息】token检验失败,拒绝连接, openId = [{}]", openId);
+            rejectConnect(session);
+            session.close();
+            return;
+        }
         sessionMap.put(openId, session);
 
         //发送未读消息数给新连接的客户端
@@ -69,7 +69,7 @@ public class ChatEndpoint {
         wsMessage.setSendTime(new Date());
         sendMessage(openId, Response.ok(wsMessage));
 
-//        log.info("【websocket消息】有新的连接, openId = [{}],用户昵称= [{}],未读消息数={}", openId, user.getNickName(), unreadCount);
+        log.info("【websocket消息】有新的连接, openId = [{}],用户昵称= [{}],未读消息数={}", openId, user.getNickName(), unreadCount);
     }
 
     /**