SpringBoot整合WebSocket，及注入Bean的方式

最近用到WebSocket协议，所以需要SpringBoot整合WebSocket。首先说下WebSocket协议，它是双向的，在客户端-服务器通信的场景中使用的全双协议，与HTTP不同，它以ws://开头，它是一个有状态协议，意味着客户端和服务器之间的连接保持活动状态，直到被任何一方终止。在客户端和服务器中的任何一方关闭连接后，连接将两端终止。
SpringBoot整合WebSocket
1、添加maven依赖

<dependency>  
           <groupId>org.springframework.boot</groupId>  
           <artifactId>spring-boot-starter-websocket</artifactId>  
       </dependency>

2、WebSocketConfig配置类

@Configuration
public class WebSocketConfig {
    
    

    /**
     * 注入一个ServerEndpointExporter,
     * 该Bean会自动注册使用@ServerEndpoint注解申明的websocket endpoint
     */
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
    
    
        return new ServerEndpointExporter();
    }
}

3、WebSocketServer

/**
 * 前后端交互的类实现消息的接收推送(自己发送给自己)
 *
 * @ServerEndpoint(value = "/test/one") 前端通过此URI和后端交互，建立连接
 */
@Slf4j
@ServerEndpoint(value = "/websocket")
@Component
public class OneWebSocket {
    
    


    private static DeviceListenerService deviceListenerService;

    @Autowired
    public void setDeviceListenerService(DeviceListenerService deviceListenerService) {
    
    
        OneWebSocket.deviceListenerService = deviceListenerService;
    }


    private static DevicesDaoMapper devicesDaoMapper;

    @Autowired
    public void setDevicesDaoMapper(DevicesDaoMapper devicesDaoMapper) {
    
    
        OneWebSocket.devicesDaoMapper = devicesDaoMapper;
    }


    private AdbDevice adbDevice;
    private MiniCapDataHandler miniCapDataHandler;
    private String serialNumber;


    /**
     * 记录当前在线连接数
     */
    private static AtomicInteger onlineCount = new AtomicInteger(0);

    /**
     * 连接建立成功调用的方法
     */
    @OnOpen
    public void onOpen(Session session) {
    
    
        onlineCount.incrementAndGet(); // 在线数加1
        log.info("有新连接加入：{}，当前在线人数为：{}", session.getId(), onlineCount.get());

        
        URI sn = session.getRequestURI();
        serialNumber = sn.getQuery().split("=")[1];
        adbDevice = deviceListenerService.getDevice(serialNumber);

        getMinicapData(session);

    }

    private void getMinicapData(Session session){
    
    
        List<Map<String, Object>> mobilephoneList = devicesDaoMapper.getDevicesByDeviceidModel(adbDevice.getSerialNumber(), adbDevice.getProperty("ro.product.model"));
        int minicapport = Integer.parseInt((String) mobilephoneList.get(0).get("minicapport"));
        miniCapDataHandler = new MiniCapDataHandler("127.0.0.1",minicapport);
        miniCapDataHandler.setRunning(true);
        miniCapDataHandler.RunImageConverterThread(session);
        miniCapDataHandler.RunImageBinaryFrameCollectorThread();
    }

    /**
     * 连接关闭调用的方法
     */
    @OnClose
    public void onClose(Session session) {
    
    

        miniCapDataHandler.setRunning(false);
        onlineCount.decrementAndGet(); // 在线数减1
        log.info("有一连接关闭：{}，当前在线人数为：{}", session.getId(), onlineCount.get());
    }

    /**
     * 收到客户端消息后调用的方法
     *
     * @param message 客户端发送过来的消息
     */
    @OnMessage
    public void onMessage(Session session,
                          String message) {
    
    
        URI sn = session.getRequestURI();
        serialNumber = sn.getQuery().split("=")[1];
        adbDevice = deviceListenerService.getDevice(serialNumber);
        String[] strarray = message.split(" ");
        if("t".equals(strarray[0])){
    
    
            String x = strarray[1];
            String y = strarray[2];
            adbDevice.adbTap(x,y);
        }
        if("s".equals(strarray[0])){
    
    
            String startx = strarray[1];
            String starty = strarray[2];
            String endx = strarray[3];
            String endy = strarray[4];
            adbDevice.adbswipe(startx,starty,endx,endy,1);
        }
        if("p".equals(strarray[0])){
    
    
            String startx = strarray[1];
            String starty = strarray[2];
            String endx = strarray[3];
            String endy = strarray[4];
            adbDevice.adbswipe(startx,starty,endx,endy,2);
        }



        log.info("服务端收到客户端[{}]的消息:{}", session.getId(), message);
    
    }

    @OnError
    public void onError(Session session, Throwable error) {
    
    
        log.error("发生错误");
        error.printStackTrace();
    }

    /**
     * 服务端发送消息给客户端
     */
//    private void sendMessage(Session session,
//                             TextMessage message) {
    
    
//        try {
    
    
//            log.info("服务端给客户端[{}]发送消息{}", session.getId(), message);
//            session.getBasicRemote().sendText(message);
//
//        } catch (Exception e) {
    
    
//            log.error("服务端发送消息给客户端失败：{}", e);
//        }
//    }


}

4、注意事项，注入bean的方式，因为spring管理的都是单例。WebSocket是对对象 相冲突
注入方式，这么写就可以了。

 //  这里使用静态，让 service 属于类
  private static DeviceListenerService deviceListenerService;
 // 注入的时候，给类的 service 注入
    @Autowired
    public void setDeviceListenerService(DeviceListenerService deviceListenerService) {
    
    
        OneWebSocket.deviceListenerService = deviceListenerService;
    }