redis cache working

Author: atharvaSharma17

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,6 +49,11 @@ 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"})
 	}
+	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)
+        }
+    }
 
 	sub := utils.SubmissionInput{
 		ID:         submissionID,

pkg/utils/tokenCache.go

@@ -0,0 +1,37 @@
+package utils
+
+import (
+    "context"
+    "fmt"
+    "os"
+
+    "github.com/redis/go-redis/v9"
+)
+
+var (
+    TokenCache *redis.Client
+    Ctx        = context.Background()
+)
+
+// InitTokenCache initializes a separate Redis client for token cache
+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, // Use DB 1 to keep it separate from queue DB 0
+    })
+
+    // Test connection
+    if err := TokenCache.Ping(Ctx).Err(); err != nil {
+        panic(fmt.Sprintf("failed to connect to Redis token cache: %v", err))
+    }
+}