Possibility to change the stack size

Author: Lucas Paixão

src/lib.rs

@@ -164,7 +164,8 @@ pub fn uring_builder() -> io_uring::Builder {
 #[derive(Clone)]
 pub struct Builder {
   entries: u32,
-  threads: usize,
+  worker_threads: usize,
+  thread_stack_size: usize,
   urb: io_uring::Builder,
 }
 
@@ -177,7 +178,8 @@ pub struct Builder {
 pub fn builder() -> Builder {
   Builder {
     entries: 256,
-    threads: std::thread::available_parallelism().unwrap().get(),
+    worker_threads: std::thread::available_parallelism().unwrap().get(),
+    thread_stack_size: 20971520,
     urb: io_uring::IoUring::builder(),
   }
 }
@@ -205,7 +207,13 @@ impl Builder {
 
   /// Replaces the default number of worker threads
   pub fn worker_threads(&mut self, num: usize) -> &mut Self {
-    self.threads = num;
+    self.worker_threads = num;
+    self
+  }
+
+  /// Replaces the default number of worker threads stack size
+  pub fn thread_stack_size(&mut self, num: usize) -> &mut Self {
+    self.worker_threads = num;
     self
   }
 

src/runtime/mod.rs

@@ -120,7 +120,8 @@ impl Runtime {
     let on_thread_start = runtime.create_on_thread_start_callback();
 
     let rt = tokio::runtime::Builder::new_multi_thread()
-      .worker_threads(builder.threads)
+      .worker_threads(builder.worker_threads)
+      .thread_stack_size(builder.thread_stack_size)
       .on_thread_start(on_thread_start)
       .on_thread_park(Runtime::on_thread_park)
       .enable_all()
@@ -191,7 +192,7 @@ impl Runtime {
     self.local.spawn_local(Runtime::drive_uring_wakes(
       self.contexts.clone(),
       self.signal.clone(),
-      self.builder.threads,
+      self.builder.worker_threads,
     ));
   }