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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
|
"""
Render cluster slot map as ascii art
Block elements: https://en.wikipedia.org/wiki/Block_Elements
U+2588 █ Full block
U+2589 ▉ Left seven eighths block
U+258A ▊ Left three quarters block
U+258B ▋ Left five eighths block
U+258C ▌ Left half block
U+258D ▍ Left three eighths block
U+258E ▎ Left one quarter block
U+258F ▏ Left one eighth block
"""
import logging
UNICODE_MARK = "▏▎▍▌▋▊▉█"
TOTAL_SLOT = 16384
logger = logging.getLogger(__name__)
class SlotCount:
def __init__(self, host, count):
self.host = host
self.count = count
def __repr__(self):
return "{}({})".format(self.host, self.count)
class ColorPlate:
SLOT_COLORS = ["#ff0066", "#44ff00", "#112233", "#ff1100", "#00ffaa"]
def __init__(self):
self.color = {}
self.used_color = 0
def color_for(self, host):
if host in self.color:
return self.color[host]
self.used_color += 1
if self.used_color >= len(self.SLOT_COLORS):
self.used_color = 0
self.color[host] = self.SLOT_COLORS[self.used_color]
return self.color[host]
PLATE = ColorPlate()
def _render_block(block):
"""8 slots are a block"""
previous = []
for slot in block:
if previous and slot == previous[-1].host:
previous[-1].count += 1
else:
previous.append(SlotCount(slot, 1))
different_hosts_len = len(previous)
if different_hosts_len == 1:
return [(PLATE.color_for(previous[0].host), "█")]
if different_hosts_len == 2:
head_host, tail_host = previous[0].host, previous[1].host
head_color, tail_color = PLATE.color_for(head_host), PLATE.color_for(tail_host)
return [
(
"{} bg:{}".format(head_color, tail_color),
UNICODE_MARK[previous[0].count - 1],
)
]
# more than 2 colors
head_host, *middle_host, tail_host = previous
result = []
result.append(
(
"{}".format(PLATE.color_for(head_host.host)),
UNICODE_MARK[head_host.count - 1],
)
)
for host in middle_host + [tail_host]:
result.append(
(
"{}".format(PLATE.color_for(host.host)),
UNICODE_MARK[host.count - 1],
)
)
return result
def render_slot_map(redis_cluster_solts_response):
slot_in_host = [""] * TOTAL_SLOT
for node in redis_cluster_solts_response:
start_slot, end_slot, master_node, *_ = node
host_ip = f"{master_node[0].decode()}:{master_node[1]}"
slot_in_host[start_slot : end_slot + 1] = [host_ip] * (
end_slot - start_slot + 1
)
ascii_art_pairs = [("", " 1 ")]
for index in range(0, TOTAL_SLOT, 8):
if index % (64 * 8) == 0 and index:
ascii_art_pairs.append(("", f" {index:<5}"))
ascii_art_pairs.append(("", "\n"))
ascii_art_pairs.append(("", f"{index+1:5} "))
ascii_art_pairs.extend(_render_block(slot_in_host[index : index + 8]))
ascii_art_pairs.append(("", f" 16384"))
host_color_sample = []
for hostname, color in PLATE.color.items():
host_color_sample.append((f"bg:{color}", " "))
host_color_sample.append((f"", f" -> {hostname}"))
host_color_sample.append(("", "\n"))
return host_color_sample + ascii_art_pairs
if __name__ == "__main__":
print(
_render_block(
[
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7001",
],
ColorPlate(),
)
)
print(
_render_block(
[
"127.0.0.1:7000",
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7002",
"127.0.0.1:7002",
"127.0.0.1:7002",
"127.0.0.1:7002",
],
ColorPlate(),
)
)
print(
_render_block(
[
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7001",
"127.0.0.1:7002",
"127.0.0.1:7002",
"127.0.0.1:7002",
"127.0.0.1:7002",
],
ColorPlate(),
)
)
|
