scottk/databunker
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/databunker.git synced 2025-10-30 17:37:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for -conf argument

Browse Source
This commit is contained in:
stremovsky
2019-12-15 10:03:03 +02:00
parent d06424f0cc
commit 977940136f
1 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/bunker.go
View File

@@ -185,8 +185,14 @@ func (e mainEnv) setupRouter() *httprouter.Router {
return router
}
func readFile(cfg *Config) error {
f, err := os.Open("databunker.yaml")
func readFile(cfg *Config, filepath *string) error {
confFile := "databunker.yaml"
if filepath != nil {
if len(*filepath) > 0 {
confFile = *filepath
}
}
f, err := os.Open(confFile)
if err != nil {
return err
}
@@ -205,14 +211,17 @@ func readEnv(cfg *Config) error {
func main() {
rand.Seed(time.Now().UnixNano())
lockMemory()
var cfg Config
readFile(&cfg)
readEnv(&cfg)
//fmt.Printf("%+v\n", cfg)
initPtr := flag.Bool("init", false, "a bool")
masterKeyPtr := flag.String("masterkey", "", "master key")
dbPtr := flag.String("db", "", "database file")
confPtr := flag.String("conf", "", "configuration file")
flag.Parse()
var cfg Config
readFile(&cfg, confPtr)
readEnv(&cfg)
var err error
var masterKey []byte
if err != nil {