使用Java WebSocket进行实时通信问题-蒲公英云

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

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

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

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

服务器端（Server.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）：

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);
    }
}