-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
88 lines (77 loc) · 1.98 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
package main
import (
"./config"
"./model"
v "./pkg/version"
"./router"
"./router/middleware"
"encoding/json"
"errors"
"fmt"
"github.com/gin-gonic/gin"
"github.com/lexkong/log"
"github.com/spf13/pflag"
"github.com/spf13/viper"
"net/http"
"os"
"time"
)
var (
cfg = pflag.StringP("config", "c", "", "apiserver config file path.")
version = pflag.BoolP("version", "v", false, "show version info.")
)
func pingServer() error {
for i := 0; i < viper.GetInt("max_ping_count"); i++ {
rsp, err := http.Get(viper.GetString("url") + "/sd/health")
if err == nil && rsp.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.")
}
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
}
// init config
if err := config.Init(*cfg); err != nil {
panic(err)
}
// init db
model.DB.Init()
defer model.DB.Close()
// set gin mode
gin.SetMode(viper.GetString("runmode"))
g := gin.New()
middlewares := []gin.HandlerFunc{
middleware.RequestId(),
}
router.Load(g, middlewares...)
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.")
}()
// Start to listening the incoming requests
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 the incoming requests on http address: %s", viper.GetString("addr"))
log.Info(http.ListenAndServe(viper.GetString("addr"), g).Error())
}