chore(project) : add controller for Project and make changes in jobQueue

Author: prabalsingh24

src/rabbitmq/jobqueue.ts

@@ -26,7 +26,16 @@ export interface RunJob {
   scenario: string
 }
 
-export type JudgeJob = RunJob | SubmissionJob
+export interface ProjectJob {
+  id: number
+  source: string,
+  lang: string,
+  problem: string,
+  timelimit: number,
+  scenario: string
+}
+
+export type JudgeJob = RunJob | SubmissionJob | ProjectJob
 
 let jobQ = 'job_queue'
 let successQ = 'success_queue'
@@ -50,8 +59,17 @@ amqp.connect(`amqp://${config.AMQP.USER}:${config.AMQP.PASS}@${config.AMQP.HOST}
       jobChannel.consume(successQ, (msg) => {
         debug(`SUCCESS:CONSUME: msg.content = ${msg.content.toString()}`)
 
-        const payload = JSON.parse(msg.content.toString())        
-        const eventName = payload.testcases ? 'submit_result' : 'run_result'
+        const payload = JSON.parse(msg.content.toString())
+        let eventName;
+        if (payload.testcases) {
+          eventName = 'submit_result'
+        }
+        else if (payload.code && payload.score) {
+          eventName = 'project_result'
+        }
+        else {
+          eventName = 'run_result'
+        }
 
         successListener.emit(eventName, payload)
         jobChannel.ack(msg)

src/routes/api/project/controller.ts

@@ -0,0 +1,67 @@
+import { Request, Response } from 'express'
+import {ProjectJob, queueJob} from 'rabbitmq/jobqueue'
+import DB from 'models'
+import axios from 'axios'
+
+type ProjectResponse = {
+    id: number,
+    stdout: string,
+    stderr: string,
+    time: number,
+    code: number,
+    score: number
+}
+type RunPoolElement = {
+    res: Response
+}
+
+const RunPool: { [x: number]: RunPoolElement } = {}
+
+export default {
+    async runPOST(req: Request, res: Response) {
+        const mode = req.body.mode || 'sync'
+        const job = await DB.submissions.create({
+            lang: req.body.lang,
+            start_time: new Date(),
+            mode,
+            callback: req.body.callback
+        })
+
+        await queueJob(<ProjectJob>{
+            id: job.id,
+            source: req.body.submission,
+            problem: req.body.problem,
+            lang: req.body.lang,
+            submissionDirs: req.body.submissionDirs,
+            timelimit: req.body.timelimit,
+            scenario: 'project'
+        })
+
+
+        if (['callback', 'poll'].includes(mode)) {
+            return res.json({
+                id: job.id
+            })
+        }
+
+        // if mode === 'sync'
+        RunPool[job.id] = {
+            res
+        }
+    },
+
+    async onSuccess(result: ProjectResponse) {
+        const job = await DB.submissions.findById(result.id)
+        job.results = result
+        await job.save()
+
+        switch (job.mode) {
+            case 'callback':
+                await axios.post(job.callback, result)
+                break
+            case 'sync':
+                RunPool[job.id].res.json(result)
+                break
+        }
+    }
+}