From c2c4ee5777a5c67e4eb8f93f09db2cb8b4abceec Mon Sep 17 00:00:00 2001
From: wizardchen <wizardchen@tencent.com>
Date: Tue, 30 Dec 2025 12:06:29 +0800
Subject: [PATCH] fix: asynq redis db

---
 internal/router/task.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/internal/router/task.go b/internal/router/task.go
index 2836aad3..4df6f92f 100644
--- a/internal/router/task.go
+++ b/internal/router/task.go
@@ -3,6 +3,7 @@ package router
 import (
 	"log"
 	"os"
+	"strconv"
 	"time"
 
 	"github.com/Tencent/WeKnora/internal/types"
@@ -23,12 +24,18 @@ type AsynqTaskParams struct {
 }
 
 func getAsynqRedisClientOpt() *asynq.RedisClientOpt {
+	db := 0
+	if dbStr := os.Getenv("REDIS_DB"); dbStr != "" {
+		if parsed, err := strconv.Atoi(dbStr); err == nil {
+			db = parsed
+		}
+	}
 	opt := &asynq.RedisClientOpt{
 		Addr:         os.Getenv("REDIS_ADDR"),
 		Password:     os.Getenv("REDIS_PASSWORD"),
 		ReadTimeout:  100 * time.Millisecond,
 		WriteTimeout: 200 * time.Millisecond,
-		DB:           0,
+		DB:           db,
 	}
 	return opt
 }