content: package com.ruoyi.framework.websocket;
  
  import java.io.IOException;
  import java.util.Collection;
  import java.util.Map;
  import java.util.Set;
  import java.util.concurrent.ConcurrentHashMap;
  import javax.websocket.Session;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  /**
   * websocket 客户端用户集
   * 
   * @author ruoyi
   */
  public class WebSocketUsers
  {
      /**
       * WebSocketUsers 日志控制器
       */
      private static final Logger LOGGER = LoggerFactory.getLogger(WebSocketUsers.class);
  
      /**
       * 用户集
       */
      private static Map<String, Session> USERS = new ConcurrentHashMap<String, Session>();
  
      /**
       * 存储用户
       *
       * @param key 唯一键
       * @param session 用户信息
       */
      public static void put(String key, Session session)
      {
          USERS.put(key, session);
      }
  
      /**
       * 移除用户
       *
       * @param session 用户信息
       *
       * @return 移除结果
       */
      public static boolean remove(Session session)
      {
          String key = null;
          boolean flag = USERS.containsValue(session);
          if (flag)
          {
              Set<Map.Entry<String, Session>> entries = USERS.entrySet();
              for (Map.Entry<String, Session> entry : entries)
              {
                  Session value = entry.getValue();
                  if (value.equals(session))
                  {
                      key = entry.getKey();
                      break;
                  }
              }
          }
          else
          {
              return true;
          }
          return remove(key);
      }
  
      /**
       * 移出用户
       *
       * @param key 键
       */
      public static boolean remove(String key)
      {
          LOGGER.info("\n 正在移出用户 - {}", key);
          Session remove = USERS.remove(key);
          if (remove != null)
          {
              boolean containsValue = USERS.containsValue(remove);
              LOGGER.info("\n 移出结果 - {}", containsValue ? "失败" : "成功");
              return containsValue;
          }
          else
          {
              return true;
          }
      }
  
      /**
       * 获取在线用户列表
       *
       * @return 返回用户集合
       */
      public static Map<String, Session> getUsers()
      {
          return USERS;
      }
  
      /**
       * 群发消息文本消息
       *
       * @param message 消息内容
       */
      public static void sendMessageToUsersByText(String message)
      {
          Collection<Session> values = USERS.values();
          for (Session value : values)
          {
              sendMessageToUserByText(value, message);
          }
      }
  
      /**
       * 发送文本消息
       *
       * @param userName 自己的用户名
       * @param message 消息内容
       */
      public static void sendMessageToUserByText(Session session, String message)
      {
          if (session != null)
          {
              try
              {
                  session.getBasicRemote().sendText(message);
              }
              catch (IOException e)
              {
                  LOGGER.error("\n[发送消息异常]", e);
              }
          }
          else
          {
              LOGGER.info("\n[你已离线]");
          }
      }
  }
  