-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
97 lines (84 loc) · 2.41 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
package main
import (
"encoding/json"
"errors"
"fmt"
"github.com/gin-gonic/gin"
"github.com/lexkong/log"
"github.com/spf13/pflag"
"github.com/spf13/viper"
"go-apiserver/config"
"go-apiserver/model"
v "go-apiserver/pkg/version"
"go-apiserver/router"
"go-apiserver/router/middleware"
"net/http"
"os"
"time"
)
var (
// 命令行指定配置文件, 通过命令行指定不同的配置文件可以方便的区分不同的环境
cfg = pflag.StringP("config", "c", "", "apiserver config file path")
version = pflag.BoolP("version", "v", false, "show version info.")
)
// 程序入库函数, 主要做配置文件解析,程序初始化, 路由加载
func main() {
pflag.Parse()
if *version {
v := v.Get()
marshalled, err := json.MarshalIndent(&v, "", " ")
if err != nil {
fmt.Printf("%v\n", err)
os.Exit(1)
}
fmt.Println(string(marshalled))
return
}
// 初始化配置
if err := config.Init(*cfg); err != nil {
panic(err)
}
// 连接并最后进行关闭
model.DB.Init()
defer model.DB.Close()
// 设置gin的运行模式 debug/release/test
gin.SetMode(viper.GetString("runmode"))
// 创建engine
g := gin.New()
// 调用router.Load来加载路由
router.Load(
g,
middleware.Logging(), // 日志中间件
middleware.RequestId(), // 请求id中间件
)
// 启动的时候开一个协程验证是否成功
go func() {
if err := pingServer(); err != nil {
log.Fatal("The router has no response, or it might took too long to start up.", err)
}
log.Info("The router has been deployed successfully.")
}()
// 如果提供了http相关的配置则开启https端口
cert := viper.GetString("tls.cert")
key := viper.GetString("tls.key")
if cert != "" && key != "" {
go func() {
log.Infof("Start to listening the incoming requests on https address: %s", viper.GetString("tls.addr"))
log.Info(http.ListenAndServeTLS(viper.GetString("tls.addr"), cert, key, g).Error())
}()
}
log.Infof("Start to listening address: %s...", viper.GetString("addr"))
log.Info(http.ListenAndServe(viper.GetString("addr"), g).Error())
}
// 检查接口正常
func pingServer() error {
for i := 0; i < viper.GetInt("max_ping_count"); i++ {
resp, err := http.Get(viper.GetString("url") + "/sd/health")
if err == nil && resp.StatusCode == 200 {
return nil
}
log.Info("Waiting for the router, retry in 1 second.")
time.Sleep(time.Second)
}
return errors.New("Cannot connect to the router.")
}