ソースを参照

ws链接注册日志增加

jingyuanchao 1 年間 前
コミット
8255903833

+ 2 - 2
soc-modules/soc-modules-host/src/main/java/com/xunmei/host/server/service/impl/IotServerInfoServiceImpl.java

@@ -47,7 +47,7 @@ public class IotServerInfoServiceImpl extends ServiceImpl<IotServerInfoMapper, I
                 queryWrapper.last(Constants.LIMIT1);
                 IotServerInfo serverInfo = baseMapper.selectOne(queryWrapper);
                 if (serverInfo == null) {
-                    LogUtils.SOCKET_DAILY_HEALTH.error("未能根据registerCode查询到对应的token,registerCode:{}", registerCode);
+                    LogUtils.WEBSOCKET_MSG.error("未能根据registerCode查询到对应的token,registerCode:{}", registerCode);
                     return null;
                 }
                 if (ObjectUtil.isEmpty(serverInfo.getIotCode())) {
@@ -56,7 +56,7 @@ public class IotServerInfoServiceImpl extends ServiceImpl<IotServerInfoMapper, I
                 }
 
                 updateById(serverInfo);
-
+                LogUtils.WEBSOCKET_MSG.error("收到链接请求,已生成token,registerCode:{},token:{}", registerCode,serverInfo.getIotCode());
                 return serverInfo;
             }
         } catch (Exception e) {

+ 4 - 1
soc-modules/soc-modules-mediator/src/main/java/com/xunmei/mediator/websocket/handler/SocWebSocketHandler.java

@@ -56,7 +56,10 @@ public class SocWebSocketHandler extends AbstractWebSocketHandler {
      */
     @Override
     public void afterConnectionEstablished(WebSocketSession session) throws IOException {
-
+        if (1 == 1) {
+            LogUtils.WEBSOCKET_MSG.info("错误的注册地址,ip:{}", WebSocketUtils.getIp(session));
+            return;
+        }
         String registerCode = (String) session.getAttributes().get("registerCode");
         if (registerCode != null) {
             Map<String, Object> map = new HashMap<>();