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
|
package cli
import (
"fmt"
"log"
"net/rpc"
"os"
"os/signal"
"syscall"
"time"
tm "github.com/buger/goterm"
"github.com/fatih/color"
"github.com/spf13/cobra"
"github.com/spiral/errors"
"github.com/spiral/roadrunner/v2/plugins/informer"
"github.com/spiral/roadrunner/v2/tools"
)
// use interactive mode
var interactive bool
const InformerList string = "informer.List"
const InformerWorkers string = "informer.Workers"
func init() {
workersCommand := &cobra.Command{
Use: "workers",
Short: "Show information about active roadrunner workers",
RunE: workersHandler,
}
workersCommand.Flags().BoolVarP(
&interactive,
"interactive",
"i",
false,
"render interactive workers table",
)
root.AddCommand(workersCommand)
}
func workersHandler(_ *cobra.Command, args []string) error {
const op = errors.Op("workers cobra handler")
// get RPC client
client, err := RPCClient()
if err != nil {
return err
}
defer func() {
err := client.Close()
if err != nil {
log.Printf("error when closing RPCClient: error %v", err)
}
}()
var plugins []string
// assume user wants to show workers from particular plugin
if len(args) != 0 {
plugins = args
} else {
err = client.Call(InformerList, true, &plugins)
if err != nil {
return errors.E(op, err)
}
}
if !interactive {
return showWorkers(plugins, client)
}
// https://golang.org/pkg/os/signal/#Notify
// should be of buffer size at least 1
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt, syscall.SIGTERM, syscall.SIGINT)
tm.Clear()
tt := time.NewTicker(time.Second)
defer tt.Stop()
for {
select {
case <-c:
return nil
case <-tt.C:
tm.MoveCursor(1, 1)
err := showWorkers(plugins, client)
if err != nil {
return errors.E(op, err)
}
tm.Flush()
}
}
}
func showWorkers(plugins []string, client *rpc.Client) error {
const op = errors.Op("show workers")
for _, plugin := range plugins {
list := &informer.WorkerList{}
err := client.Call(InformerWorkers, plugin, &list)
if err != nil {
return errors.E(op, err)
}
fmt.Printf("Workers of [%s]:\n", color.HiYellowString(plugin))
tools.WorkerTable(os.Stdout, list.Workers).Render()
}
return nil
}
|
