Skip to content

Commit 23795a5

Browse files
committed
feature: ability to split worker and rest api
1 parent a519322 commit 23795a5

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

Diff for: main.go

+13
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,11 @@ import (
1212

1313
func main() {
1414
var health bool
15+
var beatOnly bool
16+
var apiOnly bool
1517
flag.BoolVar(&health, "health", false, "make healthcheck and exit.")
18+
flag.BoolVar(&beatOnly, "beat-only", false, "start only beat worker.")
19+
flag.BoolVar(&apiOnly, "api-only", false, "start only REST API.")
1620
flag.Parse()
1721

1822
if health {
@@ -25,6 +29,15 @@ func main() {
2529
os.Exit(0)
2630
}
2731

32+
if beatOnly {
33+
worker.Start()
34+
os.Exit(0)
35+
}
36+
37+
if apiOnly {
38+
log.Fatal(rest.Start())
39+
}
40+
2841
go worker.Start()
2942
log.Fatal(rest.Start())
3043
}

0 commit comments

Comments
 (0)