tyler
/
armbian-router
1
0
Fork 0
Code Issues Pull Requests Projects Releases 17 Wiki Activity

Add config path flag for non-docker running

This commit is contained in:
Tyler 2022-03-30 21:33:35 -04:00
parent c9593a5a0d
commit 8b2d22a93b
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
main.go
View File

@ -1,6 +1,7 @@
package main package main
import ( import (
"flag"
"github.com/chi-middleware/logrus-logger" "github.com/chi-middleware/logrus-logger"
"github.com/go-chi/chi/v5" "github.com/go-chi/chi/v5"
"github.com/oschwald/maxminddb-golang" "github.com/oschwald/maxminddb-golang"
@ -41,6 +42,10 @@ type City struct {
} `maxminddb:"location"` } `maxminddb:"location"`
} }
var (
configFlag = flag.String("config", "", "configuration file path")
)
func main() { func main() {
viper.SetDefault("bind", ":8080") viper.SetDefault("bind", ":8080")
@ -49,7 +54,12 @@ func main() {
viper.AddConfigPath("/etc/dlrouter/") // path to look for the config file in viper.AddConfigPath("/etc/dlrouter/") // path to look for the config file in
viper.AddConfigPath("$HOME/.dlrouter") // call multiple times to add many search paths viper.AddConfigPath("$HOME/.dlrouter") // call multiple times to add many search paths
viper.AddConfigPath(".") // optionally look for config in the working directory viper.AddConfigPath(".") // optionally look for config in the working directory
err := viper.ReadInConfig() // Find and read the config file
if *configFlag != "" {
viper.SetConfigFile(*configFlag)
}
err := viper.ReadInConfig() // Find and read the config file
if err != nil { // Handle errors reading the config file if err != nil { // Handle errors reading the config file
log.WithError(err).Fatalln("Unable to load config file") log.WithError(err).Fatalln("Unable to load config file")