GetTotalDifficulty

smartcontractkit · Jan 16, 2025 · 705a698 · 705a698
1 parent 321938c
commit 705a698
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/multinode/adaptor.go b/multinode/adaptor.go
@@ -187,16 +187,18 @@ func (m *Adapter[RPC, HEAD]) OnNewHead(ctx context.Context, requestCh <-chan str
 
 	m.chainInfoLock.Lock()
 	defer m.chainInfoLock.Unlock()
+	blockNumber := head.BlockNumber()
+	totalDifficulty := head.GetTotalDifficulty()
 	if !CtxIsHeathCheckRequest(ctx) {
-		m.highestUserObservations.BlockNumber = max(m.highestUserObservations.BlockNumber, head.BlockNumber())
-		m.highestUserObservations.TotalDifficulty = MaxTotalDifficulty(m.highestUserObservations.TotalDifficulty, head.TotalDifficulty())
+		m.highestUserObservations.BlockNumber = max(m.highestUserObservations.BlockNumber, blockNumber)
+		m.highestUserObservations.TotalDifficulty = MaxTotalDifficulty(m.highestUserObservations.TotalDifficulty, totalDifficulty)
 	}
 	select {
 	case <-requestCh: // no need to update latestChainInfo, as rpcMultiNodeAdapter already started new life cycle
 		return
 	default:
-		m.latestChainInfo.BlockNumber = head.BlockNumber()
-		m.latestChainInfo.TotalDifficulty = head.TotalDifficulty()
+		m.latestChainInfo.BlockNumber = blockNumber
+		m.latestChainInfo.TotalDifficulty = totalDifficulty
 	}
 }
 

diff --git a/multinode/types.go b/multinode/types.go
@@ -83,7 +83,7 @@ type RPCClient[
 type Head interface {
 	BlockNumber() int64
 	BlockDifficulty() *big.Int
-	TotalDifficulty() *big.Int
+	GetTotalDifficulty() *big.Int
 	IsValid() bool
 }