refactor: config

This commit is contained in:
2025-06-18 20:17:04 +03:00
parent 82a75bd616
commit 7a79028aa7

View File

@@ -1,46 +1,49 @@
package config package config
import ( import (
"fmt"
"log" "log"
"reflect" "reflect"
"fmt"
"github.com/spf13/viper" "github.com/spf13/viper"
) )
type Config struct { type Config struct {
Hostname string `mapstructure:"HOSTNAME"` Hostname string `mapstructure:"HOSTNAME"`
Port string `mapstructure:"PORT"` Port string `mapstructure:"PORT"`
DatabaseUrl string `mapstructure:"DATABASE_URL"` DatabaseUrl string `mapstructure:"POSTGRES_URL"`
RedisUrl string `mapstructure:"REDIS_URL"` RedisUrl string `mapstructure:"REDIS_URL"`
MinioUrl string `mapstructure:"MINIO_URL"` MinioUrl string `mapstructure:"MINIO_URL"`
JwtAlgorithm string `mapstructure:"JWT_ALGORITHM"` JwtAlgorithm string `mapstructure:"JWT_ALGORITHM"`
JwtSecret string `mapstructure:"JWT_SECRET"` JwtSecret string `mapstructure:"JWT_SECRET"`
JwtIssuer string `mapstructure:"JWT_ISSUER"` JwtIssuer string `mapstructure:"JWT_ISSUER"`
JwtAudience string `mapstructure:"JWT_AUDIENCE"` JwtAudience string `mapstructure:"JWT_AUDIENCE"`
JwtExpAccess string `mapstructure:"JWT_EXP_ACCESS"` JwtExpAccess string `mapstructure:"JWT_EXP_ACCESS"`
JwtExpRefresh string `mapstructure:"JWT_EXP_REFRESH"` JwtExpRefresh string `mapstructure:"JWT_EXP_REFRESH"`
Environment string `mapstructure:"ENVIRONMENT"` Environment string `mapstructure:"ENVIRONMENT"`
} }
var config *Config var config *Config
func LoadConfig() error { func LoadConfig() error {
viper.AutomaticEnv()
viper.SetConfigName(".env")
viper.SetConfigType("env")
viper.AddConfigPath(".")
// Load .env file // Load .env file
if err := viper.ReadInConfig(); err != nil { if err := viper.ReadInConfig(); err != nil {
log.Printf("Error reading config file, proceeding with environment variables. %s", err) log.Printf("Error reading .env file, proceeding with environment variables. %s", err)
} }
// Set default parameters
viper.SetDefault("HOSTNAME", "localhost") viper.SetDefault("HOSTNAME", "localhost")
viper.SetDefault("PORT", "8080") viper.SetDefault("PORT", "8080")
viper.SetDefault("DATABASE_URL", "mydb")
viper.SetDefault("REDIS_URL", "myredis")
viper.SetDefault("MINIO_URL", "myminio")
viper.SetDefault("JWT_ALGORITHM", "HS256") viper.SetDefault("JWT_ALGORITHM", "HS256")
viper.SetDefault("JWT_SECRET", "default_jwt_secret_please_change") // TODO: remove and randomly generate viper.SetDefault("JWT_SECRET", "default_jwt_secret_please_change") // TODO: remove and randomly generate
viper.SetDefault("JWT_EXP_ACCESS", "5m") viper.SetDefault("JWT_EXP_ACCESS", "5m")
@@ -50,17 +53,10 @@ func LoadConfig() error {
viper.SetDefault("ENVIRONMENT", "production") viper.SetDefault("ENVIRONMENT", "production")
// Set the file name and type for Viper
viper.SetConfigName(".env") // name of config file (without extension)
viper.SetConfigType("env") // REQUIRED if the config file does not have the extension
viper.AddConfigPath(".") // optionally look for config in the working directory
// Unmarshal the configuration into the Config struct
if err := viper.Unmarshal(&config); err != nil { if err := viper.Unmarshal(&config); err != nil {
log.Fatalf("Unable to decode into struct, %v", err) log.Fatalf("Unable to decode into struct, %v", err)
} }
// Perform validation
if err := validateConfig(); err != nil { if err := validateConfig(); err != nil {
return err return err
} }
@@ -76,7 +72,6 @@ func validateConfig() error {
field := v.Field(i) field := v.Field(i)
fieldType := t.Field(i) fieldType := t.Field(i)
// Check if the field is a string and is empty
if field.Kind() == reflect.String && field.String() == "" { if field.Kind() == reflect.String && field.String() == "" {
return fmt.Errorf("Missing required configuration: %s", fieldType.Name) return fmt.Errorf("Missing required configuration: %s", fieldType.Name)
} }
@@ -88,4 +83,3 @@ func validateConfig() error {
func GetConfig() *Config { func GetConfig() *Config {
return config return config
} }