55 lines
940 B
Go
55 lines
940 B
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
"zhub/cli"
|
|
"zhub/conf"
|
|
"zhub/zsub"
|
|
)
|
|
|
|
func main() {
|
|
server := true
|
|
confPath := "app.conf"
|
|
addr := ""
|
|
|
|
for _, arg := range os.Args[1:] {
|
|
if strings.EqualFold(arg, "cli") {
|
|
server = false
|
|
} else if strings.Index(arg, "-d=") == 0 {
|
|
addr = arg[3:]
|
|
} else if strings.Index(arg, "-c=") == 0 {
|
|
confPath = arg[3:]
|
|
}
|
|
}
|
|
conf.Load(confPath)
|
|
if len(addr) == 0 {
|
|
addr = conf.GetStr("service.zhub.servers", "127.0.0.1:1216")
|
|
}
|
|
|
|
if len(os.Args) == 3 && strings.EqualFold(os.Args[1], "-r") {
|
|
if cli, err := cli.Create(addr, "group-admin"); err != nil {
|
|
log.Println(err)
|
|
} else {
|
|
switch os.Args[2] {
|
|
case "timer":
|
|
cli.Cmd("reload-timer")
|
|
case "shutdown":
|
|
cli.Cmd("shutdown")
|
|
}
|
|
cli.Close()
|
|
time.Sleep(time.Millisecond * 10)
|
|
}
|
|
return
|
|
}
|
|
|
|
if server {
|
|
zsub.ServerStart(addr) // 服务进程启动
|
|
} else {
|
|
cli.ClientRun(addr)
|
|
}
|
|
|
|
}
|