def instrument_meinheld_server(module):
def wrap_wsgi_application_entry_point(application, *args, **kwargs):
application = WSGIApplicationWrapper(application)
args = [application] + list(args)
return (args, kwargs)
wrap_in_function(module, 'run', wrap_wsgi_application_entry_point)
def instrument_meinheld_server(module):
def wrap_wsgi_application_entry_point(application, *args, **kwargs):
application = WSGIApplicationWrapper(application)
args = [application] + list(args)
return (args, kwargs)
wrap_in_function(module, 'run', wrap_wsgi_application_entry_point)
def instrument_piston_doc(module):
def in_HandlerMethod_init(self, method, *args, **kwargs):
if isinstance(method, MethodWrapper):
method = method._bw_wrapped
return ((self, method) + args, kwargs)
wrap_in_function(module, 'HandlerMethod.__init__', in_HandlerMethod_init)
def instrument_cherrypy_wsgiserver_wsgiserver2(module):
def wrap_wsgi_application_entry_point(server, bind_addr, application,
*args, **kwargs):
application = WSGIApplicationWrapper(application)
args = [server, bind_addr, application] + list(args)
return (args, kwargs)
wrap_in_function(module, 'CherryPyWSGIServer.__init__',
wrap_wsgi_application_entry_point)
def instrument_wsgiref_simple_server(module):
def wrap_wsgi_application_entry_point(server, application,
*args, **kwargs):
application = WSGIApplicationWrapper(application)
args = [server, application] + list(args)
return (args, kwargs)
wrap_in_function(module, 'WSGIServer.set_app', wrap_wsgi_application_entry_point)
def instrument(module):
if module.__name__ == 'web.application':
wrap_out_function(module, 'application.wsgifunc', WSGIApplicationWrapper)
wrap_in_function(module, 'application._delegate', transaction_name_delegate)
wrap_pre_function(module, 'application.internalerror', wrap_handle_exception)
elif module.__name__ == 'web.template':
wrap_function_trace(module, 'render.__getattr__', template_name, 'Template/Render')
def instrument_paste_httpserver(module):
def wrap_wsgi_application_entry_point(server, application, *args,
**kwargs):
application = WSGIApplicationWrapper(application)
args = [server, application] + list(args)
return (args, kwargs)
wrap_in_function(module, 'WSGIServerBase.__init__',
wrap_wsgi_application_entry_point)
def instrument_paste_httpserver(module):
def wrap_wsgi_application_entry_point(server, application,
*args, **kwargs):
application = WSGIApplicationWrapper(application)
args = [server, application] + list(args)
return (args, kwargs)
wrap_in_function(module, 'WSGIServerBase.__init__', wrap_wsgi_application_entry_point)
def instrument_cherrypy_wsgiserver_wsgiserver2(module):
def wrap_wsgi_application_entry_point(server, bind_addr, application,
*args, **kwargs):
application = WSGIApplicationWrapper(application)
args = [server, bind_addr, application] + list(args)
return (args, kwargs)
wrap_in_function(module, 'CherryPyWSGIServer.__init__', wrap_wsgi_application_entry_point)
def instrument_gevent_wsgi(module):
def wrapper_WSGIServer___init__(*args, **kwargs):
def _bind_params(self, listener, application, *args, **kwargs):
return self, listener, application, args, kwargs
self, listener, application, _args, _kwargs = _bind_params(
*args, **kwargs)
application = WSGIApplicationWrapper(application)
_args = (self, listener, application) + _args
return _args, _kwargs
wrap_in_function(module, 'WSGIServer.__init__',
wrapper_WSGIServer___init__)
def instrument_gevent_pywsgi(module):
def wrapper_WSGIServer___init__(*args, **kwargs):
def _bind_params(self, listener, application, *args, **kwargs):
return self, listener, application, args, kwargs
self, listener, application, _args, _kwargs = _bind_params(
*args, **kwargs)
application = WSGIApplicationWrapper(application)
_args = (self, listener, application) + _args
return _args, _kwargs
wrap_in_function(module, 'WSGIServer.__init__',
wrapper_WSGIServer___init__)
def instrument_flup_server_cgi(module):
wrap_in_function(module, 'WSGIServer.__init__', wrap_wsgi_application_entry_point)
def instrument_django_core_servers_basehttp(module):
# Allow 'runserver' to be used with Django <= 1.3. To do
# this we wrap the WSGI application argument on the way in
# so that the run() method gets the wrapped instance.
#
# Although this works, if anyone wants to use it and make
# it reliable, they may need to first need to patch Django
# as explained in the ticket:
#
# https://code.djangoproject.com/ticket/16241
#
# as the Django 'runserver' is not WSGI compliant due to a
# bug in its handling of errors when writing response.
#
# The way the agent now uses a weakref dictionary for the
# transaction object may be enough to ensure the prior
# transaction is cleaned up properly when it is deleted,
# but not absolutely sure that will always work. Thus is
# still a risk of error on subsequent request saying that
# there is an active transaction.
#
# TODO Later versions of Django use the wsgiref server
# instead which will likely need to be dealt with via
# instrumentation of the wsgiref module or some other means.
import django
framework = ('Django', django.get_version())
def wrap_wsgi_application_entry_point(server, application, **kwargs):
return ((
server,
WSGIApplicationWrapper(application, framework='Django'),
), kwargs)
# XXX Because of risk of people still trying to use the
# inbuilt Django development server and since the code is
# not going to be changed, could just patch it to fix
# problem and the instrumentation we need.
if (not hasattr(module, 'simple_server')
and hasattr(module.ServerHandler, 'run')):
# Patch the server to make it work properly.
def run(self, application):
try:
self.setup_environ()
self.result = application(self.environ, self.start_response)
self.finish_response()
except Exception:
self.handle_error()
finally:
self.close()
def close(self):
if self.result is not None:
try:
self.request_handler.log_request(
self.status.split(' ', 1)[0], self.bytes_sent)
finally:
try:
if hasattr(self.result, 'close'):
self.result.close()
finally:
self.result = None
self.headers = None
self.status = None
self.environ = None
self.bytes_sent = 0
self.headers_sent = False
# Leaving this out for now to see whether weakref solves
# the problem.
#module.ServerHandler.run = run
#module.ServerHandler.close = close
# Now wrap it with our instrumentation.
wrap_in_function(module, 'ServerHandler.run',
wrap_wsgi_application_entry_point)
def instrument_flup_server_scgi_base(module):
wrap_in_function(module, 'BaseSCGIServer.__init__', wrap_wsgi_application_entry_point)
def instrument_django_core_servers_basehttp(module):
# Allow 'runserver' to be used with Django <= 1.3. To do
# this we wrap the WSGI application argument on the way in
# so that the run() method gets the wrapped instance.
#
# Although this works, if anyone wants to use it and make
# it reliable, they may need to first need to patch Django
# as explained in the ticket:
#
# https://code.djangoproject.com/ticket/16241
#
# as the Django 'runserver' is not WSGI compliant due to a
# bug in its handling of errors when writing response.
#
# The way the agent now uses a weakref dictionary for the
# transaction object may be enough to ensure the prior
# transaction is cleaned up properly when it is deleted,
# but not absolutely sure that will always work. Thus is
# still a risk of error on subsequent request saying that
# there is an active transaction.
#
# TODO Later versions of Django use the wsgiref server
# instead which will likely need to be dealt with via
# instrumentation of the wsgiref module or some other means.
import django
framework = ('Django', django.get_version())
def wrap_wsgi_application_entry_point(server, application, **kwargs):
return ((server, WSGIApplicationWrapper(application,
framework='Django'),), kwargs)
# XXX Because of risk of people still trying to use the
# inbuilt Django development server and since the code is
# not going to be changed, could just patch it to fix
# problem and the instrumentation we need.
if (not hasattr(module, 'simple_server') and
hasattr(module.ServerHandler, 'run')):
# Patch the server to make it work properly.
def run(self, application):
try:
self.setup_environ()
self.result = application(self.environ, self.start_response)
self.finish_response()
except Exception:
self.handle_error()
finally:
self.close()
def close(self):
if self.result is not None:
try:
self.request_handler.log_request(
self.status.split(' ',1)[0], self.bytes_sent)
finally:
try:
if hasattr(self.result,'close'):
self.result.close()
finally:
self.result = None
self.headers = None
self.status = None
self.environ = None
self.bytes_sent = 0
self.headers_sent = False
# Leaving this out for now to see whether weakref solves
# the problem.
#module.ServerHandler.run = run
#module.ServerHandler.close = close
# Now wrap it with our instrumentation.
wrap_in_function(module, 'ServerHandler.run',
wrap_wsgi_application_entry_point)
def instrument_flup_server_cgi(module):
wrap_in_function(module, 'WSGIServer.__init__',
wrap_wsgi_application_entry_point)
def instrument_flup_server_scgi_base(module):
wrap_in_function(module, 'BaseSCGIServer.__init__',
wrap_wsgi_application_entry_point)
