spring websocket 基于编码的方式手动进行推送
spring websocket基于注解的@SendTo和@SendToUser虽然方便，但是有局限性，例如我这样子的需求，我想手动的把消息推送给某个人，或者特定一组人，怎么办，@SendTo只能推送给所有人，@SendToUser只能推送给请求消息的那个人，这时，我们可以利用SimpMessagingTemplate这个类
SimpMessagingTemplate有俩个推送的方法
convertAndSend(destination, payload); //将消息广播到特定订阅路径中，类似@SendTo
convertAndSendToUser(user, destination, payload);//将消息推送到固定的用户订阅路径中，类似@SendToUser
实现需求之前，我们来关注下websocket的一个ApplicationListener事件，SessionConnectedEvent，你可以订阅它，然后websocket通道链接成功之后，就广播这个事件
public class MySessionConnectedEvent implements ApplicationListener<SessionConnectedEvent> {
@Override
public void onApplicationEvent(SessionConnectedEvent event) {
}
}注册下这个接收事件到spring容器中，这里通过java config的方式注册
@Configuration
public class AppConfig {
@Bean
public MySessionConnectedEvent mySessionConnectedEvent(){
return new MySessionConnectedEvent();
}
}通过这个事件，我们可以把用户的登录授权信息保存起来，放到ServletContext中，以便于我们在推送消息的时候，拿到用户信息，当然，这里你也可以把用户的登录授权信息保存到其他的缓存策略中。
public class MySessionConnectedEvent implements ApplicationListener<SessionConnectedEvent> {
@SuppressWarnings("unchecked")
@Override
public void onApplicationEvent(SessionConnectedEvent event) {
StompHeaderAccessor headers = StompHeaderAccessor.wrap(event.getMessage()); //获取消息头
String name = headers.getUser().getName(); //获取账号名
ServletContext sc = SpringBeanUtil.getServletContext(); //SpringBeanUtil的作用就是获取ServletContext
Object obj = sc.getAttribute("users");
Map users = (Map) (obj == null ? new HashMap() : obj);
users.put(name, headers.getUser());
sc.setAttribute("users", users); //将用户信息已map格式放存放起来
}
} 有了用户信息就可以使用SimpMessagingTemplate进行特定用户推送了
@Controller
@RequestMapping("/webSocket")
public class WebSocketController {
@Autowired private SimpMessagingTemplate template;
}
SimpMessagingTemplate在配置spring websocket的时候就已经放到spring容器中了
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketMessageBrokerConfig implements WebSocketMessageBrokerConfigurer {
@Override
public void registerStompEndpoints(StompEndpointRegistry registry) {
//portfolio-stomp就是websocket的端点，客户端需要注册这个端点进行链接，withSockJS允许客户端利用sockjs进行浏览器兼容性处理
registry.addEndpoint("/portfolio-stomp")/*.setAllowedOrigins("http://localhost:8080")*/.withSockJS();
}
@Override
public void configureMessageBroker(MessageBrokerRegistry registry) {
registry.enableSimpleBroker("/topic"); //设置服务器广播消息的基础路径
registry.setApplicationDestinationPrefixes("/app"); //设置客户端订阅消息的基础路径
registry.setPathMatcher(new AntPathMatcher(".")); //可以已“.”来分割路径，看看类级别的@messageMapping和方法级别的@messageMapping
}
@Override
public boolean configureMessageConverters(List messageConverters) {
return true;
}
@Override
public void configureWebSocketTransport(WebSocketTransportRegistration registry) {
// TODO Auto-generated method stub
registry.addDecoratorFactory(new MyWebSocketHandlerDecoratorFactory());
}
@Override
public void configureClientInboundChannel(ChannelRegistration registration) {
registration.setInterceptors(new MyChannelInterceptor());
}
@Override
public void configureClientOutboundChannel(ChannelRegistration registration) {
// TODO Auto-generated method stub
}
@Override
public void addArgumentResolvers(List argumentResolvers) {
// TODO Auto-generated method stub
}
@Override
public void addReturnValueHandlers(List returnValueHandlers) {
// TODO Auto-generated method stub
}
}
所以我们这里我们可以直接通过@Autowired注入进来。接下来我们就利用进行推送了
@Controller
@RequestMapping("/webSocket")
public class WebSocketController {
@Autowired private SimpMessagingTemplate template;
@SuppressWarnings("unchecked")
@RequestMapping("/httpSendMsg")
public void httpSendMsg(String msg){
ServletContext sc = SpringBeanUtil.getServletContext();
Map users = sc.getAttribute("users") == null ? null : (Map)sc.getAttribute("users");//获取放在servletContext中的用户信息
Set set = users.keySet();
Iterator iterator = set.iterator();
while(iterator.hasNext()){ //遍历根据用户进行推送
String key = iterator.next();
this.template.convertAndSendToUser(key, "/topic/greetings", msg);
}
}
} 在这里我们就成功的将消息推送到固定用户中了，这里有个想法，我们可以把spring websocket作为消息总线服务器独立出来，项目中，如果需要广播或者订阅消息就直接在消息总线中进行处理，只需要把推送路径和订阅路径做好就可以了，这里，跨域的操作，可以用setAllowedOrigins,下次可以看看