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

原创小灰灰 2024-12-02 09:51 13阅读 0赞

在Java中，我们可以使用Java WebSocket API来实现双向通信。以下是一个简单的例子，展示了如何创建WebSocket服务器和客户端进行交互。

首先，我们需要导入相关的库：

```java
import javax.websocket.ContainerProvider;
import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
```

然后，我们可以定义一个WebSocket服务端：

```java
public class WebSocketServer {
    @OnOpen
    public void onOpen(Session session) {
        System.out.println("Client connected: " + session.getUser().getName());
    }

// Handle message events
    @OnMessage
    public String handleMessage(String message) {
        System.out.println("Received message: " + message);
        
        // For demonstration purposes, send back the same message
        return message;
    }

// Handle close events
    @OnClose
    public void onClose(Session session) {
        System.out.println("Client disconnected: " + session.getUser().getName());
    }
}
```

接下来，我们可以创建一个WebSocket客户端来发送消息：

```java
public class WebSocketClient {
    private Session session;

public WebSocketClient(String url) {
        ContainerProvider.getContainerProvider()
                .newSession参用url参数开启连接
        }

// Connect to the server
        public void connect() {
            try {
                session = ContainerProvider.getContainerProvider()
                        .newSession("ws://" + getHost() + "/websocket");
                
                System.out.println("Connected to WebSocket server.");
            } catch (Exception e) {
                System.out.println("Failed to connect: " + e.getMessage());
            }
        }

// Send a message
        public void sendMessage(String message) {
            try {
                session.getBasicRemote().sendText(message);
                
                System.out.println("Message sent successfully: " + message);
            } catch (Exception e) {
                System.out.println("Failed to send message: " + e.getMessage());
            }
        }

// Close the connection
        public void close() {
            try {
                session.close();
                
                System.out.println("WebSocket connection closed.");
            } catch (Exception e) {
                System.out.println("Failed to close connection: " + e.getMessage());
            }
        }

private String getHost() {
            // Replace with your actual host or IP
            return "your_host";
        }
    }
}
```

现在，你可以创建WebSocket客户端对象，然后调用`connect()`方法连接到服务器。接着，你可以在任何时候发送消息或者关闭连接。

注意：在实际应用中，你需要替换"your_host"为你的实际主机或IP地址。