add fastdns resolver benchmark

phuslu · Oct 22, 2024 · 90ab689 · 90ab689
1 parent 585ec96
commit 90ab689
Showing 1 changed file with 42 additions and 0 deletions.
diff --git a/client_test.go b/client_test.go
@@ -147,3 +147,45 @@ func TestClientLookup(t *testing.T) {
 		}
 	}
 }
+
+func BenchmarkResolverPureGo(b *testing.B) {
+	resolver := net.Resolver{PreferGo: true}
+
+	b.ReportAllocs()
+	b.ResetTimer()
+	b.RunParallel(func(b *testing.PB) {
+		for b.Next() {
+			resolver.LookupNetIP(context.Background(), "ip4", "www.google.com")
+		}
+	})
+}
+
+func BenchmarkResolverCGO(b *testing.B) {
+	resolver := net.Resolver{PreferGo: false}
+
+	b.ReportAllocs()
+	b.ResetTimer()
+	b.RunParallel(func(b *testing.PB) {
+		for b.Next() {
+			resolver.LookupNetIP(context.Background(), "ip4", "www.google.com")
+		}
+	})
+}
+
+func BenchmarkResolverFastdns(b *testing.B) {
+	resolver := Client{
+		Addr: "1.1.1.1:53",
+		Dialer: &UDPDialer{
+			Addr:     func() (u *net.UDPAddr) { u, _ = net.ResolveUDPAddr("udp", "1.1.1.1:53"); return }(),
+			MaxConns: 1000,
+		},
+	}
+
+	b.ReportAllocs()
+	b.ResetTimer()
+	b.RunParallel(func(b *testing.PB) {
+		for b.Next() {
+			resolver.LookupNetIP(context.Background(), "ip4", "www.google.com")
+		}
+	})
+}