有如下代码, 代码是从另外的博主那里改一改得到的,因为注入了对象,所以不希望每次向Map都保存当前的websocektServer ,就改为了只保存session ,后来发现有些地方需要用到userId ,心想糟了,这不得是null啊,结果却意外的发现,能正常执行;
debug时发现每一次连接创建时的this当前对象内存地址都不一样,而关闭时是一样的。。。 由于基础不牢,一下无法理解发生了什么。创建了一个Controller测试了一下发现,每次请求进入时当前bean又是相同的地址;
我最后的临时理解就是,含有@OnOpen 注解的bean 会变成多例模式 ,每一次连接就是一个独立的对象, 代码中的map容器 保存会话的初衷并非给一个无状态的bean通过usrId找到自己使用的,而是给其他连接的对象调用查询指定用户会话使用的; 每一次连接进入时都会有特定的机制 匹配对应的对象 ,那样无论是执行@OnClose还是@OnMessage时,this都会是创建连接时的那个对象,并没有必要通过map拿会话,直接使用成员变量session属性即可; 原博主的代码中成员变量有一个session 属性,被我删掉了,后来改成了根据userid从map 中拿,原博主的方法没有错,当需要向自身 响应消息的时候,直接使用session 进行响应即可。因为在open 时就对这个属性进行了赋值 ,同时赋值的还有userid。当执行close|message 操作时由于匹配了对应的bean ,所以属性是有值的可以直接执行,我这里就直接不用了,通过userId匹配算了,反正也只有在连接成功和权限不足的时候会使用session直接响应消息,多数时刻是从map中get其他用户的session进行响应
想法: 有没有那么一种可能。我不通过会话map进行取 其他用户的会话,我直接从ioc中拿到其他用户的bean,使用他们的session属性进行消息响应,当前,前提是要加上原博主的session成员属性
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.util.concurrent.ConcurrentHashMap;
@ServerEndpoint("/websocket/infoCar/{token}")
@Component
@Lazy
public class WebSocketServer {
static Log log = LogFactory.get(WebSocketServer.class);
private static int onlineCount = 0;
private static ConcurrentHashMap<String, Session> webSocketMap = new ConcurrentHashMap<>();
private String userId ;
private final WebSocketService webSocketService = SpringUtil.getBean(WebSocketService.class);
private final CheckToken checkToken = SpringUtil.getBean(CheckToken.class);
@OnOpen
public void onOpen(Session session, @PathParam("token") String token) throws IOException {
User user = checkToken.queryUserByToken(token);
if (StringUtils.isBlank(token) || ObjectUtils.isEmpty(user)) {
session.getAsyncRemote().sendText(JSONObject.toJSONString(new WebSocketMsgView(false, CodeEnum.TOKEN_INVALID.getCode()
,WsRespTypeEnum.TOKEN_INVALID.getType(),WsRespTypeEnum.TOKEN_INVALID.getMsg())));
throw new RuntimeException("websocket通讯中token无效");
}
this.userId = String.valueOf(user.getId());
log.info("连接成功,用户:{}", userId);
System.out.println();
if (webSocketMap.containsKey(userId)) {
webSocketMap.remove(userId);
webSocketMap.put(userId, session);
} else {
webSocketMap.put(userId, session);
addOnlineCount();
}
log.info("用户连接:" + userId + ",当前在线人数为:" + getOnlineCount());
String msgJson = JSONObject.toJSONString(new WebSocketMsgView(true, 20000,
WsRespTypeEnum.OPEN_SUCCESS.getType(), WsRespTypeEnum.OPEN_SUCCESS.getMsg()));
try {
sendToSession(session,msgJson);
} catch (IOException e) {
log.error("用户:" + userId + ",网络异常!!!!!!");
}
}
@OnClose
public void onClose() {
log.info("连接关闭");
if (webSocketMap.containsKey(userId)) {
webSocketMap.remove(userId);
subOnlineCount();
}
log.info("用户退出:" + userId + ",当前在线人数为:" + getOnlineCount());
}
@OnMessage
public void onMessage(String message, Session session,@PathParam("token") String token) throws IOException {
log.info("用户消息:" + userId + ",报文:" + message);
if (StringUtils.isNotBlank(message)) {
try {
WebSocketParam webSocketParam = JSON.parseObject(message).toJavaObject(WebSocketParam.class);
User user = checkToken.queryUserByToken(webSocketParam.getToken());
WsRequest wsRequest = webSocketService.wsController(user, webSocketParam);
String recUser = wsRequest.getRecUser().toString();
String jsonResult = JSONObject.toJSONString(wsRequest.getWebSocketMsgView());
if (StringUtils.isNotBlank(recUser) && webSocketMap.containsKey(recUser)) {
sendToSession( webSocketMap.get(recUser),jsonResult);
} else {
sendInfo(userId,JSON.toJSONString(webSocketService.userUnOnline(Long.parseLong(userId)).getWebSocketMsgView()));
log.error("请求的userId:" + recUser + "不在该服务器上");
}
} catch (JSONException e) {
WebSocketMsgView webSocketMsgView = webSocketService.badParam(Long.parseLong(userId)).getWebSocketMsgView();
sendInfo(userId,JSON.toJSONString(webSocketMsgView));
log.error("报文解析异常,报文:{},会话:{},异常:",message,session,e);
} catch (Exception e){
log.error("websocketError:",e);
}
}
}
@OnError
public void onError(Session session, Throwable error) {
log.error("用户错误:" + this.userId + ",原因:" + error.getMessage());
error.printStackTrace();
}
private void sendToSession(Session session,String message) throws IOException {
session.getBasicRemote().sendText(message);
}
public void sendInfo(@PathParam("userId") String userId,String message) throws IOException {
log.info("发送消息到:" + userId + ",报文:" + message);
if (StringUtils.isNotBlank(userId) && webSocketMap.containsKey(userId)) {
sendToSession(webSocketMap.get(userId),message);
} else {
log.error("用户" + userId + ",不在线!");
}
}
public static synchronized int getOnlineCount() {
return onlineCount;
}
public static synchronized void addOnlineCount() {
WebSocketServer.onlineCount++;
}
public static synchronized void subOnlineCount() {
WebSocketServer.onlineCount--;
}
}
|