若依 springboot websocket

1. pom

xml 复制代码

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

2. 相关配置

1. WebSocketConfig

java 复制代码

package com.lxsy.framework.config.websocket.conf;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

/**
 * websocket 配置
 */
@Configuration
@EnableWebSocket
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }

}

2. SemaphoreUtils （JUC信号量）

java 复制代码

package com.lxsy.framework.config.websocket;

import lombok.extern.slf4j.Slf4j;

import java.util.concurrent.Semaphore;

/**
 * 信号量相关处理
 */
@Slf4j
public class SemaphoreUtils
{

    /**
     * 获取信号量
     * @param semaphore
     * @return
     */
    public static boolean tryAcquire(Semaphore semaphore)
    {
        boolean flag = false;

        try
        {
            flag = semaphore.tryAcquire();
        }
        catch (Exception e)
        {
            log.error("获取信号量异常", e);
        }

        return flag;
    }

    /**
     * 释放信号量
     * @param semaphore
     */
    public static void release(Semaphore semaphore)
    {
        try
        {
            semaphore.release();
        }
        catch (Exception e)
        {
            log.error("释放信号量异常", e);
        }
    }
}

3. WebSocketServer

json 复制代码

package com.lxsy.framework.config.websocket;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import jakarta.websocket.*;
import jakarta.websocket.server.ServerEndpoint;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.Semaphore;

/**
 * WebSocketServer 消息处理类，用于实现 WebSocket 服务端功能
 */
@Slf4j
@Component
@ServerEndpoint("/websocket/message")
public class WebSocketServer2 {

    /**
     * 默认最多允许同时在线人数 100
     */
    public static int socketMaxOnlineCount = 100;

    /**
     * 信号量，用于控制同时在线人数
     */
    private static Semaphore socketSemaphore = new Semaphore(socketMaxOnlineCount);

    // 内部静态类 User，用于存储用户信息
    @AllArgsConstructor
    @NoArgsConstructor
    @Data
    private static class User {
        private String id; // 用户 ID
        private String name; // 用户名
        private Session session; // 用户会话
    }

    /**
     * 存储用户信息的 ConcurrentHashMap，键为用户 ID，值为 User 对象
     */
    private static ConcurrentHashMap<String, User> users = new ConcurrentHashMap<>();

    /**
     * 消息缓存的 ConcurrentHashMap，键为用户名，值为缓存消息列表
     */
    private static ConcurrentHashMap<String, CopyOnWriteArrayList<String>> messageCache = new ConcurrentHashMap<>();

    /**
     * 连接建立成功时调用的方法
     * @param session 用户会话
     */
    @OnOpen
    public void onOpen(Session session) throws Exception {
        // 尝试获取信号量
        if (!SemaphoreUtils.tryAcquire(socketSemaphore)) {
            // 未获取到信号量，说明在线人数已达上限
            log.error("\n 当前在线人数超过限制数- {}", socketMaxOnlineCount);
            WebSocketUsers.sendMessageToUserByText(session, "当前在线人数超过限制数：" + socketMaxOnlineCount);
            session.close();
        } else {
            // 将用户会话存储到 WebSocketUsers 中
            WebSocketUsers.put(session.getId(), session);
            log.info("\n 建立连接 - {}", session);
            log.info("\n 当前人数 - {}", WebSocketUsers.getUsers().size());
            WebSocketUsers.sendMessageToUserByText(session, "连接成功");
        }
    }

    /**
     * 连接关闭时调用的方法
     *
     * @param session 用户会话
     */
    @OnClose
    public void onClose(Session session) {
        log.info("\n 关闭连接 - {}", session);
        // 移除用户信息，移除用户时，将该用户的session值置为null
        // 用于判断该用户是否离线
        for (User user : users.values()) {
            if (user.getId().equals(session.getId())) {
                user.setSession(null);
                break;
            }
        }

        boolean removeFlag = WebSocketUsers.remove(session.getId());
        if (!removeFlag) {
            // 如果移除失败，则释放信号量
            SemaphoreUtils.release(socketSemaphore);
        }
    }

    /**
     * 抛出异常时调用的方法
     *
     * @param session     用户会话
     * @param exception   异常信息
     * @throws Exception  可能抛出的异常
     */
    @OnError
    public void onError(Session session, Throwable exception) throws Exception {
        if (session.isOpen()) {
            // 如果会话仍然开启，则关闭会话
            session.close();
        }
        String sessionId = session.getId();
        log.info("\n 连接异常 - {}", sessionId);
        log.info("\n 异常信息 - {}", exception);
        // 移除用户信息并释放信号量
        WebSocketUsers.remove(sessionId);
        SemaphoreUtils.release(socketSemaphore);
    }

