SpringBoot中@MessageMapping注解的原理及使用詳解

更新時間：2023年07月21日 08:34:37 作者：硬件人某某某

這篇文章主要介紹了SpringBoot中@MessageMapping注解的原理及使用詳解,@MessageMapping注解是Spring Boot提供的一個重要的注解之一，它可以幫助我們處理WebSocket消息,需要的朋友可以參考下

@MessageMapping注解

在Web應(yīng)用程序中，實現(xiàn)實時的雙向通信是一項重要的功能。為了實現(xiàn)這種功能，需要使用WebSocket協(xié)議。

Spring框架提供了Spring WebSocket模塊來實現(xiàn)WebSocket通信。Spring Boot是基于Spring框架構(gòu)建的，它提供了一些方便的注解和自動配置來簡化WebSocket的開發(fā)。

@MessageMapping注解是Spring Boot提供的一個重要的注解之一，它可以幫助我們處理WebSocket消息。本文將介紹@MessageMapping注解的原理及使用方法。

原理

在Spring Boot中，@MessageMapping注解用于標(biāo)識WebSocket消息的處理方法。當(dāng)客戶端發(fā)送一個WebSocket消息到服務(wù)器端時，@MessageMapping注解可以將該消息路由到對應(yīng)的處理方法中進(jìn)行處理。

@MessageMapping注解的使用方法類似于@RequestMapping注解。

我們可以將@MessageMapping注解應(yīng)用在控制器的處理方法上，以處理特定的WebSocket消息。

當(dāng)客戶端發(fā)送一個特定的消息到服務(wù)器端時，服務(wù)器就會調(diào)用標(biāo)記有@MessageMapping注解的處理方法來處理該消息。

下面是一個簡單的示例：

@Controller
public class WebSocketController {
    @MessageMapping("/hello")
    @SendTo("/topic/greetings")
    public Greeting greeting(HelloMessage message) throws Exception {
        Thread.sleep(1000); // simulated delay
        return new Greeting("Hello, " + message.getName() + "!");
    }
}

在上面的例子中，我們定義了一個WebSocket控制器，并在其中定義了一個標(biāo)記有@MessageMapping注解的處理方法。

該處理方法用于處理名為“/hello”的WebSocket消息，并將處理結(jié)果發(fā)送到“/topic/greetings”主題。

在Spring Boot應(yīng)用程序中，我們可以通過使用@EnableWebSocketMessageBroker注解來啟用WebSocket消息代理。

當(dāng)該注解被應(yīng)用時，Spring Boot將自動配置WebSocket消息代理，以便我們可以使用@MessageMapping和@SendTo等注解來處理WebSocket消息。

如何使用

使用@MessageMapping注解處理WebSocket消息的過程非常簡單。

我們只需要在控制器的處理方法上添加@MessageMapping注解，并指定該方法應(yīng)該處理的WebSocket消息類型即可。

下面是一個更完整的示例：

@Controller
public class WebSocketController {
    @MessageMapping("/chat/{roomId}")
    @SendTo("/topic/chat/{roomId}")
    public ChatMessage handleChatMessage(@DestinationVariable String roomId, ChatMessage message) {
        return message;
    }
}

在上面的例子中，我們定義了一個WebSocket控制器，并在其中定義了一個標(biāo)記有@MessageMapping注解的處理方法。

該處理方法用于處理名為“/chat/{roomId}”的WebSocket消息，并將處理結(jié)果發(fā)送到“/topic/chat/{roomId}”主題。

在我們的處理方法中，我們可以使用@DestinationVariable注解來獲取WebSocket消息中的路徑變量。在這個例子中，我們使用@DestinationVariable來獲取roomId路徑變量的值。

除了@MessageMapping注解之外，Spring Boot還提供了許多其他的注解和工具，以幫助我們更容易地處理WebSocket消息。

例如，@SendTo注解可以將處理結(jié)果發(fā)送到指定的WebSocket主題；

@Payload注解可以用于標(biāo)識WebSocket消息的有效負(fù)載；

@Payload注解可以用于將WebSocket消息的有效負(fù)載反序列化為Java對象。

示例代碼

下面是一個完整的Spring Boot WebSocket應(yīng)用程序代碼示例：

@SpringBootApplication
@EnableWebSocketMessageBroker
public class WebSocketDemoApplication {
    public static void main(String[] args) {
        SpringApplication.run(WebSocketDemoApplication.class, args);
    }
    @Configuration
    @EnableWebSocket
    public class WebSocketConfig implements WebSocketConfigurer {
        @Override
        public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
            registry.addHandler(chatWebSocketHandler(), "/chat/{roomId}").setAllowedOrigins("*");
        }
        @Bean
        public WebSocketHandler chatWebSocketHandler() {
            return new ChatWebSocketHandler();
        }
    }
    @Controller
    public class WebSocketController {
        @MessageMapping("/chat/{roomId}")
        @SendTo("/topic/chat/{roomId}")
        public ChatMessage handleChatMessage(@DestinationVariable String roomId, ChatMessage message) {
            return message;
        }
    }
    public class ChatMessage {
        private String content;
        private String sender;
        // getters and setters
    }
    public class ChatWebSocketHandler extends TextWebSocketHandler {
        private final ObjectMapper objectMapper = new ObjectMapper();
        @Override
        public void handleTextMessage(WebSocketSession session, TextMessage message) throws IOException {
            ChatMessage chatMessage = objectMapper.readValue(message.getPayload(), ChatMessage.class);
            // do something with the chatMessage
        }
        @Override
        public void afterConnectionEstablished(WebSocketSession session) throws Exception {
            // do something after a WebSocket connection is established
        }
    }
}

在上面的代碼示例中，我們定義了一個名為“WebSocketDemoApplication”的Spring Boot應(yīng)用程序。

該應(yīng)用程序使用@EnableWebSocketMessageBroker注解啟用WebSocket消息代理，并定義了一個WebSocket控制器和一個WebSocket處理程序。

在WebSocket控制器中，我們定義了一個標(biāo)記有@MessageMapping注解的處理方法，該方法用于處理名為“/chat/{roomId}”的WebSocket消息，并將處理結(jié)果發(fā)送到名為“/topic/chat/{roomId}”的主題。

在WebSocket處理程序中，我們使用TextWebSocketHandler來處理WebSocket消息。

在handleTextMessage方法中，我們使用ObjectMapper來將WebSocket消息的有效負(fù)載反序列化為ChatMessage對象，然后進(jìn)行一些處理。

在afterConnectionEstablished方法中，我們可以執(zhí)行一些WebSocket連接建立后的操作。