Replace menual env replacer to auto env replacer

Signed-off-by: Asai Neko <sugar@sne.moe>
This commit is contained in:
2025-12-23 23:53:32 +08:00
parent 485d0de64b
commit fd4da4f1a1

View File

@@ -3,6 +3,7 @@ package config
import ( import (
"log" "log"
"os" "os"
"strings"
"github.com/spf13/viper" "github.com/spf13/viper"
) )
@@ -22,15 +23,8 @@ func Init() {
viper.AddConfigPath(ConfigDir()) viper.AddConfigPath(ConfigDir())
// Bind ENV // Bind ENV
viper.BindEnv("server.address", "SERVER_ADDRESS") viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
viper.BindEnv("server.debug_mode", "SERVER_DEBUG_MODE") viper.AutomaticEnv()
viper.BindEnv("server.file_logger", "SERVER_FILE_LOGGER")
viper.BindEnv("server.jwt_secret", "SERVER_JWT_SECRET")
viper.BindEnv("database.type", "DATABASE_TYPE")
viper.BindEnv("database.host", "DATABASE_HOST")
viper.BindEnv("database.name", "DATABASE_NAME")
viper.BindEnv("database.username", "DATABASE_USERNAME")
viper.BindEnv("database.password", "DATABASE_PASSWORD")
conf := &config{} conf := &config{}
if err := viper.ReadInConfig(); err != nil { if err := viper.ReadInConfig(); err != nil {