Java WebSocket 客户端数据传递方法

本文介绍如何使用观察者模式解决 Java 中 WebSocket 客户端向父级管理器传递数据的问题。通过将父级管理器注册为 WebSocket 客户端的观察者，客户端可以在接收到消息后，通过通知观察者的方式将数据传递给管理器，从而实现高效的数据共享和管理。

观察者模式实现数据传递

观察者模式是一种行为型设计模式，它定义了对象之间的一对多依赖关系，当一个对象的状态发生改变时，所有依赖于它的对象都会得到通知并自动更新。 在本例中，我们可以将 ClientManager 作为观察者，AuthClient 作为被观察者。当 AuthClient 接收到消息时，它会通知 ClientManager，并将消息数据传递给它。

1. 定义观察者接口

首先，我们需要定义一个观察者接口，该接口定义了观察者接收通知的方法。

public interface Observer {
    void update(String message);
}

2. 定义被观察者类

接下来，我们需要修改 AuthClient 类，使其实现被观察者功能。这包括：

• 添加一个观察者列表。
• 提供注册观察者的方法 (registerObserver)。
• 提供移除观察者的方法 (removeObserver)。
• 提供通知观察者的方法 (notifyObservers)。

import java.util.ArrayList;
import java.util.List;

public class AuthClient extends WebSocketClient {

    private List observers = new ArrayList<>();

    public void registerObserver(Observer observer) {
        observers.add(observer);
    }

    public void removeObserver(Observer observer) {
        observers.remove(observer);
    }

    public void notifyObservers(String message) {
        for (Observer observer : observers) {
            observer.update(message);
        }
    }

    @Override
    public void onMessage(String message) {
        switch (message) {
            // data events here
            default:
                notifyObservers(message); // 通知观察者
                break;
        }
    }
}

3. 实现观察者类

现在，我们需要修改 ClientManager 类，使其实现 Observer 接口。 在 update 方法中，我们可以处理从 AuthClient 接收到的数据。

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;

public class ClientManager implements Observer {

    private int maxClients;
    private final Map clients;
    private final HashSet tokens;

    public ClientManager(HashSet tokens, int maxClients) {
        this.tokens = tokens;
        this.maxClients = maxClients;
        this.clients = new HashMap<>(maxClients);
    }

    @Override
    public void update(String message) {
        // 处理从 AuthClient 接收到的消息
        System.out.println("ClientManager received message: " + message);
        //  在这里可以根据消息内容进行相应的处理，例如更新客户端状态等
    }

    public void addClient(String clientId, AuthClient client) {
        clients.put(clientId, client);
        client.registerObserver(this); // 注册观察者
    }

    public void removeClient(String clientId) {
        AuthClient client = clients.get(clientId);
        if (client != null) {
            client.removeObserver(this); // 移除观察者
            clients.remove(clientId);
        }
    }
}

4. 使用示例

以下是一个使用示例，展示了如何创建 AuthClient 和 ClientManager，并将 ClientManager 注册为 AuthClient 的观察者。

public class Main {
    public static void main(String[] args) {
        HashSet tokens = new HashSet<>();
        tokens.add("token1");
        tokens.add("token2");

        ClientManager clientManager = new ClientManager(tokens, 10);
        AuthClient authClient = new AuthClient();

        clientManager.addClient("client1", authClient);

        // 模拟 AuthClient 接收到消息
        authClient.onMessage("hello");
        authClient.onMessage("pending_finish");
        authClient.onMessage("finish");

        clientManager.removeClient("client1");
    }
}

总结

使用观察者模式可以有效地解决 WebSocket 客户端向父级管理器传递数据的问题。 该模式具有以下优点：

• 松耦合： AuthClient 和 ClientManager 之间解耦，AuthClient 不需要知道 ClientManager 的具体实现。
• 可扩展性： 可以轻松地添加新的观察者，而无需修改 AuthClient 的代码。
• 灵活性： 可以根据需要注册和移除观察者。

注意事项：

• 在多线程环境下，需要考虑线程安全问题。 可以使用线程安全的集合类，例如 ConcurrentHashMap 和 CopyOnWriteArrayList。
• 避免循环依赖，例如 ClientManager 观察 AuthClient，而 AuthClient 又观察 ClientManager。 这会导致无限循环。

通过使用观察者模式，我们可以实现一个高效、灵活且可扩展的 WebSocket 客户端管理系统。