mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-26 18:18:25 +00:00
b663a224a3
- The benchmark addon now manages setting up and tearing down the backend and traffic processes itself. - Use wrk instead of hey. I get more consistent results with this tool, and hey shows a strange tail-latency bump that seems artificial. - Make termination behaviour simpler. The bencmark revealed a bug where .done events were not called if the proxy was shut down by an addon.
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
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()] |