-
Notifications
You must be signed in to change notification settings - Fork 0
/
jj.go
124 lines (107 loc) · 2.28 KB
/
jj.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
package main
import (
"fmt"
"os"
"os/signal"
"strings"
"syscall"
"github.com/jj-io/jj/service"
"github.com/jj-io/jj/service/auth"
"github.com/jj-io/jj/service/mgr"
"github.com/jj-io/jj/service/mq"
"github.com/jj-io/jj/service/notify"
"gopkg.in/logex.v1"
)
var (
srvs = []*service.ServiceType{
{Name: auth.Name, Desc: auth.Desc, New: auth.NewAuthService},
{Name: mgr.Name, Desc: mgr.Desc, New: mgr.NewMgrService},
{Name: notify.Name, Desc: notify.Desc, New: notify.NewNotifyService},
{Name: mq.Name, Desc: mq.Desc, New: mq.NewMqService},
}
)
func usage() {
print(fmt.Sprintf(`usage: %v [service ...] [service config]
service:
%v
service config:
[service].[config]=xx
example:
%[1]v auth op -auth.listen=:1111 -op.listen=:2222
`, os.Args[0], strings.Join(serviceNames(), "\n\t")))
os.Exit(1)
}
func serviceNames() []string {
s := make([]string, len(srvs))
for i := range srvs {
s[i] = srvs[i].Name + ": " + srvs[i].Desc
}
return s
}
func main() {
if len(os.Args[1:]) == 1 && os.Args[1] == "-h" {
usage()
}
hasServices := false
optIdx := -1
for i, srvName := range os.Args[1:] {
if srvName[0] == '-' {
optIdx = i + 1
break
}
found := false
for _, s := range srvs {
if s.Name == srvName {
s.Use = true
hasServices = true
found = true
break
}
}
if !found {
println(fmt.Sprintf("unknown services '%v'", srvName))
usage()
}
}
// open all services if none service is specified
if !hasServices {
for _, s := range srvs {
s.Use = true
}
}
if optIdx > 0 {
for _, opt := range os.Args[optIdx:] {
for _, s := range srvs {
prefix := "-" + s.Name + "."
if strings.HasPrefix(opt, prefix) {
if !s.Use {
println("panic: service", s.Name, "is not used")
os.Exit(1)
}
s.Args = append(s.Args, "-"+opt[len(prefix):])
break
}
}
}
}
for _, s := range srvs {
if s.Use {
s.Ins = s.New(os.Args[0]+" "+s.Name, s.Args)
if i, ok := s.Ins.(service.ServiceIniter); ok {
if err := i.Init(); err != nil {
logex.Error(err)
os.Exit(1)
}
}
}
}
for _, s := range srvs {
if s.Use {
logex.Infof("running service %v", s.Name)
go s.Ins.Run()
}
}
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt, os.Kill, syscall.SIGTERM, syscall.SIGHUP)
<-c
}