From f37efecd0a22313eacad33251512ef371557fde7 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Wed, 8 Apr 2015 20:43:59 +0200 Subject: add yappi benchmark tool --- test/tools/yappibench.py | 51 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) create mode 100644 test/tools/yappibench.py (limited to 'test/tools/yappibench.py') diff --git a/test/tools/yappibench.py b/test/tools/yappibench.py new file mode 100644 index 00000000..b9e4e41d --- /dev/null +++ b/test/tools/yappibench.py @@ -0,0 +1,51 @@ +# Profile mitmdump with apachebench and +# yappi (https://code.google.com/p/yappi/) +# +# Requirements: +# - Apache Bench "ab" binary +# - pip install click yappi + +from libmproxy.main import mitmdump +from os import system +from threading import Thread +import time + +import yappi +import click + +class ApacheBenchThread(Thread): + def __init__(self, concurrency): + self.concurrency = concurrency + super(ApacheBenchThread, self).__init__() + + def run(self): + time.sleep(2) + system("ab -n 1024 -c {} -X 127.0.0.1:8080 http://example.com/".format(self.concurrency)) + +@click.command() +@click.option('--profiler', default="yappi", type=click.Choice(['yappi'])) +@click.option('--clock-type', default="cpu", type=click.Choice(['wall', 'cpu'])) +@click.option('--concurrency', default=1, type=click.INT) +def main(profiler, clock_type, concurrency): + + outfile = "callgrind.mitmdump-{}-c{}".format(clock_type, concurrency) + a = ApacheBenchThread(concurrency) + a.start() + + if profiler == "yappi": + yappi.set_clock_type(clock_type) + yappi.start(builtins=True) + + print("Start mitmdump...") + mitmdump(["-k","-q","-S", "1024example"]) + print("mitmdump stopped.") + + print("Save profile information...") + if profiler == "yappi": + yappi.stop() + stats = yappi.get_func_stats() + stats.save(outfile, type='callgrind') + print("Done.") + +if __name__ == '__main__': + main() \ No newline at end of file -- cgit v1.2.3