openstack/oslo.messaging
Code Issues Proposed changes

Merge "Topic/server arguments changed in simulator.py"

Browse Source
This commit is contained in:
Jenkins 2016-01-18 11:52:10 +00:00 committed by Gerrit Code Review
commit e9abc3e5da
1 changed files with 29 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
tools/simulator.py
View File

@ -16,6 +16,7 @@ eventlet.monkey_patch()
import argparse
import collections
import datetime
import itertools
import logging
import os
import random
@ -242,6 +243,16 @@ def rpc_server(transport, target, wait_before_answer, executor, show_stats,
server.wait()
def spawn_rpc_clients(threads, transport, targets,
*args, **kwargs):
p = eventlet.GreenPool(size=threads)
targets = itertools.cycle(targets)
for i in range(0, threads):
target = targets.next()
p.spawn_n(send_msg, i, transport, target, *args, **kwargs)
p.waitall()
def threads_spawner(threads, method, *args, **kwargs):
p = eventlet.GreenPool(size=threads)
for i in range(0, threads):
@ -319,11 +330,16 @@ def main():
parser.add_argument('-d', '--debug', dest='debug', type=bool,
default=False,
help="Turn on DEBUG logging level instead of WARN")
parser.add_argument('-tp', '--topic', dest='topic',
default="profiler_topic",
help="Topic to publish/receive messages to/from.")
parser.add_argument('-tp', '--topics', dest='topics', nargs="+",
default=["profiler_topic"],
help="Topics to publish/receive messages to/from.")
parser.add_argument('-s', '--server', dest='server',
default="profiler_server",
help="Server to publish/receive messages to/from.")
parser.add_argument('-l', dest='duration', type=int,
help='send messages for certain time')
parser.add_argument('--config-file', dest='config_file', type=str,
help="Oslo messaging config file")
subparsers = parser.add_subparsers(dest='mode',
help='notify/rpc server/client mode')
@ -371,6 +387,9 @@ def main():
_setup_logging(is_debug=args.debug)
if args.config_file:
cfg.CONF(["--config-file", args.config_file])
if args.mode in ['rpc-server', 'rpc-client']:
transport = messaging.get_transport(cfg.CONF, url=args.url)
else:
@ -378,7 +397,6 @@ def main():
url=args.url)
cfg.CONF.oslo_messaging_notifications.topics = "notif"
cfg.CONF.oslo_messaging_notifications.driver = "messaging"
target = messaging.Target(topic=args.topic, server='profiler_server')
# oslo.config defaults
cfg.CONF.heartbeat_interval = 5
@ -386,6 +404,7 @@ def main():
cfg.CONF.project = 'oslo.messaging'
if args.mode == 'rpc-server':
target = messaging.Target(topic=args.topics[0], server=args.server)
if args.url.startswith('zmq'):
cfg.CONF.rpc_zmq_matchmaker = "redis"
transport._driver.matchmaker._redis.flushdb()
@ -402,9 +421,11 @@ def main():
init_msg(args.messages)
start = datetime.datetime.now()
threads_spawner(args.threads, send_msg, transport, target,
args.wait_after_msg, args.timeout, args.is_cast,
args.messages, args.duration)
targets = [messaging.Target(topic=topic, server=args.server) for topic
in args.topics]
spawn_rpc_clients(args.threads, transport, targets,
args.wait_after_msg, args.timeout, args.is_cast,
args.messages, args.duration)
time_elapsed = (datetime.datetime.now() - start).total_seconds()
msg_count = 0
@ -419,7 +440,7 @@ def main():
log_msg = '%s bytes were sent for %d seconds. Bandwidth is %d b/s' % (
total_bytes, time_elapsed, (total_bytes / time_elapsed))
LOG.info(log_msg)
with open('./oslo_res_%s.txt' % args.topic, 'a+') as f:
with open('./oslo_res_%s.txt' % args.server, 'a+') as f:
f.write(log_msg + '\n')
LOG.info("calls finished, wait %d seconds" % args.exit_wait)