You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
3.6 KiB
93 lines
3.6 KiB
package node
|
|
|
|
import (
|
|
"math/big"
|
|
"time"
|
|
|
|
metrics "github.com/harmony-one/harmony/api/service/metrics"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
)
|
|
|
|
// UpdateBlockHeightForMetrics updates block height for metrics service.
|
|
func (node *Node) UpdateBlockHeightForMetrics(prevBlockHeight uint64) uint64 {
|
|
curBlock := node.Blockchain().CurrentBlock()
|
|
curBlockHeight := curBlock.NumberU64()
|
|
if curBlockHeight == prevBlockHeight {
|
|
return prevBlockHeight
|
|
}
|
|
utils.Logger().Info().Msgf("Updating metrics block height %d", curBlockHeight)
|
|
metrics.UpdateBlockHeight(curBlockHeight)
|
|
blockReward := node.Consensus.GetBlockReward()
|
|
if blockReward != nil {
|
|
utils.Logger().Info().Msgf("Updating metrics block reward %d", blockReward.Uint64())
|
|
metrics.UpdateBlockReward(blockReward)
|
|
}
|
|
return curBlockHeight
|
|
}
|
|
|
|
// UpdateConnectionsNumberForMetrics uppdates connections number for metrics service.
|
|
func (node *Node) UpdateConnectionsNumberForMetrics(prevNumPeers int) int {
|
|
curNumPeers := node.numPeers
|
|
if curNumPeers == prevNumPeers {
|
|
return prevNumPeers
|
|
}
|
|
utils.Logger().Info().Msgf("Updating metrics connections number %d", curNumPeers)
|
|
metrics.UpdateConnectionsNumber(curNumPeers)
|
|
return curNumPeers
|
|
}
|
|
|
|
// UpdateTxPoolSizeForMetrics updates tx pool size for metrics service.
|
|
func (node *Node) UpdateTxPoolSizeForMetrics(txPoolSize uint64) {
|
|
utils.Logger().Info().Msgf("Updating metrics tx pool size %d", txPoolSize)
|
|
metrics.UpdateTxPoolSize(txPoolSize)
|
|
}
|
|
|
|
// UpdateBalanceForMetrics uppdates node balance for metrics service.
|
|
func (node *Node) UpdateBalanceForMetrics(prevBalance *big.Int) *big.Int {
|
|
curBalance, err := node.GetBalanceOfAddress(node.Consensus.SelfAddress)
|
|
if err != nil || curBalance.Cmp(prevBalance) == 0 {
|
|
return prevBalance
|
|
}
|
|
utils.Logger().Info().Msgf("Updating metrics node balance %d", curBalance.Uint64())
|
|
metrics.UpdateNodeBalance(curBalance)
|
|
return curBalance
|
|
}
|
|
|
|
// UpdateLastConsensusTimeForMetrics uppdates last consensus reached time for metrics service.
|
|
func (node *Node) UpdateLastConsensusTimeForMetrics(prevLastConsensusTime int64) int64 {
|
|
lastConsensusTime := node.lastConsensusTime
|
|
if lastConsensusTime == prevLastConsensusTime {
|
|
return prevLastConsensusTime
|
|
}
|
|
utils.Logger().Info().Msgf("Updating metrics last consensus time reached %d", lastConsensusTime)
|
|
metrics.UpdateLastConsensus(lastConsensusTime)
|
|
return lastConsensusTime
|
|
}
|
|
|
|
// UpdateIsLeaderForMetrics updates if node is a leader now for metrics serivce.
|
|
func (node *Node) UpdateIsLeaderForMetrics() {
|
|
if node.Consensus.LeaderPubKey.SerializeToHexStr() == node.Consensus.PubKey.SerializeToHexStr() {
|
|
utils.Logger().Info().Msgf("Node %s is a leader now", node.Consensus.PubKey.SerializeToHexStr())
|
|
metrics.UpdateIsLeader(true)
|
|
} else {
|
|
utils.Logger().Info().Msgf("Node %s is not a leader now", node.Consensus.PubKey.SerializeToHexStr())
|
|
metrics.UpdateIsLeader(false)
|
|
}
|
|
}
|
|
|
|
// CollectMetrics collects metrics: block height, connections number, node balance, block reward, last consensus, accepted blocks.
|
|
func (node *Node) CollectMetrics() {
|
|
utils.Logger().Info().Msg("[Metrics Service] Update metrics")
|
|
prevNumPeers := 0
|
|
prevBlockHeight := uint64(0)
|
|
prevBalance := big.NewInt(0)
|
|
prevLastConsensusTime := int64(0)
|
|
for range time.Tick(100 * time.Millisecond) {
|
|
prevBlockHeight = node.UpdateBlockHeightForMetrics(prevBlockHeight)
|
|
prevNumPeers = node.UpdateConnectionsNumberForMetrics(prevNumPeers)
|
|
prevBalance = node.UpdateBalanceForMetrics(prevBalance)
|
|
prevLastConsensusTime = node.UpdateLastConsensusTimeForMetrics(prevLastConsensusTime)
|
|
node.UpdateTxPoolSizeForMetrics(node.TxPool.GetTxPoolSize())
|
|
node.UpdateIsLeaderForMetrics()
|
|
}
|
|
}
|
|
|