SpringBoot中的@MessageMapping注解详解

 更新时间:2023年08月12日 08:31:42   作者:2013crazy  
这篇文章主要介绍了SpringBoot中的@MessageMapping注解详解,Spring Boot 提供了对 WebSocket 的支持,其中 @MessageMapping 注解是一个常用的注解,它可以将一个 Java 方法标记为 WebSocket 的消息处理器,需要的朋友可以参考下

@MessageMapping

随着 Web 技术的发展,越来越多的应用程序开始使用 WebSocket 协议来实现实时通信。

Spring Boot 提供了对 WebSocket 的支持,其中 @MessageMapping 注解是一个常用的注解,它可以将一个 Java 方法标记为 WebSocket 的消息处理器。

本文将介绍 Spring Boot 中 @MessageMapping 注解的原理、用法和示例。

@MessageMapping 注解的原理

在 Spring Boot 中,@MessageMapping 注解用于标识一个 Java 方法是一个 WebSocket 消息处理器。

消息处理器负责接收客户端发送的消息,并进行处理。具体来说,使用 @MessageMapping 注解的方法将会被 Spring WebSocket 自动扫描,并注册为一个消息处理器。

@MessageMapping 注解的作用类似于 @RequestMapping 注解,但是它只能用于 WebSocket 消息处理器。

使用 @MessageMapping 注解时,需要指定消息的目的地(Destination),即客户端发送消息时的目标地址。

@MessageMapping 注解的用法

使用 @MessageMapping 注解的步骤如下:

1. 创建 WebSocket 配置类

首先,需要创建一个 WebSocket 配置类,并添加 @EnableWebSocket 注解。该注解用于启用 Spring WebSocket。

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(chatHandler(), "/chat");
    }
    @Bean
    public WebSocketHandler chatHandler() {
        return new ChatHandler();
    }
}

上面的代码中,创建了一个 WebSocketHandlerRegistry 对象,并将 ChatHandler() 的实例注册为 /chat 的处理器。

2. 创建消息处理器类

在消息处理器类中,需要编写消息处理器方法。消息处理器方法处理客户端发送的消息,并返回响应。消息处理器方法可以使用 Spring WebSocket 提供的注解来指定消息的目的地、消息的类型和其他参数。

@Component
public class ChatHandler implements WebSocketHandler {
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        System.out.println("Connection established: " + session.getId());
    }
    @MessageMapping("/hello")
    public void handleHelloMessage(String message) {
        System.out.println("Received message: " + message);
    }
    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        System.out.println("Transport error: " + exception.getMessage());
    }
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        System.out.println("Connection closed: " + session.getId());
    }
    @Override
    public boolean supportsPartialMessages() {
        return false;
    }
}

上面的代码中,@MessageMapping 注解用于指定消息的目的地,即 /hello。方法的参数中,String 类型的 message 表示客户端发送的消息内容。方法的返回值为 void,表示不需要返回响应。

3. 启动应用程序

最后,需要启动应用程序,并访问 ws://localhost:8080/chat,就可以开始使用 WebSocket 了。

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

@MessageMapping 注解的示例

下面是一个使用 @MessageMapping 注解的示例项目。

1. 创建项目

使用 Spring Initializr 创建一个新的 Spring Boot 项目,添加以下依赖:

  • Spring Websocket
  • Thymeleaf

2. 创建 WebSocket 配置类

创建一个 WebSocketConfig 类,并添加 @EnableWebSocket 注解。该注解用于启用 Spring WebSocket。

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(chatHandler(), "/chat").withSockJS();
    }
    @Bean
    public WebSocketHandler chatHandler() {
        return new ChatHandler();
    }
}

上面的代码中,创建了一个 WebSocketHandlerRegistry 对象,并将 ChatHandler() 的实例注册为 /chat 的处理器。使用 withSockJS() 方法可以启用 SockJS 支持。

