这是一开始写的代码,在WebSocket服务类中注入了service。
@Component
@ServerEndpoint("/websocket/{orgCode}")
@Slf4j
public class WebSocketController {
@Autowired
INoticeService service;
这样写看上去和我们平时写的时候是一样的,没有错误。但是如果使用WebSocket服务类中的代码的话就会报空指针异常。例如:
@OnOpen
public void onOpen(Session session, @PathParam("orgCode") String orgCode) {
this.session = session;
WebSocketController.orgCode=orgCode;
String orgCode1=StringUtils.trimEven0(orgCode);
//加入set中
webSocketSet.add(this);
//在线数加1
addOnlineCount();
log.info("有新窗口开始监听"+",当前在线人数为" + getOnlineCount());
try {
List<NoticeResponse> list = ns.queryAllNotice(orgCode1);
} catch (IOException e) {
log.error("websocket IO异常");
}
}
@OnClose
public void onClose() {
//从set中删除
webSocketSet.remove(this);
//在线数减1
subOnlineCount();
log.info("有一连接关闭!当前在线人数为" + getOnlineCount());
}
/**
* 收到客户端消息后调用的方法
* @param message 客户端发送过来的消息
**/
@OnMessage
public void onMessage(String message, Session session) throws Exception {
String orgCode1=StringUtils.trimEven0(orgCode);
List<NoticeResponse> list = ns.queryAllNotice(orgCode1);
}
产生的原因:spring默认是单例的,而WebSocket是多对象的,也就是每次会产生不同的对象。
在初始化项目的时候,WebSocket就会产生一个对象,这时候就会注入service了,而当客户端与WebSocket服务端连接过后,又会产生一个新对象,而spring默认是单例,只会给一个相同的对象注入一次service,因此这时候的WebSocket新对象就不会再注入service了,再去调用该service中的方法的话也就会发生空指针异常。
解决方法
1、通过ApplicationContext中的getBean方法来获取
@Component
public class ApplicationHelper implements ApplicationContextAware {
private static ApplicationContext applicationContext;
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
ApplicationHelper.applicationContext = applicationContext;
}
public static ApplicationContext getApplicationContext() {
return applicationContext;
}
public static Object getBean(String beanName) {
return applicationContext.getBean(beanName);
}
}
在WebSocket服务类中调用getBean方法即可
@Component
@ServerEndpoint("/websocket/{orgCode}")
@Slf4j
public class WebSocketController {
private NoticeService ns=(NoticeService) ApplicationHelper.getBean("noticeService");
}
2、将service在类中写成static,通过给类的属性赋值的形式注入
@Component
@ServerEndpoint("/websocket/{orgCode}")
@Slf4j
public class WebSocketController {
private static INoticeService ns;
@Autowired
public void setINoticeService(INoticeService noticeService) {
WebSocketController.ns = noticeService;
}