Spring Boot 開(kāi)發(fā)私有即時(shí)通信系統(tǒng)（WebSocket）

1/ 概述

利用Spring Boot作為基礎(chǔ)框架裳朋，Spring Security作為安全框架病线，WebSocket作為通信框架吓著，實(shí)現(xiàn)點(diǎn)對(duì)點(diǎn)聊天和群聊天。

2/ 所需依賴

Spring Boot 版本 1.5.3送挑，使用MongoDB存儲(chǔ)數(shù)據(jù)（非必須）绑莺，Maven依賴如下：

<properties>
    <java.version>1.8</java.version>
    <thymeleaf.version>3.0.0.RELEASE</thymeleaf.version>
    <thymeleaf-layout-dialect.version>2.0.0</thymeleaf-layout-dialect.version>
  </properties>

  <dependencies>

    <!-- WebSocket依賴，移除Tomcat容器 -->
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-websocket</artifactId>
      <exclusions>
        <exclusion>
          <groupId>org.springframework.boot</groupId>
          <artifactId>spring-boot-starter-tomcat</artifactId>
        </exclusion>
      </exclusions>
    </dependency>

    <!-- 使用Undertow容器 -->
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-undertow</artifactId>
    </dependency>

    <!--  Spring Security 框架 -->
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-security</artifactId>
    </dependency>

    <!-- MongoDB數(shù)據(jù)庫(kù) -->
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-data-mongodb</artifactId>
    </dependency>

    <!-- Thymeleaf 模版引擎 -->
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-thymeleaf</artifactId>
    </dependency>

    <dependency>
      <groupId>org.projectlombok</groupId>
      <artifactId>lombok</artifactId>
      <version>1.16.16</version>
    </dependency>

    <dependency>
      <groupId>com.alibaba</groupId>
      <artifactId>fastjson</artifactId>
      <version>1.2.30</version>
    </dependency>

    <!-- 靜態(tài)資源 -->
    <dependency>
      <groupId>org.webjars</groupId>
      <artifactId>webjars-locator</artifactId>
    </dependency>
    <dependency>
      <groupId>org.webjars</groupId>
      <artifactId>sockjs-client</artifactId>
      <version>1.0.2</version>
    </dependency>
    <dependency>
      <groupId>org.webjars</groupId>
      <artifactId>stomp-websocket</artifactId>
      <version>2.3.3</version>
    </dependency>
    <dependency>
      <groupId>org.webjars</groupId>
      <artifactId>bootstrap</artifactId>
      <version>3.3.7</version>
    </dependency>
    <dependency>
      <groupId>org.webjars</groupId>
      <artifactId>jquery</artifactId>
      <version>3.1.0</version>
    </dependency>

  </dependencies>

配置文件內(nèi)容：

server:
  port: 80

# 若使用MongoDB則配置如下參數(shù)
spring:
  data:
    mongodb:
      uri: mongodb://username:password@172.25.11.228:27017
      authentication-database: admin
      database: chat

大致程序結(jié)構(gòu)惕耕，僅供參考：

程序結(jié)構(gòu)

3/ 創(chuàng)建程序啟動(dòng)類(lèi)纺裁，啟用WebSocket

使用@EnableWebSocket注解

@SpringBootApplication
@EnableWebSocket
public class Application {

  public static void main(String[] args) {
    SpringApplication.run(Application.class, args);
  }

}

4/ 配置Spring Security

此章節(jié)省略。（配置好Spring Security司澎，用戶能正常登錄即可）
可以參考：Spring Boot 全棧開(kāi)發(fā)：用戶安全

