make session lock time configurable

DolceTriade · Feb 2, 2024 · af52752 · af52752
1 parent dee576e
commit af52752
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/blade/bep/lib.rs b/blade/bep/lib.rs
@@ -120,7 +120,7 @@ impl publish_build_event_server::PublishBuildEvent for BuildEventService {
                                                 if std::time::SystemTime::now()
                                                     .duration_since(end)
                                                     .unwrap_or(std::time::Duration::from_secs(0))
-                                                    > std::time::Duration::from_secs(60)
+                                                    > global.session_lock_time
                                                 {
                                                     already_over = true;
                                                 }

diff --git a/blade/main.rs b/blade/main.rs
@@ -38,6 +38,8 @@ cfg_if! {
             bytestream_overrides: Vec<String>,
             #[arg(short='r', long="retention", value_name = "RETENTION", value_parser = humantime::parse_duration)]
             retention: Option<std::time::Duration>,
+            #[arg(short='s', long="session_lock_time", value_name = "LOCK_TIME", value_parser = humantime::parse_duration, default_value="5m")]
+            session_lock_time: std::time::Duration,
         }
 
         #[actix_web::main]
@@ -66,7 +68,7 @@ cfg_if! {
                 }
             }
             let db_manager = db::new(&args.db_path)?;
-            let state = Arc::new(state::Global { db_manager, allow_local: args.allow_local, bytestream_client: bs, retention: args.retention });
+            let state = Arc::new(state::Global { db_manager, allow_local: args.allow_local, bytestream_client: bs, retention: args.retention, session_lock_time: args.session_lock_time });
             let actix_state = state.clone();
             let cleanup_state = state.clone();
             log::info!("Starting blade server at: {}", addr.to_string());

diff --git a/blade/state/lib.rs b/blade/state/lib.rs
@@ -138,6 +138,7 @@ pub struct Global {
     pub bytestream_client: bytestream::Client,
     pub allow_local: bool,
     pub retention: Option<std::time::Duration>,
+    pub session_lock_time: std::time::Duration,
 }
 
 }