mirrors/dozzle
1
0
Fork 0
mirror of https://github.com/amir20/dozzle.git synced 2025-12-27 07:31:46 +01:00
Code Issues Projects Releases Wiki Activity

Adds viper for env parsing (#21)

Browse Source
This commit is contained in:
Amir Raminfar
2019-05-13 11:23:35 -07:00
committed by GitHub
parent 88ebe6655b
commit 11ed955b37
3 changed files with 43 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
main.go
View File

@@ -8,7 +8,8 @@ import (
"github.com/gobuffalo/packr"
"github.com/gorilla/mux"
log "github.com/sirupsen/logrus"
flag "github.com/spf13/pflag"
"github.com/spf13/pflag"
"github.com/spf13/viper"
"html/template"
"net/http"
"os"
@@ -33,10 +34,18 @@ type handler struct {
}
func init() {
flag.StringVar(&addr, "addr", ":8080", "http service address")
flag.StringVar(&base, "base", "/", "base address of the application to mount")
flag.StringVar(&level, "level", "info", "logging level")
flag.Parse()
pflag.String("addr", ":8080", "http service address")
pflag.String("base", "/", "base address of the application to mount")
pflag.String("level", "info", "logging level")
pflag.Parse()
viper.AutomaticEnv()
viper.SetEnvPrefix("DOZZLE")
viper.BindPFlags(pflag.CommandLine)
addr = viper.GetString("addr")
base = viper.GetString("base")
level = viper.GetString("level")
l, _ := log.ParseLevel(level)
log.SetLevel(l)