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
|
import os.path
import psutil
import socket
import subprocess
import time
import yatest.common
def find_free_ports(count):
sockets = []
ports = []
for _ in range(count):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(('', 0))
ports.append(sock.getsockname()[1])
sockets.append(sock)
for sock in sockets:
sock.close()
return ports
def start_daemon(command, environment, is_alive_check, pid_file_name, timeout=60, daemon_name=None):
daemon_name = daemon_name or os.path.basename(command[0])
stdout_path = yatest.common.output_path('{}.out.log').format(daemon_name)
stderr_path = yatest.common.output_path('{}.err.log').format(daemon_name)
process = subprocess.Popen(
command,
stdout=open(stdout_path, 'w'),
stderr=open(stderr_path, 'w'),
env=environment)
with open(pid_file_name, 'w') as fout:
fout.write(str(process.pid))
for attempts in range(timeout):
result = process.poll()
if result is not None:
raise RuntimeError(
'Could not launch "{}" with exit code {}\nStdout: {}\nStderr: {}'
.format(daemon_name, result, stdout_path, stderr_path)
)
if is_alive_check():
return
time.sleep(1)
raise RuntimeError(
'Could not launch "{}" for {} seconds\nStdout: {}\nStderr: {}'
.format(daemon_name, timeout, stdout_path, stderr_path)
)
def pid_exists(pid):
try:
if psutil.Process(pid).status() == psutil.STATUS_ZOMBIE:
return False
except psutil.NoSuchProcess:
return False
return True
def stop_daemon(pid, signal=15):
pid = int(pid)
if not pid_exists(pid):
return False
os.kill(pid, signal)
while pid_exists(pid):
time.sleep(1)
return True
|
