def execute_sql(self, *args, **kwargs):
if not hasattr(threadlocal, 'initialized'):
init_threadlocal(threadlocal)
if not threadlocal.sql_plugin_data:
return self.execute_sql_default(*args, **kwargs)
try:
q, params = self.as_sql()
except:
return self.execute_sql_default(*args, **kwargs)
start = datetime.now()
try:
return self.execute_sql_default(*args, **kwargs)
finally:
end = datetime.now()
time_taken = (end - start).total_seconds() * 1000.0
threadlocal.sql_plugin_data.insert(
query=q,
params=params,
start=start,
end=end,
time_taken=time_taken,
using=self.using,
stack=get_stack()
)
def wrapped_method(self, *args, **kwargs):
if not hasattr(threadlocal, 'initialized'):
init_threadlocal(threadlocal)
if not threadlocal.memcache_plugin_data:
return func(self, *args, **kwargs)
func_name = func.__name__
arg = None
if len(args) > 0:
arg = args[0]
if isinstance(arg, dict):
arg = list(arg.keys())
start = datetime.datetime.now()
try:
return func(self, *args, **kwargs)
finally:
end = datetime.datetime.now()
time_taken = (end - start).total_seconds() * 1000.0
threadlocal.memcache_plugin_data.insert(
func_name=func_name,
arg=arg,
start=start,
end=end,
time_taken=time_taken,
stack=get_stack()
)
def log(*args):
s = ' '.join([(arg if isinstance(arg, str) else repr(arg))
for arg in args])
if not hasattr(threadlocal, 'initialized'):
return
if threadlocal.log_plugin_data:
stack = get_stack()
threadlocal.log_plugin_data.log(s, stack)
def req(self, api, *args, **kwargs):
if not hasattr(threadlocal, 'initialized'):
init_threadlocal(threadlocal)
if not threadlocal.thriftpy_plugin_data:
return orig_req(self, api, *args, **kwargs)
start = datetime.now()
try:
return orig_req(self, api, *args, **kwargs)
finally:
end = datetime.now()
time_taken = (end - start).total_seconds() * 1000.0
threadlocal.thriftpy_plugin_data.insert(
service_name=self._service.__name__,
func_name=api,
args=args,
kwargs=kwargs,
start=start,
end=end,
time_taken=time_taken,
stack=get_stack()
)