Spring Boot 开发集成 WebSocket，实现私有即时通信系统

###4/ 配置Spring Security
此章节省略。（配置好Spring Security，用户能正常登录即可）
可以参考：Spring Boot 全栈开发：用户安全

###5/ 配置Web Socket（结合第7节的JS看）

@Configuration
@EnableWebSocketMessageBroker
@Log4j
public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer {

  // 此处可注入自己写的Service

  @Override
  public void registerStompEndpoints(StompEndpointRegistry stompEndpointRegistry) {
    // 客户端与服务器端建立连接的点
    stompEndpointRegistry.addEndpoint("/any-socket").withSockJS();
  }

  @Override
  public void configureMessageBroker(MessageBrokerRegistry messageBrokerRegistry) {
    // 配置客户端发送信息的路径的前缀
    messageBrokerRegistry.setApplicationDestinationPrefixes("/app");
    messageBrokerRegistry.enableSimpleBroker("/topic");
  }

  @Override
  public void configureWebSocketTransport(final WebSocketTransportRegistration registration) {
    registration.addDecoratorFactory(new WebSocketHandlerDecoratorFactory() {
      @Override
      public WebSocketHandler decorate(final WebSocketHandler handler) {
 return new WebSocketHandlerDecorator(handler) {
   @Override
   public void afterConnectionEstablished(final WebSocketSession session) throws Exception {
     // 客户端与服务器端建立连接后，此处记录谁上线了
     String username = session.getPrincipal().getName();
     log.info("online: " + username);
     super.afterConnectionEstablished(session);
   }

   @Override
   public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
     // 客户端与服务器端断开连接后，此处记录谁下线了
     String username = session.getPrincipal().getName();
     log.info("offline: " + username);
     super.afterConnectionClosed(session, closeStatus);
   }
 };
      }
    });
    super.configureWebSocketTransport(registration);
  }
}

@Controller
@Log4j
public class ChatController {

  @Autowired
  private SimpMessagingTemplate template;
  
  // 注入其它Service

  // 群聊天
  @MessageMapping("/notice")
  public void notice(Principal principal, String message) {  
    // 参数说明 principal 当前登录的用户， message 客户端发送过来的内容
    // principal.getName() 可获得当前用户的username  

    // 发送消息给订阅 "/topic/notice" 且在线的用户
    template.convertAndSend("/topic/notice", message); 
  }

  // 点对点聊天
  @MessageMapping("/chat")
  public void chat(Principal principal, String message){
    // 参数说明 principal 当前登录的用户， message 客户端发送过来的内容（应该至少包含发送对象toUser和消息内容content）
    // principal.getName() 可获得当前用户的username

    // 发送消息给订阅 "/user/topic/chat" 且用户名为toUser的用户
    template.convertAndSendToUser(toUser, "/topic/chat", content);
  }

}

    var stompClient = null;

    function connect() {
 var socket = new SockJS('/any-socket');
 stompClient = Stomp.over(socket);
 stompClient.connect({}, function (frame) {
     // 订阅 /topic/notice 实现群聊
     stompClient.subscribe('/topic/notice', function (message) {
  showMessage(JSON.parse(message.body));
     });
     // 订阅 /user/topic/chat 实现点对点聊
     stompClient.subscribe('/user/topic/chat', function (message) {
  showMessage(JSON.parse(message.body));
     });
 });
    }

    function showMessage(message) {
 // 处理消息在页面的显示
    }

    $(function () {
 // 建立websocket连接
 connect();
 // 发送消息按钮事件
 $("#send").click(function () {
     if (target == "TO_ALL"){
  // 群发消息
  // 匹配后端ChatController中的 @MessageMapping("/notice")
  stompClient.send("/app/notice", {}, '消息内容');
     }else{
  // 点对点消息，消息中必须包含对方的username
  // 匹配后端ChatController中的 @MessageMapping("/chat")
  var content = "{'content':'消息内容','receiver':'anoy'}";
  stompClient.send("/app/chat", {}, content);
     }
 });
    });