refactor: config
This commit is contained in:
@@ -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
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user