/
go2o-server.go
102 lines (87 loc) · 2.06 KB
/
go2o-server.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
/**
* Copyright 2014 @ z3q.net.
* name :
* author : jarryliu
* date : 2013-12-16 21:45
* description :
* history :
*/
package main
import (
"flag"
"github.com/jrsix/gof"
"github.com/jrsix/gof/storage"
"go2o/src/app"
"go2o/src/app/daemon"
"go2o/src/cache"
"go2o/src/core"
"os"
"runtime"
"strings"
)
func main() {
var (
ch chan bool = make(chan bool)
confFile string
httpPort int
socketPort int
restPort int
mode string //启动模式: h开启http,s开启socket,a开启所有
debug bool
trace bool
runDaemon bool // 运行daemon
help bool
newApp *core.MainApp
)
flag.IntVar(&socketPort, "port2", 1001, "socket server port")
flag.IntVar(&httpPort, "port", 1002, "web server port")
flag.IntVar(&restPort, "port3", 1003, "rest api port")
flag.StringVar(&mode, "mode", "shr", "boot mode.'h'- boot http service,'s'- boot socket service")
flag.BoolVar(&debug, "debug", false, "enable debug")
flag.BoolVar(&trace, "trace", false, "enable trace")
flag.BoolVar(&help, "help", false, "command usage")
flag.StringVar(&confFile, "conf", "app.conf", "")
flag.BoolVar(&runDaemon, "d", false, "run daemon")
flag.Parse()
if help {
flag.Usage()
return
}
runtime.GOMAXPROCS(runtime.NumCPU())
newApp = core.NewMainApp(confFile)
if !newApp.Init(debug, trace) {
os.Exit(1)
}
if v := newApp.Config().GetInt("server_port"); v != 0 {
httpPort = v
}
if v := newApp.Config().GetInt("socket_port"); v != 0 {
socketPort = v
}
if v := newApp.Config().GetInt("api_service_port"); v != 0 {
restPort = v
}
gof.CurrentApp = newApp
app.Init(newApp)
cache.Initialize(storage.NewRedisStorage(newApp.Redis()))
core.RegisterTypes()
var booted bool
if runDaemon {
go daemon.Run(newApp)
}
if strings.Contains(mode, "s") {
booted = true
go app.RunSocket(newApp, socketPort, debug, trace)
}
if strings.Contains(mode, "h") {
booted = true
go app.RunWeb(newApp, httpPort, debug, trace)
}
if strings.Contains(mode, "r") {
booted = true
go app.RunRestApi(newApp, restPort)
}
if booted {
<-ch
}
}