Repository navigation
Expand file tree
/
Copy pathconfig.go
More file actions
98 lines (90 loc) · 2.9 KB
/
Copy pathconfig.go
File metadata and controls
98 lines (90 loc) · 2.9 KB
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package tests
import (
"bytes"
"fmt"
"gin-web/pkg/global"
"github.com/piupuer/go-helper/ms"
"github.com/piupuer/go-helper/pkg/constant"
"github.com/piupuer/go-helper/pkg/log"
"github.com/piupuer/go-helper/pkg/tracing"
"github.com/piupuer/go-helper/pkg/utils"
"github.com/pkg/errors"
"github.com/spf13/viper"
"os"
"strings"
)
const (
configType = "yml"
developmentConfig = "config.dev.yml"
stagingConfig = "config.stage.yml"
productionConfig = "config.prod.yml"
)
var ctx = tracing.NewId(nil)
func Config() {
if os.Getenv("TEST_CONF") == "" {
panic("[unit test]check environment TEST_CONF")
}
confDir := os.Getenv("TEST_CONF")
var box ms.ConfBox
box.Dir = confDir
global.ConfBox = box
v := viper.New()
// read development config as default config
readConfig(box, v, developmentConfig)
settings := v.AllSettings()
for index, setting := range settings {
v.SetDefault(index, setting)
}
env := strings.ToLower(os.Getenv(fmt.Sprintf("%s_MODE", global.ProProdName)))
configName := ""
if env == constant.Stage {
configName = stagingConfig
} else if env == constant.Prod {
configName = productionConfig
} else {
env = constant.Dev
}
global.Mode = env
if configName != "" {
readConfig(box, v, configName)
}
// unmarshal to global.Conf
if err := v.Unmarshal(&global.Conf); err != nil {
panic(errors.Wrapf(err, "[unit test]initialize config failed, config env: TEST_CONF: %s", box.Dir))
}
// read env to global.Conf: config.yml system.port => CFG_SYSTEM_PORT
envPrefix := strings.ToUpper(os.Getenv(fmt.Sprintf("%s_ENV", global.ProEnvName)))
if envPrefix == "" {
envPrefix = "CFG"
}
utils.EnvToInterface(
utils.WithEnvObj(&global.Conf),
utils.WithEnvPrefix(envPrefix),
)
// remove suffix "_"
if strings.TrimSpace(global.Conf.Mysql.TablePrefix) != "" && strings.HasSuffix(global.Conf.Mysql.TablePrefix, "_") {
global.Conf.Mysql.TablePrefix = strings.TrimSuffix(global.Conf.Mysql.TablePrefix, "_")
}
// change default logger
log.DefaultWrapper = log.NewWrapper(log.New(
log.WithCategory(global.Conf.Logs.Category),
log.WithLevel(global.Conf.Logs.Level),
log.WithJson(global.Conf.Logs.Json),
log.WithLineNumPrefix(global.RuntimeRoot),
log.WithLineNum(!global.Conf.Logs.LineNum.Disable),
log.WithLineNumLevel(global.Conf.Logs.LineNum.Level),
log.WithLineNumVersion(global.Conf.Logs.LineNum.Version),
log.WithLineNumSource(global.Conf.Logs.LineNum.Source),
))
log.Info("[unit test]initialize config success, config env: `TEST_CONF: %s`", box.Dir)
}
func readConfig(box ms.ConfBox, v *viper.Viper, configFile string) {
v.SetConfigType(configType)
config := box.Get(configFile)
if len(config) == 0 {
panic(fmt.Sprintf("[unit test]initialize config failed, config env: `TEST_CONF: %s`", box.Dir))
}
if err := v.ReadConfig(bytes.NewReader(config)); err != nil {
panic(errors.Wrapf(err, "[unit test]initialize config failed, config env: `TEST_CONF: %s`", box.Dir))
}
}