Clear initial sync state caches after round robin sync (#4817)

* Clear initial sync state caches after round robin sync * fix test mock * lint Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
prysmaticlabs · Feb 10, 2020 · 4598344 · 4598344
1 parent 1a5c515
commit 4598344
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/beacon-chain/blockchain/testing/mock.go b/beacon-chain/blockchain/testing/mock.go
@@ -231,3 +231,6 @@ func (ms *ChainService) Participation(epoch uint64) *precompute.Balance {
 func (ms *ChainService) IsValidAttestation(ctx context.Context, att *ethpb.Attestation) bool {
 	return ms.ValidAttestation
 }
+
+// ClearCachedStates does nothing.
+func (ms *ChainService) ClearCachedStates() {}
diff --git a/beacon-chain/sync/initial-sync/round_robin.go b/beacon-chain/sync/initial-sync/round_robin.go
@@ -43,6 +43,7 @@ const refreshTime = 6 * time.Second
 func (s *Service) roundRobinSync(genesis time.Time) error {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
+	defer s.chain.ClearCachedStates()
 
 	if cfg := featureconfig.Get(); cfg.EnableSkipSlotsCache {
 		cfg.EnableSkipSlotsCache = false

diff --git a/beacon-chain/sync/initial-sync/service.go b/beacon-chain/sync/initial-sync/service.go
@@ -25,6 +25,7 @@ var _ = shared.Service(&Service{})
 type blockchainService interface {
 	blockchain.BlockReceiver
 	blockchain.HeadFetcher
+	ClearCachedStates()
 }
 
 const (