Add max blocking threads option to client

Author: ndr-ds

CLI.md

@@ -143,6 +143,7 @@ A Byzantine-fault tolerant sidechain with low-latency finality and high throughp
 * `--wait-for-outgoing-messages` — Whether to wait until a quorum of validators has confirmed that all sent cross-chain messages have been delivered
 * `--long-lived-services` — (EXPERIMENTAL) Whether application services can persist in some cases between queries
 * `--tokio-threads <TOKIO_THREADS>` — The number of Tokio worker threads to use
+* `--tokio-blocking-threads <TOKIO_BLOCKING_THREADS>` — The number of Tokio blocking threads to use
 * `--blanket-message-policy <BLANKET_MESSAGE_POLICY>` — The policy for handling incoming messages
 
   Default value: `accept`

linera-client/src/client_options.rs

@@ -153,6 +153,10 @@ pub struct ClientOptions {
     #[arg(long, env = "LINERA_CLIENT_TOKIO_THREADS")]
     pub tokio_threads: Option<usize>,
 
+    /// The number of Tokio blocking threads to use.
+    #[arg(long)]
+    pub tokio_blocking_threads: Option<usize>,
+
     /// The policy for handling incoming messages.
     #[arg(long, default_value = "accept")]
     pub blanket_message_policy: BlanketMessagePolicy,

linera-service/src/linera/main.rs

@@ -1290,6 +1290,10 @@ fn main() -> anyhow::Result<()> {
         builder
     };
 
+    if let Some(blocking_threads) = options.tokio_blocking_threads {
+        runtime.max_blocking_threads(blocking_threads);
+    }
+
     let span = tracing::info_span!("linera::main");
     if let Some(wallet_id) = &options.with_wallet {
         span.record("wallet_id", wallet_id);

linera-service/src/proxy/main.rs

@@ -58,6 +58,10 @@ pub struct ProxyOptions {
     #[arg(long, env = "LINERA_PROXY_TOKIO_THREADS")]
     tokio_threads: Option<usize>,
 
+    /// The number of Tokio blocking threads to use.
+    #[arg(long)]
+    tokio_blocking_threads: Option<usize>,
+
     /// Storage configuration for the blockchain history, chain states and binary blobs.
     #[arg(long = "storage")]
     storage_config: StorageConfigNamespace,
@@ -398,6 +402,10 @@ fn main() -> Result<()> {
         builder
     };
 
+    if let Some(blocking_threads) = options.tokio_blocking_threads {
+        runtime.max_blocking_threads(blocking_threads);
+    }
+
     runtime.enable_all().build()?.block_on(options.run())
 }
 

linera-service/src/server.rs

@@ -283,6 +283,10 @@ struct ServerOptions {
     /// The number of Tokio worker threads to use.
     #[arg(long, env = "LINERA_SERVER_TOKIO_THREADS")]
     tokio_threads: Option<usize>,
+
+    /// The number of Tokio blocking threads to use.
+    #[arg(long)]
+    tokio_blocking_threads: Option<usize>,
 }
 
 #[derive(Debug, PartialEq, Eq, Deserialize)]
@@ -537,6 +541,10 @@ fn main() {
         builder
     };
 
+    if let Some(blocking_threads) = options.tokio_blocking_threads {
+        runtime.max_blocking_threads(blocking_threads);
+    }
+
     runtime
         .enable_all()
         .build()