basic app

config/config.go

@@ -0,0 +1,57 @@
+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
+}