feat: save scores to database

Author: Mikadows

src/migrations/1626038854102-AddQualityAndLintScores.ts

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

src/submissions/submission.entity.ts

@@ -76,6 +76,14 @@ export class Submission extends BaseEntity {
   @UpdateDateColumn()
   updatedAt: Date;
 
+  @jsonMember
+  @Column({ nullable: true })
+  lintScore: number;
+
+  @jsonMember
+  @Column({ nullable: true })
+  qualityScore: number;
+
   @ManyToOne(() => User, (user) => user.submissions, {
     nullable: false,
     eager: true,

src/submissions/submissions.controller.ts

@@ -33,22 +33,10 @@ export class SubmissionsController {
     @Request() req: ValidatedJWTReq,
     @Body() createSubmissionDTO: CreateSubmissionDTO,
   ): Promise<SubmissionResultDTO> {
-    const submission = await this.submissionsService.create({
-      ...createSubmissionDTO,
-      user: req.user,
-    });
-
-    // Send job to worker
-    await this.amqpConnection.publish('jobs_ex', 'jobs_rk', {
-      id: submission.id,
-      code: createSubmissionDTO.code,
-      language: createSubmissionDTO.language,
-    });
-
     let lintScore = { score: 100 };
     let qualityScore = { score: 100 };
     switch (createSubmissionDTO.language) {
-      case 'cpython3':
+      case 'python':
         qualityScore = this.qualityService.run(createSubmissionDTO.code, 'py');
         // lintScore = this.lintService.lintPython3(createSubmissionDTO.code);
         break;
@@ -63,6 +51,22 @@ export class SubmissionsController {
         qualityScore = { score: 0 };
     }
 
+    const submission = await this.submissionsService.create(
+      {
+        ...createSubmissionDTO,
+        user: req.user,
+      },
+      lintScore.score,
+      qualityScore.score,
+    );
+
+    // Send job to worker
+    await this.amqpConnection.publish('jobs_ex', 'jobs_rk', {
+      id: submission.id,
+      code: createSubmissionDTO.code,
+      language: createSubmissionDTO.language,
+    });
+
     return {
       submission,
       lint: lintScore,

src/submissions/submissions.service.ts

@@ -26,9 +26,15 @@ export class SubmissionsService {
     private benchmarkService: BenchmarkService,
   ) {}
 
-  async create(insertSubmissionDTO: InsertSubmissionDTO): Promise<Submission> {
+  async create(
+    insertSubmissionDTO: InsertSubmissionDTO,
+    lintScore: number,
+    qualityScore: number,
+  ): Promise<Submission> {
     const submission = new Submission(insertSubmissionDTO);
     submission.status = 'waiting';
+    submission.lintScore = lintScore;
+    submission.qualityScore = qualityScore;
 
     const benchmark = await this.benchmarkService.findOne(
       insertSubmissionDTO.benchmarkId,