-
Notifications
You must be signed in to change notification settings - Fork 0
/
outputs.go
61 lines (48 loc) · 1.21 KB
/
outputs.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
package main
import (
"fmt"
"io"
"github.com/davecgh/go-spew/spew"
)
type Outputter interface {
Print(nodes []*RedisInstance)
}
type debugOutputter struct {
writer io.Writer
}
func newDebugOutputter(writer io.Writer) Outputter {
return &debugOutputter{
writer: writer,
}
}
func (d *debugOutputter) Print(nodes []*RedisInstance) {
spew.Dump(nodes)
}
type graphvizOutputter struct {
writer io.Writer
}
func newGraphvizOutputter(writer io.Writer) Outputter {
return &graphvizOutputter{
writer: writer,
}
}
func (d *graphvizOutputter) Print(nodes []*RedisInstance) {
d.writer.Write([]byte("digraph redis {\n"))
for _, node := range nodes {
switch node.Role {
case Master:
fmt.Fprintf(d.writer, "\t%q[color=red];\n", node.Address)
case Slave:
fmt.Fprintf(d.writer, "\t%q[color=yellow];\n", node.Address)
case Sentinel:
fmt.Fprintf(d.writer, "\t%q[color=blue];\n", node.Address)
}
if node.Master != nil {
fmt.Fprintf(d.writer, "\t%q -> %q[label=\"REPLICAOF\"];\n", node.Address, node.Master.Address)
}
for _, sentinelNode := range node.SentinelMasters {
fmt.Fprintf(d.writer, "\t%q -> %q[label=\"WATCHES\"];\n", node.Address, sentinelNode.Address)
}
}
d.writer.Write([]byte("}\n"))
}