Bump Badger version to v3.2011.1

badger_store.go

@@ -20,7 +20,7 @@ import (
 	"errors"
 	"time"
 
-	"github.com/dgraph-io/badger/v2"
+	"github.com/dgraph-io/badger/v3"
 	"github.com/hashicorp/raft"
 )
 
@@ -252,19 +252,30 @@ func (b *BadgerStore) StoreLog(log *raft.Log) error {
 
 // StoreLogs stores a set of raft logs.
 func (b *BadgerStore) StoreLogs(logs []*raft.Log) error {
-	return b.conn.Update(func(txn *badger.Txn) error {
-		for _, log := range logs {
-			key := append(prefixLogs, uint64ToBytes(log.Index)...)
-			val, err := encodeMsgPack(log)
-			if err != nil {
-				return err
-			}
-			if err := txn.Set(key, val.Bytes()); err != nil {
-				return err
+	// we manage the transaction manually in order to avoid ErrTxnTooBig errors
+	txn := b.conn.NewTransaction(true)
+	for i, log := range logs {
+		key := append(prefixLogs, uint64ToBytes(log.Index)...)
+		val, err := encodeMsgPack(log)
+		if err != nil {
+			return err
+		}
+		if err := txn.Set(key, val.Bytes()); err != nil {
+			if err == badger.ErrTxnTooBig {
+				err = txn.Commit()
+				if err != nil {
+					return err
+				}
+				return b.StoreLogs(logs[i:])
 			}
+			return err
 		}
-		return nil
-	})
+	}
+	err := txn.Commit()
+	if err != nil {
+		return err
+	}
+	return nil
 }
 
 // DeleteRange deletes logs within a given range inclusively.

badger_store_test.go

@@ -23,7 +23,7 @@ import (
 	"reflect"
 	"testing"
 
-	"github.com/dgraph-io/badger/v2"
+	"github.com/dgraph-io/badger/v3"
 	"github.com/hashicorp/raft"
 )
 

go.mod

@@ -4,7 +4,7 @@ go 1.14
 
 require (
 	github.com/AndreasBriese/bbloom v0.0.0-20190825152654-46b345b51c96 // indirect
-	github.com/dgraph-io/badger/v2 v2.0.3
+	github.com/dgraph-io/badger/v3 v3.2011.1
 	github.com/dgryski/go-farm v0.0.0-20191112170834-c2139c5d712b // indirect
 	github.com/hashicorp/go-msgpack v0.5.5
 	github.com/hashicorp/raft v1.1.1