-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
107 lines (101 loc) · 2.7 KB
/
main.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
package main
import (
"fmt"
"github.com/gin-gonic/gin"
"github.com/mogfee/blog-server/global"
"github.com/mogfee/blog-server/internal/model"
"github.com/mogfee/blog-server/internal/routers"
"github.com/mogfee/blog-server/pkg/logger"
"github.com/mogfee/blog-server/pkg/setting"
"github.com/mogfee/blog-server/pkg/tracer"
"github.com/opentracing/opentracing-go"
"gopkg.in/natefinch/lumberjack.v2"
"io"
"log"
"net/http"
"os"
"time"
)
func init() {
if err := setupSetting(); err != nil {
log.Fatalf("init.setupSetting err:%v", err)
}
if err := setupDBEngine(); err != nil {
log.Fatalf("init.setupDBEngine err:%v", err)
}
if err := setupLogger(); err != nil {
log.Fatalf("init.setupLogger err:%v", err)
}
if err := setupTracer(); err != nil {
log.Fatalf("init.setupTracer err:%v", err)
}
}
func main() {
gin.SetMode(global.ServerSetting.RunModel)
router := routers.NewRouter()
s := http.Server{
Addr: ":" + global.ServerSetting.HttpPort,
Handler: router,
ReadTimeout: global.ServerSetting.ReadTimeout,
WriteTimeout: global.ServerSetting.WriteTimeout,
MaxHeaderBytes: 1 << 20,
}
log.Fatal(s.ListenAndServe())
}
func setupSetting() error {
settings, err := setting.NewSetting()
if err != nil {
return err
}
if err := settings.ReadSection("Server", &global.ServerSetting); err != nil {
return err
}
if err := settings.ReadSection("App", &global.AppSetting); err != nil {
return err
}
if err := settings.ReadSection("Database", &global.DatabaseSetting); err != nil {
return err
}
if err := settings.ReadSection("Jaeger", &global.JaegerSetting); err != nil {
return err
}
if err := settings.ReadSection("Monitor", &global.MonitorSetting); err != nil {
return err
}
global.ServerSetting.ReadTimeout *= time.Second
global.ServerSetting.WriteTimeout *= time.Second
global.DatabaseSetting.ConnMaxLifetime *= time.Second
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 {
var out io.Writer
if global.ServerSetting.RunModel != "debug" {
out = &lumberjack.Logger{
Filename: global.AppSetting.LogSavePath + "/" + global.AppSetting.LogFileName + global.AppSetting.LogFileExt,
MaxSize: 600,
MaxAge: 10,
LocalTime: true,
}
} else {
out = os.Stdout
}
global.Logger = logger.NewLogger(out, "", log.LstdFlags)
return nil
}
func setupTracer() error {
tracer, _, err := tracer.NewJaegerTracer(global.AppSetting.ServerName, fmt.Sprintf("%s:%d", global.JaegerSetting.Host, global.JaegerSetting.Port))
if err != nil {
return err
}
opentracing.SetGlobalTracer(tracer)
global.Tracer = tracer
return nil
}