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 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 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); } }