Function TestWriteBatch_Concurrent

driver/hybriddb/write_batch_test.go:271–325

(t *testing.T)

}

func TestWriteBatch_Concurrent(t *testing.T) {
	tempDir := t.TempDir()
	cfg := &config.Config{
		LevelDB: config.LevelDBConfig{
			CacheSize:       64 * 1024 * 1024,
			BlockSize:       4 * 1024,
			WriteBufferSize: 4 * 1024 * 1024,
			MaxOpenFiles:    1000,
			Compression:     true,
		},
	}

	db, err := Store{}.Open(tempDir, cfg)
	require.NoError(t, err)
	defer db.Close()

	numBatches := 10
	opsPerBatch := 100
	done := make(chan bool, numBatches)

	for i := 0; i < numBatches; i++ {
		go func(batchID int) {
			wb := db.NewWriteBatch()
			defer wb.Close()

			for j := 0; j < opsPerBatch; j++ {
				key := []byte(string(rune(batchID)) + "-" + string(rune(j)))
				value := []byte(string(rune(batchID)) + "-value-" + string(rune(j)))

				wb.Put(key, value)
			}

			err := wb.Commit()
			assert.NoError(t, err)
			done <- true
		}(i)
	}

	// Wait for all batches to complete
	for i := 0; i < numBatches; i++ {
		<-done
	}

	// Verify all data was written
	for i := 0; i < numBatches; i++ {
		for j := 0; j < opsPerBatch; j++ {
			key := []byte(string(rune(i)) + "-" + string(rune(j)))
			expectedValue := []byte(string(rune(i)) + "-value-" + string(rune(j)))

			value, err := db.Get(key)
			require.NoError(t, err)
			assert.Equal(t, expectedValue, value)
		}
	}
}