重写

jianjianai · jianjianai · commit bb6e1fff6943 · 2023-04-01T21:03:25.000+08:00
diff --git a/src/main/java/cn/jja8/newbinggogo/NewBingGoGoServer.java b/src/main/java/cn/jja8/newbinggogo/NewBingGoGoServer.java
@@ -10,8 +10,11 @@
 import java.net.URL;
 import java.util.HashMap;
 import java.util.Map;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
 
 public class NewBingGoGoServer extends NanoWSD {
+    ScheduledExecutorService scheduledExecutorService = Executors.newSingleThreadScheduledExecutor();
     public static void main(String[] args) {
         if(args.length<1){
             System.err.print("需要指定运行端口号！");
@@ -67,7 +70,7 @@ public Response serve(IHTTPSession session) {
 
     @Override
     protected WebSocket openWebSocket(IHTTPSession handshake) {
-        return new NewBingGoGoServerWebSocket(handshake);
+        return new NewBingGoGoServerWebSocket(handshake,scheduledExecutorService);
     }
 
     /*
diff --git a/src/main/java/cn/jja8/newbinggogo/NewBingGoGoServerWebSocket.java b/src/main/java/cn/jja8/newbinggogo/NewBingGoGoServerWebSocket.java
@@ -7,24 +7,41 @@
 import java.net.URI;
 import java.net.URISyntaxException;
 import java.util.LinkedList;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.ScheduledFuture;
+import java.util.concurrent.TimeUnit;
 
 public class NewBingGoGoServerWebSocket extends NanoWSD.WebSocket {
     NewBingGoGoClientWebSocket newBingGoGoClientWebSocket;
     LinkedList<String> messList = new LinkedList<>();
-    public NewBingGoGoServerWebSocket(NanoHTTPD.IHTTPSession handshakeRequest) {
+    ScheduledExecutorService scheduledExecutorService;
+    ScheduledFuture<?> task;
+
+    public NewBingGoGoServerWebSocket(NanoHTTPD.IHTTPSession handshakeRequest, ScheduledExecutorService scheduledExecutorService) {
         super(handshakeRequest);
         URI url;
         try {
             url = new URI("wss://sydney.bing.com/sydney/ChatHub");
         } catch (URISyntaxException e) {
             throw new RuntimeException(e);//这个异常这辈子都不会出的
         }
-
+        this.scheduledExecutorService = scheduledExecutorService;
         newBingGoGoClientWebSocket = new NewBingGoGoClientWebSocket(url,this,messList);
     }
 
     @Override
     protected void onOpen() {
+        task = scheduledExecutorService.scheduleWithFixedDelay(() -> {
+            if (!isOpen()) {
+                task.cancel(false);
+                return;
+            }
+            try {
+                ping(new byte[1]);
+            } catch (IOException e) {
+                task.cancel(false);
+            }
+        },2,2, TimeUnit.SECONDS);
         newBingGoGoClientWebSocket.connect();
     }
 
