commit d120f4bf3028be9d2fe47a8e0e428428c37f11a0
author Junfan Zhang <zuston@apache.org> Fri Feb 16 11:14:15 2024 +0800
committer GitHub <noreply@github.com> Fri Feb 16 11:14:15 2024 +0800
tree 08917bd1bb2c3108e5da8e0b2cbb7ee4bf82c369
parent 9291a27b749b8b5f0560745cd87226a1a12e515c

[#1407] fix(rust): use grpc runtime worker threads and adjust default runtime config (#1517)

### What changes were proposed in this pull request?

Server will start multiple grpc service to improve throughout, which will 
be executed by grpc runtime blocking threads, this is hard to configure the number of 
blocking threads.

Let's use the worker threads to start all tasks.

### Why are the changes needed?

Subtask for #1407 

### Does this PR introduce _any_ user-facing change?

No.

### How was this patch tested?

Online tests.

rust/experimental/server/src/config.rs

diff --git a/rust/experimental/server/src/config.rs b/rust/experimental/server/src/config.rs
index a273e2d..d197649 100644
--- a/rust/experimental/server/src/config.rs
+++ b/rust/experimental/server/src/config.rs
@@ -87,8 +87,8 @@
     fn default() -> Self {
         RuntimeConfig {
             read_thread_num: 10,
-            write_thread_num: 10,
-            grpc_thread_num: 20,
+            write_thread_num: 40,
+            grpc_thread_num: 100,
             http_thread_num: 5,
             default_thread_num: 5,
         }

rust/experimental/server/src/main.rs

diff --git a/rust/experimental/server/src/main.rs b/rust/experimental/server/src/main.rs
index 01ec4ae..f5e93c5 100644
--- a/rust/experimental/server/src/main.rs
+++ b/rust/experimental/server/src/main.rs
@@ -240,13 +240,9 @@
             .max_decoding_message_size(usize::MAX)
             .max_encoding_message_size(usize::MAX);
         let service_tx = tx.subscribe();
-        runtime_manager.grpc_runtime.spawn_blocking(move || {
-            tokio::runtime::Builder::new_current_thread()
-                .enable_all()
-                .build()
-                .unwrap()
-                .block_on(grpc_serve(service, addr, service_tx));
-        });
+        runtime_manager
+            .grpc_runtime
+            .spawn(async move { grpc_serve(service, addr, service_tx).await });
     }
 
     graceful_wait_for_signal(tx);