springboot集成websocket

Posted 2021-02-13 kanhin

websocket是全双工通信协议，目前html5支持，如果是app端的话可能不支持，建议app端实现通过tcp握手长连接实现通信，这里暂不研究。

首先websocket是一个协议，需要了解一下

 

第一步先引入starter

1 <dependency>
2             <groupId>org.springframework.boot</groupId>
3             <artifactId>spring-boot-starter-websocket</artifactId>
4             <version>${websocket.version}</version>
5         </dependency>

 

 

编写一个websocket终端类

 

 1 @ServerEndpoint(value = "/websocket")
 2 @Component
 3 public class FinePetroWebSocket {
 4 
 5     private static ApplicationContext applicationContext;
 6 
 7     public static void setApplicationContext(ApplicationContext context) {
 8         applicationContext = context;
 9     }
10 
11     /**
12      * concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。
13      */
14     private static CopyOnWriteArraySet<FinePetroWebSocket> webSocketSet = new CopyOnWriteArraySet<>();
15 
16     /**
17      * 与某个客户端的连接会话，需要通过它来给客户端发送数据
18      */
19     private Session session;
20 
21     /**
22      * 存储session的map 
23      */
24     private static Map<String, Session> sessionMap = new ConcurrentHashMap<>();
25 
26    
27 
28 
29     /**
30      * 连接建立成功调用的方法
31      */
32     @OnOpen
33     public void onOpen(Session session) {
34         this.session = session;
35         //加入set中
36         webSocketSet.add(this);
37 
38 
39     }
40 
41     /**
42      * 连接关闭调用的方法
43      */
44     @OnClose
45     public void onClose() {
46         //从set中删除
47         webSocketSet.remove(this);
48     }
49 
50     /**
51      * 收到客户端消息后调用的方法
52      *
53      * @param message 客户端发送过来的消息
54      */
55     @OnMessage
56     public void onMessage(String message, Session session) {
57        
58        
59     }
60 
61     private void sendMessage(Session session, String message) throws IOException {
62         session.getBasicRemote().sendText(message);
63     }
64 
65     
66    
67 
68 
69 }

 

1 @Configuration
2 public class WebSocketConfig {
3     @Bean
4     public ServerEndpointExporter serverEndpointExporter() {
5         return new ServerEndpointExporter();
6     }
7 
8 }

 

 

 

 

这样就可以实现通讯了 ， 但是注意这里的一个坑就是websocket每个链接使用一个对象，这里就不算是单例的，所以注入的service会为null，所以要使用static静态变量，在application中调用。

@Component默认是单例模式的，但springboot还是会为每个websocket连接初始化一个bean，所以可以用一个静态set保存起来。