From db620ceae30089b5575d6439ffabbbd024d5709b Mon Sep 17 00:00:00 2001
From: cfngc4594 <cfngc4594@gmail.com>
Date: Sun, 13 Apr 2025 17:28:13 +0800
Subject: [PATCH] feat(judge): add memory usage tracking for accepted solutions

---
 src/actions/judge.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/actions/judge.ts b/src/actions/judge.ts
index 32a7d07..de4379d 100644
--- a/src/actions/judge.ts
+++ b/src/actions/judge.ts
@@ -501,12 +501,14 @@ async function run(
       }
     }
   }
+  const maxMemoryUsage = (await container.stats({ stream: false, "one-shot": true })).memory_stats.max_usage;
   finalSubmission = await prisma.submission.update({
     where: { id: submissionId },
     data: {
       status: Status.AC,
       message: "All testcases passed",
       executionTime: maxExecutionTime,
+      memoryUsage: maxMemoryUsage / 1024 / 1024,
     },
     include: {
       TestcaseResult: true,