Merge pull request #3636 from JackyWYX/bc_validatorsnapshotcache_fix

[blockchain] added cache in ReadValidatorSnapshot
harmony-one · Apr 3, 2021 · 7875921 · 7875921
2 parents 52ac53e + cff288a
commit 7875921
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/core/blockchain.go b/core/blockchain.go
@@ -2332,7 +2332,12 @@ func (bc *BlockChain) ReadValidatorSnapshot(
 	if cached, ok := bc.validatorSnapshotCache.Get(key); ok {
 		return cached.(*staking.ValidatorSnapshot), nil
 	}
-	return rawdb.ReadValidatorSnapshot(bc.db, addr, epoch)
+	vs, err := rawdb.ReadValidatorSnapshot(bc.db, addr, epoch)
+	if err != nil {
+		return nil, err
+	}
+	bc.validatorSnapshotCache.Add(key, vs)
+	return vs, nil
 }
 
 // WriteValidatorSnapshot writes the snapshot of provided validator