From 2303ff55329a20ad833fda5434153a6b9359883a Mon Sep 17 00:00:00 2001
From: FongMi <b8892778@gmail.com>
Date: Wed, 12 Feb 2025 00:31:30 +0800
Subject: [PATCH] Clean code

---
 catvod/src/main/java/com/github/catvod/net/OkDns.java | 8 ++++----
 1 file changed, 4 insertions(+), 4 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 04a6fec42..eac49cd4f 100644
--- a/catvod/src/main/java/com/github/catvod/net/OkDns.java
+++ b/catvod/src/main/java/com/github/catvod/net/OkDns.java
@@ -8,17 +8,18 @@ import java.util.HashMap;
 import java.util.List;
 
 import okhttp3.Dns;
+import okhttp3.dnsoverhttps.DnsOverHttps;
 
 public class OkDns implements Dns {
 
     private final HashMap<String, String> map;
-    private Dns doh;
+    private DnsOverHttps doh;
 
     public OkDns() {
         this.map = new HashMap<>();
     }
 
-    public void setDoh(Dns doh) {
+    public void setDoh(DnsOverHttps doh) {
         this.doh = doh;
     }
 
@@ -33,8 +34,7 @@ public class OkDns implements Dns {
     @NonNull
     @Override
     public List<InetAddress> lookup(@NonNull String hostname) throws UnknownHostException {
-        if (map.containsKey(hostname)) return Dns.SYSTEM.lookup(map.get(hostname));
-        return doh != null ? doh.lookup(hostname) : Dns.SYSTEM.lookup(hostname);
+        return (doh != null ? doh : Dns.SYSTEM).lookup(map.containsKey(hostname) ? map.get(hostname) : hostname);
     }
 }