ObjectPool IsUdpEnabled

mili-tan · Nov 7, 2023 · eeffa25 · eeffa25
1 parent 5a8416d
commit eeffa25
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Arashi.Aoi/Routes/DnsQueryRoutes.cs b/Arashi.Aoi/Routes/DnsQueryRoutes.cs
@@ -21,10 +21,12 @@ class DnsQueryRoutes
         public static IPEndPoint BackUpEndPoint = IPEndPoint.Parse(Config.BackUpStream);
 
         public static ObjectPool<DnsClient> UpPool = new(() =>
-            new DnsClient(UpEndPoint.Address, Config.TimeOut, UpEndPoint.Port != 0 ? UpEndPoint.Port : 53));
+            new DnsClient(UpEndPoint.Address, Config.TimeOut, UpEndPoint.Port != 0 ? UpEndPoint.Port : 53)
+                {IsUdpEnabled = !Config.OnlyTcpEnable, IsTcpEnabled = true});
 
         public static ObjectPool<DnsClient> BackUpPool = new(() =>
-            new DnsClient(BackUpEndPoint.Address, Config.TimeOut, BackUpEndPoint.Port != 0 ? UpEndPoint.Port : 53));
+            new DnsClient(BackUpEndPoint.Address, Config.TimeOut, BackUpEndPoint.Port != 0 ? UpEndPoint.Port : 53)
+                {IsUdpEnabled = !Config.OnlyTcpEnable, IsTcpEnabled = true});
 
 
         public static void DnsQueryRoute(IEndpointRouteBuilder endpoints)