Python library for GoReplay Middleware
To install GorMW, simply:
$ pip install gor
or from source:
python setup.py install
Initialize a AsyncioGor based middleware and start it in the following way:
from gor.middleware import AsyncioGor proxy = AsyncioGor() proxy.run()
Basic idea is that you write callbacks which respond to request, response, replay, or message events, which contains request meta information and actuall http paylod. Depending on your needs you may compare, override or filter incoming requests and responses.
You can respond to the incoming events using on function, by providing callbacks:
def on_request(proxy, msg, **kwargs): # do anything you want with msg # msg is a GorMessage object pass proxy = AsyncioGor() proxy.on('request', on_request) proxy.run()
You can provide request ID as additional argument to on function, which allow you to map related requests and responses. Below is example of middleware which checks that original and replayed response have same HTTP status code. Have a try with the following command and sample middleware.
gor --input-raw :14000 --middleware "/path/to/middleware.py" --output-http-track-response --input-raw-track-response --output-http "http://127.0.0.1:14001"
# coding: utf-8 import sys from gor.middleware import AsyncioGor def on_request(proxy, msg, **kwargs): proxy.on('response', on_response, idx=msg.id, req=msg) def on_response(proxy, msg, **kwargs): proxy.on('replay', on_replay, idx=kwargs['req'].id, req=kwargs['req'], resp=msg) def on_replay(proxy, msg, **kwargs): replay_status = proxy.http_status(msg.http) resp_status = proxy.http_status(kwargs['resp'].http) if replay_status != resp_status: sys.stderr.write('replay status [%s] diffs from response status [%s]\n' % (replay_status, resp_status)) else: sys.stderr.write('replay status is same as response status\n') sys.stderr.flush() if __name__ == '__main__': proxy = AsyncioGor() proxy.on('request', on_request) proxy.run()
Mutiple middleware choices
This library provides multiple middleware to choice, currently includes
AsyncioGor, implements based on python3 asyncio
MultiProcessGor, implements based multi processing
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.