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

 public class WebsocketUtil {
  /**
   * 記錄當(dāng)前在線的Session
   */
  private static final Map<String, Session> ONLINE_SESSION = new ConcurrentHashMap<> ();

  /**
   * 添加session
   * @param userId
   * @param session
   */
  public static void addSession(String userId, Session session){
    // 此處只允許一個用戶的session鏈接。一個用戶的多個連接，我們視為無效。
    ONLINE_SESSION.putIfAbsent ( userId, session );
  }

  /**
   * 關(guān)閉session
   * @param userId
   */
  public static void removeSession(String userId){
    ONLINE_SESSION.remove ( userId );
  }

  /**
   * 給單個用戶推送消息
   * @param session
   * @param message
   */
  public static void sendMessage(Session session, String message){
    if(session == null){
      return;
    }

    // 同步
    RemoteEndpoint.Async async = session.getAsyncRemote ();
    async.sendText ( message );
  }

  /**
   * 向所有在線人發(fā)送消息
   * @param message
   */
  public static void sendMessageForAll(String message) {
    //jdk8 新方法
    ONLINE_SESSION.forEach((sessionId, session) -> sendMessage(session, message));
  }
}

/**
 * websocket接口處理類
 */
@Component
@ServerEndpoint ( value = "/chat/{userid}" )
public class WebsocketController {

  /**
   * 連接事件，加入注解
   * @param userId
   * @param session
   */
  @OnOpen
  public void onOpen( @PathParam ( value = "userid" ) String userId, Session session ) {
    String message ="[" + userId + "]加入聊天室??！";

    // 添加到session的映射關(guān)系中
    WebsocketUtil.addSession ( userId, session );
    // 廣播通知，某用戶上線了
    WebsocketUtil.sendMessageForAll ( message );
  }

  /**
   * 連接事件，加入注解
   * 用戶斷開鏈接
   * @param userId
   * @param session
   */
  @OnClose
  public void onClose(@PathParam ( value = "userId" ) String userId, Session session ) {
    String message ="[" + userId + "]退出了聊天室...";

    // 刪除映射關(guān)系
    WebsocketUtil.removeSession ( userId );
    // 廣播通知，用戶下線了
    WebsocketUtil.sendMessageForAll ( message );
  }

  /**
   * 當(dāng)接收到用戶上傳的消息
   * @param userId
   * @param session
   */
  @OnMessage
  public void onMessage(@PathParam ( value = "userId" ) String userId, Session session ,String message) {
    String msg ="[" + userId + "]:"+message;

    // 直接廣播
    WebsocketUtil.sendMessageForAll ( msg );
  }

  /**
   * 處理用戶活連接異常
   * @param session
   * @param throwable
   */
  @OnError
  public void onError(Session session, Throwable throwable) {
    try {
      session.close();
    } catch (IOException e) {
      e.printStackTrace();
    }
    throwable.printStackTrace();
  }
}

public class DemoApplication {

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

  /**
   * 會自動注冊使用了@ServerEndpoint注解聲明的Websocket endpoint
   * 要注意，如果使用獨立的servlet容器，
   * 而不是直接使用springboot的內(nèi)置容器，
   * 就不要注入ServerEndpointExporter，因為它將由容器自己提供和管理。
   */
  @Bean
  public ServerEndpointExporter serverEndpointExporter() {
    return new ServerEndpointExporter();
  }
}