diff options
Diffstat (limited to 'test/bench/benchmark.py')
-rw-r--r-- | test/bench/benchmark.py | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/test/bench/benchmark.py b/test/bench/benchmark.py new file mode 100644 index 00000000..8d208088 --- /dev/null +++ b/test/bench/benchmark.py @@ -0,0 +1,51 @@ +import asyncio +import cProfile +from mitmproxy import ctx + + +class Benchmark: + """ + A simple profiler addon. + """ + def __init__(self): + self.pr = cProfile.Profile() + self.started = False + + async def procs(self): + ctx.log.error("starting benchmark") + backend = await asyncio.create_subprocess_exec("devd", "-q", "-p", "10001", ".") + traf = await asyncio.create_subprocess_exec( + "wrk", + "-c50", + "-d5s", + "http://localhost:%s/benchmark.py" % ctx.master.server.address[1], + stdout=asyncio.subprocess.PIPE + ) + stdout, _ = await traf.communicate() + open(ctx.options.benchmark_save_path + ".bench", mode="wb").write(stdout) + ctx.log.error(stdout.decode("ascii")) + backend.kill() + ctx.master.shutdown() + + def load(self, loader): + loader.add_option( + "benchmark_save_path", + str, + "/tmp/profile", + "Destination for the .prof and and .bench result files" + ) + ctx.options.update( + mode="reverse:http://devd.io:10001", + ) + self.pr.enable() + + def running(self): + if not self.started: + self.started = True + asyncio.get_event_loop().create_task(self.procs()) + + def done(self): + self.pr.dump_stats(ctx.options.benchmark_save_path + ".prof") + + +addons = [Benchmark()]
\ No newline at end of file |