Merge pull request #20 from aayushk231/worker

feat: add the worker and helper functions

Author: Soham-Maha

.gitignore

@@ -21,6 +21,8 @@ tmp/
 # IDE specific files
 .vscode
 .idea
+*.rest
+dump.txt
 
 # .env file
 .env

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

@@ -0,0 +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);
+
+-- name: GetStatsForFinalSubEntry :many
+SELECT 
+    runtime, 
+    memory,   
+    status
+FROM submission_results
+WHERE submission_id = $1;

pkg/db/submission.sql.go

@@ -1,66 +1,131 @@
 package utils
 
 import (
-    "context"
-    "fmt"
-    "os"
+	"context"
+	"fmt"
+	"log"
+	"math/big"
+	"os"
 
-    "github.com/redis/go-redis/v9"
+	"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"
 )
 
 var (
-    TokenCache *redis.Client
-    Ctx        = context.Background()
+	TokenCache *redis.Client
+	Ctx        = context.Background()
 )
 
 func InitTokenCache() {
-    redisHost := os.Getenv("REDIS_HOST")
-    if redisHost == "" {
-        redisHost = "localhost"
-    }
-
-    redisPort := os.Getenv("REDIS_PORT")
-    if redisPort == "" {
-        redisPort = "6379"
-    }
-
-    TokenCache = redis.NewClient(&redis.Options{
-        Addr: fmt.Sprintf("%s:%s", redisHost, redisPort),
-        DB:   1,
-    })
-
-    if err := TokenCache.Ping(Ctx).Err(); err != nil {
-        panic(fmt.Sprintf("failed to connect to Redis token cache: %v", err))
-    }
+	redisHost := os.Getenv("REDIS_HOST")
+	if redisHost == "" {
+		redisHost = "localhost"
+	}
+
+	redisPort := os.Getenv("REDIS_PORT")
+	if redisPort == "" {
+		redisPort = "6379"
+	}
+
+	TokenCache = redis.NewClient(&redis.Options{
+		Addr: fmt.Sprintf("%s:%s", redisHost, redisPort),
+		DB:   1,
+	})
+
+	if err := TokenCache.Ping(Ctx).Err(); err != nil {
+		panic(fmt.Sprintf("failed to connect to Redis token cache: %v", err))
+	}
 }
 
-func StoreToken(token string, submissionID string) error {
-    if err := TokenCache.Set(Ctx, token, submissionID, 0).Err(); err != nil {
-        return err
-    }
-    return TokenCache.SAdd(Ctx, fmt.Sprintf("submission_tokens:%s", submissionID), token).Err()
+func StoreToken(token string, submissionID string, testcaseID string) error {
+	if err := TokenCache.Set(Ctx, token, fmt.Sprintf("%s:%s", submissionID, testcaseID), 0).Err(); err != nil {
+		return err
+	}
+	return TokenCache.SAdd(Ctx, fmt.Sprintf("submission_tokens:%s", submissionID), token).Err()
 }
-// get sub 
+
+// get sub
 func GetSubmissionIDByToken(token string) (string, error) {
-    submissionID, err := TokenCache.Get(Ctx, token).Result()
-    if err == redis.Nil {
-        return "", fmt.Errorf("token not found")
-    }
-    return submissionID, err
+	submissionID, err := TokenCache.Get(Ctx, token).Result()
+	if err == redis.Nil {
+		return "", fmt.Errorf("token not found")
+	}
+	return submissionID, err
 }
-//can be used to delete all tokens 
+
+// can be used to delete all tokens
 func DeleteTokensBySubmissionID(submissionID string) error {
-    setKey := fmt.Sprintf("submission_tokens:%s", submissionID)
-    tokens, err := TokenCache.SMembers(Ctx, setKey).Result()
-    if err != nil {
-        return err
-    }
-
-    if len(tokens) > 0 {
-        if err := TokenCache.Del(Ctx, tokens...).Err(); err != nil {
-            return err
-        }
-    }
-
-    return TokenCache.Del(Ctx, setKey).Err()
+	setKey := fmt.Sprintf("submission_tokens:%s", submissionID)
+	tokens, err := TokenCache.SMembers(Ctx, setKey).Result()
+	if err != nil {
+		return err
+	}
+
+	if len(tokens) > 0 {
+		if err := TokenCache.Del(Ctx, tokens...).Err(); err != nil {
+			return err
+		}
+	}
+
+	return TokenCache.Del(Ctx, setKey).Err()
+}
+
+func GetTokenCount(ctx context.Context, subID string) (int64, error) {
+	tokenCount, err := TokenCache.SCard(ctx, fmt.Sprintf("submission_tokens:%s", subID)).Result()
+	if err != nil {
+		return 0, err
+	}
+	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
 }