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,下次可以看看