在websocket中使用@ServerEndpoint无法注入@Autowired、@Value
问题分析
Spring管理采用单例模式(singleton),而 WebSocket 是多对象的,即每个客户端对应后台的一个 WebSocket 对象,也可以理解成 new 了一个 WebSocket,这样当然是不能获得自动注入的对象了,因为这两者刚好冲突。
@Autowired 注解注入对象操作是在启动时执行的,而不是在使用时,而 WebSocket 是只有连接使用时才实例化对象,且有多个连接就有多个对象。
所以我们可以得出结论,这个 Service 根本就没有注入到 WebSocket 当中。
使用static静态变量
读取nacos的配置,通过Environment来获取,必须使用static,将需要注入的 Service 改为静态,让它属于当前类,然后通过 set方法进行注入即可解决。
private static Environment environment;
@Autowired
private void setEnvironment(Environment environment){
WebSocketServer.environment = environment;
}
在连接websocket的时候获取nacos的字段值
@OnOpen
public void onOpen(Session session, @PathParam(value = "userId") String userId) {
try {
token = environment.getProperty("platform.login.token");
String stationIDStr = environment.getProperty("platform.station.id");
if (StrUtil.isEmptyIfStr(stationIDStr)){
stationId = 1;
}else {
stationId = Integer.valueOf(stationIDStr);
}
this.session = session;
this.userId = userId;
webSockets.add(this);
sessionPool.put(userId, session);
log.info("[websocket消息]有新的连接,总数为:" + webSockets.size());
} catch (Exception e) {
}
}