Skip to content

Commit

Permalink
chore(mempool_node): change consensus_manager server type
Browse files Browse the repository at this point in the history
commit-id:7923c2da
  • Loading branch information
Itay-Tsabary-Starkware committed Sep 26, 2024
1 parent 862a5c4 commit d8bc065
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 11 deletions.
14 changes: 5 additions & 9 deletions crates/consensus_manager/src/communication.rs
Original file line number Diff line number Diff line change
@@ -1,23 +1,19 @@
use async_trait::async_trait;
use starknet_consensus_manager_types::communication::{
ConsensusManagerRequest,
ConsensusManagerRequestAndResponseSender,
ConsensusManagerResponse,
};
use starknet_mempool_infra::component_definitions::ComponentRequestHandler;
use starknet_mempool_infra::component_server::LocalActiveComponentServer;
use tokio::sync::mpsc::Receiver;
use starknet_mempool_infra::component_server::EmptyServer;

use crate::consensus_manager::ConsensusManager;

pub type LocalConsensusManagerServer =
LocalActiveComponentServer<ConsensusManager, ConsensusManagerRequest, ConsensusManagerResponse>;
pub type ConsensusManagerServer = EmptyServer<ConsensusManager>;

pub fn create_local_consensus_manager_server(
pub fn create_consensus_manager_server(
consensus_manager: ConsensusManager,
rx_consensus_manager: Receiver<ConsensusManagerRequestAndResponseSender>,
) -> LocalConsensusManagerServer {
LocalActiveComponentServer::new(consensus_manager, rx_consensus_manager)
) -> ConsensusManagerServer {
EmptyServer::new(consensus_manager)
}

#[async_trait]
Expand Down
4 changes: 2 additions & 2 deletions crates/mempool_node/src/servers.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ use std::pin::Pin;
use futures::{Future, FutureExt};
use starknet_batcher::communication::{create_local_batcher_server, LocalBatcherServer};
use starknet_consensus_manager::communication::{
create_local_consensus_manager_server,
create_consensus_manager_server,
LocalConsensusManagerServer,
};
use starknet_gateway::communication::{create_gateway_server, GatewayServer};
Expand Down Expand Up @@ -40,7 +40,7 @@ pub fn create_servers(
None
};
let consensus_manager_server = if config.components.consensus_manager.execute {
Some(Box::new(create_local_consensus_manager_server(
Some(Box::new(create_consensus_manager_server(
components.consensus_manager.expect("Consensus Manager is not initialized."),
communication.take_consensus_manager_rx(),
)))
Expand Down

0 comments on commit d8bc065

Please sign in to comment.