修改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.util.SpringContextHolder;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Component;
import javax.annotation.PostConstruct;
@ -228,9 +229,11 @@ public class WebSocketServer {
public static Long getUserId(Session session) {
String queryString = session.getQueryString();
if (queryString != null) {
return UserTokenManager.getUserId
(queryString.substring(DefaultNumberConstants.FOURTEEN_NUMBER));
if (queryString != null &&
StringUtils.isNotBlank(StringUtils.substring(queryString, DefaultNumberConstants.FOURTEEN_NUMBER))) {
String substring = queryString.substring(DefaultNumberConstants.FOURTEEN_NUMBER);
log.info("=================== the sub string as {} ===================", substring);
return UserTokenManager.getUserId(substring);
}
return null;
}

Loading…
Cancel
Save