溫馨提示×

SpringBoot怎么實現(xiàn)點對點發(fā)送消息

小億
110
2024-01-29 09:47:43
欄目: 編程語言

在Spring Boot中，可以使用WebSocket實現(xiàn)點對點發(fā)送消息。WebSocket是一種在單個TCP連接上進行全雙工通信的協(xié)議，它可以在Web瀏覽器和服務(wù)器之間建立一個持久連接，使得服務(wù)器可以主動向客戶端推送數(shù)據(jù)。

以下是使用Spring Boot實現(xiàn)點對點發(fā)送消息的步驟：

  1. 添加WebSocket依賴：在pom.xml文件中添加以下依賴：
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
  1. 創(chuàng)建WebSocket配置類：創(chuàng)建一個配置類，用于配置WebSocket相關(guān)的信息。示例代碼如下：
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new WebSocketHandler(), "/ws").setAllowedOrigins("*");
    }
}

上述示例代碼中，WebSocketHandler()是自定義的WebSocket處理器，/ws是WebSocket的端點路徑。

  1. 創(chuàng)建WebSocket處理器：創(chuàng)建一個自定義的WebSocket處理器，用于處理WebSocket連接的建立、關(guān)閉和消息的發(fā)送。示例代碼如下：
public class WebSocketHandler extends TextWebSocketHandler {

    private static final List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        sessions.add(session);
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        // 處理收到的消息
        for (WebSocketSession s : sessions) {
            s.sendMessage(message);
        }
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        sessions.remove(session);
    }
}

上述示例代碼中，afterConnectionEstablished()方法在WebSocket連接建立時調(diào)用，handleTextMessage()方法用于處理收到的文本消息，afterConnectionClosed()方法在WebSocket連接關(guān)閉時調(diào)用。

  1. 創(chuàng)建Controller類：創(chuàng)建一個Controller類，用于處理與WebSocket相關(guān)的請求。示例代碼如下：
@RestController
@RequestMapping("/api")
public class MessageController {

    @Autowired
    private SimpMessagingTemplate messagingTemplate;

    @PostMapping("/send/{userId}")
    public void sendMessage(@PathVariable("userId") String userId, @RequestBody String message) {
        messagingTemplate.convertAndSendToUser(userId, "/ws", message);
    }
}

上述示例代碼中，sendMessage()方法用于發(fā)送消息到指定的WebSocket連接。

至此，點對點發(fā)送消息的實現(xiàn)就完成了。你可以通過向/api/send/{userId}發(fā)送POST請求，傳遞userId和message參數(shù)來發(fā)送消息。

請注意，上述示例只是演示了基本的點對點發(fā)送消息的實現(xiàn)方式。在實際應(yīng)用中，你可能需要根據(jù)具體的需求進行進一步的開發(fā)和優(yōu)化。

0 踩