add batch

storage/operation/badgerimpl/dbstore.go

@@ -21,3 +21,7 @@ func (b *dbStore) Reader() storage.Reader {
 func (b *dbStore) WithReaderBatchWriter(fn func(storage.ReaderBatchWriter) error) error {
 	return WithReaderBatchWriter(b.db, fn)
 }
+
+func (b *dbStore) NewBatch() storage.Batch {
+	return NewReaderBatchWriter(b.db)
+}

storage/operation/pebbleimpl/dbstore.go

@@ -21,3 +21,7 @@ func (b *dbStore) Reader() storage.Reader {
 func (b *dbStore) WithReaderBatchWriter(fn func(storage.ReaderBatchWriter) error) error {
 	return WithReaderBatchWriter(b.db, fn)
 }
+
+func (b *dbStore) NewBatch() storage.Batch {
+	return NewReaderBatchWriter(b.db)
+}

storage/operations.go

@@ -104,6 +104,19 @@ type DB interface {
 	// atomic batch updates to the database.
 	// Any error returned are considered fatal and the batch is not committed.
 	WithReaderBatchWriter(func(ReaderBatchWriter) error) error
+
+	// NewBatch create a new batch for writing.
+	NewBatch() Batch
+}
+
+// Batch is an interface for a batch of writes to a storage backend.
+// The batch is pending until it is committed.
+// Useful for dynamically adding writes to the batch
+type Batch interface {
+	ReaderBatchWriter
+
+	// Commit applies the batched updates to the database.
+	Commit() error
 }
 
 // OnlyWriter is an adapter to convert a function that takes a Writer