update

dcodesdev · Sep 21, 2024 · 2de29ba · 2de29ba
1 parent 37c1695
commit 2de29ba
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/crates/rustfinity-runner/Dockerfile b/crates/rustfinity-runner/Dockerfile
@@ -11,6 +11,7 @@ LABEL rustfinity-runner="true"
 RUN apt-get update && apt-get install -y \
     pkg-config \
     libssl-dev \
+    heaptrack \
     && rm -rf /var/lib/apt/lists/*
 
 # Build the Runner CLI

diff --git a/crates/rustfinity-runner/Dockerfile.staging b/crates/rustfinity-runner/Dockerfile.staging
@@ -11,6 +11,7 @@ LABEL rustfinity-runner="true"
 RUN apt-get update && apt-get install -y \
     pkg-config \
     libssl-dev \
+    heaptrack \
     && rm -rf /var/lib/apt/lists/*
 
 # Build the Runner CLI

diff --git a/crates/rustfinity-runner/src/command.rs b/crates/rustfinity-runner/src/command.rs
@@ -71,16 +71,15 @@ async fn memory_benchmark(challenge: &str, test_binary_path: &str) -> anyhow::Re
     let last_index = details.len() - 1;
     let details = details[last_index];
 
-    let mut memory = "";
+    let mut memory = String::new();
 
-    //let memory: Vec<_> = details.split("peak heap memory consumption:").collect();
     details.lines().for_each(|line| {
         if line.contains("peak heap memory consumption:") {
-            memory = line;
+            memory = line.replace("peak", "Peak");
         }
     });
 
-    Ok(memory.to_string())
+    Ok(memory)
 }
 
 async fn run_tests(code_base64: &str, challenge: &str) -> anyhow::Result<String> {