refactor(backend,worker): update environment variable keys for consistency and improve configuration handling across backend and worker modules

worker/internal/utils/asyncq.go

@@ -3,7 +3,7 @@ package utils
 import "github.com/hibiken/asynq"
 
 func RedisURI2AsynqOpt(uri string) asynq.RedisConnOpt {
-	opt, err := asynq.ParseRedisURI(GetEnv("REDIS_URL"))
+	opt, err := asynq.ParseRedisURI(GetEnv("redis.url"))
 	if err != nil {
 		panic(err)
 	}

worker/internal/utils/env.go

@@ -1,6 +1,8 @@
 package utils
 
 import (
+	"strings"
+
 	"github.com/spf13/viper"
 )
 
@@ -15,12 +17,14 @@ func InitEnv() {
 		return
 	}
 	v = viper.New()
-	v.SetConfigName(".env")
-	v.SetConfigType("env")
+	v.SetConfigName("config.yaml")
+	v.SetConfigType("yaml")
+	v.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
 	v.AddConfigPath(".")
 	v.AddConfigPath("../")
 	v.AutomaticEnv()
-	if err := v.ReadInConfig(); err != nil {
+	err := v.ReadInConfig()
+	if err != nil {
 		if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
 			// 只有当错误不是"配置文件未找到"时才 panic
 			panic(err)

worker/internal/utils/redis.go

@@ -10,7 +10,7 @@ var rdb *redis.Client = InitRedis()
 var ctx = context.Background()
 
 func InitRedis() *redis.Client {
-	opt, err := redis.ParseURL(GetEnv("REDIS_URL"))
+	opt, err := redis.ParseURL(GetEnv("redis.url"))
 	if err != nil {
 		panic(err)
 	}

worker/main.go

@@ -23,8 +23,8 @@ func main() {
 	zap.ReplaceGlobals(logger)
 
 	srv := asynq.NewServer(
-		utils.RedisURI2AsynqOpt(utils.GetEnv("REDIS_URL")),
-		asynq.Config{Concurrency: cast.ToInt(utils.GetEnvWithDefault("WORKER_CONCURRENCY", "4"))},
+		utils.RedisURI2AsynqOpt(utils.GetEnv("redis.url")),
+		asynq.Config{Concurrency: cast.ToInt(utils.GetEnvWithDefault("worker.concurrency", "4"))},
 	)
 
 	mux := asynq.NewServeMux()