Add max blocking threads option to client

Author: ndr-ds

CLI.md

@@ -127,6 +127,13 @@ A Byzantine-fault tolerant sidechain with low-latency finality and high throughp
   Default value: `10`
 * `--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
+<<<<<<< HEAD
+||||||| parent of e85007b6307 (Add max blocking threads option to client)
+* `--tokio-threads <TOKIO_THREADS>` — The number of Tokio worker threads to use
+=======
+* `--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
+>>>>>>> e85007b6307 (Add max blocking threads option to client)
 * `--blanket-message-policy <BLANKET_MESSAGE_POLICY>` — The policy for handling incoming messages
 
   Default value: `accept`

linera-client/src/client_options.rs

@@ -88,6 +88,14 @@ pub struct ClientContextOptions {
     #[arg(long)]
     pub long_lived_services: bool,
 
+    /// The number of Tokio worker threads to use.
+    #[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

@@ -1572,6 +1572,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.inner.with_wallet {
         span.record("wallet_id", wallet_id);

linera-service/src/proxy/main.rs

@@ -64,6 +64,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,
@@ -386,6 +390,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

@@ -249,6 +249,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)]
@@ -476,6 +480,10 @@ fn main() {
         builder
     };
 
+    if let Some(blocking_threads) = options.tokio_blocking_threads {
+        runtime.max_blocking_threads(blocking_threads);
+    }
+
     runtime
         .enable_all()
         .build()