Benchmark cache

1 files changed, 19 insertions, 1 deletions

diff --git a/cache/cache_test.go b/cache/cache_test.go
index 4b6521a..a81cef1 100644
--- a/cache/cache_test.go
+++ b/cache/cache_test.go
@@ -112,7 +112,7 @@ func TestCacheMaxSize(t *testing.T) {
 		{3, 2, 2},
 	}
 	for i, tt := range tests {
-		c, err := New(tt.maxSize, time.Duration(10*time.Minute))
+		c, err := New(tt.maxSize, 10*time.Minute)
 		if err != nil {
 			t.Fatal(err)
 		}
@@ -141,3 +141,21 @@ func TestCacheMaxSize(t *testing.T) {
 		}
 	}
 }
+
+func BenchmarkNewKey(b *testing.B) {
+	for n := 0; n < b.N; n++ {
+		_ = NewKey("key", 1, 1)
+	}
+}
+
+func BenchmarkCache(b *testing.B) {
+	c, err := New(1000, 10*time.Minute)
+	if err != nil {
+		b.Fatal(err)
+	}
+	b.ResetTimer()
+	for n := 0; n < b.N; n++ {
+		c.Set(uint32(n), &dns.Msg{})
+		_, _ = c.Get(uint32(n))
+	}
+}