Merge pull request #17 from equestrian2296/master

Feat: Added token manager (redis cache)

Author: Soham-Maha

cmd/api/main.go

@@ -20,6 +20,7 @@ func main() {
 	logger.InitLogger()
 	utils.LoadConfig()
 	utils.InitCache()
+	utils.InitTokenCache()
 	utils.InitDB()
 	validator.InitValidator()
 

pkg/controllers/submission.go

@@ -49,20 +49,13 @@ func SubmitCode(c echo.Context) error {
 		fmt.Println("CreateBatchSubmission error:", err)
 		return c.JSON(http.StatusInternalServerError, map[string]string{"error": "Failed to create batch submission"})
 	}
-
-	sub := utils.SubmissionInput{
-		ID:         submissionID,
-		QuestionID: req.QuestionID,
-		LanguageID: req.LanguageID,
-		SourceCode: req.SourceCode,
-		UserID:     req.UserID,
-	}
-	if err := utils.SaveSubmission(sub); err != nil {
-		return c.JSON(http.StatusInternalServerError, map[string]string{"error": "Failed to save submission record"})
-	}
+	for _, token := range tokens {
+        if err := utils.TokenCache.Set(utils.Ctx, token, submissionID, 0).Err(); err != nil {
+            fmt.Printf("Failed to cache token %s: %v\n", token, err)
+        }
+    }
 
 	return c.JSON(http.StatusOK, map[string]interface{}{
 		"submission_id": submissionID,
-		"tokens":        tokens,
 	})
-}
+}

pkg/utils/submission.go

@@ -5,7 +5,7 @@ import (
 	"fmt"
 	"github.com/CodeChefVIT/cookoff-10.0-be/pkg/db"
 	"github.com/google/uuid"
-
+	
 )
 
 type SubmissionInput struct {
@@ -16,6 +16,7 @@ type SubmissionInput struct {
 	UserID     string
 }
 // place holder code -- renove when queue is setup
+
 func SaveSubmission(sub SubmissionInput) error {
 	if Queries == nil {
 		return fmt.Errorf("DB Queries not initialized")

pkg/utils/tokenCache.go

@@ -0,0 +1,66 @@
+package utils
+
+import (
+    "context"
+    "fmt"
+    "os"
+
+    "github.com/redis/go-redis/v9"
+)
+
+var (
+    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))
+    }
+}
+
+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()
+}
+// 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
+}
+//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()
+}