SpringBoot使用websocket（二）

最新推荐文章于 2025-06-20 16:14:57 发布

原创最新推荐文章于 2025-06-20 16:14:57 发布 · 7.4k 阅读

24 ·

CC 4.0 BY-SA版权

文章标签：

#java #springboot #websocket

JAVA 同时被 3 个专栏收录

41 篇文章

订阅专栏

springboot

22 篇文章

订阅专栏

websocket

2 篇文章

订阅专栏

本文详细介绍如何在SpringBoot项目中使用WebSocket实现一对一的消息发送功能。通过配置WebSocket依赖，创建消息对象和WebSocket服务，实现客户端上线、离线、错误处理及消息发送。文章提供了完整的代码示例，并演示了两个客户端之间的消息传递过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本文讲解如何SpringBoot如何使用WebSocket进行一对一发送消息，如果想学习SpringBoot结合Websocket群发消息，可以前往查看我的上一篇文章：springboot结合websocket（一）

导入依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<dependency>
    <groupId>org.projectlombok</groupId>
    <artifactId>lombok</artifactId>
</dependency>
<dependency>
    <groupId>com.google.code.gson</groupId>
    <artifactId>gson</artifactId>
</dependency>

新建Websocket配置类

/**
 * @author Gjing
 **/
@Configuration
public class WebsocketConfiguration {

    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }
}

新建消息对象

/**
 * @author Gjing
 **/
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class Message {

    private String userId;

    private String message;
}

新建websocket服务

/**
 * @author Gjing
 **/
@ServerEndpoint("/test-one")
@Component
@Slf4j
public class MyOneToOneServer {
    /**
     * 用于存放所有在线客户端
     */
    private static Map<String, Session> clients = new ConcurrentHashMap<>();

    private Gson gson = new Gson();

    @OnOpen
    public void onOpen(Session session) {
        log.info("有新的客户端上线: {}", session.getId());
        clients.put(session.getId(), session);
    }

    @OnClose
    public void onClose(Session session) {
        String sessionId = session.getId();
        log.info("有客户端离线: {}", sessionId);
        clients.remove(sessionId);
    }

    @OnError
    public void onError(Session session, Throwable throwable) {
        throwable.printStackTrace();
        if (clients.get(session.getId()) != null) {
            clients.remove(session.getId());
        }
    }

    @OnMessage
    public void onMessage(String message) {
        log.info("收到客户端发来的消息: {}", message);
        this.sendTo(gson.fromJson(message, Message.class));
    }

    /**
     * 发送消息
     *
     * @param message 消息对象
     */
    private void sendTo(Message message) {
        Session s = clients.get(message.getUserId());
        if (s != null) {
            try {
                s.getBasicRemote().sendText(message.getMessage());
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}