premature exit bug fix

.env.example

@@ -16,9 +16,11 @@ DKN_P2P_LISTEN_ADDR=/ip4/0.0.0.0/tcp/4001
 DKN_RELAY_NODES=
 # Comma-separated static bootstrap nodes
 DKN_BOOTSTRAP_NODES=
+
+## DRIA (profiling only, do not uncomment) ##
 # Set to a number of seconds to wait before exiting, only use in profiling build!
 # Otherwise, leave this empty.
-DKN_EXIT_TIMEOUT=
+# DKN_EXIT_TIMEOUT=
 
 ## Open AI (if used, required) ##
 OPENAI_API_KEY=

Cargo.toml

@@ -7,7 +7,7 @@ default-members = ["compute"]
 
 [workspace.package]
 edition = "2021"
-version = "0.2.13"
+version = "0.2.14"
 license = "Apache-2.0"
 readme = "README.md"
 

compute/src/main.rs

@@ -31,10 +31,13 @@ async fn main() -> Result<()> {
     let token = CancellationToken::new();
     let cancellation_token = token.clone();
     tokio::spawn(async move {
-        if let Ok(timeout_str) = env::var("DKN_EXIT_TIMEOUT").map(|s| s.trim().to_string()) {
-            let duration_secs = timeout_str.parse().unwrap_or(120);
+        if let Ok(Ok(duration_secs)) =
+            env::var("DKN_EXIT_TIMEOUT").map(|s| s.to_string().parse::<u64>())
+        {
             log::warn!("Waiting for {} seconds before exiting.", duration_secs);
             tokio::time::sleep(tokio::time::Duration::from_secs(duration_secs)).await;
+
+            log::warn!("Exiting due to DKN_EXIT_TIMEOUT.");
             cancellation_token.cancel();
         } else if let Err(err) = wait_for_termination(cancellation_token.clone()).await {
             log::error!("Error waiting for termination: {:?}", err);