add allocation id to recover in config

Author: Hitenjain14

code/go/0chain.net/blobber/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"github.com/0chain/blobber/code/go/0chain.net/blobbercore/allocation"
+	"github.com/0chain/blobber/code/go/0chain.net/blobbercore/config"
 	"github.com/0chain/blobber/code/go/0chain.net/core/common"
 	"github.com/0chain/blobber/code/go/0chain.net/core/logging"
 	"github.com/0chain/blobber/code/go/0chain.net/core/node"
@@ -61,7 +62,7 @@ func main() {
 
 	if recoverTrie {
 		logging.Logger.Info("Recovering trie")
-		allocation.RecoverTrie()
+		allocation.RecoverTrie(config.Configuration.RecoverAllocations)
 	}
 
 	// todo: activate this when gRPC functionalities are implemented

code/go/0chain.net/blobbercore/allocation/workers.go

@@ -349,33 +349,27 @@ func deleteAllocation(ctx context.Context, a *Allocation) (err error) {
 	return err
 }
 
-func RecoverTrie() {
+func RecoverTrie(recoverAllocs []string) {
 	var (
-		allocs []*Allocation
-		err    error
-		offset int64
+		err error
 	)
 
-	for {
+	for _, allocID := range recoverAllocs {
+		var alloc *Allocation
 		err = datastore.GetStore().WithNewTransaction(func(ctx context.Context) error {
-			allocs, err = Repo.GetAllocations(ctx, offset)
+			alloc, err = Repo.GetAllocationFromDB(ctx, allocID)
 			return err
 		})
 		if err != nil {
 			logging.Logger.Error("recover_trie_fetch_alloc", zap.Error(err))
 			return
 		}
-		if len(allocs) == 0 {
-			return
-		}
-		offset += int64(len(allocs))
-		// recover trie of each allocation
-		for _, a := range allocs {
-			logging.Logger.Info("recover_trie", zap.String("allocation_id", a.ID))
-			err = a.recoverTrie()
-			if err != nil {
-				logging.Logger.Error("recover_trie", zap.Error(err))
-			}
+
+		// recover trie of allocation
+		logging.Logger.Info("recover_trie", zap.String("allocation_id", alloc.ID))
+		err = alloc.recoverTrie()
+		if err != nil {
+			logging.Logger.Error("recover_trie", zap.Error(err))
 		}
 	}
 }

code/go/0chain.net/blobbercore/config/config.go

@@ -157,9 +157,10 @@ type Config struct {
 	MinConfirmation int
 
 	// MountPoint is where allocation files are stored. This is basically arranged in RAID5.
-	MountPoint    string
-	AllocDirLevel []int
-	FileDirLevel  []int
+	MountPoint         string
+	AllocDirLevel      []int
+	FileDirLevel       []int
+	RecoverAllocations []string
 	// AutomacitUpdate Whether to automatically update blobber updates to blockchain
 	AutomaticUpdate       bool
 	BlobberUpdateInterval time.Duration
@@ -291,6 +292,7 @@ func ReadConfig(deploymentMode int) {
 	if w := Configuration.DelegateWallet; len(w) != 64 {
 		log.Fatal("invalid delegate wallet:", w)
 	}
+	Configuration.RecoverAllocations = viper.GetStringSlice("recover_allocations")
 
 	Configuration.MinSubmit = viper.GetInt("min_submit")
 	if Configuration.MinSubmit < 1 {

config/0chain_blobber.yaml

@@ -27,6 +27,9 @@ finalize_allocations_interval: 24h
 # maximum limit on the number of combined directories and files on each allocation
 max_dirs_files: 50000
 
+#recover allocations
+recover_allocations: [""]
+
 # maximum limit on the number of objects in a directory
 max_objects_dir: 1000
 # limit of objects per gb(storage v2)