Scheduler

config.go

@@ -0,0 +1,54 @@
+package main
+
+import (
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/viper"
+)
+
+var config struct {
+	ServerUrl string
+	Token     string
+	Retries   int
+	Workers   int
+}
+
+const (
+	ConfServerUrl = "server_url"
+	ConfToken     = "token"
+	ConfRetries   = "retries"
+	ConfWorkers   = "workers"
+)
+
+func prepareConfig() {
+	viper.SetDefault(ConfRetries, 3)
+	viper.SetDefault(ConfWorkers, 50)
+}
+
+func readConfig() {
+	viper.AddConfigPath(".")
+	viper.SetConfigName("config")
+	err := viper.ReadInConfig()
+	if err != nil {
+		logrus.Fatal(err)
+	}
+
+	config.ServerUrl = viper.GetString(ConfServerUrl)
+	if config.ServerUrl == "" {
+		logrus.Fatal("config: server_url not set!")
+	}
+
+	config.Token = viper.GetString(ConfToken)
+	if config.Token == "" {
+		logrus.Fatal("config: token not set!")
+	}
+
+	config.Retries = viper.GetInt(ConfRetries)
+	if config.Retries < 0 {
+		config.Retries = 1 << 31
+	}
+
+	config.Workers = viper.GetInt(ConfWorkers)
+	if config.Workers <= 0 {
+		logrus.Fatal("config: illegal value %d for workers!", config.Workers)
+	}
+}