3. 创建消息处理器类

在消息处理器类中,需要编写消息处理器方法。消息处理器方法处理客户端发送的消息,并返回响应。消息处理器方法可以使用 Spring WebSocket 提供的注解来指定消息的目的地、消息的类型和其他参数。

@Component
public class ChatHandler implements WebSocketHandler {
    private List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        System.out.println("Connection established: " + session.getId());
        sessions.add(session);
        broadcast("User " + session.getId() + " joined the chat");
    }
    @MessageMapping("/chat")
    public void handleChatMessage(ChatMessage message) {
        System.out.println("Received message: " + message);
        broadcast(message);
    }
    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        System.out.println("Transport error: " + exception.getMessage());
    }
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        System.out.println("Connection closed: " + session.getId() + " (" + closeStatus.getCode() + ")");
        sessions.remove(session);
        broadcast("User " + session.getId() + " left the chat");
    }
    @Override
    public boolean supportsPartialMessages() {
        return false;
    }
    private void broadcast(Object message) {
        for (WebSocketSession session : sessions) {
            try {
                session.sendMessage(new TextMessage(new ObjectMapper().writeValueAsString(message)));
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

上面的代码中,@MessageMapping 注解用于指定消息的目的地,即 /chat。方法的参数中,ChatMessage 类型的 message 表示客户端发送的消息内容。方法的返回值为 void,表示不需要返回响应。使用 sessions 列表来存储所有连接的 WebSocketSession。

同时,还实现了 afterConnectionEstablished()、handleTransportError() 和 afterConnectionClosed() 方法,分别用于处理连接建立、传输错误和连接关闭的事件。broadcast() 方法用于向所有 WebSocketSession 广播消息。

4. 创建 Thymeleaf 模板文件

在 src/main/resources/templates 目录下,创建一个名为 chat.html 的 Thymeleaf 模板文件。

<!DOCTYPE html>
<html>
<head>
    <title>Chat</title>
    <script src="/webjars/sockjs-client/1.5.0/dist/sockjs.min.js"></script>
    <script src="/webjars/stomp-websocket/2.3.3/dist/stomp.min.js"></script>
    <script src="/js/chat.js"></script>
</head>
<body>
<h1>Chat</h1>
<form onsubmit="return false;">
    <input type="text" id="name" placeholder="Name">
    <input type="text" id="message" placeholder="Message">
    <button onclick="sendMessage()">Send</button>
</form>
<div id="messages"></div>
</body>
</html>

上面的代码中,使用了 SockJS 和 STOMP 协议来实现 WebSocket 的兼容性。同时,引入了 chat.js 文件,用于处理客户端发送消息和接收消息的逻辑。

5. 创建客户端脚本文件

在 src/main/resources/static/js 目录下,创建一个名为 chat.js 的客户端脚本文件。

var stompClient = null;
function connect() {
    var socket = new SockJS('/chat');
    stompClient = Stomp.over(socket);
    stompClient.connect({}, function (frame) {
        console.log('Connected: ' + frame);
        stompClient.subscribe('/topic/messages', function (message) {
            showMessage(JSON.parse(message.body));
        });
    });
}
function disconnect() {
    if (stompClient !== null) {
        stompClient.disconnect();
    }
    console.log("Disconnected");
}
function sendMessage() {
    var name = document.getElementById('name').value;
    var message = document.getElementById('message').value;
    stompClient.send("/app/chat", {}, JSON.stringify({'name': name, 'message': message}));
}
function showMessage(message) {
    var div = document.createElement('div');
    div.appendChild(document.createTextNode(message.name + ': ' + message.message));
    document.getElementById('messages').appendChild(div);
}
connect();

上面的代码中,使用 connect() 方法来建立 WebSocket 连接。使用Stomp.over() 方法创建一个 STOMP 客户端对象。使用 stompClient.connect() 方法来连接到 WebSocket 服务器,并使用 stompClient.subscribe() 方法来订阅 /topic/messages 目的地,以接收广播消息。

使用 sendMessage() 方法来发送消息。该方法将 name 和 message 作为 JSON 对象发送到 /app/chat 目的地。

使用 showMessage() 方法来显示消息。该方法将消息添加到 messages 元素中。

6. 运行应用程序

最后,运行应用程序,访问 //localhost:8080/chat,就可以开始使用聊天功能了。

总结

本文介绍了 Spring Boot 中 @MessageMapping 注解的原理、用法和示例。通过使用 @MessageMapping 注解,可以很方便地编写 WebSocket 消息处理器,实现实时通信功能。

到此这篇关于SpringBoot中的@MessageMapping注解详解的文章就介绍到这了,更多相关SpringBoot的@MessageMapping内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • 使用Python+wxPython打造一个Skill管理器

    使用Python+wxPython打造一个Skill管理器

    如果你用过 Claude Code 或 OpenCode,一定知道 Skills 机制,下面我们就来详细介绍一下如何使用Python打造一个简单的Skill管理器,感兴趣的小伙伴可以了解下
    2026-03-03
  • Python如何获取域名的SSL证书信息和到期时间

    Python如何获取域名的SSL证书信息和到期时间

    在当今互联网时代,SSL证书的重要性不言而喻,它不仅为用户提供了安全的连接,还能提高网站的搜索引擎排名,那我们怎么才能通过Python获取域名的SSL证书信息和到期时间呢,下面小编就来简单讲讲吧
    2025-03-03
  • Python ORM框架SQLAlchemy学习笔记之关系映射实例

    Python ORM框架SQLAlchemy学习笔记之关系映射实例

    这篇文章主要介绍了Python ORM框架SQLAlchemy学习笔记之关系映射实例,Classic (经典模式)和Modern (现代模式),分别介绍了,需要的朋友可以参考下
    2014-06-06
  • Python周期任务神器之Schedule模块使用详解

    Python周期任务神器之Schedule模块使用详解

    这篇文章主要为大家详细介绍了Python中的周期任务神器—Schedule模块的安装和初级、进阶使用方法,文中的示例代码讲解详细,需要的可以参考一下
    2022-04-04
  • 解决List.append() 在 Python 中不起作用的问题

    解决List.append() 在 Python 中不起作用的问题

    在 Python 中,我们通常使用 List.append() 方法向列表末尾添加元素,然而,在某些情况下,你可能会遇到 List.append() 方法不起作用的问题,本文将详细讨论这个问题并提供解决方法,需要的朋友可以参考下
    2023-06-06
  • 利用Python第三方库实现预测NBA比赛结果

    利用Python第三方库实现预测NBA比赛结果

    今天给大家带来的是关于Python的相关知识,文章围绕着利用Python实现预测NBA比赛结果展开,文中有非常详细的介绍,需要的朋友可以参考下
    2021-06-06
  • python 如何获取元素在array中的下标

    python 如何获取元素在array中的下标

    这篇文章主要介绍了python 获取元素在array中的下标操作,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2021-05-05
  • python实现音乐下载器

    python实现音乐下载器

    这篇文章主要为大家详细介绍了python实现音乐下载器,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-04-04
  • uwsgi+anaconda+django启动问题的坑及解决

    uwsgi+anaconda+django启动问题的坑及解决

    作者记录了自己在使用uWSGI启动Django项目时遇到的问题,并最终找到了解决方案,问题出在Anaconda环境下安装uWSGI时,解决方法是使用conda安装uWSGI
    2026-02-02
  • 一道题学会Python函数中形参和实参

    一道题学会Python函数中形参和实参

    在Python编程中,函数的形参和实参是函数定义和调用的重要概念,本文主要介绍了一道题学会Python函数中形参和实参,具有一定的参考价值,感兴趣的可以了解一下
    2024-01-01

最新评论