Springboot项目WebSocket服务中不能注入(@Autowired)报null错误

需求介绍

Springboot中,Java服务端与浏览器客户端建立WebSocket连接,客户端向服务端发送信息后,@OnMessage,将信息转换为POJO对象,并调用Mabatis的Mapper层接口,将对象信息写入数据库。

报错信息

ERROR 23288 --- [nio-8080-exec-3] o.a.t.websocket.pojo.PojoEndpointBase    : No error handling configured for [com.evan.server.WebSocketServer] and the following error occurred

java.lang.NullPointerException: null

错误的代码

@ServerEndpoint("/websocketdemo/{cid}")
@Component
public class WebSocket {
	@Autowired
    private static ActionLogMapper actionLogMapper;

    @OnMessage
    public void onMessage(Session session, String string) {
        //JSON转换为POJO对象
        VideoMessage videoMessage = JSON.parseObject(string, VideoMessage.class);
        //...省略的对videoMessage的处理
        //转换为ActionLog对象,存入数据库
        SimpleDateFormat df = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss");
        String date = df.format(new Date());
		//对象转换
        ActionLog actionLog = new ActionLog();
        actionLog.setAction(videoMessage.getMsg());
        actionLog.setFace_id(videoMessage.getFaceId());
        actionLog.setTime(date);
        actionLog.setFigure_path("D://xx001.jpg");
        System.out.println(actionLog.toString());
        //存入数据库
        actionLogMapper.crateAction(actionLog);
    }
}

正确的代码

@ServerEndpoint("/websocketdemo/{cid}")
@Component
public class WebSocket {

    private static ActionLogMapper actionLogMapper;

    @Autowired
    public void setActionLogMapper(ActionLogMapper actionLogMapper) {
        WebSocket.actionLogMapper = actionLogMapper;
    }

    @OnMessage
    public void onMessage(Session session, String string) {
        //JSON转换为POJO对象
        VideoMessage videoMessage = JSON.parseObject(string, VideoMessage.class);
        //...省略的对videoMessage的处理
        //转换为ActionLog对象,存入数据库
        SimpleDateFormat df = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss");
        String date = df.format(new Date());
		//对象转换
        ActionLog actionLog = new ActionLog();
        actionLog.setAction(videoMessage.getMsg());
        actionLog.setFace_id(videoMessage.getFaceId());
        actionLog.setTime(date);
        actionLog.setFigure_path("D://xx001.jpg");
        System.out.println(actionLog.toString());
        //存入数据库
        actionLogMapper.crateAction(actionLog);
    }
}

一般的Controller层写法

上边的错误代码是仿照Controller层写的。Controller先注入Mapper层,接口收到请求后,调用Mapper层,将数据写入数据库。但是这样的模式在WebSocket中行不通,会出现上述的Bug。

@RestController
@RequestMapping("/actionlog")
public class ActionLogController {

    @Autowired
    ActionLogMapper actionLogMapper;

    @PostMapping("/create")
    public String createAction(...) {
        //获取系统时间
        SimpleDateFormat df = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss");
        String date = df.format(new Date());

        ActionLog actionLog = new ActionLog();
		//设置对象参数,调用mapper接口写入数据库
        actionLogMapper.crateAction(actionLog);
        return "succ";
    }
}

原因

本质原因:spring管理的都是单例(singleton),和 websocket (多对象)相冲突。
详细解释:项目启动时初始化,会初始化 websocket (非用户连接的),spring 同时会为其注入 mapper,该对象的 mapper不是 null,被成功注入。但是,由于 spring 默认管理的是单例,所以只会注入一次 mapper。当新用户进入聊天时,系统又会创建一个新的 websocket 对象,这时矛盾出现了:spring 管理的都是单例,不会给第二个 websocket 对象注入 mapper,所以导致只要是用户连接创建的 websocket 对象,都不能再注入了。

像 controller 里面有 service, service 里面有 dao。因为 controller,service ,dao 都有是单例,所以注入时不会报 null。但是 websocket 不是单例,所以使用spring注入一次后,后面的对象就不会再注入了,会报null。

参考:

https://blog.csdn.net/m0_37202351/article/details/86255132

SpringBootWebSocket,可以通过实现WebSocketHandler或者使用@ServerEndpoint注解来创建WebSocket服务端。如果在WebSocketHandler无法获取到Spring注入的对象,可以尝试将WebSocketHandler注入Spring容器,这样就可以在WebSocketHandler使用@Autowired注解来注入需要的对象了。 在使用@ServerEndpoint注解的情况下,由于WebSocket是通过独立的线程处理的,所以无法直接使用@Autowired注解来注入对象。可以通过使用Spring提供的ApplicationContextAware接口来获取Spring上下文,并通过Spring上下文来获取需要的对象。 具体实现方式可以参考下面的代码示例: ```java @ServerEndpoint("/websocket") public class MyWebSocket { private static ApplicationContext applicationContext; @Autowired private SomeService someService; @OnOpen public void onOpen(Session session) { // 在第一次连接时获取Spring上下文 if (applicationContext == null) { applicationContext = SpringContextUtils.getApplicationContext(); } // 此时可以使用@Autowired注解注入的someService对象 someService.doSomething(); } } @Component public class SpringContextUtils implements ApplicationContextAware { private static ApplicationContext applicationContext; @Override public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { SpringContextUtils.applicationContext = applicationContext; } public static ApplicationContext getApplicationContext() { return applicationContext; } } ``` 以上是一个简单的示例,可以在WebSocket通过@Autowired注解来注入Spring管理的对象,具体实现方式可以根据需要进行调整。
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值