/
hive.go
80 lines (64 loc) · 1.32 KB
/
hive.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
package main
import (
"fmt"
"github.com/jtanderson/hive/drone"
"github.com/jtanderson/hive/queen"
"github.com/olekukonko/tablewriter"
"log"
"os"
"os/signal"
"syscall"
)
func main() {
if len(os.Args) == 1 {
showHelp()
return
}
command := os.Args[1]
switch command {
case "queen":
runQueen()
return
case "drone":
runDrone()
return
}
fmt.Println("That command was not recognized.")
showHelp()
}
func runDrone() {
d := new(drone.Drone)
drone.StartService(d)
ch := make(chan os.Signal)
signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM)
log.Println(<-ch)
log.Println("Stopping drone...")
drone.StopService(d)
}
func runQueen() {
q := new(queen.Queen)
/* Delete this madness eventually */
q.EnlistDrone("127.0.0.1:1234")
d := q.GetDrone(0)
args := &drone.Args{7, 8}
var reply string
err := d.GetClient().Call("Drone.MyCall", args, &reply)
if err != nil {
log.Fatal("dialing:", err)
}
fmt.Println("Called MyCall to get:", reply)
/* ------------------- */
showEnlisted(q)
}
func showEnlisted(q *queen.Queen) {
table := tablewriter.NewWriter(os.Stdout)
table.SetHeader([]string{"Id", "Address", "Status"})
for _, v := range q.GetDrones() {
table.Append([]string{v.Id, v.Address, ""})
}
table.Render()
}
func showHelp() {
fmt.Println("Usage:")
fmt.Println("\thive command [options]")
}