springboot整合WebSocket遇到的问题(注入service为null)

这是一开始写的代码,在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;
    }

参考链接

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值