feat: add tracing

Cargo.toml

@@ -75,5 +75,7 @@ tempfile = "3.3.0"
 thiserror = "1.0"
 tokio = { version = "1.37.0", features = ["full"] }
 tokio-test = "0.4.4"
+tracing = "0.1.37"
+tracing-subscriber = "0.3.16"
 url = "2.5.0"
 validator = "0.12"

crates/gateway/Cargo.toml

@@ -32,6 +32,7 @@ starknet-types-core.workspace = true
 test_utils = { path = "../test_utils", version = "0.0"}
 thiserror.workspace = true
 tokio.workspace = true
+tracing.workspace = true
 validator.workspace = true
 
 [dev-dependencies]

crates/gateway/src/gateway.rs

@@ -11,6 +11,7 @@ use starknet_api::transaction::TransactionHash;
 use starknet_mempool_infra::component_runner::{ComponentRunner, ComponentStartError};
 use starknet_mempool_types::communication::SharedMempoolClient;
 use starknet_mempool_types::mempool_types::{Account, MempoolInput};
+use tracing::info;
 
 use crate::compilation::compile_contract_class;
 use crate::config::{GatewayConfig, GatewayNetworkConfig, RpcStateReaderConfig};
@@ -146,7 +147,7 @@ pub fn create_gateway(
 #[async_trait]
 impl ComponentRunner for Gateway {
     async fn start(&mut self) -> Result<(), ComponentStartError> {
-        println!("Gateway::start()");
+        info!("Gateway::start()");
         self.run().await.map_err(|_| ComponentStartError::InternalComponentError)
     }
 }

crates/mempool_infra/Cargo.toml

@@ -19,6 +19,7 @@ hyper.workspace = true
 papyrus_config.workspace = true
 thiserror.workspace = true
 tokio.workspace = true
+tracing.workspace = true
 
 [dev-dependencies]
 assert_matches.workspace = true

crates/mempool_infra/src/component_server.rs

@@ -11,6 +11,7 @@ use hyper::{Body, Request as HyperRequest, Response as HyperResponse, Server, St
 use serde::{Deserialize, Serialize};
 use tokio::sync::mpsc::Receiver;
 use tokio::sync::Mutex;
+use tracing::{error, info};
 
 use crate::component_definitions::{
     ComponentRequestAndResponseSender, ComponentRequestHandler, ServerError,
@@ -152,8 +153,8 @@ impl<T: ComponentRunner + Send + Sync> EmptyServer<T> {
 impl<T: ComponentRunner + Send + Sync> ComponentServerStarter for EmptyServer<T> {
     async fn start(&mut self) {
         match self.component.start().await {
-            Ok(_) => println!("ComponentServer::start() completed."),
-            Err(err) => println!("ComponentServer::start() failed: {:?}", err),
+            Ok(_) => info!("ComponentServer::start() completed."),
+            Err(err) => error!("ComponentServer::start() failed: {:?}", err),
         }
     }
 }

crates/mempool_node/Cargo.toml

@@ -20,6 +20,8 @@ starknet_mempool_types = { path = "../mempool_types", version = "0.0" }
 serde.workspace = true
 papyrus_config.workspace = true
 tokio.workspace = true
+tracing.workspace = true
+tracing-subscriber = { workspace = true, features = ["env-filter"] }
 validator.workspace = true
 
 [dev-dependencies]

crates/mempool_node/src/main.rs

@@ -7,17 +7,35 @@ use starknet_mempool_node::communication::{create_node_channels, create_node_cli
 use starknet_mempool_node::components::create_components;
 use starknet_mempool_node::config::MempoolNodeConfig;
 use starknet_mempool_node::servers::{create_servers, run_server_components};
+use tracing::metadata::LevelFilter;
+use tracing::{error, info};
+use tracing_subscriber::prelude::*;
+use tracing_subscriber::{fmt, EnvFilter};
+
+const DEFAULT_LEVEL: LevelFilter = LevelFilter::INFO;
+
+fn configure_tracing() {
+    let fmt_layer = fmt::layer().compact().with_target(false);
+    let level_filter_layer =
+        EnvFilter::builder().with_default_directive(DEFAULT_LEVEL.into()).from_env_lossy();
+
+    // This sets a single subscriber to all of the threads. We may want to implement different
+    // subscriber for some threads and use set_global_default instead of init.
+    tracing_subscriber::registry().with(fmt_layer).with(level_filter_layer).init();
+}
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
+    configure_tracing();
+
     let config = MempoolNodeConfig::load_and_process(args().collect());
     if let Err(ConfigError::CommandInput(clap_err)) = config {
         clap_err.exit();
     }
 
     let config = config?;
     if let Err(error) = config_validate(&config) {
-        println!("Error: {}", error);
+        error!("{}", error);
         exit(1);
     }
 
@@ -26,7 +44,7 @@ async fn main() -> anyhow::Result<()> {
     let components = create_components(&config, &clients);
     let servers = create_servers(&config, channels, components);
 
-    println!("Info: Starting components!");
+    info!("Starting components!");
     run_server_components(&config, servers).await?;
 
     Ok(())

crates/mempool_node/src/servers.rs

@@ -5,6 +5,7 @@ use futures::{Future, FutureExt};
 use starknet_gateway::communication::{create_gateway_server, GatewayServer};
 use starknet_mempool::communication::{create_mempool_server, MempoolServer};
 use starknet_mempool_infra::component_server::ComponentServerStarter;
+use tracing::error;
 
 use crate::communication::MempoolNodeCommunication;
 use crate::components::Components;
@@ -57,15 +58,15 @@ pub async fn run_server_components(
 
     tokio::select! {
         res = gateway_handle => {
-            println!("Error: Gateway Server stopped.");
+            error!("Gateway Server stopped.");
             res?
         }
         res = mempool_handle => {
-            println!("Error: Mempool Server stopped.");
+            error!("Mempool Server stopped.");
             res?
         }
     };
-    println!("Error: Servers ended with unexpected Ok.");
+    error!("Servers ended with unexpected Ok.");
 
     Ok(())
 }