/
main.go
137 lines (115 loc) · 3.05 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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package main
import (
"context"
"flag"
"fmt"
"log"
"net/http"
"net/url"
"os"
"os/signal"
"strings"
"time"
"github.com/e2u/goboot"
"github.com/e2u/mcd/cache"
"github.com/gorilla/mux"
"github.com/urfave/negroni"
)
const (
Version = "1.0.0"
)
var (
ListenPort int
RunEnv string
WL *WhiteList // 白名单
WLModTime time.Time // 白名单最后更新时间
Cache cache.Storager // 源文件缓存
TL *TrustServer //信任服务器列表
)
type Controller struct {
}
// 初始化缓存
func initCache() error {
Cache = cache.NewFileStorage(goboot.Config.MustString("file.storage.path", "/tmp/mcd"))
return nil
}
// 初始化白名单
func initWhitelist() error {
WL = NewWhiteList()
wlfile := goboot.Config.MustString("resources.whitelist")
go func() {
for {
if s, err := os.Stat(wlfile); err == nil && !s.ModTime().Equal(WLModTime) {
WL.LoadFromFile(wlfile)
WLModTime = s.ModTime()
goboot.Log.Infof("whitelist %v reload.", wlfile)
}
time.Sleep(10 * time.Second)
}
}()
return nil
}
// 初始化信任服务器列表
func initTrustServerList() error {
TL = NewTrustServer()
tls := goboot.Config.MustStringArray("trust.server.list", ",")
for _, tl := range tls {
func() {
as := strings.Split(tl, "=")
u, err := url.Parse(as[1])
if err != nil {
panic(err.Error())
}
TL.Set(as[0], u)
}()
}
return nil
}
// 初始化函数
func init() {
flag.StringVar(&RunEnv, "env", "dev", "app run env: [dev|dev-prod|prod]")
flag.IntVar(&ListenPort, "port", 9000, "http listen port: [9000|9001]")
flag.Parse()
goboot.Init(RunEnv)
goboot.OnAppStart(initWhitelist, 10)
goboot.OnAppStart(initCache, 20)
goboot.OnAppStart(initTrustServerList, 30)
goboot.Startup()
}
func main() {
if false {
return
}
goboot.Log.Infof("run mode: %v\n", RunEnv)
c := &Controller{}
// n := negroni.Classic()
n := negroni.New()
n.Use(negroni.HandlerFunc(headerMiddleware))
n.Use(negroni.HandlerFunc(corsRequestMiddleware))
r := mux.NewRouter()
r.KeepContext = false
r.HandleFunc("/{rcType:(?:js|css)}", c.MergeHandler).Methods("GET", "POST", "HEAD", "OPTIONS")
r.HandleFunc("/update", c.UpdateHandler).Methods("GET", "POST")
r.HandleFunc("/tags", c.Tags)
r.HandleFunc("/scss", c.CSSSpriteHandler).Methods("GET", "POST", "HEAD", "OPTIONS")
r.HandleFunc("/scss-image", c.CSSSpriteImageHandler).Methods("GET", "POST", "HEAD", "OPTIONS")
r.HandleFunc("/scss-review", c.CSSSpriteReviewHandler).Methods("GET", "POST", "HEAD", "OPTIONS")
n.UseHandler(r)
srv := &http.Server{
Handler: n,
Addr: fmt.Sprintf("0.0.0.0:%d", ListenPort),
// Good practice: enforce timeouts for servers you create!
WriteTimeout: 15 * time.Second,
ReadTimeout: 15 * time.Second,
}
go func() {
log.Fatal(srv.ListenAndServe())
}()
stopChan := make(chan os.Signal)
signal.Notify(stopChan, os.Interrupt)
<-stopChan // wait for SIGINT
log.Println("Shutting down server...")
ctx, _ := context.WithTimeout(context.Background(), 15*time.Second)
log.Fatal(srv.Shutdown(ctx))
log.Println("Server gracefully stopped")
}