fix: add remaining code for worker

Author: aayushk231

database/queries/submission.sql

@@ -34,3 +34,32 @@ SELECT
     user_id
 FROM submissions
 WHERE id = $1;
+
+-- name: UpdateSubmission :exec
+UPDATE submissions
+SET 
+    runtime = $1, 
+    memory = $2, 
+    status = $3,
+    testcases_passed = $4,
+    testcases_failed = $5
+WHERE id = $6;
+
+-- name: UpdateScore :exec
+WITH best_submissions AS (
+    SELECT 
+        s.user_id AS user_id,
+        s.question_id,
+        MAX((s.testcases_passed) * q.points / (s.testcases_passed + s.testcases_failed)::numeric) AS best_score
+    FROM submissions s
+    INNER JOIN questions q ON s.question_id = q.id
+    INNER JOIN users u on s.user_id = u.id 
+    WHERE s.user_id = (select user_id from submissions where id = $1) AND q.round = u.round_qualified
+    GROUP BY s.user_id, s.question_id
+)
+UPDATE users
+SET score = (
+    SELECT SUM(best_score)
+    FROM best_submissions
+)
+WHERE users.id = (select user_id from submissions s where s.id = $1);

database/queries/submission_results.sql

@@ -1,3 +1,11 @@
 -- name: CreateSubmissionResult :exec
 INSERT INTO submission_results (id, testcase_id, submission_id, runtime, memory, points_awarded, status, description)
-VALUES ($1, $2, $3, $4, $5, $6, $7, $8);
+VALUES ($1, $2, $3, $4, $5, $6, $7, $8);
+
+-- name: GetStatsForFinalSubEntry :many
+SELECT 
+    runtime, 
+    memory,   
+    status
+FROM submission_results
+WHERE submission_id = $1;

pkg/db/submission.sql.go

@@ -3,8 +3,13 @@ package utils
 import (
 	"context"
 	"fmt"
+	"log"
+	"math/big"
 	"os"
 
+	"github.com/CodeChefVIT/cookoff-10.0-be/pkg/db"
+	"github.com/google/uuid"
+	"github.com/jackc/pgx/v5/pgtype"
 	"github.com/redis/go-redis/v9"
 )
 
@@ -74,3 +79,53 @@ func GetTokenCount(ctx context.Context, subID string) (int64, error) {
 	}
 	return tokenCount, nil
 }
+
+func UpdateSubmission(ctx context.Context, idUUID uuid.UUID) error {
+	status := "DONE"
+
+	data, err := Queries.GetStatsForFinalSubEntry(ctx, idUUID)
+	if err != nil {
+		log.Println("Error Fetching submission results: ", err)
+		return err
+	}
+	var runtime float64
+	var memory int64
+	var passed, failed int
+	for _, v := range data {
+		temp, err := v.Runtime.Float64Value()
+		if err != nil {
+			log.Println("Failed to convert runtime to float kms")
+			return err
+		}
+		runtime += temp.Float64
+		memory += v.Memory.Int.Int64()
+		if v.Status == "success" {
+			passed += 1
+		} else {
+			failed += 1
+		}
+	}
+
+	err = Queries.UpdateSubmission(ctx, db.UpdateSubmissionParams{
+		Runtime:         pgtype.Numeric{Int: big.NewInt(int64(runtime)), Valid: true},
+		Memory:          pgtype.Numeric{Int: big.NewInt(int64(memory)), Valid: true},
+		Status:          &status,
+		TestcasesPassed: pgtype.Int4{Int32: int32(passed), Valid: true},
+		TestcasesFailed: pgtype.Int4{Int32: int32(failed), Valid: true},
+		ID:              idUUID,
+	})
+
+	if err != nil {
+		log.Println("Error updating submission: ", err)
+		return err
+	}
+
+	err = Queries.UpdateScore(ctx, idUUID)
+	if err != nil {
+		log.Println("Error updating the score: ", err)
+		return err
+	}
+
+	log.Printf("Submission ID: %v\n", idUUID)
+	return nil
+}

pkg/db/submission_results.sql.go

@@ -3,6 +3,7 @@ package workers
 import (
 	"context"
 	"encoding/json"
+	"fmt"
 	"log"
 	"math/big"
 	"strings"
@@ -151,20 +152,20 @@ func ProcessJudge0CallbackTask(ctx context.Context, t *asynq.Task) error {
 		return err
 	}
 
-	// tokenCount, err := submission.Tokens.GetTokenCount(ctx, value)
-	// if err != nil {
-	// 	log.Println("Error getting token count: ", err)
-	// 	return err
-	// }
+	tokenCount, err := utils.GetTokenCount(ctx, value)
+	if err != nil {
+		log.Println("Error getting token count: ", err)
+		return err
+	}
 
-	// fmt.Println("Token :- ", tokenCount)
+	fmt.Println("Token :- ", tokenCount)
 
-	// if tokenCount == 0 {
-	// 	err = submission.UpdateSubmission(ctx, idUUID)
-	// 	if err != nil {
-	// 		return err
-	// 	}
-	// }
+	if tokenCount == 0 {
+		err = utils.UpdateSubmission(ctx, idUUID)
+		if err != nil {
+			return err
+		}
+	}
 
 	return nil
 }