def setUp():
gossip = MockGossip()
clientHanlerTestCase.initialize(
handlers.PeersGetRequest(gossip),
client_peers_pb2.ClientPeersGetRequest,
client_peers_pb2.ClientPeersGetResponse,
)
def setUp(self):
gossip = MockGossip()
self.initialize(
handlers.PeersGetRequest(gossip),
client_peers_pb2.ClientPeersGetRequest,
client_peers_pb2.ClientPeersGetResponse,
)
def add(dispatcher,
gossip,
context_manager,
executor,
completer,
block_store,
batch_tracker,
merkle_db,
get_current_root,
receipt_store,
event_broadcaster,
permission_verifier,
thread_pool,
sig_pool,
block_publisher,
metrics_registry=None):
# -- Transaction Processor -- #
dispatcher.add_handler(
validator_pb2.Message.TP_RECEIPT_ADD_DATA_REQUEST,
tp_state_handlers.TpReceiptAddDataHandler(context_manager),
thread_pool)
dispatcher.add_handler(
validator_pb2.Message.TP_EVENT_ADD_REQUEST,
tp_state_handlers.TpEventAddHandler(context_manager), thread_pool)
dispatcher.add_handler(
validator_pb2.Message.TP_STATE_DELETE_REQUEST,
tp_state_handlers.TpStateDeleteHandler(context_manager), thread_pool)
dispatcher.add_handler(
validator_pb2.Message.TP_STATE_GET_REQUEST,
tp_state_handlers.TpStateGetHandler(context_manager), thread_pool)
dispatcher.add_handler(
validator_pb2.Message.TP_STATE_SET_REQUEST,
tp_state_handlers.TpStateSetHandler(context_manager), thread_pool)
dispatcher.add_handler(
validator_pb2.Message.TP_REGISTER_REQUEST,
processor_handlers.ProcessorRegisterHandler(executor.processors),
thread_pool)
dispatcher.add_handler(
validator_pb2.Message.TP_UNREGISTER_REQUEST,
processor_handlers.ProcessorUnRegisterHandler(executor.processors),
thread_pool)
# -- Client -- #
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
BatchListPermissionVerifier(permission_verifier=permission_verifier),
sig_pool)
# Submit
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
ClientBatchSubmitBackpressureHandler(
block_publisher.can_accept_batch,
block_publisher.get_current_queue_info,
metrics_registry=metrics_registry), thread_pool)
dispatcher.add_handler(validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
signature_verifier.BatchListSignatureVerifier(),
sig_pool)
dispatcher.add_handler(validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
structure_verifier.BatchListStructureVerifier(),
thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
CompleterBatchListBroadcastHandler(completer, gossip), thread_pool)
dispatcher.add_handler(validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
client_handlers.BatchSubmitFinisher(batch_tracker),
thread_pool)
dispatcher.add_handler(validator_pb2.Message.CLIENT_BATCH_STATUS_REQUEST,
client_handlers.BatchStatusRequest(batch_tracker),
thread_pool)
# State
dispatcher.add_handler(
validator_pb2.Message.CLIENT_STATE_LIST_REQUEST,
client_handlers.StateListRequest(merkle_db, block_store), thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_STATE_GET_REQUEST,
client_handlers.StateGetRequest(merkle_db, block_store), thread_pool)
# Blocks
dispatcher.add_handler(validator_pb2.Message.CLIENT_BLOCK_LIST_REQUEST,
client_handlers.BlockListRequest(block_store),
thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BLOCK_GET_BY_ID_REQUEST,
client_handlers.BlockGetByIdRequest(block_store), thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BLOCK_GET_BY_NUM_REQUEST,
client_handlers.BlockGetByNumRequest(block_store), thread_pool)
# Batches
dispatcher.add_handler(validator_pb2.Message.CLIENT_BATCH_LIST_REQUEST,
client_handlers.BatchListRequest(block_store),
thread_pool)
dispatcher.add_handler(validator_pb2.Message.CLIENT_BATCH_GET_REQUEST,
client_handlers.BatchGetRequest(block_store),
thread_pool)
# Transactions
dispatcher.add_handler(
validator_pb2.Message.CLIENT_TRANSACTION_LIST_REQUEST,
client_handlers.TransactionListRequest(block_store), thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_TRANSACTION_GET_REQUEST,
client_handlers.TransactionGetRequest(block_store), thread_pool)
# Receipts
dispatcher.add_handler(validator_pb2.Message.CLIENT_RECEIPT_GET_REQUEST,
ClientReceiptGetRequestHandler(receipt_store),
thread_pool)
# Events
dispatcher.add_handler(
validator_pb2.Message.CLIENT_EVENTS_SUBSCRIBE_REQUEST,
ClientEventsSubscribeValidationHandler(event_broadcaster), thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_EVENTS_SUBSCRIBE_REQUEST,
ClientEventsSubscribeHandler(event_broadcaster), thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_EVENTS_UNSUBSCRIBE_REQUEST,
ClientEventsUnsubscribeHandler(event_broadcaster), thread_pool)
dispatcher.add_handler(validator_pb2.Message.CLIENT_EVENTS_GET_REQUEST,
ClientEventsGetRequestHandler(event_broadcaster),
thread_pool)
dispatcher.add_handler(validator_pb2.Message.CLIENT_PEERS_GET_REQUEST,
client_handlers.PeersGetRequest(gossip),
thread_pool)
def add(
dispatcher,
gossip,
completer,
block_store,
batch_tracker,
merkle_db,
get_current_root,
receipt_store,
event_broadcaster,
permission_verifier,
thread_pool,
client_thread_pool,
sig_pool,
journal,
public_key,
):
# -- Client -- #
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
BatchListPermissionVerifier(
permission_verifier=permission_verifier
),
sig_pool)
# Submit
dispatcher.set_preprocessor(
validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
client_handlers.client_batch_submit_request_preprocessor,
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
ClientBatchSubmitBackpressureHandler(
public_key,
journal.is_batch_pool_full),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
signature_verifier.BatchListSignatureVerifier(),
sig_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
structure_verifier.BatchListStructureVerifier(),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
CompleterBatchListBroadcastHandler(
completer, gossip),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_SUBMIT_REQUEST,
client_handlers.BatchSubmitFinisher(batch_tracker),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_STATUS_REQUEST,
client_handlers.BatchStatusRequest(batch_tracker),
client_thread_pool)
# State
dispatcher.add_handler(
validator_pb2.Message.CLIENT_STATE_LIST_REQUEST,
client_handlers.StateListRequest(
merkle_db,
block_store),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_STATE_GET_REQUEST,
client_handlers.StateGetRequest(
merkle_db,
block_store),
client_thread_pool)
# Blocks
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BLOCK_LIST_REQUEST,
client_handlers.BlockListRequest(block_store),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BLOCK_GET_BY_ID_REQUEST,
client_handlers.BlockGetByIdRequest(block_store),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BLOCK_GET_BY_NUM_REQUEST,
client_handlers.BlockGetByNumRequest(block_store),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BLOCK_GET_BY_BATCH_ID_REQUEST,
client_handlers.BlockGetByBatchRequest(block_store),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BLOCK_GET_BY_TRANSACTION_ID_REQUEST,
client_handlers.BlockGetByTransactionRequest(block_store),
client_thread_pool)
# Batches
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_LIST_REQUEST,
client_handlers.BatchListRequest(block_store),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_BATCH_GET_REQUEST,
client_handlers.BatchGetRequest(block_store),
client_thread_pool)
# Transactions
dispatcher.add_handler(
validator_pb2.Message.CLIENT_TRANSACTION_LIST_REQUEST,
client_handlers.TransactionListRequest(
block_store),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_TRANSACTION_GET_REQUEST,
client_handlers.TransactionGetRequest(
block_store),
client_thread_pool)
# Receipts
dispatcher.add_handler(
validator_pb2.Message.CLIENT_RECEIPT_GET_REQUEST,
ClientReceiptGetRequestHandler(receipt_store),
client_thread_pool)
# Events
dispatcher.add_handler(
validator_pb2.Message.CLIENT_EVENTS_SUBSCRIBE_REQUEST,
ClientEventsSubscribeValidationHandler(event_broadcaster),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_EVENTS_SUBSCRIBE_REQUEST,
ClientEventsSubscribeHandler(event_broadcaster),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_EVENTS_UNSUBSCRIBE_REQUEST,
ClientEventsUnsubscribeHandler(event_broadcaster),
client_thread_pool)
dispatcher.add_handler(
validator_pb2.Message.CLIENT_EVENTS_GET_REQUEST,
ClientEventsGetRequestHandler(event_broadcaster),
client_thread_pool)
# Peers
dispatcher.add_handler(
validator_pb2.Message.CLIENT_PEERS_GET_REQUEST,
client_handlers.PeersGetRequest(gossip),
thread_pool)
# Status
dispatcher.add_handler(
validator_pb2.Message.CLIENT_STATUS_GET_REQUEST,
client_handlers.StatusGetRequest(gossip),
thread_pool)
# Ping
dispatcher.add_handler(
validator_pb2.Message.PING_RESPONSE,
PingResponseHandler(),
thread_pool)
