feat: Stop AfreecaTvChatCrawler on missing live broadcast info

Author: Lede-dev

src/main/java/net/ledestudio/acc/http/AccHttpRequester.java

@@ -1,5 +1,6 @@
 package net.ledestudio.acc.http;
 
+import com.google.gson.JsonElement;
 import com.google.gson.JsonObject;
 import com.google.gson.JsonParser;
 import org.apache.http.HttpEntity;
@@ -9,6 +10,7 @@
 import org.apache.http.impl.client.HttpClients;
 import org.apache.http.util.EntityUtils;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
@@ -41,7 +43,7 @@ public AccHttpRequester(@NotNull String afreecaTvLiveUrl) {
         this.bno = split[1];
     }
 
-    public @NotNull CompletableFuture<AccHttpRequestResult> request() {
+    public @NotNull CompletableFuture<@Nullable AccHttpRequestResult> request() {
         return CompletableFuture.supplyAsync(() -> {
             String url = "https://live.afreecatv.com/afreeca/player_live_api.php";
 
@@ -58,7 +60,14 @@ public AccHttpRequester(@NotNull String afreecaTvLiveUrl) {
                 String responseStr = EntityUtils.toString(entity, StandardCharsets.UTF_8);
 
                 JsonObject root = JsonParser.parseString(responseStr).getAsJsonObject();
+                if (root == null || !root.has("CHANNEL")) {
+                    return null;
+                }
+
                 JsonObject channel = root.getAsJsonObject("CHANNEL");
+                if (channel == null || !channel.has("CHDOMAIN")) {
+                    return null;
+                }
 
                 // mapping result
                 AccHttpRequestResult result = new AccHttpRequestResult();
@@ -75,8 +84,7 @@ public AccHttpRequester(@NotNull String afreecaTvLiveUrl) {
                 System.err.println("An error occurred during HTTP request execution: " + e.getMessage());
             }
 
-            // return empty result
-            return new AccHttpRequestResult();
+            return null;
         });
     }
 

src/main/java/net/ledestudio/acc/service/AfreecaTvChatCrawler.java

@@ -17,9 +17,12 @@
 import java.security.NoSuchAlgorithmException;
 import java.util.Set;
 import java.util.concurrent.*;
+import java.util.logging.Logger;
 
 public class AfreecaTvChatCrawler {
 
+    private static final Logger LOGGER = Logger.getLogger(AfreecaTvChatCrawler.class.getSimpleName());
+
     private AccClient client;
     private AccHttpRequestResult result;
 
@@ -65,9 +68,12 @@ public void connect() {
         Executors.newSingleThreadExecutor().execute(() -> {
             try {
                 // Request AfreecaTv Live Information
-                final AccHttpRequester requester = new AccHttpRequester(url);
-                final CompletableFuture<AccHttpRequestResult> future = requester.request();
-                this.result = future.get();
+                this.result = getLiveBroadcastResult();
+
+                if (result == null) {
+                    LOGGER.warning("Unable to find live broadcast information.");
+                    return;
+                }
 
                 // Create Draft
                 Draft_6455 draft = new Draft_6455(Lists.newArrayList(),
@@ -103,9 +109,16 @@ public void connect() {
     public void reconnect() {
         if (client != null) {
             try {
+                this.result = getLiveBroadcastResult();
+                if (result == null) {
+                    LOGGER.warning("Unable to find live broadcast information.");
+                    close();
+                    return;
+                }
+
                 client.reconnectBlocking();
                 sendPacket();
-            } catch (InterruptedException e) {
+            } catch (InterruptedException | ExecutionException e) {
                 e.printStackTrace();
             }
         }
@@ -120,6 +133,12 @@ public void close() {
         }
     }
 
+    private @Nullable AccHttpRequestResult getLiveBroadcastResult() throws ExecutionException, InterruptedException {
+        final AccHttpRequester requester = new AccHttpRequester(url);
+        final CompletableFuture<AccHttpRequestResult> future = requester.request();
+        return future.get();
+    }
+
     private void sendPacket() {
         if (client == null) {
             return;
@@ -136,7 +155,6 @@ private void sendPacket() {
         // Repeatedly Sending Ping Packet
         scheduler = Executors.newSingleThreadScheduledExecutor();
         scheduler.scheduleAtFixedRate(() -> {
-            System.out.println("Send Ping Packet");
             client.send(AccConstants.createPingPacket());
         }, AccConstants.PING_PACKET_PERIOD_SECONDS, AccConstants.PING_PACKET_PERIOD_SECONDS, TimeUnit.SECONDS);
     }

src/test/java/net/ledestudio/acc/AccHttpTest.java

@@ -14,6 +14,11 @@ public void accHttpRequesterUrlConstructorTest() throws ExecutionException, Inte
         AccHttpRequester requester = new AccHttpRequester("https://play.afreecatv.com/seokwngud/262914674");
         CompletableFuture<AccHttpRequestResult> future = requester.request();
         AccHttpRequestResult result = future.get();
+
+        if (result == null) {
+            System.out.println("Cannot found live broadcast information");
+            return;
+        }
         System.out.println(result.toString());
     }
 

src/test/java/net/ledestudio/acc/AccWebSocketTest.java

@@ -9,6 +9,7 @@
 import org.java_websocket.client.WebSocketClient;
 import org.java_websocket.drafts.Draft_6455;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 import org.junit.jupiter.api.Test;
 
 import java.time.format.DateTimeFormatter;
@@ -19,8 +20,14 @@ public class AccWebSocketTest {
     @Test
     public void accWebSocketClientConnectTest() throws ExecutionException, InterruptedException {
         AccHttpRequester requester = new AccHttpRequester("https://play.afreecatv.com/seokwngud/262914674");
-        CompletableFuture<AccHttpRequestResult> future = requester.request();
-        WebSocketClient client = new AccClient(future.get(), new Draft_6455());
+        CompletableFuture<@Nullable AccHttpRequestResult> future = requester.request();
+        AccHttpRequestResult result = future.get();
+        if (result == null) {
+            System.out.println("Cannot found live broadcast information");
+            return;
+        }
+
+        WebSocketClient client = new AccClient(result, new Draft_6455());
         client.connectBlocking();
     }