mirror/dolt
1
0
Fork 0
mirror of https://github.com/dolthub/dolt.git synced 2026-01-25 18:49:36 -06:00
Code Issues Packages Projects Releases 123 Wiki Activity

make Dolt benchmark do a ChunkStore.Commit()

Browse Source
This commit is contained in:
Andy Arthur
2022-09-08 16:43:21 -07:00
parent 3afe8b41f6
commit 6bedfcdeb3
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
go/store/prolly/benchmark/benchmark_batch_writes_test.go
View File

@@ -89,7 +89,7 @@ func BenchmarkImportDolt(b *testing.B) {
desc := val.NewTupleDescriptor(val.Type{Enc: val.Uint64Enc})
m, err := prolly.NewMapFromTuples(ctx, tree.NewNodeStore(cs), desc, desc)
require.NoError(b, err)
return &doltWriter{mut: m.Mutate()}
return &doltWriter{mut: m.Mutate(), cs: cs}
}
b.Run("Dolt", func(b *testing.B) {
@@ -127,6 +127,7 @@ func (wr *bboltWriter) Flush() error {
type doltWriter struct {
mut prolly.MutableMap
cs *nbs.NomsBlockStore
}
func (wr *doltWriter) Put(key, value []byte) error {
@@ -139,7 +140,13 @@ func (wr *doltWriter) Flush() error {
return err
}
wr.mut = m.Mutate()
return nil
h, err := wr.cs.Root(context.Background())
if err != nil {
return err
}
_, err = wr.cs.Commit(context.Background(), h, h)
return err
}
func benchmarkBatchWrite(b *testing.B, wr writer) {