5/ 配置Web Socket（結(jié)合第7節(jié)的JS看）

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

  // 此處可注入自己寫(xiě)的Service

  @Override
  public void registerStompEndpoints(StompEndpointRegistry stompEndpointRegistry) {
    // 客戶端與服務(wù)器端建立連接的點(diǎn)
    stompEndpointRegistry.addEndpoint("/any-socket").withSockJS();
  }

  @Override
  public void configureMessageBroker(MessageBrokerRegistry messageBrokerRegistry) {
    // 配置客戶端發(fā)送信息的路徑的前綴
    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 {
            // 客戶端與服務(wù)器端建立連接后欺缘，此處記錄誰(shuí)上線了
            String username = session.getPrincipal().getName();
            log.info("online: " + username);
            super.afterConnectionEstablished(session);
          }

          @Override
          public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
            // 客戶端與服務(wù)器端斷開(kāi)連接后，此處記錄誰(shuí)下線了
            String username = session.getPrincipal().getName();
            log.info("offline: " + username);
            super.afterConnectionClosed(session, closeStatus);
          }
        };
      }
    });
    super.configureWebSocketTransport(registration);
  }
}

6/ 點(diǎn)對(duì)點(diǎn)消息挤安，群消息

@Controller
@Log4j
public class ChatController {

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

  // 群聊天
  @MessageMapping("/notice")
  public void notice(Principal principal, String message) {  
    // 參數(shù)說(shuō)明 principal 當(dāng)前登錄的用戶谚殊， message 客戶端發(fā)送過(guò)來(lái)的內(nèi)容
    // principal.getName() 可獲得當(dāng)前用戶的username  

    // 發(fā)送消息給訂閱 "/topic/notice" 且在線的用戶
    template.convertAndSend("/topic/notice", message); 
  }

  // 點(diǎn)對(duì)點(diǎn)聊天
  @MessageMapping("/chat")
  public void chat(Principal principal, String message){
    // 參數(shù)說(shuō)明 principal 當(dāng)前登錄的用戶， message 客戶端發(fā)送過(guò)來(lái)的內(nèi)容（應(yīng)該至少包含發(fā)送對(duì)象toUser和消息內(nèi)容content）
    // principal.getName() 可獲得當(dāng)前用戶的username

    // 發(fā)送消息給訂閱 "/user/topic/chat" 且用戶名為toUser的用戶
    template.convertAndSendToUser(toUser, "/topic/chat", content);
  }

}

7/ 客戶端與服務(wù)器端交互

    var stompClient = null;

    function connect() {
        var socket = new SockJS('/any-socket');
        stompClient = Stomp.over(socket);
        stompClient.connect({}, function (frame) {
            // 訂閱 /topic/notice 實(shí)現(xiàn)群聊
            stompClient.subscribe('/topic/notice', function (message) {
                showMessage(JSON.parse(message.body));
            });
            // 訂閱 /user/topic/chat 實(shí)現(xiàn)點(diǎn)對(duì)點(diǎn)聊
            stompClient.subscribe('/user/topic/chat', function (message) {
                showMessage(JSON.parse(message.body));
            });
        });
    }

    function showMessage(message) {
        // 處理消息在頁(yè)面的顯示
    }

    $(function () {
        // 建立websocket連接
        connect();
        // 發(fā)送消息按鈕事件
        $("#send").click(function () {
            if (target == "TO_ALL"){
                // 群發(fā)消息
                // 匹配后端ChatController中的 @MessageMapping("/notice")
                stompClient.send("/app/notice", {}, '消息內(nèi)容');
            }else{
                // 點(diǎn)對(duì)點(diǎn)消息蛤铜，消息中必須包含對(duì)方的username
                // 匹配后端ChatController中的 @MessageMapping("/chat")
                var content = "{'content':'消息內(nèi)容','receiver':'anoy'}";
                stompClient.send("/app/chat", {}, content);
            }
        });
    });

8/ 效果測(cè)試

登錄三個(gè)用戶：Anoyi嫩絮、Jock、超級(jí)管理員围肥。
群消息測(cè)試剿干，超級(jí)管理員群發(fā)消息：

超級(jí)管理員

Anoyi

Jock

點(diǎn)對(duì)點(diǎn)消息測(cè)試，Anoyi給Jock發(fā)送消息穆刻，只有Jock收到消息置尔，Anoyi和超級(jí)管理員收不到消息：