java实现websocket怎样处理消息 - 问答

在Java中，我们可以使用Java API for WebSocket (JSR 356)来实现WebSocket服务器。要处理WebSocket消息，你需要遵循以下步骤：

添加依赖首先，确保你的项目中包含了Java WebSocket API的依赖。如果你使用的是Maven，可以在pom.xml文件中添加以下依赖：

<dependency>
    <groupId>javax.websocket</groupId>
    <artifactId>javax.websocket-api</artifactId>
    <version>1.1</version>
</dependency>

创建WebSocket端点类接下来，创建一个实现javax.websocket.Endpoint接口的类，该类将作为WebSocket服务器的端点。在这个类中，你可以重写onOpen、onMessage、onClose和onError方法来处理WebSocket连接的生命周期事件和消息。

import javax.websocket.Endpoint;
import javax.websocket.EndpointConfig;
import javax.websocket.MessageHandler;
import javax.websocket.PongMessage;
import javax.websocket.Session;
import java.io.IOException;
import java.nio.ByteBuffer;

public class MyWebSocketEndpoint extends Endpoint {

    @Override
    public void onOpen(Session session, EndpointConfig config) {
        System.out.println("WebSocket连接已打开: " + session.getId());

        // 添加消息处理器
        session.addMessageHandler(new MessageHandler.Whole<String>() {
            @Override
            public void onMessage(String message) {
                System.out.println("收到客户端消息: " + message);
                // 处理消息并发送响应
                session.getAsyncRemote().sendText("服务器已收到消息: " + message, result -> {
                    if (result.getException() != null) {
                        System.out.println("发送响应失败: " + result.getException());
                    }
                });
            }
        });
    }

    @Override
    public void onClose(Session session, CloseReason closeReason) {
        System.out.println("WebSocket连接已关闭: " + session.getId());
    }

    @Override
    public void onError(Session session, Throwable throwable) {
        System.out.println("WebSocket发生错误: " + throwable.getMessage());
    }
}

创建WebSocket服务器现在，你需要创建一个WebSocket服务器来监听客户端的连接请求。你可以使用javax.websocket.server.ServerContainer接口来实现这一点。通常，你可以在Servlet容器（如Tomcat）中注册一个WebSocket端点，或者使用嵌入式服务器（如Jetty）来运行WebSocket服务器。

以下是使用Servlet 3.1容器（如Tomcat）注册WebSocket端点的示例：

import javax.websocket.ContainerProvider;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;

@ServerEndpoint("/my-websocket")
public class MyWebSocketEndpoint {

    @OnOpen
    public void onOpen(Session session) {
        System.out.println("WebSocket连接已打开: " + session.getId());
    }

    @OnMessage
    public void onMessage(String message, Session session) {
        System.out.println("收到客户端消息: " + message);
        // 处理消息并发送响应
        try {
            session.getBasicRemote().sendText("服务器已收到消息: " + message, result -> {
                if (result.getException() != null) {
                    System.out.println("发送响应失败: " + result.getException());
                }
            });
        } catch (IOException e) {
            System.out.println("发送响应失败: " + e.getMessage());
        }
    }

    @OnClose
    public void onClose(Session session) {
        System.out.println("WebSocket连接已关闭: " + session.getId());
    }

    @OnError
    public void onError(Session session, Throwable throwable) {
        System.out.println("WebSocket发生错误: " + throwable.getMessage());
    }
}

部署和运行将你的应用程序部署到支持WebSocket的Servlet容器（如Tomcat）中，并启动服务器。现在，当客户端连接到/my-websocket端点时，MyWebSocketEndpoint类中的onMessage方法将被调用，你可以在该方法中处理接收到的消息。

0 赞

0 踩