-
Notifications
You must be signed in to change notification settings - Fork 2
/
motoboto_benchmark_subprocess_main.py
93 lines (75 loc) · 2.62 KB
/
motoboto_benchmark_subprocess_main.py
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
# -*- coding: utf-8 -*-
"""
motoboto_benchmark_subprocess_main.py
Manage
"""
import logging
import os
import os.path
import signal
import subprocess
import sys
from threading import Event
from common import parse_command_line, initialize_logging
def _create_signal_handler(halt_event):
def __signal_handler(*_args):
halt_event.set()
return __signal_handler
def main():
"""
main processing module
"""
options = parse_command_line()
initialize_logging(options.log_name)
log = logging.getLogger("main")
log.info("program starts")
halt_event = Event()
signal.signal(signal.SIGTERM, _create_signal_handler(halt_event))
log.info("using test script %r" % (options.test_script, ))
program_dir = os.path.dirname(__file__)
program_path = os.path.join(program_dir, "customer_process.py")
customer_process_list = list()
for file_name in os.listdir(options.user_identity_dir):
if options.max_users is not None \
and len(customer_process_list) >= options.max_users:
log.info("breaking at %s users" % (options.max_users, ))
break
log.info("user identity %r" % (file_name, ))
user_identity_path = os.path.join(options.user_identity_dir, file_name)
args = [
sys.executable,
program_path,
options.test_script,
user_identity_path
]
environment = {
"PYTHONPATH" : os.environ["PYTHONPATH"],
"NIMBUSIO_LOG_DIR" : os.environ["NIMBUSIO_LOG_DIR"],
"NIMBUS_IO_SERVICE_HOST" : os.environ["NIMBUS_IO_SERVICE_HOST"],
"NIMBUS_IO_SERVICE_PORT" : os.environ["NIMBUS_IO_SERVICE_PORT"],
"NIMBUS_IO_SERVICE_DOMAIN" : \
os.environ["NIMBUS_IO_SERVICE_DOMAIN"],
"NIMBUS_IO_SERVICE_SSL" : os.environ.get(
"NIMBUS_IO_SERVICE_SSL", "0"
)
}
process = subprocess.Popen(args, env=environment)
customer_process_list.append(process)
log.info("waiting")
try:
halt_event.wait(options.test_duration)
except KeyboardInterrupt:
log.info("KeyBoardInterrupt")
halt_event.set()
log.info("terminating processes")
for process in customer_process_list:
process.terminate()
log.info("waiting for processes")
for process in customer_process_list:
process.wait()
if process.returncode != 0:
log.error("process returncode %s" % (process.returncode, ))
log.info("program ends")
return 0
if __name__ == "__main__":
sys.exit(main())