From 8d45c23811006e9bcc2dcf9bf321a2bbcf01f4d9 Mon Sep 17 00:00:00 2001
From: FongMi <b8892778@gmail.com>
Date: Thu, 20 Feb 2025 19:05:15 +0800
Subject: [PATCH] lookup realtime

---
 .../main/java/com/github/catvod/net/OkDns.java  | 17 +++--------------
 1 file changed, 3 insertions(+), 14 deletions(-)

diff --git a/catvod/src/main/java/com/github/catvod/net/OkDns.java b/catvod/src/main/java/com/github/catvod/net/OkDns.java
index 941d6a003..3b24b52a6 100644
--- a/catvod/src/main/java/com/github/catvod/net/OkDns.java
+++ b/catvod/src/main/java/com/github/catvod/net/OkDns.java
@@ -4,8 +4,6 @@ import androidx.annotation.NonNull;
 
 import java.net.InetAddress;
 import java.net.UnknownHostException;
-import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.List;
 import java.util.concurrent.ConcurrentHashMap;
 
@@ -14,7 +12,7 @@ import okhttp3.dnsoverhttps.DnsOverHttps;
 
 public class OkDns implements Dns {
 
-    private final ConcurrentHashMap<String, List<InetAddress>> map;
+    private final ConcurrentHashMap<String, String> map;
     private DnsOverHttps doh;
 
     public OkDns() {
@@ -35,23 +33,14 @@ public class OkDns implements Dns {
             String[] splits = host.split("=");
             String oldHost = splits[0];
             String newHost = splits[1];
-            if (!map.containsKey(oldHost)) map.put(oldHost, new ArrayList<>());
-            map.get(oldHost).addAll(getAllByName(newHost));
-        }
-    }
-
-    private List<InetAddress> getAllByName(String host) {
-        try {
-            return Arrays.asList(InetAddress.getAllByName(host));
-        } catch (Exception e) {
-            return new ArrayList<>();
+            map.put(oldHost, newHost);
         }
     }
 
     @NonNull
     @Override
     public List<InetAddress> lookup(@NonNull String hostname) throws UnknownHostException {
-        return map.containsKey(hostname) ? map.get(hostname) : (doh != null ? doh : Dns.SYSTEM).lookup(hostname);
+        return (doh != null ? doh : Dns.SYSTEM).lookup(map.containsKey(hostname) ? map.get(hostname) : hostname);
     }
 }