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

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

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

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

```java
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服务器：

```java
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());
    }
}
```

这个例子中，当一个客户端连接到WebSocket服务器时，服务器会调用`onConnectClose`方法处理连接关闭事件。同时，你也可以在`handleOnConnectClose`方法中添加你的业务逻辑。