def handle_publish(self, key: HathorEvents, args: EventArguments) -> None:
""" This method is called when pubsub publishes an event that we subscribed
"""
from hathor.p2p.protocol import HathorProtocol
data = args.__dict__
if key == HathorEvents.NETWORK_NEW_TX_ACCEPTED:
if data['tx'].is_block:
self.blocks += 1
self.total_block_weight = sum_weights(data['tx'].weight,
self.total_block_weight)
self.hash_rate = self.calculate_new_hashrate(data['tx'])
self.best_block_weight = self.tx_storage.get_weight_best_block(
)
else:
self.transactions += 1
self.total_tx_weight = sum_weights(data['tx'].weight,
self.total_tx_weight)
elif key == HathorEvents.NETWORK_PEER_CONNECTED:
self.peers += 1
elif key == HathorEvents.NETWORK_PEER_DISCONNECTED:
# Check if peer was ready before disconnecting
if data['protocol'].state.state_name == HathorProtocol.PeerState.READY.name:
self.peers -= 1
else:
raise ValueError('Invalid key')
def set_stratum_data(self) -> None:
""" Set stratum metrics data for the mining process
"""
if not self.stratum_factory:
return
stratum_stats = self.stratum_factory.get_stats()
completed_jobs = 0
blocks_found = 0
estimated_hash_rate = 0.0
for stats in stratum_stats:
completed_jobs += stats.completed_jobs
blocks_found += stats.blocks_found
estimated_hash_rate = sum_weights(estimated_hash_rate, stats.estimated_hash_rate)
self.completed_jobs = completed_jobs
self.blocks_found = blocks_found
self.estimated_hash_rate = estimated_hash_rate