Add insert benchmark

Signed-off-by: Sam Batschelet <[email protected]>

Author: hexfusion

tstate/tstate_test.go

@@ -55,7 +55,7 @@ func TestGetValue(t *testing.T) {
 	_, err := ts.GetValue(ctx, TestKey)
 	require.ErrorIs(err, ErrKeyNotSpecified, "No error thrown.")
 	// SetScope
-	ts.SetScope(ctx, [][]byte{TestKey}, map[Key][]byte{ToStateKeyArray(TestKey): TestVal})
+	ts.SetScope(ctx, [][]byte{TestKey}, map[*Key][]byte{ToStateKeyArray(TestKey): TestVal})
 	val, err := ts.GetValue(ctx, TestKey)
 	require.NoError(err, "Error getting value.")
 	require.Equal(TestVal, val, "Value was not saved correctly.")
@@ -66,7 +66,7 @@ func TestGetValueNoStorage(t *testing.T) {
 	ctx := context.TODO()
 	ts := New(10)
 	// SetScope but dont add to storage
-	ts.SetScope(ctx, [][]byte{TestKey}, map[Key][]byte{})
+	ts.SetScope(ctx, [][]byte{TestKey}, map[*Key][]byte{})
 	_, err := ts.GetValue(ctx, TestKey)
 	require.ErrorIs(database.ErrNotFound, err, "No error thrown.")
 }
@@ -79,7 +79,7 @@ func TestInsertNew(t *testing.T) {
 	err := ts.Insert(ctx, TestKey, TestVal)
 	require.ErrorIs(ErrKeyNotSpecified, err, "No error thrown.")
 	// SetScope
-	ts.SetScope(ctx, [][]byte{TestKey}, map[Key][]byte{})
+	ts.SetScope(ctx, [][]byte{TestKey}, map[*Key][]byte{})
 	// Insert key
 	err = ts.Insert(ctx, TestKey, TestVal)
 	require.NoError(err, "Error thrown.")
@@ -94,7 +94,7 @@ func TestInsertUpdate(t *testing.T) {
 	ctx := context.TODO()
 	ts := New(10)
 	// SetScope and add
-	ts.SetScope(ctx, [][]byte{TestKey}, map[Key][]byte{ToStateKeyArray(TestKey): TestVal})
+	ts.SetScope(ctx, [][]byte{TestKey}, map[*Key][]byte{ToStateKeyArray(TestKey): TestVal})
 	require.Equal(0, ts.OpIndex(), "SetStorage operation was not added.")
 	// Insert key
 	newVal := []byte("newVal")
@@ -163,15 +163,15 @@ func TestSetScope(t *testing.T) {
 	ts := New(10)
 	ctx := context.TODO()
 	keys := [][]byte{[]byte("key1"), []byte("key2"), []byte("key3")}
-	ts.SetScope(ctx, keys, map[Key][]byte{})
+	ts.SetScope(ctx, keys, map[*Key][]byte{})
 	require.Equal(keys, ts.scope, "Scope not updated correctly.")
 }
 
 func TestRemoveInsertRollback(t *testing.T) {
 	require := require.New(t)
 	ts := New(10)
 	ctx := context.TODO()
-	ts.SetScope(ctx, [][]byte{TestKey}, map[Key][]byte{})
+	ts.SetScope(ctx, [][]byte{TestKey}, map[*Key][]byte{})
 	// Insert
 	err := ts.Insert(ctx, TestKey, TestVal)
 	require.NoError(err, "Error from insert.")
@@ -219,7 +219,7 @@ func TestRestoreInsert(t *testing.T) {
 	ctx := context.TODO()
 	keys := [][]byte{[]byte("key1"), []byte("key2"), []byte("key3")}
 	vals := [][]byte{[]byte("val1"), []byte("val2"), []byte("val3")}
-	ts.SetScope(ctx, keys, map[Key][]byte{})
+	ts.SetScope(ctx, keys, map[*Key][]byte{})
 	for i, key := range keys {
 		err := ts.Insert(ctx, key, vals[i])
 		require.NoError(err, "Error inserting.")
@@ -249,7 +249,7 @@ func TestRestoreDelete(t *testing.T) {
 	ctx := context.TODO()
 	keys := [][]byte{[]byte("key1"), []byte("key2"), []byte("key3")}
 	vals := [][]byte{[]byte("val1"), []byte("val2"), []byte("val3")}
-	ts.SetScope(ctx, keys, map[Key][]byte{
+	ts.SetScope(ctx, keys, map[*Key][]byte{
 		ToStateKeyArray(keys[0]): vals[0],
 		ToStateKeyArray(keys[1]): vals[1],
 		ToStateKeyArray(keys[2]): vals[2],
@@ -288,7 +288,7 @@ func TestWriteChanges(t *testing.T) {
 	tracer, _ := trace.New(&trace.Config{Enabled: false})
 	keys := [][]byte{[]byte("key1"), []byte("key2"), []byte("key3")}
 	vals := [][]byte{[]byte("val1"), []byte("val2"), []byte("val3")}
-	ts.SetScope(ctx, keys, map[Key][]byte{})
+	ts.SetScope(ctx, keys, map[*Key][]byte{})
 	// Add
 	for i, key := range keys {
 		err := ts.Insert(ctx, key, vals[i])
@@ -306,7 +306,7 @@ func TestWriteChanges(t *testing.T) {
 	}
 	// Remove
 	ts = New(10)
-	ts.SetScope(ctx, keys, map[Key][]byte{
+	ts.SetScope(ctx, keys, map[*Key][]byte{
 		ToStateKeyArray(keys[0]): vals[0],
 		ToStateKeyArray(keys[1]): vals[1],
 		ToStateKeyArray(keys[2]): vals[2],
@@ -334,21 +334,21 @@ func BenchmarkFetchAndSetScope(b *testing.B) {
 	}
 }
 
+func BenchmarkInsert(b *testing.B) {
+	for _, size := range []int{4, 8, 16, 32, 64, 128} {
+		b.Run(fmt.Sprintf("insert_%d_keys", size), func(b *testing.B) {
+			benchmarkInsert(b, size)
+		})
+	}
+}
+
 func benchmarkFetchAndSetScope(b *testing.B, size int) {
 	require := require.New(b)
-	ts := New(10)
+	ts := New(size)
 	db := NewTestDB()
 	ctx := context.TODO()
 
-	keys := [][]byte{}
-	vals := [][]byte{}
-
-	// each k/v is unique to simulate worst case
-	for i := 0; i <= size; i++ {
-		keys = append(keys, randomBytes(MapKeyLength))
-		vals = append(vals, randomBytes(8))
-	}
-
+	keys, vals := initializeSet(size)
 	for i, key := range keys {
 		err := db.Insert(ctx, key, vals[i])
 		require.NoError(err, "Error during insert.")
@@ -363,6 +363,43 @@ func benchmarkFetchAndSetScope(b *testing.B, size int) {
 	b.StopTimer()
 }
 
+func benchmarkInsert(b *testing.B, size int) {
+	require := require.New(b)
+	ts := New(size)
+	ctx := context.TODO()
+
+	keys, vals := initializeSet(size)
+
+	storage := map[*Key][]byte{}
+	for i, key := range keys {
+		storage[ToStateKeyArray(key)] = vals[i]
+	}
+
+	ts.SetScope(ctx, keys, storage)
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		for i, key := range keys {
+			err := ts.Insert(ctx, key, vals[i])
+			require.NoError(err, "Error during insert.")
+		}
+	}
+	b.ReportAllocs()
+	b.StopTimer()
+}
+
+func initializeSet(size int) ([][]byte, [][]byte) {
+	keys := [][]byte{}
+	vals := [][]byte{}
+
+	for i := 0; i <= size; i++ {
+		keys = append(keys, randomBytes(MapKeyLength))
+		vals = append(vals, randomBytes(8))
+	}
+
+	return keys, vals
+}
+
 func randomBytes(size int) []byte {
 	bytes := make([]byte, size)
 	rand.Read(bytes)