修改websocket代码

master
bynt 2 years ago
parent 1d3cb94e81
commit 818cf08000

@ -13,6 +13,7 @@ import com.baiye.modules.system.service.UserMessageService;
import com.baiye.modules.system.service.dto.UserMessageDto; import com.baiye.modules.system.service.dto.UserMessageDto;
import com.baiye.util.SpringContextHolder; import com.baiye.util.SpringContextHolder;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import javax.annotation.PostConstruct; import javax.annotation.PostConstruct;
@ -228,9 +229,11 @@ public class WebSocketServer {
public static Long getUserId(Session session) { public static Long getUserId(Session session) {
String queryString = session.getQueryString(); String queryString = session.getQueryString();
if (queryString != null) { if (queryString != null &&
return UserTokenManager.getUserId StringUtils.isNotBlank(StringUtils.substring(queryString, DefaultNumberConstants.FOURTEEN_NUMBER))) {
(queryString.substring(DefaultNumberConstants.FOURTEEN_NUMBER)); String substring = queryString.substring(DefaultNumberConstants.FOURTEEN_NUMBER);
log.info("=================== the sub string as {} ===================", substring);
return UserTokenManager.getUserId(substring);
} }
return null; return null;
} }

Loading…
Cancel
Save