57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package config
|
|
|
|
import (
|
|
log "github.com/sirupsen/logrus"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
type Config struct {
|
|
Host string
|
|
ClientPort string
|
|
ServerPort string
|
|
LogLevel string
|
|
NumCPUs int
|
|
CheckPeriod int
|
|
TasksConfig TasksConfig
|
|
}
|
|
|
|
type TasksConfig struct {
|
|
RetryCount int
|
|
RetryMinutes int
|
|
}
|
|
|
|
func NewConfig() *Config {
|
|
return &Config{
|
|
Host: getEnv("APP_HOST", "0.0.0.0"),
|
|
ClientPort: getEnv("APP_PORT", "9050"),
|
|
ServerPort: getEnv("APP_SERVER_PORT", "9060"),
|
|
LogLevel: getEnv("APP_LOG_LEVEL", "debug"),
|
|
NumCPUs: getEnvInt("APP_NUMCPUS", -1),
|
|
CheckPeriod: getEnvInt("APP_CHECK_PERIOD", 6),
|
|
|
|
TasksConfig: TasksConfig{
|
|
RetryCount: getEnvInt("TASK_RETRY_COUNT", 3),
|
|
RetryMinutes: getEnvInt("TASK_RETRY_MINUTES", 5),
|
|
},
|
|
}
|
|
}
|
|
|
|
func getEnv(key, fallback string) string {
|
|
if value, ok := os.LookupEnv(key); ok {
|
|
return value
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
func getEnvInt(key string, fallback int) int {
|
|
if value, ok := os.LookupEnv(key); ok {
|
|
num, err := strconv.Atoi(value)
|
|
if err != nil {
|
|
log.WithField("default", -1).Warn("Config | Can't parse value as int")
|
|
return fallback
|
|
}
|
|
return num
|
|
}
|
|
return fallback
|
|
}
|