Cache filtered block tree

beacon-chain/blockchain/forkchoice/process_block.go

@@ -93,6 +93,14 @@ func (s *Store) OnBlock(ctx context.Context, signed *ethpb.SignedBeaconBlock) er
 		return errors.Wrap(err, "could not save state")
 	}
 
+	if featureconfig.Get().EnableBlockTreeCache {
+		tree, err := s.getFilterBlockTree(ctx)
+		if err != nil {
+			return errors.Wrap(err, "could not calculate filtered block tree")
+		}
+		s.filteredBlockTree = tree
+	}
+
 	// Update justified check point.
 	if postState.CurrentJustifiedCheckpoint.Epoch > s.justifiedCheckpt.Epoch {
 		if err := s.updateJustified(ctx, postState); err != nil {

beacon-chain/blockchain/forkchoice/service.go

@@ -52,6 +52,7 @@ type Store struct {
 	initSyncState         map[[32]byte]*pb.BeaconState
 	initSyncStateLock     sync.RWMutex
 	nextEpochBoundarySlot uint64
+	filteredBlockTree     map[[32]byte]*ethpb.BeaconBlock
 }
 
 // NewForkChoiceService instantiates a new service instance that will
@@ -262,9 +263,15 @@ func (s *Store) Head(ctx context.Context) ([]byte, error) {
 	defer span.End()
 
 	head := s.JustifiedCheckpt().Root
-	filteredBlocks, err := s.getFilterBlockTree(ctx)
-	if err != nil {
-		return nil, err
+	filteredBlocks := make(map[[32]byte]*ethpb.BeaconBlock)
+	var err error
+	if featureconfig.Get().EnableBlockTreeCache {
+		filteredBlocks = s.filteredBlockTree
+	} else {
+		filteredBlocks, err = s.getFilterBlockTree(ctx)
+		if err != nil {
+			return nil, err
+		}
 	}
 
 	justifiedSlot := helpers.StartSlot(s.justifiedCheckpt.Epoch)

shared/featureconfig/config.go

@@ -44,6 +44,7 @@ type Flags struct {
 	EnableShuffledIndexCache bool // EnableShuffledIndexCache to cache expensive shuffled index computation.
 	EnableSkipSlotsCache     bool // EnableSkipSlotsCache caches the state in skipped slots.
 	EnableSlasherConnection  bool // EnableSlasher enable retrieval of slashing events from a slasher instance.
+	EnableBlockTreeCache     bool // EnableBlockTreeCache enable fork choice service to maintain latest filtered block tree.
 }
 
 var featureConfig *Flags
@@ -128,6 +129,10 @@ func ConfigureBeaconChain(ctx *cli.Context) {
 		log.Warn("Enable slasher connection.")
 		cfg.EnableSlasherConnection = true
 	}
+	if ctx.GlobalBool(cacheFilteredBlockTree.Name) {
+		log.Warn("Enabled filtered block tree cache for fork choice.")
+		cfg.EnableBlockTreeCache = true
+	}
 	Init(cfg)
 }
 

shared/featureconfig/flags.go

@@ -78,6 +78,11 @@ var (
 			"triggered the genesis as the genesis time. This flag should be used for local " +
 			"development and testing only.",
 	}
+	cacheFilteredBlockTree = cli.BoolFlag{
+		Name: "cache-filtered-block-tree",
+		Usage: "Cache filtered block tree by maintaining it rather than continually recalculating on the fly," +
+			"this is used for fork choice.",
+	}
 )
 
 // Deprecated flags list.
@@ -190,4 +195,5 @@ var BeaconChainFlags = append(deprecatedFlags, []cli.Flag{
 	enableSkipSlotsCache,
 	saveDepositData,
 	enableSlasherFlag,
+	cacheFilteredBlockTree,
 }...)