Merge pull request #216 from rfyiamcool/fix/update_batch_committed

fix: update committed flag after batch commit
rosedblabs · Jun 15, 2023 · 7c5b2b6 · 7c5b2b6
2 parents 67e7a9c + f6cf0be
commit 7c5b2b6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/batch.go b/batch.go
@@ -5,6 +5,7 @@ import (
 	"sync"
 
 	"github.com/bwmarrin/snowflake"
+
 	"github.com/rosedblabs/wal"
 )
 
@@ -202,6 +203,7 @@ func (b *Batch) Commit() error {
 
 	batchId := b.batchId.Generate()
 	positions := make(map[string]*wal.ChunkPosition)
+
 	// write to wal
 	for _, record := range b.pendingWrites {
 		record.BatchId = uint64(batchId)
@@ -238,5 +240,6 @@ func (b *Batch) Commit() error {
 		}
 	}
 
+	b.committed = true
 	return nil
 }