关于gocq的ws链接 断线重连的提议

[DeepseaXX]

我一直在用cq的ws链接，这种方式在断线后不会自动重新连接。想修改但是我的Java水平实在捉急。提问了一下ChatGPT这是我得到的回答…… 大佬能否帮忙完成这个更改？ 另外如果可以的话，还想实现在配置文件里的开关选项和断线重连时间的参数设置………

相对目录

src/main/java/moe/dituon/petpet/websocket/gocq/GoCQAPIWebSocketClient.java

ChatGPT回答如下

你可以通过在 `onClose` 和 `onError` 方法中实现自动重连逻辑。为了实现每5秒自动重连，你可以使用 `ScheduledExecutorService` 来设置定时任务，直到连接成功为止。

这里是修改后的代码：

```java
import moe.dituon.petpet.plugin.PluginPetService;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.handshake.ServerHandshake;

import java.net.URI;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

public class GoCQAPIWebSocketClient extends WebSocketClient {
    private static final int RECONNECT_INTERVAL_SECONDS = 5; // 每5秒重连一次
    private ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor();
    public GoCQRequester requester = new GoCQRequester(this);

    public GoCQAPIWebSocketClient(URI serverURI) {
        super(serverURI);
        connect();
    }

    @Override
    public void onOpen(ServerHandshake handshakedata) {
        PluginPetService.LOGGER.info("GoCQ Event WebSocket 连接成功");
        // 如果连接成功，关闭之前的重连任务
        if (!scheduler.isShutdown()) {
            scheduler.shutdownNow();
        }
    }

    @Override
    public void onClose(int code, String reason, boolean remote) {
        PluginPetService.LOGGER.info("GoCQ Event WebSocket 连接关闭 (" + code + "): " + reason);
        // 断开连接后，尝试重新连接
        attemptReconnect();
    }

    @Override
    public void onError(Exception ex) {
        PluginPetService.LOGGER.warning("an error occurred: " + ex);
        // 出现错误后，尝试重新连接
        attemptReconnect();
    }

    @Override
    public void onMessage(String message) {
        try {
            GoCQResponseDTO response = GoCQResponseDTO.parse(message);
            Long id = Long.parseLong(response.getEcho());
            ThreadLockObject<GoCQMemberDTO> lock = requester.getThreadLock(id);
            synchronized (lock) {
                lock.set(response.toGetGroupMemberResponseDTO().getData());
                lock.notify();
            }
        } catch (Exception ignored) {
        }
    }

    // 尝试每5秒重新连接
    private void attemptReconnect() {
        if (!scheduler.isShutdown()) {
            scheduler.shutdownNow();  // 停止之前的调度任务
        }

        scheduler = Executors.newSingleThreadScheduledExecutor();
        scheduler.scheduleAtFixedRate(() -> {
            PluginPetService.LOGGER.info("尝试重新连接 WebSocket...");
            try {
                reconnect();
            } catch (Exception e) {
                PluginPetService.LOGGER.warning("重新连接失败: " + e.getMessage());
            }
        }, RECONNECT_INTERVAL_SECONDS, RECONNECT_INTERVAL_SECONDS, TimeUnit.SECONDS);
    }
}

修改内容：

1. 引入 ScheduledExecutorService，用于定时任务。
2. 在 onClose 和 onError 中，当连接断开或出错时，启动一个定时任务，每5秒尝试调用 reconnect() 进行重新连接。
3. 在连接成功的 onOpen 方法中，取消定时任务以避免重复重连。

通过这种方式，当 WebSocket 断开或出错时，它会每5秒自动尝试重新连接，直到连接成功。