Add Write method to batch write

batch.go

@@ -19,6 +19,7 @@ package badger
 import (
 	"sync"
 
+	"github.com/dgraph-io/badger/v2/pb"
 	"github.com/dgraph-io/badger/v2/y"
 	"github.com/pkg/errors"
 )
@@ -89,6 +90,23 @@ func (wb *WriteBatch) callback(err error) {
 	wb.err = err
 }
 
+func (wb *WriteBatch) Write(kvList *pb.KVList) error {
+	wb.Lock()
+	defer wb.Unlock()
+	for _, kv := range kvList.Kv {
+		e := Entry{Key: kv.Key, Value: kv.Value}
+		if len(kv.UserMeta) > 0 {
+			e.UserMeta = kv.UserMeta[0]
+		}
+		y.AssertTrue(kv.Version != 0)
+		e.version = kv.Version
+		if err := wb.handleEntry(&e); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
 // SetEntryAt is the equivalent of Txn.SetEntry but it also allows setting version for the entry.
 // SetEntryAt can be used only in managed mode.
 func (wb *WriteBatch) SetEntryAt(e *Entry, ts uint64) error {
@@ -99,11 +117,8 @@ func (wb *WriteBatch) SetEntryAt(e *Entry, ts uint64) error {
 	return wb.SetEntry(e)
 }
 
-// SetEntry is the equivalent of Txn.SetEntry.
-func (wb *WriteBatch) SetEntry(e *Entry) error {
-	wb.Lock()
-	defer wb.Unlock()
-
+// Should be called with lock acquired.
+func (wb *WriteBatch) handleEntry(e *Entry) error {
 	if err := wb.txn.SetEntry(e); err != ErrTxnTooBig {
 		return err
 	}
@@ -120,6 +135,13 @@ func (wb *WriteBatch) SetEntry(e *Entry) error {
 	return nil
 }
 
+// SetEntry is the equivalent of Txn.SetEntry.
+func (wb *WriteBatch) SetEntry(e *Entry) error {
+	wb.Lock()
+	defer wb.Unlock()
+	return wb.handleEntry(e)
+}
+
 // Set is equivalent of Txn.Set().
 func (wb *WriteBatch) Set(k, v []byte) error {
 	e := &Entry{Key: k, Value: v}