Bootstrap

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

;