springboot集成websocket点对点推送、广播推送

privateString message;
privateT data;
}
三、因为要实现点对点的推送，所以需要创建一个监听器来获取到websocket的session，如下：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
importorg.springframework.beans.factory.annotation.Autowired;
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
importlombok.extern.slf4j.Slf4j;
importorg.springframework.context.annotation.Bean;
importorg.springframework.context.annotation.Configuration;
importorg.springframework.messaging.Message;
importorg.springframework.messaging.MessageChannel;
//STOMP监听类
@Bean
publicSTOMPConnectEventListener applicationStartListener(){
returnnewSTOMPConnectEventListener();
}
@Override
publicvoidregisterStompEndpoints(StompEndpointRegistry stompEndpointRegistry) {
publicclassSTOMPConnectEventListenerimplementsApplicationListener<SessionConnectEvent> {
@Autowired
privateRedisHelper redisHelper;
@Override
publicvoidonApplicationEvent(SessionConnectEvent event) {
importorg.springframework.messaging.simp.config.ChannelRegistration;
importorg.springframework.messaging.simp.config.MessageBrokerRegistry;
importorg.springframework.messaging.simp.stomp.StompCommand;
String sessionId = sha.getSessionId();
redisHelper.redisTemplate.opsForValue().set("websocket:"+userid,sessionId);
}
}
四、最重要的配置类
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
二、推送到前端的消息实体类
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
importlombok.AllArgsConstructor;
importlombok.Builder;
importlombok.Data;
importlombok.NoArgsConstructor;
importjava.io.Serializable;
StompHeaderAccessor sha = StompHeaderAccessor.wrap(event.getMessage());
//login get from browser
if(sha.getNativeHeader("userid")==null){
return;
}
String userid = sha.getNativeHeader("userid").get(0);
importorg.springframework.web.socket.config.annotation.*;
@Configuration
@EnableWebSocketMessageBroker
@Slf4j
publicclassWebSocketConfigextendsAbstractWebSocketMessageBrokerConfigurer {
importorg.springframework.messaging.simp.stomp.StompHeaderAccessor;
importorg.springframework.messaging.support.ChannelInterceptorAdapter;
importorg.springframework.messaging.support.MessageHeaderAccessor;
springboot
一、什么都不用说，导入个依赖先
1
2
3
4
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
importorg.springframework.context.ApplicationListener;
importorg.springframework.messaging.simp.stomp.StompHeaderAccessor;
importorg.springframework.web.socket.messaging.SessionConnectEvent;
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
publicclassNotifyBean<T>implementsSerializable {
privatestaticfinallongserialVersionUID = 1L;
privateinttype;