Repository navigation
Expand file tree
/
Copy pathserver.py
More file actions
53 lines (39 loc) · 1.58 KB
/
Copy pathserver.py
File metadata and controls
53 lines (39 loc) · 1.58 KB
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
#!/usr/bin/env python3
from utils import *
import atexit
import json
import datetime
class CodisServer(Process):
def __init__(self, port, master_port=None, requirepass=None):
self.config = self._open_config(port, master_port, requirepass)
self.port = port
self.logfile = "redis-{}.log".format(port)
self.command = "codis-server {}".format(self.config)
Process.__init__(self, self.command, self.logfile)
dict = {"port": port, "pid": self.proc.pid}
print(" >> codis.server = " + json.dumps(dict, sort_keys=True))
@staticmethod
def _open_config(port, master_port=None, requirepass=None):
config = 'redis-{}.conf'.format(port)
with open(config, "w+") as f:
f.write('port {}\n'.format(port))
f.write('save ""\n')
f.write('dbfilename "{}.rdb"\n'.format(port))
if master_port is not None:
f.write('slaveof 127.0.0.1 {}\n'.format(master_port))
if requirepass is not None:
f.write('masterauth {}\n'.format(requirepass))
f.write('requirepass {}\n'.format(requirepass))
f.write('protected-mode no\n')
return config
if __name__ == "__main__":
children = []
atexit.register(kill_all, children)
passwd = None
for port in range(16380, 16384):
children.append(CodisServer(port, requirepass=passwd))
children.append(CodisServer(port + 1000, port, requirepass=passwd))
check_alive(children, 3)
while True:
print(datetime.datetime.now())
time.sleep(5)