使用Java WebSocket实现双向通信的实例-蒲公英云

使用Java WebSocket实现双向通信的实例

原创电玩女神 2024-10-09 22:24 114阅读 0赞

以下是一个简单的Java WebSocket实现双向通信的例子。这个例子使用了Java Websocket API和Spring Boot框架。

首先，我们需要创建一个WebSocket服务类：

import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import java.io.IOException;
public class WebSocketService {
    public void sendMessage(WebSocketSession session, String message) {
        try {
            session.getAsyncRemote().sendText(message);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    // 当连接关闭时，我们调用此方法
    public void onConnectClose(WebSocketSession session) {
        System.out.println("Connection closed: " + session.getId());
    }
}

然后，在Spring Boot的配置类中配置WebSocket服务器：

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.EnableWebSocketMessageBroker;
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig {
    // 设置消息中间件类型，这里设置为"topic"
    @Bean
    public MessageBrokerConfiguration messageBrokerConfig() {
        MessageBrokerConfiguration config = new MessageBrokerConfiguration();
        config.setMessageBrokerType(MessageBrokerTypes.TOPIC);
        return config;
    }
    // 自定义连接关闭处理器
    @Bean
    public WebSocketServiceContainer postConnectCloseProcessor() {
        WebSocketServiceContainer processor = new WebSocketServiceContainer();
        processor.addWebSocketService(new WebSocketService());
        processor.getServices().forEach(service -> {
            if (service instanceof WebSocketService) {
                ((WebSocketService) service).setOnConnectCloseListener(processor::handleOnConnectClose));
            }
        }));
        return processor;
    }
    // 自定义连接关闭处理器处理逻辑
    private void handleOnConnectClose(WebSocketSession session) {
        System.out.println("Connection closed with reason: " + session.getAsyncRemote().sendText(session.getId() + " - Reason: Closed manually")) + "\nSession ID: " + session.getId());
    }
}