refactor: add bench test

vicanso · Aug 13, 2023 · fbb996a · fbb996a
1 parent 7c7ec73
commit fbb996a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/middleware/lru_store_test.go b/middleware/lru_store_test.go
@@ -48,3 +48,12 @@ func TestLruStore(t *testing.T) {
 	buf, _ = store.Get(ctx, key1)
 	assert.Empty(buf)
 }
+
+func BenchmarkLruStore(b *testing.B) {
+	store := NewLruStore(128)
+	ctx := context.Background()
+	for i := 0; i < b.N; i++ {
+		_ = store.Set(ctx, "key", []byte("Hello world!"), time.Second)
+		_, _ = store.Get(ctx, "key")
+	}
+}