test page lock after everything

worldcoin · Jan 28, 2025 · 0f48603 · 0f48603
1 parent 2090c2f
commit 0f48603
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/iris-mpc-common/src/config/mod.rs b/iris-mpc-common/src/config/mod.rs
@@ -114,6 +114,9 @@ pub struct Config {
 
     #[serde(default)]
     pub load_chunks_buffer_size: usize,
+
+    #[serde(default)]
+    pub page_lock_after: bool,
 }
 
 fn default_load_chunks_parallelism() -> usize {

diff --git a/iris-mpc/src/bin/server.rs b/iris-mpc/src/bin/server.rs
@@ -62,6 +62,7 @@ use std::{
     },
     time::{Duration, Instant},
 };
+use std::thread::sleep;
 use telemetry_batteries::tracing::{datadog::DatadogBattery, TracingShutdownHandle};
 use tokio::{
     sync::{mpsc, oneshot, Semaphore},
@@ -1093,8 +1094,13 @@ async fn server_main(config: Config) -> eyre::Result<()> {
 
                         let device_manager_clone = actor.device_manager.clone();
 
+                        let page_lock_after = config.page_lock_after;
                         // prepare the handle for the rest of the page locks
                         let page_lock_handle = spawn_blocking(move || {
+                            if page_lock_after {
+                                tracing::info!("Sleeping before page lock");
+                                sleep(Duration::from_secs(120));
+                            }
                             tracing::info!("Page locking host memory for code slices");
                             let now = Instant::now();
                             for db in [&left_codes, &right_codes] {