Ver Fonte

Merge pull request #52 from OliveTin/issue-38-config-flag

feature: issue #38 - Config Directories
James Read há 4 anos atrás
pai
commit
5115aa1142
1 ficheiros alterados com 11 adições e 1 exclusões
  1. 11 1
      cmd/OliveTin/main.go

+ 11 - 1
cmd/OliveTin/main.go

@@ -1,6 +1,8 @@
 package main
 
 import (
+	"flag"
+
 	log "github.com/sirupsen/logrus"
 
 	grpcapi "github.com/jamesread/OliveTin/internal/grpcapi"
@@ -30,10 +32,18 @@ func init() {
 
 	log.SetLevel(log.DebugLevel) // Default to debug, to catch cfg issues
 
+	var configDir string;
+	flag.StringVar(&configDir, "configdir", ".", "Config directory path")
+	flag.Parse()
+
+	log.WithFields(log.Fields{
+		"value": configDir,
+	}).Debugf("Value of -configdir flag")
+
 	viper.AutomaticEnv()
 	viper.SetConfigName("config.yaml")
 	viper.SetConfigType("yaml")
-	viper.AddConfigPath(".")
+	viper.AddConfigPath(configDir)
 	viper.AddConfigPath("/config") // For containers.
 	viper.AddConfigPath("/etc/OliveTin/")