add benchmarks

abdulsattar · Jan 25, 2024 · 44de5bf · 44de5bf
1 parent c1f5e8a
commit 44de5bf
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/kv_test.go b/kv_test.go
@@ -0,0 +1,28 @@
+package main
+
+import (
+	"fmt"
+	"testing"
+)
+
+func BenchmarkWrites(b *testing.B) {
+	kv, err := NewKV()
+	defer func(kv *KV) {
+		err := kv.Close()
+		if err != nil {
+			fmt.Println(err)
+			panic(err)
+		}
+	}(kv)
+	if err != nil {
+		fmt.Println(err)
+		panic(err)
+	}
+	for i := 0; i < b.N; i++ {
+		err := kv.Set(fmt.Sprintf("key%d", i), "value")
+		if err != nil {
+			fmt.Println(err)
+			panic(err)
+		}
+	}
+}