diff options
author | Maximilian Hils <git@maximilianhils.com> | 2015-05-31 01:51:13 +0200 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2015-05-31 01:51:13 +0200 |
commit | 0f8e8ee1ad667c8d747d5876f9b630ecbc01d7e9 (patch) | |
tree | 54fbd572b3a00af966f7c4348d64ac660b614d8f /test/tools/benchtool.py | |
parent | 26923dd0e41fe9d5db6fa0677ee0bc627f5a7db2 (diff) | |
download | mitmproxy-0f8e8ee1ad667c8d747d5876f9b630ecbc01d7e9.tar.gz mitmproxy-0f8e8ee1ad667c8d747d5876f9b630ecbc01d7e9.tar.bz2 mitmproxy-0f8e8ee1ad667c8d747d5876f9b630ecbc01d7e9.zip |
add bench tool for windows
Diffstat (limited to 'test/tools/benchtool.py')
-rw-r--r-- | test/tools/benchtool.py | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/test/tools/benchtool.py b/test/tools/benchtool.py new file mode 100644 index 00000000..5cfcba30 --- /dev/null +++ b/test/tools/benchtool.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="none", type=click.Choice(['none', '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 |