-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.go
109 lines (101 loc) · 2.53 KB
/
setup.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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package setup
import (
"github.com/hayuzi/blogserver/global"
"github.com/hayuzi/blogserver/internal/model"
"github.com/hayuzi/blogserver/pkg/logger"
"github.com/hayuzi/blogserver/pkg/setting"
"github.com/hayuzi/blogserver/pkg/tracer"
"github.com/hayuzi/blogserver/pkg/transfer"
"gopkg.in/natefinch/lumberjack.v2"
"log"
"strings"
"time"
)
func Init(config string) {
err := setupSetting(config)
if err != nil {
log.Fatalf("init.setupSetting err: %v", err)
}
err = setupDBEngine()
if err != nil {
log.Fatalf("init.setupDBEngine err: %v", err)
}
err = setupLogger()
if err != nil {
log.Fatalf("init.setupLogger err: %v", err)
}
err = setupTracer()
if err != nil {
log.Fatalf("init.setupTracer err: %v", err)
}
err = setupGinValidateTrans()
if err != nil {
log.Fatalf("init.setupGinValidateTrans err: %v", err)
}
}
func setupSetting(configPath string) error {
st, err := setting.NewSetting(strings.Split(configPath, ",")...)
if err != nil {
return err
}
err = st.ReadSection("Server", &global.ServerSetting)
if err != nil {
return err
}
err = st.ReadSection("App", &global.AppSetting)
if err != nil {
return err
}
err = st.ReadSection("DataBase", &global.DatabaseSetting)
if err != nil {
return err
}
err = st.ReadSection("JWT", &global.JWTSetting)
if err != nil {
return err
}
err = st.ReadSection("Jaeger", &global.JaegerSetting)
if err != nil {
return err
}
global.ServerSetting.ReadTimeout *= time.Second
global.ServerSetting.WriteTimeout *= time.Second
global.JWTSetting.Expire *= time.Minute
return nil
}
func setupDBEngine() error {
var err error
global.DBEngine, err = model.NewDBEngine(global.DatabaseSetting)
if err != nil {
return err
}
return nil
}
func setupLogger() error {
ljLogger := &lumberjack.Logger{
Filename: global.AppSetting.LogSavePath + "/" + global.AppSetting.LogFileName + global.AppSetting.LogFileExt,
MaxSize: 600, //最大占用空间为600MB
MaxAge: 10, //最大生存周期为10天
LocalTime: true,
}
// WithCaller(3) 回溯到入口文件
//global.Logger = logger.NewLogger(ljLogger, "", log.LstdFlags).WithCaller(3)
global.Logger = logger.NewLogger(ljLogger, "", log.LstdFlags)
return nil
}
func setupTracer() error {
jaegerTracer, _, err := tracer.NewJaegerTracer(global.ServerSetting.ServiceName, global.JaegerSetting.HostPort)
if err != nil {
return err
}
global.Tracer = jaegerTracer
return nil
}
func setupGinValidateTrans() error {
trans, err := transfer.NewGinValidateTrans()
if err != nil {
return err
}
global.Trans = trans
return nil
}