Set configurable message_threads

Author: gfodor

janus.plugin.sfu.cfg.example

@@ -10,3 +10,6 @@ max_room_size = 30
 
 # If present, the maximum number of concurrent users allowed to join any room on the server. Zero means no limit.
 max_ccu = 1000
+
+# Number of threads to run message processing on. If zero, use the # of logical CPUs.
+message_threads = 0

src/config.rs

@@ -11,6 +11,7 @@ pub struct Config {
     pub auth_key: Option<Vec<u8>>,
     pub max_room_size: usize,
     pub max_ccu: usize,
+    pub message_threads: usize,
 }
 
 impl Default for Config {
@@ -19,6 +20,7 @@ impl Default for Config {
             auth_key: None,
             max_room_size: usize::max_value(),
             max_ccu: usize::max_value(),
+            message_threads: 0,
         }
     }
 }
@@ -45,6 +47,7 @@ impl Config {
             auth_key: auth_key,
             max_room_size: section.get("max_room_size").and_then(|x| x.parse().ok()).unwrap_or(defaults.max_room_size),
             max_ccu: section.get("max_ccu").and_then(|x| x.parse().ok()).unwrap_or(defaults.max_ccu),
+            message_threads: section.get("message_threads").and_then(|x| x.parse().ok()).unwrap_or(defaults.message_threads),
         })
     }
 }

src/lib.rs

@@ -242,12 +242,13 @@ extern "C" fn init(callbacks: *mut PluginCallbacks, config_path: *const c_char)
             Config::default()
         }
     };
+    let message_threads = config.message_threads;
     CONFIG.set(config).expect("Big problem: config already initialized!");
     match unsafe { callbacks.as_ref() } {
         Some(c) => {
             unsafe { CALLBACKS = Some(c) };
             let mut senders = Vec::new();
-            let cpus = num_cpus::get(); // Run one thread per logical CPU
+            let cpus = if message_threads == 0 { num_cpus::get() } else { message_threads };
 
             for i in 0..cpus {
                 let (messages_tx, messages_rx) = mpsc::sync_channel(0);