    /**
     * 服务器接收到客户端消息时调用的方法
     *
     * @param message 消息内容
     * @param session 用户会话
     */
    @OnMessage
    public void onMessage(String message, Session session) {
        try {
            // 使用 Jackson 解析 JSON 消息
            ObjectMapper objectMapper = new ObjectMapper();
            JsonNode jsonNode = objectMapper.readTree(message);
            if (jsonNode.has("username") && !jsonNode.get("username").asText().isEmpty()) {
                String username = jsonNode.get("username").asText();
                // 检查用户是否已存在
                boolean userExists = false;
                for (User user : users.values()) {
                    if (user.getName().equals(username)) {
                        userExists = true;
                        // 更新用户的 ID 和会话
                        user.setId(session.getId());
                        user.setSession(session);

                        // 检查该用户是否有缓存消息
                        searchCache(username, user);

                        log.info("\n用户 {} 已更新", username);
                        log.info("\n当前用户列表 {}", users);
                        break;
                    }
                }
                if (!userExists) {
                    // 新建用户并存储到 users 中
                    User user = new User(session.getId(), username, session);
                    users.put(session.getId(), user);
                    log.info("\n用户 {} 已连接", username);
                    log.info("\n当前用户列表 {}", users);

                    // 检查是否有缓存消息
                    searchCache(username, user);
                }
            } else if (jsonNode.has("targetusername") && jsonNode.has("sendmessage")) {
                String targetusername = jsonNode.get("targetusername").asText();
                String sendmessage = jsonNode.get("sendmessage").asText();

                boolean userExists = false;
                // 查找目标用户，如果用户的session为空，则表示用户离线
                // 将通知存放到缓存消息列表中，如果不为空则表示用户在线，直接发送通知
                for (User user : users.values()) {
                    if (user.getName().equals(targetusername)) {
                        if (user.getSession() != null) {
                            userExists = true;
                            WebSocketUsers.sendMessageToUserByText(user.getSession(), sendmessage);
                        }
                        break;
                    }
                }

                log.info("\n目标用户 {}", targetusername);
                log.info("\n发送消息 {}", sendmessage);
                log.info("\n当前用户列表 {}", users);

                if (!userExists) {
                    // 如果目标用户不在线，则缓存通知
                    messageCache.computeIfAbsent(targetusername, k -> new CopyOnWriteArrayList<String>()).add(sendmessage);
                    log.info("\n消息已缓存，等待用户 {} 上线", targetusername);
                }
            }

        } catch (Exception e) {
            // 如果解析失败，则按普通消息处理
            String msg = message.replace("你", "我").replace("吗", "");
            WebSocketUsers.sendMessageToUserByText(session, msg);
        }
    }

    /**
     * 检查一个用户的缓存通知并发送
     *
     * @param username 用户名
     * @param user     用户对象
     */
    public void searchCache(String username, User user) {
        if (messageCache.containsKey(username)) {
            CopyOnWriteArrayList<String> cachedMessages = messageCache.get(username);
            for (String cachedMessage : cachedMessages) {
                WebSocketUsers.sendMessageToUserByText(user.getSession(), cachedMessage);
                log.info("\n向用户 {} 发送缓存消息 - {}", username, cachedMessage);
            }
            // 清理缓存通知
            messageCache.remove(username);
            log.info("\n已清理用户 {} 的缓存消息", username);
        }
    }
}

4. WebSocketUsers

java 复制代码

package com.lxsy.framework.config.websocket;

import jakarta.websocket.Session;
import lombok.extern.slf4j.Slf4j;

import java.io.IOException;
import java.util.Collection;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

/**
 * websocket 客户端用户集
 */
@Slf4j
public class WebSocketUsers {

    /**
     * 用户集
     */
    private static Map<String, Session> USERS = new ConcurrentHashMap<>();

    /**
     * 存储用户
     */
    public static void put(String key, Session session) {
        USERS.put(key, session);
    }

    /**
     * 移除用户
     */
    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) {
        //log.info("\n 正在移出用户 - {}", key);
        Session remove = USERS.remove(key);
        if (remove != null) {
            boolean containsValue = USERS.containsValue(remove);
            //log.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);
        }
    }

    /**
     * 发送文本消息
     */
    public static void sendMessageToUserByText(Session session, String message) {
        if (session != null) {
            try {
                session.getBasicRemote().sendText(message);
            } catch (IOException e) {
                log.error("\n[发送消息异常]", e);
            }
        } else {
            //log.info("\n[你已离线]");
        }
    }
}