Merge pull request #2759 from nspcc-dev/avoid-db-lookup-in-hasblock

core: optimize HasBlock check for recent blocks
nspcc-dev · Oct 24, 2022 · a17d9f8 · a17d9f8
2 parents f42b8e7 + 0c3b036
commit a17d9f8
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/core/blockchain.go b/pkg/core/blockchain.go
@@ -1714,6 +1714,16 @@ func (bc *Blockchain) HasTransaction(hash util.Uint256) bool {
 // HasBlock returns true if the blockchain contains the given
 // block hash.
 func (bc *Blockchain) HasBlock(hash util.Uint256) bool {
+	var height = bc.BlockHeight()
+	bc.headerHashesLock.RLock()
+	for i := int(height); i >= int(height)-4 && i >= 0; i-- {
+		if hash.Equals(bc.headerHashes[i]) {
+			bc.headerHashesLock.RUnlock()
+			return true
+		}
+	}
+	bc.headerHashesLock.RUnlock()
+
 	if header, err := bc.GetHeader(hash); err == nil {
 		return header.Index <= bc.BlockHeight()
 	}