/
config.go
76 lines (61 loc) · 1.45 KB
/
config.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
/*
Copyright 2020 Adevinta
*/
package main
import (
"io/ioutil"
"os"
"github.com/BurntSushi/toml"
"github.com/adevinta/vulnerability-db-api/pkg/storage/postgresql"
"github.com/labstack/gommon/log"
)
type config struct {
API apiConfig `toml:"api"`
PSQL postgresql.ConnStr `toml:"postgresql"`
PSQLRead postgresql.ConnStr `toml:"postgresql_read"`
Log logConfig `toml:"log"`
}
type apiConfig struct {
MaxSize int `toml:"max_size"`
DefaultSize int `toml:"default_size"`
}
type logConfig struct {
Level string `toml:"level"`
}
func parseConfig(cfgFilePath string) (*config, error) {
cfgFile, err := os.Open(cfgFilePath)
if err != nil {
return nil, err
}
defer cfgFile.Close()
cfgData, err := ioutil.ReadAll(cfgFile)
var conf config
if _, err := toml.Decode(string(cfgData[:]), &conf); err != nil {
return nil, err
}
if envVar := os.Getenv("VULNERABILITYDBAPI_DB_HOST"); envVar != "" {
conf.PSQL.Host = envVar
}
if envVar := os.Getenv("VULNERABILITYDBAPI_DB_PORT"); envVar != "" {
conf.PSQL.Port = envVar
}
if envVar := os.Getenv("VULNERABILITYDBAPI_DB_USER"); envVar != "" {
conf.PSQL.User = envVar
}
if envVar := os.Getenv("VULNERABILITYDBAPI_DB_NAME"); envVar != "" {
conf.PSQL.DB = envVar
}
return &conf, nil
}
func parseLogLvl(lvl string) log.Lvl {
switch lvl {
case "ERROR":
return log.ERROR
case "WARN":
return log.WARN
case "DEBUG":
return log.DEBUG
default:
return log.INFO
}
}