使用Java WebSocket进行实时通信问题

原创深碍√TFBOYSˉ_ 2024-12-09 16:36 1阅读 0赞

WebSocket是一种在单个TCP连接上进行全双工通信的协议。在Java中，可以使用Java Web Socket API来实现WebSocket通信。

以下是一个简单的示例，展示了如何创建一个WebSocket服务器和客户端：

**服务器端（Server.java）：**

```java
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;

import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;

@ServerEndpoint("/websocket")
public class Server {

@OnOpen
    public void onOpen(Session session) {
        System.out.println("Client connected: " + session.getId());
        // Send welcome message to client
        sendMessage(session, "Welcome to the WebSocket server!"));
    }

@OnMessage
    public String onMessage(String message, Session session) {
        System.out.println("Received message from client: " + message);
        
        // Echo back the message to client
        return sendMessage(session, message);
    }

private String sendMessage(Session session, String message)) {
        try {
            session.getBasicRemote().sendText(message);
            return "Message sent successfully!";
        } catch (IOException e) {
            System.out.println("Error sending message: " + e.getMessage());
            return "Error sending message!";
        }
    }
}
```

**客户端（Client.java）：**

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

public class Client {

public static void main(String[] args) {
        // Connect to WebSocket server
        connectToServer("localhost", "/websocket"));

// Send message to server
        sendMessageToServer("Hello, WebSocket Server!");
    }

private static void connectToServer(String host, String path) {
        try (Session session = javax.websocket.Container.connectURI(host, path), "Client")) {
            System.out.println("Connected to WebSocket server: " + session.getId()));
            // Register close listener to handle disconnection
            session.addCloseListener(new CloseListener() {
                @Override
                public void onClose(Session session) {
                    System.out.println("WebSocket connection closed: " + session.getId()));
                }
            }));
        } catch (Exception e) {
            System.out.println("Error connecting to WebSocket server: " + e.getMessage());
        }
    }

private static void sendMessageToServer(String message) {
        try (Session session = javax.websocket.Container.connectURI("localhost", "/websocket"), "Client")) {
            System.out.println("Sending message to WebSocket server: " + message);
            
            // Send message using sendText method
            session.getBasicRemote().sendText(message);

// Register close listener to handle disconnection
            session.addCloseListener(new CloseListener() {
                @Override
                public void onClose(Session session) {
                    System.out.println("WebSocket connection closed: " + session.getId()));
                }
            }));
        } catch (Exception e) {
            System.out.println("Error sending message to WebSocket server: " + e.getMessage());
        }
    }

private interface CloseListener {
        void onClose(Session session);
    }
}
```

这个例子中，服务器端监听WebSocket连接，并在接收到消息时进行回应。客户端连接到服务器并发送一条消息。

请注意，这个示例需要在运行Java Web Socket服务器的环境中进行。