feat: shift starting node base on iteration

Author: travisperson

cmd/filecoin-chain-archiver/cmds/create.go

@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"io"
 	"io/ioutil"
+	"math/rand"
 	"net/url"
 	"strings"
 	"syscall"
@@ -241,11 +242,23 @@ var cmdCreate = &cli.Command{
 			return err
 		}
 
+		var iteration int
+		if cctx.IsSet("interval") {
+			iteration = int(uint64(height)/uint64(flagInterval)) % len(nodes)
+		} else {
+			iteration = rand.Int() % len(nodes)
+		}
+
+		logger.Infow("iteration", "value", iteration)
+		cm.ShiftStartNode(iteration)
+
 		node, peerID, err := cm.GetNodeWithTipSet(ctx, tsk, filterList)
 		if err != nil {
 			return err
 		}
 
+		logger.Infow("node", "peer_id", peerID)
+
 		lock, locked, err := nl.Lock(ctx, peerID)
 		if err != nil {
 			return err

pkg/consensus/consensus.go

@@ -85,6 +85,18 @@ func (cm *ConsensusManager) GetTipset(ctx context.Context, height abi.ChainEpoch
 	return pick, nil
 }
 
+func (cm *ConsensusManager) ShiftStartNode(iteration int) {
+	nodes := make([]api.FullNode, len(cm.nodes))
+
+	for i := 0; i < len(cm.nodes); i++ {
+		source := (i + iteration) % len(cm.nodes)
+		logger.Debugw("shift start node", "assignment", i, "source")
+		nodes[i] = cm.nodes[source]
+	}
+
+	cm.nodes = nodes
+}
+
 func (cm *ConsensusManager) GetNodeWithTipSet(ctx context.Context, tsk types.TipSetKey, filterList []string) (api.FullNode, string, error) {
 	peerFilter := make(map[string]struct{})
 	for _, peer := range filterList {