feat(submission): add memory usage support (#105)

Author: angristan

src/migrations/1626732208784-AddMemUsageSubmission.ts

@@ -0,0 +1,14 @@
+import {MigrationInterface, QueryRunner} from "typeorm";
+
+export class AddMemUsageSubmission1626732208784 implements MigrationInterface {
+    name = 'AddMemUsageSubmission1626732208784'
+
+    public async up(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`ALTER TABLE "submissions" ADD "memUsage" integer NOT NULL DEFAULT '0'`);
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`ALTER TABLE "submissions" DROP COLUMN "memUsage"`);
+    }
+
+}

src/submissions/dto/job-status.dto.ts

@@ -22,4 +22,7 @@ export class JobStatusDTO {
 
   @jsonMember
   exec_duration: number;
+
+  @jsonMember
+  mem_usage: number;
 }

src/submissions/submission.entity.ts

@@ -66,6 +66,11 @@ export class Submission extends BaseEntity {
   @Column({ default: 0 })
   execDuration: number;
 
+  @jsonMember
+  @ApiProperty()
+  @Column({ default: 0 })
+  memUsage: number;
+
   @jsonMember
   @ApiProperty()
   @CreateDateColumn()

src/submissions/submissions.service.ts

@@ -56,6 +56,7 @@ export class SubmissionsService {
     stdout?: string,
     stderr?: string,
     execDuration?: number,
+    memUsage?: number,
   ): Promise<Submission | undefined> {
     const submission = await this.submissionsRepository.findOne({
       id,
@@ -77,6 +78,9 @@ export class SubmissionsService {
       if (execDuration) {
         submission.execDuration = execDuration;
       }
+      if (memUsage) {
+        submission.memUsage = memUsage;
+      }
       await submission.save();
       return submission;
     }
@@ -124,6 +128,7 @@ export class SubmissionsService {
         jobStatus.stdout,
         jobStatus.stderr,
         jobStatus.exec_duration,
+        jobStatus.mem_usage,
       );
 
       // if (submission) {