1.首先新建模塊：

<dependency>
 <groupId>org.springframework.boot</groupId>
 <artifactId>spring-boot-starter</artifactId>
</dependency>
<dependency>
 <groupId>org.springframework.boot</groupId>
 <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

• 或者在創(chuàng)建模塊的時候按順序點擊添加：

  image

2.建立config、controller包：

image

3.WebSocketConfig、WebSocketServer、WebSocketController：

• WebSocketConfig

  import org.springframework.context.annotation.Bean;
  import org.springframework.context.annotation.Configuration;
  import org.springframework.web.socket.server.standard.ServerEndpointExporter;
  
  /**
   * WebSocket的配置類
   * 開啟了WebSocket支持
   */
  @Configuration
  public class WebSocketConfig {
  
      @Bean
      public ServerEndpointExporter serverEndpointExporter() {
          return new ServerEndpointExporter();
      }
  }
  
  

• WebSocketServer

  import org.springframework.stereotype.Component;
  import javax.websocket.*;
  import javax.websocket.server.ServerEndpoint;
  import java.io.IOException;
  import java.util.concurrent.CopyOnWriteArraySet;
  
  //客戶端向服務(wù)器端建立WebSocket連接的url
  @ServerEndpoint("/websocket")
  @Component
  public class WebSocketServer {
  
      //靜態(tài)變量，用來記錄當前在線連接數(shù),可選
      private static int onlineCount = 0;
  
      //concurrent包的線程安全Set，用來存放每個客戶端對應(yīng)的MyWebSocket對象，必須
      private static CopyOnWriteArraySet<WebSocketServer> webSocketSet
        = new CopyOnWriteArraySet<WebSocketServer>();
  
      //與某個客戶端的連接會話，需要通過它來給客戶端發(fā)送數(shù)據(jù),必須
      private Session session;
  
      /**
       * 連接建立成功調(diào)用的方法
       */
      @OnOpen
      public void onOpen(Session session) {
          this.session = session;
          webSocketSet.add(this);     //將客戶端加入set中
          addOnlineCount();           //在線數(shù)加1
          System.out.println("有新窗口開始監(jiān)聽,當前在線人數(shù)為"
                  + getOnlineCount());
          try {
              sendMessage("連接成功");
          } catch (IOException e) {
              System.out.println("WebSocket IO異常");
          }
      }
  
      /**
       * 連接關(guān)閉調(diào)用的方法
       */
      @OnClose
      public void onClose() {
          webSocketSet.remove(this);  //從set中刪除
          subOnlineCount();           //在線數(shù)減1
          System.out.println("有連接關(guān)閉！當前在線人數(shù)為" + getOnlineCount());
      }
  
      /**
       * 收到客戶端消息后調(diào)用的方法
       *
       * @param message 客戶端發(fā)送過來的消息
       */
      @OnMessage
      public void onMessage(String message, Session session) {
          System.out.println("收到客戶端的信息:" + message);
          //群發(fā)消息
          for (WebSocketServer item : webSocketSet) {
              try {
                  item.sendMessage(message);
              } catch (IOException e) {
                 e.printStackTrace();
              }
          }
      }
  
      /**
       * @param session
       * @param error
       */
      @OnError
      public void onError(Session session, Throwable error) {
          System.out.println("發(fā)生錯誤");
          error.printStackTrace();
      }
  
      /**
       * 實現(xiàn)服務(wù)器主動推送
       */
      public void sendMessage(String message) throws IOException {
          this.session.getBasicRemote().sendText(message);
      }
      /**
       * 群發(fā)自定義消息
       */
      public static void sendInfo(String message) throws IOException {
          System.out.println("推送消息內(nèi)容:" + message);
          for (WebSocketServer item : webSocketSet) {
              try {
                  item.sendMessage(message);
              } catch (IOException e) {
                  continue;
              }
          }
      }
  
     public static synchronized int getOnlineCount() {
          return onlineCount;
      }
  
      public static synchronized void addOnlineCount() {
          WebSocketServer.onlineCount++;
      }
  
      public static synchronized void subOnlineCount() {
          WebSocketServer.onlineCount--;
      }
  }
  
  

• WebSocketController

  import com.soft1721.websocket.config.WebSocketServer;
  import org.springframework.web.bind.annotation.GetMapping;
  import org.springframework.web.bind.annotation.RestController;
  
  import java.io.IOException;
  
  @RestController
  public class WebSocketController {
  
      //推送數(shù)據(jù)接口
      @GetMapping("/socket/push")
      public String pushMsg(String message) {
          try {
              WebSocketServer.sendInfo(message);
          } catch (IOException e) {
              e.printStackTrace();
          }
          return "success";
      }
  }
  
  

4.接下來主方法WebsocketApplication

5.運行Swagger查看結(jié)果