func main() {
var (
port int
ch chan bool = make(chan bool)
conf string
)
flag.IntVar(&port, "port", 1005, "")
flag.StringVar(&conf, "conf", "app.conf", "")
flag.Parse()
gof.CurrentApp = core.NewMainApp(conf)
dps.Init(gof.CurrentApp)
cache.Initialize(gof.CurrentApp.Storage())
tcpserve.DebugOn = true
go tcpserve.ListenTcp(fmt.Sprintf(":%d", port))
go func(mainCh chan bool) {
ch := make(chan os.Signal)
signal.Notify(ch, syscall.SIGTERM, syscall.SIGKILL)
for {
switch <-ch {
case syscall.SIGKILL, syscall.SIGTERM:
log.Println("[ Tcp][ Term] - tcp serve has term!")
close(mainCh)
}
}
}(ch)
log.Println("[ TCP][ SERVE] - socket is served ... ")
<-ch
}
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)
}
go handleSignal(ch)
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
}
}
