test: 添加dns解析慢的单元测试

Author: meethigher

src/test/java/top/meethigher/proxy/http/DnsResolverTest.java

@@ -0,0 +1,78 @@
+package top.meethigher.proxy.http;
+
+import io.vertx.core.Vertx;
+import io.vertx.core.VertxOptions;
+import io.vertx.core.dns.DnsClient;
+import io.vertx.core.dns.DnsClientOptions;
+
+import java.net.InetAddress;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.atomic.AtomicLong;
+
+public class DnsResolverTest {
+
+    public static void main(String[] args) throws Exception {
+        String hostname = "reqres.in";
+        int iterations = 10;
+
+        System.out.println("Testing with InetAddress...");
+        // 当前没有该host缓存时，会直接sun.net.spi.nameservice.NameService.lookupAllHostAddr
+        testWithInetAddress(hostname, iterations);
+
+        System.out.println("Testing with Vert.x DnsClient...");
+        // io.vertx.core.dns.impl.DnsClientImpl.Query.run
+        testWithVertx(hostname, iterations);
+
+
+    }
+
+    private static void testWithVertx(String hostname, int iterations) throws InterruptedException {
+        Vertx vertx = Vertx.vertx(new VertxOptions()
+                // .setAddressResolverOptions(new AddressResolverOptions()
+                //         .setCacheMinTimeToLive(60)  // DNS 缓存时间，减少重复解析
+                //         .setCacheMaxTimeToLive(600)
+                //         .setQueryTimeout(5000))
+
+        );
+
+        DnsClient dnsClient = vertx.createDnsClient(
+                new DnsClientOptions()
+                        .setQueryTimeout(10000)
+        );
+
+        CountDownLatch latch = new CountDownLatch(iterations);
+        AtomicLong totalTime = new AtomicLong();
+
+        for (int i = 0; i < iterations; i++) {
+            long start = System.nanoTime();
+            dnsClient.lookup(hostname, result -> {
+                long duration = System.nanoTime() - start;
+                if (result.succeeded()) {
+                    System.out.println("Vert.x resolved: " + result.result() + " in " + duration / 1_000_000.0 + " ms");
+                } else {
+                    System.err.println("Vert.x failed to resolve: " + result.cause());
+                }
+                totalTime.addAndGet(duration);
+                latch.countDown();
+            });
+        }
+
+        latch.await();
+        System.out.println("Vert.x Average Time: " + (totalTime.get() / iterations) / 1_000_000.0 + " ms");
+        vertx.close();
+    }
+
+    private static void testWithInetAddress(String hostname, int iterations) throws Exception {
+        long totalTime = 0;
+
+        for (int i = 0; i < iterations; i++) {
+            long start = System.nanoTime();
+            InetAddress address = InetAddress.getByName(hostname);
+            long duration = System.nanoTime() - start;
+            System.out.println("InetAddress resolved: " + address.getHostAddress() + " in " + duration / 1_000_000.0 + " ms");
+            totalTime += duration;
+        }
+
+        System.out.println("InetAddress Average Time: " + (totalTime / iterations) / 1_000_000.0 + " ms");
+    }
+}

src/test/java/top/meethigher/proxy/http/ReverseHttpProxyTest.java

@@ -8,6 +8,8 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.util.Arrays;
+import java.util.Collections;
 import java.util.LinkedHashMap;
 import java.util.Map;
 import java.util.concurrent.ExecutionException;
@@ -210,4 +212,43 @@ private void proxyUrlCommon(Map<String, String> cases) throws InterruptedExcepti
         Assert.assertTrue(actual);
     }
 
+
+    /**
+     * 代理服务开启h2c（HTTP/2 cleartext）
+     * 后端服务也支持http2
+     */
+    @Test
+    public void testServerHttp2Proxy() throws Exception {
+        Router router = Router.router(vertx);
+        HttpClientOptions httpClientOptions = new HttpClientOptions()
+                .setTrustAll(true)
+                .setVerifyHost(false)
+                // httpclient支持与后端服务进行协商使用使用http2
+                .setUseAlpn(true)
+                .setProtocolVersion(HttpVersion.HTTP_2);
+        HttpServerOptions httpServerOptions = new HttpServerOptions()
+                // 服务端支持与客户端进行协商，使用h2c（HTTP/2 cleartext）
+                // 常规情况下，h2只在开启了tls使用。如果不开启tls，需要指定使用的是h2c
+                .setAlpnVersions(Collections.unmodifiableList(Arrays.asList(HttpVersion.HTTP_1_1, HttpVersion.HTTP_2)))
+                .setUseAlpn(true)
+                .setHttp2ClearTextEnabled(true);
+
+        HttpServer httpServer = vertx.createHttpServer(httpServerOptions);
+        HttpClient httpClient = vertx.createHttpClient(httpClientOptions);
+
+        ReverseHttpProxy.create(router, httpServer, httpClient)
+                .addRoute(new ProxyRoute()
+                        .setSourceUrl("/*")
+                        .setTargetUrl("https://reqres.in"))
+                .port(8080)
+                .start();
+
+        LockSupport.park();
+    }
+
+    @Test
+    public void testVertxDnsResolve() {
+        // 比较okhttp与vertx http的dns解析。发现vertx特别慢。因此定位问题
+        // 复现代码参考https://github.com/meethigher/bug-test/tree/vertx-http-dns
+    }
 }

src/test/java/top/meethigher/proxy/tcp/ReverseTcpProxyTest.java

@@ -9,7 +9,7 @@ public class ReverseTcpProxyTest {
 
     @Test
     public void testVertxTCPReverseProxy() throws Exception {
-        ReverseTcpProxy proxy = ReverseTcpProxy.create(Vertx.vertx(), "10.0.0.1", 888);
+        ReverseTcpProxy proxy = ReverseTcpProxy.create(Vertx.vertx(), "meethigher.top", 443);
         proxy.port(8080).start();
         TimeUnit.MINUTES.sleep(10);
         proxy.stop();

src/test/resources/logback-test.xml

@@ -6,7 +6,7 @@
     </appender>
 
     <!-- 设置根日志级别为TRACE -->
-    <root level="INFO">
+    <root level="TRACE">
         <appender-ref ref="CONSOLE"/>
     </root>