fix(queue): add Redis database selection support for queue.redis configuration (#849)

* Initial plan

* Add Redis DB configuration support for queue.redis

- Add db field to SectionRedisQueue configuration structure
- Update default configuration with queue.redis.db setting (default: 0)
- Update config loading to read queue.redis.db parameter
- Pass db parameter to redisdb.NewWorker in main.go using WithDB option
- Add comprehensive tests for Redis DB configuration
- Update test configuration files with db field

Co-authored-by: appleboy <21979+appleboy@users.noreply.github.com>

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: appleboy <21979+appleboy@users.noreply.github.com>

Author: Copilot

config/config.go

@@ -88,6 +88,7 @@ queue:
     with_tls: false
     username: ""
     password: ""
+    db: 0
 
 ios:
   enabled: false
@@ -263,6 +264,7 @@ type SectionRedisQueue struct {
 	Addr       string `yaml:"addr"`
 	Username   string `yaml:"username"`
 	Password   string `yaml:"password"`
+	DB         int    `yaml:"db"`
 	StreamName string `yaml:"stream_name"`
 	Group      string `yaml:"group"`
 	Consumer   string `yaml:"consumer"`
@@ -487,6 +489,7 @@ func loadConfigFromViper() (*ConfYaml, error) {
 	conf.Queue.Redis.WithTLS = viper.GetBool("queue.redis.with_tls")
 	conf.Queue.Redis.Username = viper.GetString("queue.redis.username")
 	conf.Queue.Redis.Password = viper.GetString("queue.redis.password")
+	conf.Queue.Redis.DB = viper.GetInt("queue.redis.db")
 
 	// Stat Engine
 	conf.Stat.Engine = viper.GetString("stat.engine")

config/config_test.go

@@ -160,6 +160,7 @@ func (suite *ConfigTestSuite) TestValidateConfDefault() {
 	assert.Equal(suite.T(), "", suite.ConfGorushDefault.Queue.Redis.Username)
 	assert.Equal(suite.T(), "", suite.ConfGorushDefault.Queue.Redis.Password)
 	assert.Equal(suite.T(), false, suite.ConfGorushDefault.Queue.Redis.WithTLS)
+	assert.Equal(suite.T(), 0, suite.ConfGorushDefault.Queue.Redis.DB)
 
 	// log
 	assert.Equal(suite.T(), "string", suite.ConfGorushDefault.Log.Format)
@@ -299,6 +300,43 @@ func TestLoadConfigFromEnv(t *testing.T) {
 	assert.Equal(t, "x-auth-key:0987654321", ConfGorush.Core.FeedbackHeader[1])
 }
 
+func TestRedisDBConfiguration(t *testing.T) {
+	// Test loading Redis DB configuration from file
+	conf, err := LoadConf("testdata/redis_db_config.yml")
+	if err != nil {
+		t.Fatalf("failed to load redis_db_config.yml: %v", err)
+	}
+
+	// Test queue.redis.db is properly loaded
+	assert.Equal(t, "redis", conf.Queue.Engine)
+	assert.Equal(t, 5, conf.Queue.Redis.DB)
+
+	// Test stat.redis.db is properly loaded
+	assert.Equal(t, "redis", conf.Stat.Engine)
+	assert.Equal(t, 3, conf.Stat.Redis.DB)
+}
+
+func TestRedisDBConfigurationFromEnv(t *testing.T) {
+	// Test loading Redis DB configuration from environment variables
+	os.Setenv("GORUSH_QUEUE_REDIS_DB", "7")
+	os.Setenv("GORUSH_STAT_REDIS_DB", "9")
+
+	conf, err := LoadConf()
+	if err != nil {
+		t.Fatalf("failed to load config: %v", err)
+	}
+
+	// Test queue.redis.db is properly loaded from env
+	assert.Equal(t, 7, conf.Queue.Redis.DB)
+
+	// Test stat.redis.db is properly loaded from env
+	assert.Equal(t, 9, conf.Stat.Redis.DB)
+
+	// Clean up
+	os.Unsetenv("GORUSH_QUEUE_REDIS_DB")
+	os.Unsetenv("GORUSH_STAT_REDIS_DB")
+}
+
 func TestLoadWrongDefaultYAMLConfig(t *testing.T) {
 	defaultConf = []byte(`a`)
 	_, err := LoadConf()

config/testdata/config.yml

@@ -73,6 +73,7 @@ queue:
     username: ""
     password: ""
     with_tls: false
+    db: 0
 
 ios:
   enabled: false

config/testdata/redis_db_config.yml

@@ -0,0 +1,20 @@
+queue:
+  engine: "redis"
+  redis:
+    addr: 127.0.0.1:6379
+    group: gorush
+    consumer: gorush
+    stream_name: gorush
+    username: ""
+    password: ""
+    with_tls: false
+    db: 5
+
+stat:
+  engine: "redis"
+  redis:
+    cluster: false
+    addr: "localhost:6379"
+    username: ""
+    password: ""
+    db: 3

main.go

@@ -344,6 +344,7 @@ func main() {
 			redisdb.WithAddr(cfg.Queue.Redis.Addr),
 			redisdb.WithUsername(cfg.Queue.Redis.Username),
 			redisdb.WithPassword(cfg.Queue.Redis.Password),
+			redisdb.WithDB(cfg.Queue.Redis.DB),
 			redisdb.WithStreamName(cfg.Queue.Redis.StreamName),
 			redisdb.WithGroup(cfg.Queue.Redis.Group),
 			redisdb.WithConsumer(cfg.Queue.Redis.Consumer),