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安装第三方包的三种方法图文详解

    python安装第三方包的三种方法图文详解

    安装Python第三方包有多种方法,下面这篇文章主要给大家介绍了关于python安装第三方包的三种方法,文中通过图文的非常详细,需要的朋友可以参考下
    2024-03-03
  • python实现简易猜数小游戏

    python实现简易猜数小游戏

    大家好,本篇文章主要讲的是python实现简易猜数小游戏,感兴趣的同学赶快来看一看吧,对你有帮助的话记得收藏一下,方便下次浏览
    2022-01-01
  • django xadmin action兼容自定义model权限教程

    django xadmin action兼容自定义model权限教程

    这篇文章主要介绍了django xadmin action兼容自定义model权限教程,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-03-03
  • 只需要100行Python代码就可以实现的贪吃蛇小游戏

    只需要100行Python代码就可以实现的贪吃蛇小游戏

    贪吃蛇小游戏相信80、90后小时候肯定都玩过,那么你知道如果通过Python来实现吗?今天就来教大家,文中有非常详细的代码示例,对正在学习python的小伙伴们很有帮助,需要的朋友可以参考下
    2021-05-05
  • Python函数和模块的使用总结

    Python函数和模块的使用总结

    在本文中小编给大家整理了关于Python函数和模块的使用的相关知识点以及实例代码内容,需要的朋友们跟着学习下。
    2019-05-05
  • Python的PIL库中getpixel方法的使用

    Python的PIL库中getpixel方法的使用

    这篇文章主要介绍了Python的PIL库中getpixel方法的使用,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-04-04
  • 初学者快看,Python下划线的五个作用介绍

    初学者快看,Python下划线的五个作用介绍

    大家好,本篇文章主要讲的是初学者快看,Python下划线的五个作用介绍,感兴趣的同学赶快来看一看吧,对你有帮助的话记得收藏一下,方便下次浏览
    2021-12-12
  • Python实现大乐透号码随机生成

    Python实现大乐透号码随机生成

    全国有很多彩民,其中购买最多的彩种分别是体彩大乐透和福彩双色球。本篇文章将介绍Python实现彩票自由的全流程:随机选取号码+查看是否中奖,需要的可以参考一下
    2022-05-05
  • Python实现朗读在线音频和本地音频

    Python实现朗读在线音频和本地音频

    在日常的Python软件开发中,我们经常会遇到一个非常重要的功能需求——让程序能够读取并显示文本内容,下面我们就来学习一下Python实现朗读音频的具体操作吧
    2024-03-03
  • tensorflow训练中出现nan问题的解决

    tensorflow训练中出现nan问题的解决

    本篇文章主要介绍了tensorflow训练中出现nan问题的解决,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2018-02-02

最新评论