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<String, Principal> users = (Map<String, Principal>) (obj == null ? new HashMap<String,Principal>() : 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<MessageConverter> 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<HandlerMethodArgumentResolver> argumentResolvers) {
        // TODO Auto-generated method stub

    }

    @Override
    public void addReturnValueHandlers(List<HandlerMethodReturnValueHandler> 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<String,Principal> users = sc.getAttribute("users") == null ? null : (Map<String,Principal>)sc.getAttribute("users");//获取放在servletContext中的用户信息

        Set<String> set = users.keySet();
        Iterator<String> iterator = set.iterator();
        while(iterator.hasNext()){ //遍历根据用户进行推送
            String key = iterator.next();
            this.template.convertAndSendToUser(key, "/topic/greetings", msg);
        }

    }
}

在这里我们就成功的将消息推送到固定用户中了，这里有个想法，我们可以把spring websocket作为消息总线服务器独立出来，项目中，如果需要广播或者订阅消息就直接在消息总线中进行处理，只需要把推送路径和订阅路径做好就可以了，这里，跨域的操作，可以用setAllowedOrigins,下次可以看看