def test_change_state_from_same_state(self):
# GIVEN
mock_channel_service = MockChannelService()
channel_state_machine = ChannelStateMachine(mock_channel_service)
channel_state_machine.complete_init_components()
channel_state_machine.block_sync()
channel_state_machine.complete_sync()
channel_state_machine.complete_subscribe()
util.logger.spam(f"\nstate is {channel_state_machine.state}")
# WHEN
channel_state_machine.turn_to_leader()
util.logger.spam(
f"\ntimer called({mock_channel_service.block_manager.timer_called})"
)
channel_state_machine.turn_to_leader()
util.logger.spam(f"\nstate is {channel_state_machine.state}")
# THEN
self.assertEqual(1, mock_channel_service.block_manager.timer_called)
class ChannelService:
def __init__(self, channel_name, amqp_target, amqp_key):
self.__block_manager: BlockManager = None
self.__score_container: CommonSubprocess = None
self.__score_info: dict = None
self.__peer_auth: Signer = None
self.__peer_manager: PeerManager = None
self.__broadcast_scheduler: BroadcastScheduler = None
self.__radio_station_stub = None
self.__consensus = None
# self.__proposer: Proposer = None
# self.__acceptor: Acceptor = None
self.__timer_service = TimerService()
self.__node_subscriber: NodeSubscriber = None
loggers.get_preset().channel_name = channel_name
loggers.get_preset().update_logger()
channel_queue_name = conf.CHANNEL_QUEUE_NAME_FORMAT.format(
channel_name=channel_name, amqp_key=amqp_key)
self.__inner_service = ChannelInnerService(amqp_target,
channel_queue_name,
conf.AMQP_USERNAME,
conf.AMQP_PASSWORD,
channel_service=self)
logging.info(
f"ChannelService : {channel_name}, Queue : {channel_queue_name}")
ChannelProperty().name = channel_name
ChannelProperty().amqp_target = amqp_target
StubCollection().amqp_key = amqp_key
StubCollection().amqp_target = amqp_target
command_arguments.add_raw_command(command_arguments.Type.Channel,
channel_name)
command_arguments.add_raw_command(command_arguments.Type.AMQPTarget,
amqp_target)
command_arguments.add_raw_command(command_arguments.Type.AMQPKey,
amqp_key)
ObjectManager().channel_service = self
self.__state_machine = ChannelStateMachine(self)
@property
def block_manager(self):
return self.__block_manager
@property
def score_container(self):
return self.__score_container
@property
def score_info(self):
return self.__score_info
@property
def radio_station_stub(self):
return self.__radio_station_stub
@property
def peer_auth(self):
return self.__peer_auth
@property
def peer_manager(self):
return self.__peer_manager
@property
def broadcast_scheduler(self):
return self.__broadcast_scheduler
@property
def consensus(self):
return self.__consensus
@property
def acceptor(self):
return self.__acceptor
@property
def timer_service(self):
return self.__timer_service
@property
def state_machine(self):
return self.__state_machine
@property
def inner_service(self):
return self.__inner_service
def serve(self):
async def _serve():
await StubCollection().create_peer_stub()
results = await StubCollection().peer_stub.async_task(
).get_channel_info_detail(ChannelProperty().name)
await self.init(*results)
self.__timer_service.start()
self.__state_machine.complete_init_components()
logging.info(
f'channel_service: init complete channel: {ChannelProperty().name}, '
f'state({self.__state_machine.state})')
loop = MessageQueueService.loop
loop.create_task(_serve())
loop.add_signal_handler(signal.SIGINT, self.close)
loop.add_signal_handler(signal.SIGTERM, self.close)
try:
loop.run_forever()
finally:
loop.run_until_complete(loop.shutdown_asyncgens())
loop.close()
self.cleanup()
def close(self):
MessageQueueService.loop.stop()
def cleanup(self):
logging.info("Cleanup Channel Resources.")
if self.__block_manager:
self.__block_manager.stop()
self.__block_manager = None
logging.info("Cleanup BlockManager.")
if self.__score_container:
self.__score_container.stop()
self.__score_container.wait()
self.__score_container = None
logging.info("Cleanup ScoreContainer.")
if self.__broadcast_scheduler:
self.__broadcast_scheduler.stop()
self.__broadcast_scheduler.wait()
self.__broadcast_scheduler = None
logging.info("Cleanup BroadcastScheduler.")
if self.__consensus:
self.__consensus.stop()
self.__consensus.wait()
logging.info("Cleanup Consensus.")
if self.__timer_service.is_run():
self.__timer_service.stop()
self.__timer_service.wait()
logging.info("Cleanup TimerService.")
async def init(self, peer_port, peer_target, rest_target,
radio_station_target, peer_id, group_id, node_type,
score_package):
loggers.get_preset().peer_id = peer_id
loggers.get_preset().update_logger()
ChannelProperty().peer_port = peer_port
ChannelProperty().peer_target = peer_target
ChannelProperty().rest_target = rest_target
ChannelProperty().radio_station_target = radio_station_target
ChannelProperty().peer_id = peer_id
ChannelProperty().group_id = group_id
ChannelProperty().node_type = conf.NodeType(node_type)
ChannelProperty().score_package = score_package
self.__peer_manager = PeerManager(ChannelProperty().name)
self.__init_peer_auth()
self.__init_broadcast_scheduler()
self.__init_block_manager()
self.__init_radio_station_stub()
await self.__init_score_container()
await self.__inner_service.connect(conf.AMQP_CONNECTION_ATTEMPS,
conf.AMQP_RETRY_DELAY,
exclusive=True)
# if conf.CONSENSUS_ALGORITHM == conf.ConsensusAlgorithm.lft:
# util.logger.spam(f"init consensus !")
# # load consensus
# self.__init_consensus()
# # load proposer
# self.__init_proposer(peer_id=peer_id)
# # load acceptor
# self.__init_acceptor(peer_id=peer_id)
if self.is_support_node_function(conf.NodeFunction.Vote):
if conf.ENABLE_REP_RADIO_STATION:
self.connect_to_radio_station()
else:
await self.__load_peers_from_file()
# subscribe to other peers
self.__subscribe_to_peer_list()
# broadcast AnnounceNewPeer to other peers
# If allow broadcast AnnounceNewPeer here, complained peer can be leader again.
else:
self.__init_node_subscriber()
self.block_manager.init_epoch()
async def evaluate_network(self):
await self.set_peer_type_in_channel()
if self.block_manager.peer_type == loopchain_pb2.BLOCK_GENERATOR:
self.__state_machine.subscribe_network()
else:
self.__state_machine.block_sync()
async def subscribe_network(self):
# Subscribe to radiostation and block_sync_target_stub
if self.is_support_node_function(conf.NodeFunction.Vote):
if conf.ENABLE_REP_RADIO_STATION:
await self.subscribe_to_radio_station()
if self.block_manager.peer_type == loopchain_pb2.PEER:
await self.__subscribe_call_to_stub(
peer_stub=self.block_manager.subscribe_target_peer_stub,
peer_type=loopchain_pb2.PEER)
else:
await self.subscribe_to_radio_station()
self.generate_genesis_block()
if conf.CONSENSUS_ALGORITHM == conf.ConsensusAlgorithm.lft:
if not self.__consensus.is_run():
self.__consensus.change_epoch(
precommit_block=self.__block_manager.get_blockchain(
).last_block)
self.__consensus.start()
elif conf.ALLOW_MAKE_EMPTY_BLOCK:
if not self.block_manager.block_generation_scheduler.is_run():
self.block_manager.block_generation_scheduler.start()
self.__state_machine.complete_sync()
def __init_peer_auth(self):
try:
self.__peer_auth = Signer.from_channel(ChannelProperty().name)
except Exception as e:
logging.exception(f"peer auth init fail cause : {e}")
util.exit_and_msg(f"peer auth init fail cause : {e}")
def __init_block_manager(self):
logging.debug(
f"__load_block_manager_each channel({ChannelProperty().name})")
try:
self.__block_manager = BlockManager(
name="loopchain.peer.BlockManager",
channel_manager=self,
peer_id=ChannelProperty().peer_id,
channel_name=ChannelProperty().name,
level_db_identity=ChannelProperty().peer_target)
except leveldb.LevelDBError as e:
util.exit_and_msg("LevelDBError(" + str(e) + ")")
# def __init_consensus(self):
# consensus = Consensus(self, ChannelProperty().name)
# self.__consensus = consensus
# self.__block_manager.consensus = consensus
# consensus.register_subscriber(self.__block_manager)
#
# def __init_proposer(self, peer_id: str):
# proposer = Proposer(
# name="loopchain.consensus.Proposer",
# peer_id=peer_id,
# channel=ChannelProperty().name,
# channel_service=self
# )
# self.__consensus.register_subscriber(proposer)
# self.__proposer = proposer
#
# def __init_acceptor(self, peer_id: str):
# acceptor = Acceptor(
# name="loopchain.consensus.Acceptor",
# consensus=self.__consensus,
# peer_id=peer_id,
# channel=ChannelProperty().name,
# channel_service=self
# )
# self.__consensus.register_subscriber(acceptor)
# self.__acceptor = acceptor
def __init_broadcast_scheduler(self):
scheduler = BroadcastScheduler(
channel=ChannelProperty().name,
self_target=ChannelProperty().peer_target)
scheduler.start()
self.__broadcast_scheduler = scheduler
future = scheduler.schedule_job(BroadcastCommand.SUBSCRIBE,
ChannelProperty().peer_target)
future.result(conf.TIMEOUT_FOR_FUTURE)
def __init_radio_station_stub(self):
if self.is_support_node_function(conf.NodeFunction.Vote):
if conf.ENABLE_REP_RADIO_STATION:
self.__radio_station_stub = StubManager.get_stub_manager_to_server(
ChannelProperty().radio_station_target,
loopchain_pb2_grpc.RadioStationStub,
conf.CONNECTION_RETRY_TIMEOUT_TO_RS,
ssl_auth_type=conf.GRPC_SSL_TYPE)
else:
self.__radio_station_stub = RestStubManager(
ChannelProperty().radio_station_target,
ChannelProperty().name)
async def __init_score_container(self):
"""create score container and save score_info and score_stub
"""
for i in range(conf.SCORE_LOAD_RETRY_TIMES):
try:
self.__score_info = await self.__run_score_container()
except BaseException as e:
util.logger.spam(
f"channel_manager:load_score_container_each score_info load fail retry({i})"
)
logging.error(e)
traceback.print_exc()
time.sleep(conf.SCORE_LOAD_RETRY_INTERVAL
) # This blocking main thread is intended.
else:
break
def __init_node_subscriber(self):
self.__node_subscriber = NodeSubscriber(
channel=ChannelProperty().name,
rs_target=ChannelProperty().radio_station_target)
async def __run_score_container(self):
if not conf.USE_EXTERNAL_SCORE or conf.EXTERNAL_SCORE_RUN_IN_LAUNCHER:
process_args = [
'python3', '-m', 'loopchain', 'score', '--channel',
ChannelProperty().name, '--score_package',
ChannelProperty().score_package
]
process_args += command_arguments.get_raw_commands_by_filter(
command_arguments.Type.AMQPTarget,
command_arguments.Type.AMQPKey, command_arguments.Type.Develop,
command_arguments.Type.ConfigurationFilePath,
command_arguments.Type.RadioStationTarget)
self.__score_container = CommonSubprocess(process_args)
await StubCollection().create_icon_score_stub(ChannelProperty().name)
await StubCollection().icon_score_stubs[ChannelProperty().name
].connect()
await StubCollection().icon_score_stubs[ChannelProperty().name
].async_task().hello()
return None
async def __load_score(self):
channel_name = ChannelProperty().name
score_package_name = ChannelProperty().score_package
util.logger.spam(f"peer_service:__load_score --init--")
logging.info("LOAD SCORE AND CONNECT TO SCORE SERVICE!")
params = dict()
params[message_code.MetaParams.ScoreLoad.
repository_path] = conf.DEFAULT_SCORE_REPOSITORY_PATH
params[message_code.MetaParams.ScoreLoad.
score_package] = score_package_name
params[
message_code.MetaParams.ScoreLoad.base] = conf.DEFAULT_SCORE_BASE
params[message_code.MetaParams.ScoreLoad.peer_id] = ChannelProperty(
).peer_id
meta = json.dumps(params)
logging.debug(f"load score params : {meta}")
util.logger.spam(f"peer_service:__load_score --1--")
score_stub = StubCollection().score_stubs[channel_name]
response = await score_stub.async_task().score_load(meta)
logging.debug("try score load on score service: " + str(response))
if not response:
return None
if response.code != message_code.Response.success:
util.exit_and_msg("Fail Get Score from Score Server...")
return None
logging.debug("Get Score from Score Server...")
score_info = json.loads(response.meta)
logging.info("LOAD SCORE DONE!")
util.logger.spam(f"peer_service:__load_score --end--")
return score_info
async def __load_peers_from_file(self):
channel_info = await StubCollection().peer_stub.async_task(
).get_channel_infos()
for peer_info in channel_info[ChannelProperty().name]["peers"]:
self.__peer_manager.add_peer(peer_info)
self.__broadcast_scheduler.schedule_job(BroadcastCommand.SUBSCRIBE,
peer_info["peer_target"])
self.show_peers()
def is_support_node_function(self, node_function):
return conf.NodeType.is_support_node_function(
node_function,
ChannelProperty().node_type)
def get_channel_option(self) -> dict:
channel_option = conf.CHANNEL_OPTION
return channel_option[ChannelProperty().name]
def generate_genesis_block(self):
if self.block_manager.peer_type != loopchain_pb2.BLOCK_GENERATOR:
return
block_chain = self.block_manager.get_blockchain()
if block_chain.block_height > -1:
logging.debug("genesis block was already generated")
return
block_chain.generate_genesis_block()
def connect_to_radio_station(self, is_reconnect=False):
response = self.__radio_station_stub.call_in_times(
method_name="ConnectPeer",
message=loopchain_pb2.ConnectPeerRequest(
channel=ChannelProperty().name,
peer_object=b'',
peer_id=ChannelProperty().peer_id,
peer_target=ChannelProperty().peer_target,
group_id=ChannelProperty().group_id),
retry_times=conf.CONNECTION_RETRY_TIMES_TO_RS,
is_stub_reuse=True,
timeout=conf.CONNECTION_TIMEOUT_TO_RS)
# start next ConnectPeer timer
self.__timer_service.add_timer_convenient(
timer_key=TimerService.TIMER_KEY_CONNECT_PEER,
duration=conf.CONNECTION_RETRY_TIMER,
callback=self.connect_to_radio_station,
callback_kwargs={"is_reconnect": True})
if is_reconnect:
return
if response and response.status == message_code.Response.success:
peer_list_data = pickle.loads(response.peer_list)
self.__peer_manager.load(peer_list_data, False)
peers, peer_list = self.__peer_manager.get_peers_for_debug()
logging.debug("peer list update: " + peers)
# add connected peer to processes audience
for each_peer in peer_list:
util.logger.spam(
f"peer_service:connect_to_radio_station peer({each_peer.target}-{each_peer.status})"
)
if each_peer.status == PeerStatus.connected:
self.__broadcast_scheduler.schedule_job(
BroadcastCommand.SUBSCRIBE, each_peer.target)
def __subscribe_to_peer_list(self):
peer_object = self.peer_manager.get_peer(ChannelProperty().peer_id)
peer_request = loopchain_pb2.PeerRequest(
channel=ChannelProperty().name,
peer_target=ChannelProperty().peer_target,
peer_id=ChannelProperty().peer_id,
group_id=ChannelProperty().group_id,
node_type=ChannelProperty().node_type,
peer_order=peer_object.order)
self.__broadcast_scheduler.schedule_broadcast("Subscribe",
peer_request)
async def subscribe_to_radio_station(self):
await self.__subscribe_call_to_stub(self.__radio_station_stub,
loopchain_pb2.PEER)
async def subscribe_to_peer(self, peer_id, peer_type):
peer = self.peer_manager.get_peer(peer_id)
peer_stub = self.peer_manager.get_peer_stub_manager(peer)
await self.__subscribe_call_to_stub(peer_stub, peer_type)
self.__broadcast_scheduler.schedule_job(BroadcastCommand.SUBSCRIBE,
peer_stub.target)
async def __subscribe_call_to_stub(self, peer_stub, peer_type):
if self.is_support_node_function(conf.NodeFunction.Vote):
await peer_stub.call_async(
"Subscribe",
loopchain_pb2.PeerRequest(
channel=ChannelProperty().name,
peer_target=ChannelProperty().peer_target,
peer_type=peer_type,
peer_id=ChannelProperty().peer_id,
group_id=ChannelProperty().group_id,
node_type=ChannelProperty().node_type),
)
else:
await self.__subscribe_call_from_citizen()
async def __subscribe_call_from_citizen(self):
def _handle_exception(future: asyncio.Future):
logging.debug(
f"error: {type(future.exception())}, {str(future.exception())}"
)
if isinstance(future.exception(), NotImplementedError):
asyncio.ensure_future(
self.__subscribe_call_by_rest_stub(subscribe_event))
elif isinstance(future.exception(), ConnectionError):
logging.warning(f"Waiting for next subscribe request...")
if self.__state_machine.state != "SubscribeNetwork":
self.__state_machine.subscribe_network()
subscribe_event = asyncio.Event()
util.logger.spam(
f"try subscribe_call_by_citizen target({ChannelProperty().rest_target})"
)
# try websocket connection, and handle exception in callback
asyncio.ensure_future(
self.__node_subscriber.subscribe(
block_height=self.block_manager.get_blockchain().block_height,
event=subscribe_event)).add_done_callback(_handle_exception)
await subscribe_event.wait()
async def __subscribe_call_by_rest_stub(self, event):
if conf.REST_SSL_TYPE == conf.SSLAuthType.none:
peer_target = ChannelProperty().rest_target
else:
peer_target = f"https://{ChannelProperty().rest_target}"
response = None
try:
response = await self.__radio_station_stub.call_async(
"Subscribe", {
'channel': ChannelProperty().name,
'peer_target': peer_target
})
except Exception as e:
logging.warning(
f"Due to Subscription fail to RadioStation(mother peer), "
f"automatically retrying subscribe call")
if response and response[
'response_code'] == message_code.Response.success:
logging.debug(
f"Subscription to RadioStation(mother peer) is successful.")
event.set()
self.start_check_last_block_rs_timer()
def __check_last_block_to_rs(self):
last_block = self.__radio_station_stub.call_async("GetLastBlock")
if last_block['height'] <= self.__block_manager.get_blockchain(
).block_height:
return
# RS peer didn't announced new block
self.stop_check_last_block_rs_timer()
if self.__state_machine.state != "SubscribeNetwork":
self.__state_machine.subscribe_network()
def shutdown_peer(self, **kwargs):
logging.debug(f"channel_service:shutdown_peer")
StubCollection().peer_stub.sync_task().stop(message=kwargs['message'])
def set_peer_type(self, peer_type):
"""Set peer type when peer init only
:param peer_type:
:return:
"""
self.__block_manager.set_peer_type(peer_type)
def save_peer_manager(self, peer_manager):
"""peer_list 를 leveldb 에 저장한다.
:param peer_manager:
"""
level_db_key_name = str.encode(conf.LEVEL_DB_KEY_FOR_PEER_LIST)
try:
dump = peer_manager.dump()
level_db = self.__block_manager.get_level_db()
level_db.Put(level_db_key_name, dump)
except AttributeError as e:
logging.warning("Fail Save Peer_list: " + str(e))
async def set_peer_type_in_channel(self):
peer_type = loopchain_pb2.PEER
peer_leader = self.peer_manager.get_leader_peer(
is_complain_to_rs=self.is_support_node_function(
conf.NodeFunction.Vote))
logging.debug(f"channel({ChannelProperty().name}) peer_leader: " +
str(peer_leader))
logger_preset = loggers.get_preset()
if self.is_support_node_function(
conf.NodeFunction.Vote) and ChannelProperty(
).peer_id == peer_leader.peer_id:
logger_preset.is_leader = True
logging.debug(
f"Set Peer Type Leader! channel({ChannelProperty().name})")
peer_type = loopchain_pb2.BLOCK_GENERATOR
else:
logger_preset.is_leader = False
logger_preset.update_logger()
if conf.CONSENSUS_ALGORITHM == conf.ConsensusAlgorithm.lft:
self.consensus.leader_id = peer_leader.peer_id
if peer_type == loopchain_pb2.BLOCK_GENERATOR:
self.block_manager.set_peer_type(peer_type)
self.__ready_to_height_sync(True)
elif peer_type == loopchain_pb2.PEER:
self.__ready_to_height_sync(False)
def __ready_to_height_sync(self, is_leader: bool = False):
block_chain = self.block_manager.get_blockchain()
block_chain.init_block_chain(is_leader)
if block_chain.block_height > -1:
self.block_manager.rebuild_block()
async def block_height_sync_channel(self):
# leader 로 시작하지 않았는데 자신의 정보가 leader Peer 정보이면 block height sync 하여
# 최종 블럭의 leader 를 찾는다.
peer_manager = self.peer_manager
peer_leader = peer_manager.get_leader_peer()
self_peer_object = peer_manager.get_peer(ChannelProperty().peer_id)
is_delay_announce_new_leader = False
peer_old_leader = None
if peer_leader:
block_sync_target = peer_leader.target
block_sync_target_stub = StubManager.get_stub_manager_to_server(
block_sync_target,
loopchain_pb2_grpc.PeerServiceStub,
time_out_seconds=conf.CONNECTION_RETRY_TIMEOUT,
ssl_auth_type=conf.GRPC_SSL_TYPE)
else:
block_sync_target = ChannelProperty().radio_station_target
block_sync_target_stub = self.__radio_station_stub
if block_sync_target != ChannelProperty().peer_target:
if block_sync_target_stub is None:
logging.warning(
"You maybe Older from this network... or No leader in this network!"
)
is_delay_announce_new_leader = True
peer_old_leader = peer_leader
peer_leader = self.peer_manager.leader_complain_to_rs(
conf.ALL_GROUP_ID, is_announce_new_peer=False)
if peer_leader is not None and ChannelProperty(
).node_type == conf.NodeType.CommunityNode:
block_sync_target_stub = StubManager.get_stub_manager_to_server(
peer_leader.target,
loopchain_pb2_grpc.PeerServiceStub,
time_out_seconds=conf.CONNECTION_RETRY_TIMEOUT,
ssl_auth_type=conf.GRPC_SSL_TYPE)
if self.is_support_node_function(conf.NodeFunction.Vote) and \
(not peer_leader or peer_leader.peer_id == ChannelProperty().peer_id):
peer_leader = self_peer_object
self.block_manager.set_peer_type(loopchain_pb2.BLOCK_GENERATOR)
else:
_, future = self.block_manager.block_height_sync(
block_sync_target_stub)
await future
self.show_peers()
if is_delay_announce_new_leader and ChannelProperty(
).node_type == conf.NodeType.CommunityNode:
self.peer_manager.announce_new_leader(
peer_old_leader.peer_id,
peer_leader.peer_id,
self_peer_id=ChannelProperty().peer_id)
def show_peers(self):
logging.debug(f"peer_service:show_peers ({ChannelProperty().name}): ")
for peer in self.peer_manager.get_IP_of_peers_in_group():
logging.debug("peer_target: " + peer)
async def reset_leader(self, new_leader_id, block_height=0):
logging.info(
f"RESET LEADER channel({ChannelProperty().name}) leader_id({new_leader_id})"
)
leader_peer = self.peer_manager.get_peer(new_leader_id, None)
if block_height > 0 and block_height != self.block_manager.get_blockchain(
).last_block.header.height + 1:
util.logger.warning(
f"height behind peer can not take leader role. block_height({block_height}), "
f"last_block.header.height("
f"{self.block_manager.get_blockchain().last_block.header.height})"
)
return
if leader_peer is None:
logging.warning(
f"in peer_service:reset_leader There is no peer by peer_id({new_leader_id})"
)
return
util.logger.spam(
f"peer_service:reset_leader target({leader_peer.target})")
self_peer_object = self.peer_manager.get_peer(
ChannelProperty().peer_id)
self.peer_manager.set_leader_peer(leader_peer, None)
peer_leader = self.peer_manager.get_leader_peer()
peer_type = loopchain_pb2.PEER
if self_peer_object.target == peer_leader.target:
logging.debug("Set Peer Type Leader!")
peer_type = loopchain_pb2.BLOCK_GENERATOR
self.state_machine.turn_to_leader()
if conf.CONSENSUS_ALGORITHM != conf.ConsensusAlgorithm.lft:
if conf.ENABLE_REP_RADIO_STATION:
self.peer_manager.announce_new_leader(
self.peer_manager.get_leader_peer().peer_id,
new_leader_id,
is_broadcast=True,
self_peer_id=ChannelProperty().peer_id)
else:
logging.debug("Set Peer Type Peer!")
self.state_machine.turn_to_peer()
# 새 leader 에게 subscribe 하기
# await self.subscribe_to_radio_station()
await self.subscribe_to_peer(peer_leader.peer_id,
loopchain_pb2.BLOCK_GENERATOR)
self.block_manager.set_peer_type(peer_type)
self.block_manager.epoch.set_epoch_leader(peer_leader.peer_id)
def set_new_leader(self, new_leader_id, block_height=0):
logging.info(
f"SET NEW LEADER channel({ChannelProperty().name}) leader_id({new_leader_id})"
)
# complained_leader = self.peer_manager.get_leader_peer()
leader_peer = self.peer_manager.get_peer(new_leader_id, None)
if block_height > 0 and block_height != self.block_manager.get_blockchain(
).last_block.height + 1:
logging.warning(f"height behind peer can not take leader role.")
return
if leader_peer is None:
logging.warning(
f"in channel_service:set_new_leader::There is no peer by peer_id({new_leader_id})"
)
return
util.logger.spam(
f"channel_service:set_new_leader::leader_target({leader_peer.target})"
)
self_peer_object = self.peer_manager.get_peer(
ChannelProperty().peer_id)
self.peer_manager.set_leader_peer(leader_peer, None)
peer_leader = self.peer_manager.get_leader_peer()
if self_peer_object.target == peer_leader.target:
loggers.get_preset().is_leader = True
loggers.get_preset().update_logger()
logging.debug("I'm Leader Peer!")
else:
loggers.get_preset().is_leader = False
loggers.get_preset().update_logger()
logging.debug("I'm general Peer!")
# 새 leader 에게 subscribe 하기
# await self.subscribe_to_radio_station()
# await self.subscribe_to_peer(peer_leader.peer_id, loopchain_pb2.BLOCK_GENERATOR)
def genesis_invoke(self, block: Block) -> ('Block', dict):
method = "icx_sendTransaction"
transactions = []
for tx in block.body.transactions.values():
tx_serializer = TransactionSerializer.new(
tx.version,
self.block_manager.get_blockchain().tx_versioner)
transaction = {
"method": method,
"params": {
"txHash": tx.hash.hex()
},
"genesisData": tx_serializer.to_full_data(tx)
}
transactions.append(transaction)
request = {
'block': {
'blockHeight': block.header.height,
'blockHash': block.header.hash.hex(),
'timestamp': block.header.timestamp
},
'transactions': transactions
}
request = convert_params(request, ParamType.invoke)
stub = StubCollection().icon_score_stubs[ChannelProperty().name]
response = stub.sync_task().invoke(request)
response_to_json_query(response)
block_builder = BlockBuilder.from_new(
block,
self.block_manager.get_blockchain().tx_versioner)
block_builder.commit_state = {
ChannelProperty().name: response['stateRootHash']
}
new_block = block_builder.build()
return new_block, response["txResults"]
def score_invoke(self, _block: Block) -> dict or None:
method = "icx_sendTransaction"
transactions = []
for tx in _block.body.transactions.values():
tx_serializer = TransactionSerializer.new(
tx.version,
self.block_manager.get_blockchain().tx_versioner)
transaction = {
"method": method,
"params": tx_serializer.to_full_data(tx)
}
transactions.append(transaction)
request = {
'block': {
'blockHeight':
_block.header.height,
'blockHash':
_block.header.hash.hex(),
'prevBlockHash':
_block.header.prev_hash.hex()
if _block.header.prev_hash else '',
'timestamp':
_block.header.timestamp
},
'transactions': transactions
}
request = convert_params(request, ParamType.invoke)
stub = StubCollection().icon_score_stubs[ChannelProperty().name]
response = stub.sync_task().invoke(request)
response_to_json_query(response)
block_builder = BlockBuilder.from_new(
_block,
self.__block_manager.get_blockchain().tx_versioner)
block_builder.commit_state = {
ChannelProperty().name: response['stateRootHash']
}
new_block = block_builder.build()
return new_block, response["txResults"]
def score_change_block_hash(self, block_height, old_block_hash,
new_block_hash):
change_hash_info = json.dumps({
"block_height": block_height,
"old_block_hash": old_block_hash,
"new_block_hash": new_block_hash
})
stub = StubCollection().score_stubs[ChannelProperty().name]
stub.sync_task().change_block_hash(change_hash_info)
def score_write_precommit_state(self, block: Block):
logging.debug(
f"call score commit {ChannelProperty().name} {block.header.height} {block.header.hash.hex()}"
)
request = {
"blockHeight": block.header.height,
"blockHash": block.header.hash.hex(),
}
request = convert_params(request, ParamType.write_precommit_state)
stub = StubCollection().icon_score_stubs[ChannelProperty().name]
stub.sync_task().write_precommit_state(request)
return True
def score_remove_precommit_state(self, block: Block):
invoke_fail_info = json.dumps({
"block_height": block.height,
"block_hash": block.block_hash
})
stub = StubCollection().score_stubs[ChannelProperty().name]
stub.sync_task().remove_precommit_state(invoke_fail_info)
return True
def get_object_has_queue_by_consensus(self):
if conf.CONSENSUS_ALGORITHM == conf.ConsensusAlgorithm.lft:
object_has_queue = self.__consensus
else:
object_has_queue = self.__block_manager
self.start_leader_complain_timer()
return object_has_queue
def start_leader_complain_timer(self):
# util.logger.debug(f"start_leader_complain_timer in channel service.")
self.__timer_service.add_timer_convenient(
timer_key=TimerService.TIMER_KEY_LEADER_COMPLAIN,
duration=conf.TIMEOUT_FOR_LEADER_COMPLAIN,
is_repeat=True,
callback=self.state_machine.leader_complain)
def stop_leader_complain_timer(self):
# util.logger.debug(f"stop_leader_complain_timer in channel service.")
self.__timer_service.stop_timer(TimerService.TIMER_KEY_LEADER_COMPLAIN)
def start_subscribe_timer(self):
self.__timer_service.add_timer_convenient(
timer_key=TimerService.TIMER_KEY_SUBSCRIBE,
duration=conf.SUBSCRIBE_RETRY_TIMER,
is_repeat=True,
callback=self.subscribe_network)
def stop_subscribe_timer(self):
self.__timer_service.stop_timer(TimerService.TIMER_KEY_SUBSCRIBE)
def start_check_last_block_rs_timer(self):
self.__timer_service.add_timer_convenient(
timer_key=TimerService.
TIMER_KEY_GET_LAST_BLOCK_KEEP_CITIZEN_SUBSCRIPTION,
duration=conf.GET_LAST_BLOCK_TIMER,
is_repeat=True,
callback=self.__check_last_block_to_rs)
def stop_check_last_block_rs_timer(self):
self.__timer_service.stop_timer(
TimerService.TIMER_KEY_GET_LAST_BLOCK_KEEP_CITIZEN_SUBSCRIPTION)
def start_shutdown_timer(self):
error = f"Shutdown by Subscribe retry timeout({conf.SHUTDOWN_TIMER} sec)"
self.__timer_service.add_timer_convenient(
timer_key=TimerService.TIMER_KEY_SHUTDOWN_WHEN_FAIL_SUBSCRIBE,
duration=conf.SHUTDOWN_TIMER,
callback=self.shutdown_peer,
callback_kwargs={"message": error})
def stop_shutdown_timer(self):
self.__timer_service.stop_timer(
TimerService.TIMER_KEY_SHUTDOWN_WHEN_FAIL_SUBSCRIBE)
class ChannelService:
def __init__(self, channel_name, amqp_target, amqp_key, rollback=False):
self.__block_manager: BlockManager = None
self.__score_container: CommonSubprocess = None
self.__score_info: dict = None
self.__peer_auth: Signer = None
self.__broadcast_scheduler: BroadcastScheduler = None
self.__rs_client: RestClient = None
self.__timer_service = TimerService()
self.__node_subscriber: NodeSubscriber = None
self._rollback: bool = rollback
loggers.get_preset().channel_name = channel_name
loggers.get_preset().update_logger()
channel_queue_name = conf.CHANNEL_QUEUE_NAME_FORMAT.format(channel_name=channel_name, amqp_key=amqp_key)
self.__inner_service = ChannelInnerService(
amqp_target, channel_queue_name, conf.AMQP_USERNAME, conf.AMQP_PASSWORD, channel_service=self)
logging.info(f"ChannelService : {channel_name}, Queue : {channel_queue_name}")
ChannelProperty().name = channel_name
ChannelProperty().amqp_target = amqp_target
ChannelProperty().crep_root_hash = Hash32.fromhex(conf.CHANNEL_OPTION[channel_name].get('crep_root_hash'))
StubCollection().amqp_key = amqp_key
StubCollection().amqp_target = amqp_target
command_arguments.add_raw_command(command_arguments.Type.Channel, channel_name)
command_arguments.add_raw_command(command_arguments.Type.AMQPTarget, amqp_target)
command_arguments.add_raw_command(command_arguments.Type.AMQPKey, amqp_key)
ObjectManager().channel_service = self
self.__state_machine = ChannelStateMachine(self)
@property
def block_manager(self):
return self.__block_manager
@property
def score_container(self):
return self.__score_container
@property
def score_info(self):
return self.__score_info
@property
def rs_client(self):
return self.__rs_client
@property
def broadcast_scheduler(self):
return self.__broadcast_scheduler
@property
def timer_service(self):
return self.__timer_service
@property
def state_machine(self):
return self.__state_machine
@property
def inner_service(self):
return self.__inner_service
@property
def node_subscriber(self):
return self.__node_subscriber
def serve(self):
async def _serve():
await StubCollection().create_peer_stub()
results = await StubCollection().peer_stub.async_task().get_node_info_detail()
self._init_properties(**results)
await self._init()
self.__timer_service.start()
self.__state_machine.complete_init_components()
logging.info(f'channel_service: init complete channel: {ChannelProperty().name}, '
f'state({self.__state_machine.state})')
loop = self.__inner_service.loop
serve_coroutine = _serve() if not self._rollback else self._serve_manual_rollback()
loop.create_task(serve_coroutine)
loop.add_signal_handler(signal.SIGINT, self.close, signal.SIGINT)
loop.add_signal_handler(signal.SIGTERM, self.close, signal.SIGTERM)
try:
loop.run_forever()
except Exception as e:
traceback.print_exception(type(e), e, e.__traceback__)
finally:
loop.run_until_complete(loop.shutdown_asyncgens())
self._cancel_tasks(loop)
self._cleanup()
loop.close()
async def _serve_manual_rollback(self):
"""Initialize minimum channel resources and manual rollback
:return: None
"""
await StubCollection().create_peer_stub()
results = await StubCollection().peer_stub.async_task().get_node_info_detail()
self._init_properties(**results)
self.__init_block_manager()
await self.__init_score_container()
await self.__inner_service.connect(conf.AMQP_CONNECTION_ATTEMPTS, conf.AMQP_RETRY_DELAY, exclusive=True)
await asyncio.sleep(0.01) # sleep to complete peer service initialization
message = self._manual_rollback()
self.shutdown_peer(message=message)
def _manual_rollback(self) -> str:
logging.debug("_manual_rollback() start manual rollback")
if self.block_manager.blockchain.block_height >= 0:
self.block_manager.rebuild_block()
if self.block_manager.request_rollback():
message = "rollback finished"
else:
message = "rollback cancelled"
logging.debug("_manual_rollback() end manual rollback")
return message
def close(self, signum=None):
logging.info(f"close() signum = {repr(signum)}")
if self.__inner_service:
self.__inner_service.cleanup()
self.__inner_service.loop.stop()
@staticmethod
def _cancel_tasks(loop):
for task in asyncio.Task.all_tasks(loop):
if task.done():
continue
task.cancel()
try:
loop.run_until_complete(task)
except asyncio.CancelledError as e:
logging.info(f"_cancel_tasks() task : {task}, error : {e}")
def _cleanup(self):
logging.info("_cleanup() Channel Resources.")
if self.__timer_service.is_run():
self.__timer_service.stop()
self.__timer_service.wait()
logging.info("_cleanup() TimerService.")
if self.__score_container:
self.__score_container.stop()
self.__score_container.wait()
self.__score_container = None
logging.info("_cleanup() ScoreContainer.")
if self.__broadcast_scheduler:
self.__broadcast_scheduler.stop()
self.__broadcast_scheduler.wait()
self.__broadcast_scheduler = None
logging.info("_cleanup() BroadcastScheduler.")
if self.__block_manager:
self.__block_manager.stop()
self.__block_manager = None
logging.info("_cleanup() BlockManager.")
@staticmethod
def _init_properties(**kwargs):
"""Initialize properties
:param kwargs: takes (peer_id, peer_port, peer_target, rest_target)
within parameters
:return: None
"""
loggers.get_preset().peer_id = kwargs.get('peer_id')
loggers.get_preset().update_logger()
ChannelProperty().peer_port = kwargs.get('peer_port')
ChannelProperty().peer_target = kwargs.get('peer_target')
ChannelProperty().rest_target = kwargs.get('rest_target')
ChannelProperty().peer_id = kwargs.get('peer_id')
ChannelProperty().peer_address = ExternalAddress.fromhex_address(ChannelProperty().peer_id)
ChannelProperty().node_type = conf.NodeType.CitizenNode
ChannelProperty().rs_target = None
async def _init(self):
"""Initialize channel resources
:return: None
"""
await self.__init_peer_auth()
self.__init_broadcast_scheduler()
self.__init_block_manager()
await self.__init_score_container()
await self.__inner_service.connect(conf.AMQP_CONNECTION_ATTEMPTS, conf.AMQP_RETRY_DELAY, exclusive=True)
await self.__init_sub_services()
async def evaluate_network(self):
await self._init_rs_client()
self.__block_manager.blockchain.init_crep_reps()
await self._select_node_type()
self.__ready_to_height_sync()
self.__state_machine.block_sync()
async def subscribe_network(self):
await self._select_node_type()
if self.is_support_node_function(conf.NodeFunction.Vote):
await self.set_peer_type_in_channel()
else:
await self._init_rs_target()
if ChannelProperty().rs_target is None:
return
self.__init_node_subscriber()
await self.subscribe_to_parent()
self.__state_machine.complete_subscribe()
if self.is_support_node_function(conf.NodeFunction.Vote):
self.turn_on_leader_complain_timer()
def update_nid(self):
nid = self.__block_manager.blockchain.find_nid()
self.__inner_service.update_sub_services_properties(nid=int(nid, 16))
def _get_node_type_by_peer_list(self):
epoch = self.block_manager.epoch
if epoch:
reps = self.__block_manager.blockchain.find_preps_addresses_by_roothash(
epoch.reps_hash)
else:
reps = self.__block_manager.blockchain.find_preps_addresses_by_roothash(
ChannelProperty().crep_root_hash)
if ChannelProperty().peer_address in reps:
return conf.NodeType.CommunityNode
return conf.NodeType.CitizenNode
def _is_role_switched(self) -> bool:
new_node_type = self._get_node_type_by_peer_list()
if new_node_type == ChannelProperty().node_type:
utils.logger.debug(f"By peer manager, maintains the current node type({ChannelProperty().node_type})")
return False
return True
async def _select_node_type(self):
if self._is_role_switched():
new_node_type = self._get_node_type_by_peer_list()
utils.logger.info(f"Role switching to new node type: {new_node_type.name}")
ChannelProperty().node_type = new_node_type
self.__inner_service.update_sub_services_properties(node_type=ChannelProperty().node_type.value)
def switch_role(self):
self.__block_manager.blockchain.reset_leader_made_block_count(need_check_switched_role=True)
if self._is_role_switched():
self.__state_machine.switch_role()
async def reset_network(self):
utils.logger.info("Reset network")
self.__timer_service.clean(except_key=TimerService.TIMER_KEY_BROADCAST_SEND_UNCONFIRMED_BLOCK)
self.__rs_client = None
self.__state_machine.evaluate_network()
async def __init_peer_auth(self):
try:
node_key: bytes = await StubCollection().peer_stub.async_task().get_node_key()
self.__peer_auth = Signer.from_prikey(node_key)
ChannelProperty().peer_auth = self.__peer_auth
except Exception as e:
utils.exit_and_msg(f"peer auth init fail cause : {e}")
def __init_block_manager(self):
logging.debug(f"__init_block_manager() : channel({ChannelProperty().name})")
channel_name = ChannelProperty().name
develop = command_arguments.command_values.get(command_arguments.Type.Develop, False)
store_id = f"{ChannelProperty().peer_port}_{channel_name}" if develop else channel_name
try:
self.__block_manager = BlockManager(
channel_service=self,
peer_id=ChannelProperty().peer_id,
channel_name=channel_name,
store_id=store_id
)
except KeyValueStoreError as e:
utils.exit_and_msg("KeyValueStoreError(" + str(e) + ")")
def __init_broadcast_scheduler(self):
scheduler = BroadcastSchedulerFactory.new(channel=ChannelProperty().name,
self_target=ChannelProperty().peer_target)
scheduler.start()
self.__broadcast_scheduler = scheduler
def _get_radiostations(self):
radiostations: list = self.get_channel_option().get('radiostations')
if not radiostations:
logging.warning(f"no configurations for radiostations.")
return None
radiostations = utils.convert_local_ip_to_private_ip(radiostations)
try:
radiostations.remove(ChannelProperty().rest_target)
except ValueError:
pass
return radiostations
async def _init_rs_target(self, refresh_all: bool = False):
if refresh_all:
radiostations = self._get_radiostations()
if radiostations is None:
return
await self.__rs_client.init(radiostations)
else:
try:
self.__rs_client.init_next_target()
except StopIteration:
return await self._init_rs_target(refresh_all=True)
ChannelProperty().rs_target = self.__rs_client.target
self.__inner_service.update_sub_services_properties(relay_target=ChannelProperty().rs_target)
async def _init_rs_client(self):
self.__rs_client = RestClient(channel=ChannelProperty().name)
await self._init_rs_target(refresh_all=True)
async def __init_score_container(self):
"""create score container and save score_info and score_stub
"""
try:
self.__score_info = await self.__run_score_container()
except BaseException as e:
logging.error(e)
traceback.print_exc()
utils.exit_and_msg(f"run_score_container failed!!")
async def __init_sub_services(self):
self.__inner_service.init_sub_services()
await StubCollection().create_channel_tx_creator_stub(ChannelProperty().name)
await StubCollection().create_channel_tx_receiver_stub(ChannelProperty().name)
def __init_node_subscriber(self):
self.__node_subscriber = NodeSubscriber(
channel=ChannelProperty().name,
rs_target=ChannelProperty().rs_target
)
async def __run_score_container(self):
if conf.RUN_ICON_IN_LAUNCHER:
process_args = ['python3', '-m', 'loopchain', 'score',
'--channel', ChannelProperty().name]
process_args += command_arguments.get_raw_commands_by_filter(
command_arguments.Type.AMQPTarget,
command_arguments.Type.AMQPKey,
command_arguments.Type.Develop,
command_arguments.Type.ConfigurationFilePath,
command_arguments.Type.RadioStationTarget
)
self.__score_container = CommonSubprocess(process_args)
await StubCollection().create_icon_score_stub(ChannelProperty().name)
await StubCollection().icon_score_stubs[ChannelProperty().name].connect()
await StubCollection().icon_score_stubs[ChannelProperty().name].async_task().hello()
return None
def is_support_node_function(self, node_function):
return conf.NodeType.is_support_node_function(node_function, ChannelProperty().node_type)
def get_channel_option(self) -> dict:
return conf.CHANNEL_OPTION[ChannelProperty().name]
def generate_genesis_block(self):
if self.__block_manager.blockchain.block_height > -1:
logging.debug("genesis block was already generated")
return
reps = self.block_manager.blockchain.find_preps_addresses_by_roothash(ChannelProperty().crep_root_hash)
self.__block_manager.blockchain.generate_genesis_block(reps)
async def subscribe_to_parent(self):
def _handle_exception(future: asyncio.Future):
exc = future.exception()
logging.debug(f"error: {type(exc)}, {str(exc)}")
if ChannelProperty().node_type != conf.NodeType.CitizenNode:
logging.debug(f"This node is not Citizen anymore.")
return
if isinstance(exc, AnnounceNewBlockError):
self.__state_machine.block_sync()
return
if exc:
if (self.__state_machine.state != "SubscribeNetwork"
or isinstance(exc, UnregisteredException)):
self.__state_machine.subscribe_network()
else:
logging.warning(f"Waiting for next subscribe request...")
utils.logger.spam(f"try subscribe_call_by_citizen target({ChannelProperty().rest_target})")
subscribe_event = asyncio.Event()
# try websocket connection, and handle exception in callback
task = asyncio.ensure_future(
self.__node_subscriber.start(
block_height=self.__block_manager.blockchain.block_height,
event=subscribe_event
),
loop=self.__inner_service.loop
)
task.add_done_callback(_handle_exception)
await subscribe_event.wait()
def shutdown_peer(self, **kwargs):
logging.debug(f"shutdown_peer() kwargs = {kwargs}")
StubCollection().peer_stub.sync_task().stop(message=kwargs['message'])
def set_peer_type(self, peer_type):
"""Set peer type when peer init only
:param peer_type:
:return:
"""
self.__block_manager.set_peer_type(peer_type)
def save_peer_manager(self, peer_manager):
"""Save peer_list to leveldb
:param peer_manager:
"""
level_db_key_name = str.encode(conf.LEVEL_DB_KEY_FOR_PEER_LIST)
try:
dump = peer_manager.dump()
key_value_store = self.__block_manager.blockchain.blockchain_store
key_value_store.put(level_db_key_name, dump)
except AttributeError as e:
logging.warning("Fail Save Peer_list: " + str(e))
async def set_peer_type_in_channel(self):
peer_type = loopchain_pb2.PEER
leader_id = self.__block_manager.get_next_leader()
utils.logger.info(f"channel({ChannelProperty().name}) peer_leader: {leader_id}")
logger_preset = loggers.get_preset()
if ChannelProperty().peer_id == leader_id:
logger_preset.is_leader = True
utils.logger.info(f"Set Peer Type Leader! channel({ChannelProperty().name})")
peer_type = loopchain_pb2.BLOCK_GENERATOR
else:
logger_preset.is_leader = False
logger_preset.update_logger()
self.__block_manager.set_peer_type(peer_type)
def _is_genesis_node(self):
return ('genesis_data_path' in self.get_channel_option()
and self.is_support_node_function(conf.NodeFunction.Vote))
def __ready_to_height_sync(self):
if self.block_manager.blockchain.block_height >= 0:
self.block_manager.rebuild_block()
else:
if self._is_genesis_node():
self.generate_genesis_block()
if not self.is_support_node_function(conf.NodeFunction.Vote) and not ChannelProperty().rs_target:
utils.exit_and_msg(f"There's no radiostation target to sync block.")
def reset_leader(self, new_leader_id, block_height=0, complained=False):
"""
:param new_leader_id:
:param block_height:
:param complained:
:return:
"""
blockchain = self.__block_manager.blockchain
prep_targets = blockchain.find_preps_targets_by_roothash(self.__block_manager.epoch.reps_hash)
if ChannelProperty().peer_id not in prep_targets:
if self.is_support_node_function(conf.NodeFunction.Vote):
utils.logger.warning(f"This peer needs to switch to citizen.")
return
leader_peer_target = prep_targets.get(new_leader_id, None)
if block_height > 0 and block_height != self.block_manager.blockchain.last_block.header.height + 1:
utils.logger.warning(f"height behind peer can not take leader role. block_height({block_height}), "
f"last_block.header.height("
f"{self.block_manager.blockchain.last_block.header.height})")
return
if leader_peer_target is None:
logging.warning(f"in peer_service:reset_leader There is no peer by peer_id({new_leader_id})")
return
utils.logger.spam(f"reset_leader target({leader_peer_target}), complained={complained}")
if complained:
self.__block_manager.blockchain.reset_leader_made_block_count()
self.__block_manager.epoch.new_round(new_leader_id)
if ChannelProperty().peer_id == new_leader_id:
utils.logger.debug("Set Peer Type Leader!")
peer_type = loopchain_pb2.BLOCK_GENERATOR
self.state_machine.turn_to_leader()
else:
utils.logger.debug("Set Peer Type Peer!")
peer_type = loopchain_pb2.PEER
self.state_machine.turn_to_peer()
self.__block_manager.set_peer_type(peer_type)
self.turn_on_leader_complain_timer()
def score_write_precommit_state(self, block: Block):
logging.debug(f"call score commit {ChannelProperty().name} {block.header.height} {block.header.hash.hex()}")
new_block_hash = block.header.hash
try:
old_block_hash = self.__block_manager.get_old_block_hash(block.header.height, new_block_hash)
except KeyError:
old_block_hash = new_block_hash
logging.debug(f"Block Hash : {old_block_hash} -> {new_block_hash}")
request = {
"blockHeight": block.header.height,
"oldBlockHash": old_block_hash.hex(),
"newBlockHash": new_block_hash.hex()
}
request = convert_params(request, ParamType.write_precommit_state)
stub = StubCollection().icon_score_stubs[ChannelProperty().name]
precommit_result: dict = stub.sync_task().write_precommit_state(request)
if "error" in precommit_result:
raise WritePrecommitStateError(precommit_result['error'])
self.__block_manager.pop_old_block_hashes(block.header.height)
return True
def callback_leader_complain_timeout(self):
if self.state_machine.state == "BlockGenerate":
_, new_leader_id = self.block_manager.get_leader_ids_for_complaint()
if new_leader_id == ChannelProperty().peer_id:
utils.logger.debug(
f"Cannot convert the state to the `LeaderComplain` from the `BlockGenerate`"
f", because I'm the BlockGenerate and the next leader candidate.")
return
self.state_machine.leader_complain()
def turn_on_leader_complain_timer(self):
"""Turn on a leader complaint timer by the configuration name of `ALLOW_MAKE_EMPTY_BLOCK`.
"""
if conf.ALLOW_MAKE_EMPTY_BLOCK:
self.reset_leader_complain_timer()
else:
self.start_leader_complain_timer_if_tx_exists()
def reset_leader_complain_timer(self):
utils.logger.spam(f"reset_leader_complain_timer in channel service. ("
f"{self.__block_manager.epoch.round}/{self.__block_manager.epoch.complain_duration})")
if self.__timer_service.get_timer(TimerService.TIMER_KEY_LEADER_COMPLAIN):
utils.logger.spam(f"Try to stop leader complaint timer for reset.")
self.stop_leader_complain_timer()
self.start_leader_complain_timer()
def start_leader_complain_timer_if_tx_exists(self):
if not self.block_manager.get_tx_queue().is_empty_in_status(TransactionStatusInQueue.normal):
utils.logger.debug("Start leader complain timer because unconfirmed tx exists.")
self.start_leader_complain_timer()
def start_leader_complain_timer(self, duration=None):
if duration is None:
duration = self.__block_manager.epoch.complain_duration
utils.logger.spam(
f"start_leader_complain_timer in channel service. ({self.block_manager.epoch.round}/{duration})")
if self.state_machine.state in ("Vote", "LeaderComplain"):
self.__timer_service.add_timer_convenient(timer_key=TimerService.TIMER_KEY_LEADER_COMPLAIN,
duration=duration,
is_repeat=True, callback=self.callback_leader_complain_timeout)
def stop_leader_complain_timer(self):
utils.logger.spam(f"stop_leader_complain_timer in channel service.")
self.__timer_service.stop_timer(TimerService.TIMER_KEY_LEADER_COMPLAIN)
def start_subscribe_timer(self):
self.__timer_service.add_timer_convenient(timer_key=TimerService.TIMER_KEY_SUBSCRIBE,
duration=conf.SUBSCRIBE_RETRY_TIMER,
is_repeat=True, callback=self.subscribe_network)
def stop_subscribe_timer(self):
self.__timer_service.stop_timer(TimerService.TIMER_KEY_SUBSCRIBE)
def start_shutdown_timer_when_fail_subscribe(self):
error = f"Shutdown by Subscribe retry timeout({conf.SHUTDOWN_TIMER} sec)"
self.__timer_service.add_timer_convenient(timer_key=TimerService.TIMER_KEY_SHUTDOWN_WHEN_FAIL_SUBSCRIBE,
duration=conf.SHUTDOWN_TIMER, callback=self.shutdown_peer,
callback_kwargs={"message": error})
def stop_shutdown_timer_when_fail_subscribe(self):
self.__timer_service.stop_timer(TimerService.TIMER_KEY_SHUTDOWN_WHEN_FAIL_SUBSCRIBE)
def start_block_monitoring_timer(self):
self.__timer_service.add_timer_convenient(timer_key=TimerService.TIMER_KEY_BLOCK_MONITOR,
duration=conf.TIMEOUT_FOR_BLOCK_MONITOR,
callback=self.state_machine.subscribe_network)
def reset_block_monitoring_timer(self):
if self.__timer_service.get_timer(TimerService.TIMER_KEY_BLOCK_MONITOR):
self.__timer_service.reset_timer(TimerService.TIMER_KEY_BLOCK_MONITOR)
def stop_block_monitoring_timer(self):
self.__timer_service.stop_timer(TimerService.TIMER_KEY_BLOCK_MONITOR)
def stop_ws_heartbeat_timer(self):
self.__timer_service.stop_timer(TimerService.TIMER_KEY_WS_HEARTBEAT)
class ChannelService:
def __init__(self, channel_name, amqp_target, amqp_key):
self.__block_manager: BlockManager = None
self.__score_container: CommonSubprocess = None
self.__score_info: dict = None
self.__peer_auth: Signer = None
self.__peer_manager: PeerManager = None
self.__broadcast_scheduler: BroadcastScheduler = None
self.__radio_station_stub = None
self.__consensus = None
self.__timer_service = TimerService()
self.__node_subscriber: NodeSubscriber = None
self.__channel_infos: dict = None
loggers.get_preset().channel_name = channel_name
loggers.get_preset().update_logger()
channel_queue_name = conf.CHANNEL_QUEUE_NAME_FORMAT.format(
channel_name=channel_name, amqp_key=amqp_key)
self.__inner_service = ChannelInnerService(amqp_target,
channel_queue_name,
conf.AMQP_USERNAME,
conf.AMQP_PASSWORD,
channel_service=self)
logging.info(
f"ChannelService : {channel_name}, Queue : {channel_queue_name}")
ChannelProperty().name = channel_name
ChannelProperty().amqp_target = amqp_target
StubCollection().amqp_key = amqp_key
StubCollection().amqp_target = amqp_target
command_arguments.add_raw_command(command_arguments.Type.Channel,
channel_name)
command_arguments.add_raw_command(command_arguments.Type.AMQPTarget,
amqp_target)
command_arguments.add_raw_command(command_arguments.Type.AMQPKey,
amqp_key)
ObjectManager().channel_service = self
self.__state_machine = ChannelStateMachine(self)
@property
def block_manager(self):
return self.__block_manager
@property
def score_container(self):
return self.__score_container
@property
def score_info(self):
return self.__score_info
@property
def radio_station_stub(self):
return self.__radio_station_stub
@property
def peer_auth(self):
return self.__peer_auth
@property
def peer_manager(self):
return self.__peer_manager
@property
def broadcast_scheduler(self):
return self.__broadcast_scheduler
@property
def consensus(self):
return self.__consensus
@property
def timer_service(self):
return self.__timer_service
@property
def state_machine(self):
return self.__state_machine
@property
def inner_service(self):
return self.__inner_service
def serve(self):
async def _serve():
await StubCollection().create_peer_stub()
channel_name = ChannelProperty().name
self.__channel_infos = (await StubCollection(
).peer_stub.async_task().get_channel_infos())[channel_name]
results = await StubCollection().peer_stub.async_task(
).get_channel_info_detail(channel_name)
await self.init(*results)
self.__timer_service.start()
self.__state_machine.complete_init_components()
logging.info(
f'channel_service: init complete channel: {ChannelProperty().name}, '
f'state({self.__state_machine.state})')
loop = MessageQueueService.loop
# loop.set_debug(True)
loop.create_task(_serve())
loop.add_signal_handler(signal.SIGINT, self.close)
loop.add_signal_handler(signal.SIGTERM, self.close)
try:
loop.run_forever()
finally:
loop.run_until_complete(loop.shutdown_asyncgens())
loop.close()
self.cleanup()
def close(self):
if self.__inner_service:
self.__inner_service.cleanup()
logging.info("Cleanup ChannelInnerService.")
MessageQueueService.loop.stop()
def cleanup(self):
logging.info("Cleanup Channel Resources.")
if self.__block_manager:
self.__block_manager.stop()
self.__block_manager = None
logging.info("Cleanup BlockManager.")
if self.__score_container:
self.__score_container.stop()
self.__score_container.wait()
self.__score_container = None
logging.info("Cleanup ScoreContainer.")
if self.__broadcast_scheduler:
self.__broadcast_scheduler.stop()
self.__broadcast_scheduler.wait()
self.__broadcast_scheduler = None
logging.info("Cleanup BroadcastScheduler.")
if self.__consensus:
self.__consensus.stop()
self.__consensus.wait()
logging.info("Cleanup Consensus.")
if self.__timer_service.is_run():
self.__timer_service.stop()
self.__timer_service.wait()
logging.info("Cleanup TimerService.")
async def init(self, peer_port, peer_target, rest_target,
radio_station_target, peer_id, group_id, node_type,
score_package):
loggers.get_preset().peer_id = peer_id
loggers.get_preset().update_logger()
ChannelProperty().peer_port = peer_port
ChannelProperty().peer_target = peer_target
ChannelProperty().rest_target = rest_target
ChannelProperty().radio_station_target = radio_station_target
ChannelProperty().peer_id = peer_id
ChannelProperty().group_id = group_id
ChannelProperty().node_type = conf.NodeType(node_type)
ChannelProperty().score_package = score_package
self.__peer_manager = PeerManager(ChannelProperty().name)
await self.__init_peer_auth()
self.__init_broadcast_scheduler()
self.__init_block_manager()
await self.__init_score_container()
await self.__inner_service.connect(conf.AMQP_CONNECTION_ATTEMPS,
conf.AMQP_RETRY_DELAY,
exclusive=True)
await self.__init_sub_services()
self.block_manager.init_epoch()
async def __init_network(self):
self.__inner_service.update_sub_services_properties(
node_type=ChannelProperty().node_type.value)
self.__init_radio_station_stub()
if self.is_support_node_function(conf.NodeFunction.Vote):
if conf.ENABLE_REP_RADIO_STATION:
self.connect_to_radio_station()
else:
await self.__load_peers_from_file()
else:
self.__init_node_subscriber()
async def evaluate_network(self):
self.__ready_to_height_sync()
# Do not consider to change peer list by IISS this time.
await self.__select_node_type()
await self.__init_network()
self.__state_machine.block_sync()
async def subscribe_network(self):
if self.is_support_node_function(conf.NodeFunction.Vote):
await self.set_peer_type_in_channel()
else:
await self.subscribe_to_radio_station()
if conf.CONSENSUS_ALGORITHM == conf.ConsensusAlgorithm.lft:
if not self.__consensus.is_run():
self.__consensus.change_epoch(
precommit_block=self.__block_manager.get_blockchain(
).last_block)
self.__consensus.start()
elif conf.ALLOW_MAKE_EMPTY_BLOCK:
if not self.block_manager.block_generation_scheduler.is_run():
self.block_manager.block_generation_scheduler.start()
self.__state_machine.complete_subscribe()
self.start_leader_complain_timer_if_tx_exists()
def update_sub_services_properties(self):
nid = self.block_manager.get_blockchain().find_nid()
self.__inner_service.update_sub_services_properties(nid=int(nid, 16))
def __get_role_switch_block_height(self):
# Currently, only one way role switch is supported from Citizen to Rep
if ChannelProperty().node_type != conf.NodeType.CitizenNode:
return -1
return self.get_channel_option().get('role_switch_block_height', -1)
def __get_node_type_by_peer_list(self):
# FIXME: this is temporary codes. get peer list with IISS API
# IISS peer list include just peer_id and a URL that a server provide peer details
channels = util.load_json_data(conf.CHANNEL_MANAGE_DATA_PATH)
for peer_info in channels[ChannelProperty().name]["peers"]:
if peer_info['id'] == ChannelProperty().peer_id:
return conf.NodeType.CommunityNode
return conf.NodeType.CitizenNode
async def __clean_network(self):
if self.__node_subscriber is not None:
await self.__node_subscriber.close()
self.__node_subscriber: NodeSubscriber = None
self.__timer_service.clean()
peer_ids = set()
for peer_list in self.__peer_manager.peer_list.values():
for peer_id in peer_list.keys():
peer_ids.add(peer_id)
for peer_id in peer_ids:
self.__peer_manager.remove_peer(peer_id)
self.__radio_station_stub = None
async def __select_node_type(self):
# If block height is under zero this node has not synchronized blocks yet.
block_height = self.__block_manager.get_blockchain().block_height
if block_height < 0:
util.logger.debug(
f"Currently, Can't select node type without block height. block height={block_height}"
)
return
switch_block_height = self.__get_role_switch_block_height()
if switch_block_height < 0 or block_height < switch_block_height:
util.logger.debug(
f"Does not need to select node type. role switch block height={switch_block_height}"
)
return
node_type: conf.NodeType = self.__get_node_type_by_peer_list()
if node_type == ChannelProperty().node_type:
util.logger.info(
f"Node type equals previous note type ({node_type}). force={force}"
)
return
util.logger.info(f"Selected node type {node_type}")
ChannelProperty().node_type = node_type
await StubCollection().peer_stub.async_task().change_node_type(
node_type.value)
def reset_network_by_block_height(self, height):
if height == self.__get_role_switch_block_height():
self.__state_machine.switch_role()
async def reset_network(self):
util.logger.info("Reset network")
await self.__clean_network()
self.__state_machine.evaluate_network()
async def __init_peer_auth(self):
try:
node_key: bytes = await StubCollection().peer_stub.async_task(
).get_node_key(ChannelProperty().name)
self.__peer_auth = Signer.from_prikey(node_key)
except KeyError:
self.__peer_auth = Signer.from_channel(ChannelProperty().name)
except Exception as e:
logging.exception(f"peer auth init fail cause : {e}")
util.exit_and_msg(f"peer auth init fail cause : {e}")
def __init_block_manager(self):
logging.debug(
f"__load_block_manager_each channel({ChannelProperty().name})")
try:
self.__block_manager = BlockManager(
name="loopchain.peer.BlockManager",
channel_manager=self,
peer_id=ChannelProperty().peer_id,
channel_name=ChannelProperty().name,
level_db_identity=ChannelProperty().peer_target)
except leveldb.LevelDBError as e:
util.exit_and_msg("LevelDBError(" + str(e) + ")")
def __init_broadcast_scheduler(self):
scheduler = BroadcastSchedulerFactory.new(
channel=ChannelProperty().name,
self_target=ChannelProperty().peer_target)
scheduler.start()
self.__broadcast_scheduler = scheduler
scheduler.schedule_job(BroadcastCommand.SUBSCRIBE,
ChannelProperty().peer_target,
block=True,
block_timeout=conf.TIMEOUT_FOR_FUTURE)
def __init_radio_station_stub(self):
if self.is_support_node_function(conf.NodeFunction.Vote):
if conf.ENABLE_REP_RADIO_STATION:
self.__radio_station_stub = StubManager.get_stub_manager_to_server(
ChannelProperty().radio_station_target,
loopchain_pb2_grpc.RadioStationStub,
conf.CONNECTION_RETRY_TIMEOUT_TO_RS,
ssl_auth_type=conf.GRPC_SSL_TYPE)
else:
self.__radio_station_stub = RestStubManager(
ChannelProperty().radio_station_target,
ChannelProperty().name)
async def __init_score_container(self):
"""create score container and save score_info and score_stub
"""
for i in range(conf.SCORE_LOAD_RETRY_TIMES):
try:
self.__score_info = await self.__run_score_container()
except BaseException as e:
util.logger.spam(
f"channel_manager:load_score_container_each score_info load fail retry({i})"
)
logging.error(e)
traceback.print_exc()
time.sleep(conf.SCORE_LOAD_RETRY_INTERVAL
) # This blocking main thread is intended.
else:
break
async def __init_sub_services(self):
self.__inner_service.init_sub_services()
await StubCollection().create_channel_tx_creator_stub(
ChannelProperty().name)
await StubCollection().create_channel_tx_receiver_stub(
ChannelProperty().name)
def __init_node_subscriber(self):
self.__node_subscriber = NodeSubscriber(
channel=ChannelProperty().name,
rs_target=ChannelProperty().radio_station_target)
async def __run_score_container(self):
if conf.RUN_ICON_IN_LAUNCHER:
process_args = [
'python3', '-m', 'loopchain', 'score', '--channel',
ChannelProperty().name, '--score_package',
ChannelProperty().score_package
]
process_args += command_arguments.get_raw_commands_by_filter(
command_arguments.Type.AMQPTarget,
command_arguments.Type.AMQPKey, command_arguments.Type.Develop,
command_arguments.Type.ConfigurationFilePath,
command_arguments.Type.RadioStationTarget)
self.__score_container = CommonSubprocess(process_args)
await StubCollection().create_icon_score_stub(ChannelProperty().name)
await StubCollection().icon_score_stubs[ChannelProperty().name
].connect()
await StubCollection().icon_score_stubs[ChannelProperty().name
].async_task().hello()
return None
async def __load_score(self):
channel_name = ChannelProperty().name
score_package_name = ChannelProperty().score_package
util.logger.spam(f"peer_service:__load_score --init--")
logging.info("LOAD SCORE AND CONNECT TO SCORE SERVICE!")
params = dict()
params[message_code.MetaParams.ScoreLoad.
repository_path] = conf.DEFAULT_SCORE_REPOSITORY_PATH
params[message_code.MetaParams.ScoreLoad.
score_package] = score_package_name
params[
message_code.MetaParams.ScoreLoad.base] = conf.DEFAULT_SCORE_BASE
params[message_code.MetaParams.ScoreLoad.peer_id] = ChannelProperty(
).peer_id
meta = json.dumps(params)
logging.debug(f"load score params : {meta}")
util.logger.spam(f"peer_service:__load_score --1--")
score_stub = StubCollection().score_stubs[channel_name]
response = await score_stub.async_task().score_load(meta)
logging.debug("try score load on score service: " + str(response))
if not response:
return None
if response.code != message_code.Response.success:
util.exit_and_msg("Fail Get Score from Score Server...")
return None
logging.debug("Get Score from Score Server...")
score_info = json.loads(response.meta)
logging.info("LOAD SCORE DONE!")
util.logger.spam(f"peer_service:__load_score --end--")
return score_info
async def __load_peers_from_file(self):
channel_info = await StubCollection().peer_stub.async_task(
).get_channel_infos()
for peer_info in channel_info[ChannelProperty().name]["peers"]:
self.__peer_manager.add_peer(peer_info)
self.__broadcast_scheduler.schedule_job(BroadcastCommand.SUBSCRIBE,
peer_info["peer_target"])
self.show_peers()
def is_support_node_function(self, node_function):
return conf.NodeType.is_support_node_function(
node_function,
ChannelProperty().node_type)
def get_channel_option(self) -> dict:
channel_option = conf.CHANNEL_OPTION
return channel_option[ChannelProperty().name]
def get_channel_infos(self) -> dict:
return self.__channel_infos
def get_rep_ids(self) -> list:
return [
ExternalAddress.fromhex_address(peer.get('id'),
allow_malformed=True)
for peer in self.get_channel_infos()['peers']
]
def generate_genesis_block(self):
blockchain = self.block_manager.get_blockchain()
if blockchain.block_height > -1:
logging.debug("genesis block was already generated")
return
reps = self.get_rep_ids()
blockchain.generate_genesis_block(reps)
def connect_to_radio_station(self, is_reconnect=False):
response = self.__radio_station_stub.call_in_times(
method_name="ConnectPeer",
message=loopchain_pb2.ConnectPeerRequest(
channel=ChannelProperty().name,
peer_object=b'',
peer_id=ChannelProperty().peer_id,
peer_target=ChannelProperty().peer_target,
group_id=ChannelProperty().group_id),
retry_times=conf.CONNECTION_RETRY_TIMES_TO_RS,
is_stub_reuse=True,
timeout=conf.CONNECTION_TIMEOUT_TO_RS)
# start next ConnectPeer timer
self.__timer_service.add_timer_convenient(
timer_key=TimerService.TIMER_KEY_CONNECT_PEER,
duration=conf.CONNECTION_RETRY_TIMER,
callback=self.connect_to_radio_station,
callback_kwargs={"is_reconnect": True})
if is_reconnect:
return
if response and response.status == message_code.Response.success:
try:
peer_list_data = PeerListData.load(response.peer_list)
except Exception as e:
traceback.print_exc()
logging.error(
f"Invalid peer list. Check your Radio Station. exception={e}"
)
return
self.__peer_manager.set_peer_list(peer_list_data)
peers, peer_list = self.__peer_manager.get_peers_for_debug()
logging.debug("peer list update: " + peers)
# add connected peer to processes audience
for each_peer in peer_list:
util.logger.spam(
f"peer_service:connect_to_radio_station peer({each_peer.target}-{each_peer.status})"
)
if each_peer.status == PeerStatus.connected:
self.__broadcast_scheduler.schedule_job(
BroadcastCommand.SUBSCRIBE, each_peer.target)
def __subscribe_to_peer_list(self):
peer_object = self.peer_manager.get_peer(ChannelProperty().peer_id)
peer_request = loopchain_pb2.PeerRequest(
channel=ChannelProperty().name,
peer_target=ChannelProperty().peer_target,
peer_id=ChannelProperty().peer_id,
group_id=ChannelProperty().group_id,
node_type=ChannelProperty().node_type,
peer_order=peer_object.order)
self.__broadcast_scheduler.schedule_broadcast("Subscribe",
peer_request)
async def subscribe_to_radio_station(self):
await self.__subscribe_call_to_stub(self.__radio_station_stub,
loopchain_pb2.PEER)
async def subscribe_to_peer(self, peer_id, peer_type):
peer = self.peer_manager.get_peer(peer_id)
peer_stub = self.peer_manager.get_peer_stub_manager(peer)
await self.__subscribe_call_to_stub(peer_stub, peer_type)
self.__broadcast_scheduler.schedule_job(BroadcastCommand.SUBSCRIBE,
peer_stub.target)
async def __subscribe_call_to_stub(self, peer_stub, peer_type):
if self.is_support_node_function(conf.NodeFunction.Vote):
await peer_stub.call_async(
"Subscribe",
loopchain_pb2.PeerRequest(
channel=ChannelProperty().name,
peer_target=ChannelProperty().peer_target,
peer_type=peer_type,
peer_id=ChannelProperty().peer_id,
group_id=ChannelProperty().group_id,
node_type=ChannelProperty().node_type),
)
else:
await self.__subscribe_call_from_citizen()
async def __subscribe_call_from_citizen(self):
def _handle_exception(future: asyncio.Future):
logging.debug(
f"error: {type(future.exception())}, {str(future.exception())}"
)
if ChannelProperty().node_type != conf.NodeType.CitizenNode:
logging.debug(f"This node is not Citizen anymore.")
return
elif isinstance(future.exception(), ConnectionError):
logging.warning(f"Waiting for next subscribe request...")
if self.__state_machine.state != "SubscribeNetwork":
self.__state_machine.subscribe_network()
subscribe_event = asyncio.Event()
util.logger.spam(
f"try subscribe_call_by_citizen target({ChannelProperty().rest_target})"
)
# try websocket connection, and handle exception in callback
asyncio.ensure_future(
self.__node_subscriber.subscribe(
block_height=self.block_manager.get_blockchain().block_height,
event=subscribe_event)).add_done_callback(_handle_exception)
await subscribe_event.wait()
def shutdown_peer(self, **kwargs):
logging.debug(f"channel_service:shutdown_peer")
StubCollection().peer_stub.sync_task().stop(message=kwargs['message'])
def set_peer_type(self, peer_type):
"""Set peer type when peer init only
:param peer_type:
:return:
"""
self.__block_manager.set_peer_type(peer_type)
def save_peer_manager(self, peer_manager):
"""peer_list 를 leveldb 에 저장한다.
:param peer_manager:
"""
level_db_key_name = str.encode(conf.LEVEL_DB_KEY_FOR_PEER_LIST)
try:
dump = peer_manager.dump()
level_db = self.__block_manager.get_level_db()
level_db.Put(level_db_key_name, dump)
except AttributeError as e:
logging.warning("Fail Save Peer_list: " + str(e))
async def set_peer_type_in_channel(self):
peer_type = loopchain_pb2.PEER
blockchain = self.block_manager.get_blockchain()
last_block = blockchain.last_unconfirmed_block or blockchain.last_block
leader_id = None
if last_block and last_block.header.next_leader is not None:
leader_id = last_block.header.next_leader.hex_hx()
peer = self.peer_manager.get_peer(leader_id)
if peer is None:
leader_id = None
else:
self.peer_manager.set_leader_peer(peer)
if leader_id is None:
leader_id = self.peer_manager.get_leader_peer().peer_id
logging.debug(
f"channel({ChannelProperty().name}) peer_leader: {leader_id}")
logger_preset = loggers.get_preset()
if ChannelProperty().peer_id == leader_id:
logger_preset.is_leader = True
logging.debug(
f"Set Peer Type Leader! channel({ChannelProperty().name})")
peer_type = loopchain_pb2.BLOCK_GENERATOR
else:
logger_preset.is_leader = False
logger_preset.update_logger()
if conf.CONSENSUS_ALGORITHM == conf.ConsensusAlgorithm.lft:
self.consensus.leader_id = leader_id
self.block_manager.set_peer_type(peer_type)
def __ready_to_height_sync(self):
blockchain = self.block_manager.get_blockchain()
blockchain.init_blockchain()
if blockchain.block_height == -1 and 'genesis_data_path' in conf.CHANNEL_OPTION[
ChannelProperty().name]:
self.generate_genesis_block()
elif blockchain.block_height > -1:
self.block_manager.rebuild_block()
def show_peers(self):
logging.debug(f"peer_service:show_peers ({ChannelProperty().name}): ")
for peer in self.peer_manager.get_IP_of_peers_in_group():
logging.debug("peer_target: " + peer)
def reset_leader(self, new_leader_id, block_height=0, complained=False):
"""
:param new_leader_id:
:param block_height:
:param complained:
:return:
"""
if self.peer_manager.get_leader_id(conf.ALL_GROUP_ID) == new_leader_id:
return
utils.logger.info(
f"RESET LEADER channel({ChannelProperty().name}) leader_id({new_leader_id}), "
f"complained={complained}")
leader_peer = self.peer_manager.get_peer(new_leader_id, None)
if block_height > 0 and block_height != self.block_manager.get_blockchain(
).last_block.header.height + 1:
util.logger.warning(
f"height behind peer can not take leader role. block_height({block_height}), "
f"last_block.header.height("
f"{self.block_manager.get_blockchain().last_block.header.height})"
)
return
if leader_peer is None:
logging.warning(
f"in peer_service:reset_leader There is no peer by peer_id({new_leader_id})"
)
return
util.logger.spam(
f"peer_service:reset_leader target({leader_peer.target}), complained={complained}"
)
self_peer_object = self.peer_manager.get_peer(
ChannelProperty().peer_id)
self.peer_manager.set_leader_peer(leader_peer, None)
if complained:
self.block_manager.epoch.new_round(leader_peer.peer_id)
else:
self.block_manager.epoch = Epoch.new_epoch(leader_peer.peer_id)
logging.info(
f"Epoch height({self.block_manager.epoch.height}), leader ({self.block_manager.epoch.leader_id})"
)
if self_peer_object.peer_id == leader_peer.peer_id:
logging.debug("Set Peer Type Leader!")
peer_type = loopchain_pb2.BLOCK_GENERATOR
self.state_machine.turn_to_leader()
else:
logging.debug("Set Peer Type Peer!")
peer_type = loopchain_pb2.PEER
self.state_machine.turn_to_peer()
self.block_manager.set_peer_type(peer_type)
def set_new_leader(self, new_leader_id, block_height=0):
logging.info(
f"SET NEW LEADER channel({ChannelProperty().name}) leader_id({new_leader_id})"
)
# complained_leader = self.peer_manager.get_leader_peer()
leader_peer = self.peer_manager.get_peer(new_leader_id, None)
if block_height > 0 and block_height != self.block_manager.get_blockchain(
).last_block.height + 1:
logging.warning(f"height behind peer can not take leader role.")
return
if leader_peer is None:
logging.warning(
f"in channel_service:set_new_leader::There is no peer by peer_id({new_leader_id})"
)
return
util.logger.spam(
f"channel_service:set_new_leader::leader_target({leader_peer.target})"
)
self_peer_object = self.peer_manager.get_peer(
ChannelProperty().peer_id)
self.peer_manager.set_leader_peer(leader_peer, None)
peer_leader = self.peer_manager.get_leader_peer()
if self_peer_object.target == peer_leader.target:
loggers.get_preset().is_leader = True
loggers.get_preset().update_logger()
logging.debug("I'm Leader Peer!")
else:
loggers.get_preset().is_leader = False
loggers.get_preset().update_logger()
logging.debug("I'm general Peer!")
# 새 leader 에게 subscribe 하기
# await self.subscribe_to_radio_station()
# await self.subscribe_to_peer(peer_leader.peer_id, loopchain_pb2.BLOCK_GENERATOR)
def genesis_invoke(self, block: Block) -> ('Block', dict):
method = "icx_sendTransaction"
transactions = []
for tx in block.body.transactions.values():
tx_serializer = TransactionSerializer.new(
tx.version,
self.block_manager.get_blockchain().tx_versioner)
transaction = {
"method": method,
"params": {
"txHash": tx.hash.hex()
},
"genesisData": tx_serializer.to_full_data(tx)
}
transactions.append(transaction)
request = {
'block': {
'blockHeight': block.header.height,
'blockHash': block.header.hash.hex(),
'timestamp': block.header.timestamp
},
'transactions': transactions
}
request = convert_params(request, ParamType.invoke)
stub = StubCollection().icon_score_stubs[ChannelProperty().name]
response = stub.sync_task().invoke(request)
response_to_json_query(response)
tx_receipts = response["txResults"]
block_builder = BlockBuilder.from_new(
block,
self.block_manager.get_blockchain().tx_versioner)
block_builder.reset_cache()
block_builder.peer_id = block.header.peer_id
block_builder.signature = block.header.signature
block_builder.commit_state = {
ChannelProperty().name: response['stateRootHash']
}
block_builder.state_hash = Hash32(
bytes.fromhex(response['stateRootHash']))
block_builder.receipts = tx_receipts
block_builder.reps = self.get_rep_ids()
new_block = block_builder.build()
return new_block, tx_receipts
def score_invoke(self, _block: Block) -> dict or None:
method = "icx_sendTransaction"
transactions = []
for tx in _block.body.transactions.values():
tx_serializer = TransactionSerializer.new(
tx.version,
self.block_manager.get_blockchain().tx_versioner)
transaction = {
"method": method,
"params": tx_serializer.to_full_data(tx)
}
transactions.append(transaction)
request = {
'block': {
'blockHeight':
_block.header.height,
'blockHash':
_block.header.hash.hex(),
'prevBlockHash':
_block.header.prev_hash.hex()
if _block.header.prev_hash else '',
'timestamp':
_block.header.timestamp
},
'transactions': transactions
}
request = convert_params(request, ParamType.invoke)
stub = StubCollection().icon_score_stubs[ChannelProperty().name]
response = stub.sync_task().invoke(request)
response_to_json_query(response)
tx_receipts = response["txResults"]
block_builder = BlockBuilder.from_new(
_block,
self.__block_manager.get_blockchain().tx_versioner)
block_builder.reset_cache()
block_builder.peer_id = _block.header.peer_id
block_builder.signature = _block.header.signature
block_builder.commit_state = {
ChannelProperty().name: response['stateRootHash']
}
block_builder.state_hash = Hash32(
bytes.fromhex(response['stateRootHash']))
block_builder.receipts = tx_receipts
block_builder.reps = self.get_rep_ids()
new_block = block_builder.build()
return new_block, tx_receipts
def score_change_block_hash(self, block_height, old_block_hash,
new_block_hash):
change_hash_info = json.dumps({
"block_height": block_height,
"old_block_hash": old_block_hash,
"new_block_hash": new_block_hash
})
stub = StubCollection().score_stubs[ChannelProperty().name]
stub.sync_task().change_block_hash(change_hash_info)
def score_write_precommit_state(self, block: Block):
logging.debug(
f"call score commit {ChannelProperty().name} {block.header.height} {block.header.hash.hex()}"
)
request = {
"blockHeight": block.header.height,
"blockHash": block.header.hash.hex(),
}
request = convert_params(request, ParamType.write_precommit_state)
stub = StubCollection().icon_score_stubs[ChannelProperty().name]
stub.sync_task().write_precommit_state(request)
return True
def score_remove_precommit_state(self, block: Block):
invoke_fail_info = json.dumps({
"block_height": block.height,
"block_hash": block.block_hash
})
stub = StubCollection().score_stubs[ChannelProperty().name]
stub.sync_task().remove_precommit_state(invoke_fail_info)
return True
def reset_leader_complain_timer(self):
if self.__timer_service.get_timer(
TimerService.TIMER_KEY_LEADER_COMPLAIN):
self.stop_leader_complain_timer()
self.start_leader_complain_timer()
def start_leader_complain_timer_if_tx_exists(self):
if not self.block_manager.get_tx_queue().is_empty_in_status(
TransactionStatusInQueue.normal):
util.logger.debug(
"Start leader complain timer because unconfirmed tx exists.")
self.start_leader_complain_timer()
def start_leader_complain_timer(self, duration=None):
if not duration:
duration = self.block_manager.epoch.complain_duration
# util.logger.spam(
# f"start_leader_complain_timer in channel service. ({self.block_manager.epoch.round}/{duration})")
if self.state_machine.state not in ("BlockGenerate", "BlockSync",
"Watch"):
self.__timer_service.add_timer_convenient(
timer_key=TimerService.TIMER_KEY_LEADER_COMPLAIN,
duration=duration,
is_repeat=True,
callback=self.state_machine.leader_complain)
def stop_leader_complain_timer(self):
util.logger.spam(f"stop_leader_complain_timer in channel service.")
self.__timer_service.stop_timer(TimerService.TIMER_KEY_LEADER_COMPLAIN)
def start_subscribe_timer(self):
self.__timer_service.add_timer_convenient(
timer_key=TimerService.TIMER_KEY_SUBSCRIBE,
duration=conf.SUBSCRIBE_RETRY_TIMER,
is_repeat=True,
callback=self.subscribe_network)
def stop_subscribe_timer(self):
self.__timer_service.stop_timer(TimerService.TIMER_KEY_SUBSCRIBE)
def start_shutdown_timer(self):
error = f"Shutdown by Subscribe retry timeout({conf.SHUTDOWN_TIMER} sec)"
self.__timer_service.add_timer_convenient(
timer_key=TimerService.TIMER_KEY_SHUTDOWN_WHEN_FAIL_SUBSCRIBE,
duration=conf.SHUTDOWN_TIMER,
callback=self.shutdown_peer,
callback_kwargs={"message": error})
def stop_shutdown_timer(self):
self.__timer_service.stop_timer(
TimerService.TIMER_KEY_SHUTDOWN_WHEN_FAIL_SUBSCRIBE)