-
Notifications
You must be signed in to change notification settings - Fork 7
/
main.go
114 lines (101 loc) · 2.76 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
/*
* Copyright (C) 2020 The ontology Authors
* This file is part of The ontology library.
*
* The ontology is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* The ontology is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with The ontology. If not, see <http://www.gnu.org/licenses/>.
*/
package main
import (
"flag"
"fmt"
"github.com/ontio/layer2/operator/cmd"
"github.com/ontio/layer2/operator/config"
"github.com/ontio/layer2/operator/core"
"github.com/ontio/layer2/operator/log"
"github.com/urfave/cli"
"os"
"os/signal"
"runtime"
"syscall"
)
var ConfigPath string
var mgr *core.Layer2Operator
func init() {
flag.StringVar(&ConfigPath, "cfg", config.DEFAULT_CONFIG_FILE_NAME, "config file of server")
}
func setupApp() *cli.App {
app := cli.NewApp()
app.Usage = "Ontology Layer2 Service"
app.Action = startServer
app.Version = config.Version
app.Copyright = "Copyright in 2019 The Ontology Authors"
app.Flags = []cli.Flag{
cmd.LogLevelFlag,
cmd.ConfigPathFlag,
}
app.Commands = []cli.Command{
}
app.Before = func(context *cli.Context) error {
runtime.GOMAXPROCS(runtime.NumCPU())
return nil
}
return app
}
func startServer(ctx *cli.Context) {
// get all cmd flag
logLevel := ctx.GlobalInt(cmd.GetFlagName(cmd.LogLevelFlag))
log.InitLog(logLevel, log.PATH, log.Stdout)
configPath := ctx.GlobalString(cmd.GetFlagName(cmd.ConfigPathFlag))
if configPath != "" {
ConfigPath = configPath
}
// read config
servConfig := config.NewServiceConfig(ConfigPath)
if servConfig == nil {
log.Errorf("startServer - create config failed!")
return
}
initOperatorServer(servConfig)
waitToExit()
}
func waitToExit() {
exit := make(chan bool, 0)
sc := make(chan os.Signal, 1)
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP)
go func() {
for sig := range sc {
log.Infof("waitToExit - Layer2 Operator received exit signal:%v.", sig.String())
mgr.Stop()
close(exit)
break
}
}()
<-exit
}
func initOperatorServer(servConfig *config.ServiceConfig) {
var err error
mgr, err = core.NewLayer2Operator(servConfig)
if err != nil || mgr == nil {
log.Error("initOperatorServer failed!")
return
}
mgr.Start()
}
func main() {
log.Infof("main - Layer2 Operator Starting...")
if err := setupApp().Run(os.Args); err != nil {
fmt.Fprintln(os.Stderr, err)
os.Exit(1)
}
}