springboot集成websocket

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

 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 }