feat(pkg): improve BlockBatchIterator (#173)

taikoxyz · Mar 6, 2023 · 4fab06a · 4fab06a
1 parent 6c85058
commit 4fab06a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/chain_iterator/block_batch_iterator.go b/pkg/chain_iterator/block_batch_iterator.go
@@ -295,10 +295,10 @@ func (i *BlockBatchIterator) ensureCurrentNotReorged() error {
 
 	// Reorg detected, rewind back `ReorgRewindDepth` blocks
 	var newCurrentHeight uint64
-	if current.Number.Uint64() <= ReorgRewindDepth {
+	if i.current.Number.Uint64() <= ReorgRewindDepth {
 		newCurrentHeight = 0
 	} else {
-		newCurrentHeight = current.Number.Uint64() - ReorgRewindDepth
+		newCurrentHeight = i.current.Number.Uint64() - ReorgRewindDepth
 	}
 
 	i.current, err = i.client.HeaderByNumber(i.ctx, new(big.Int).SetUint64(newCurrentHeight))