From 74ab1eff68fdad4183ce2a7490cfbc9ac1f4513f Mon Sep 17 00:00:00 2001 From: Pjotr Prins Date: Tue, 21 Jun 2016 09:10:40 +0000 Subject: Benchmarking: allow output select with LOG_BENCH --- wqflask/utility/benchmark.py | 29 ++++++++++++++++------------- wqflask/utility/logger.py | 2 +- wqflask/utility/tools.py | 1 + 3 files changed, 18 insertions(+), 14 deletions(-) (limited to 'wqflask/utility') diff --git a/wqflask/utility/benchmark.py b/wqflask/utility/benchmark.py index d5b32703..887b3d88 100644 --- a/wqflask/utility/benchmark.py +++ b/wqflask/utility/benchmark.py @@ -3,34 +3,37 @@ from __future__ import print_function, division, absolute_import import collections import inspect import time +from utility.tools import LOG_BENCH class Bench(object): entries = collections.OrderedDict() - def __init__(self, name=None): + def __init__(self, name=None, write_output=True): self.name = name + self.write_output = write_output def __enter__(self): - if self.name: - print("Starting benchmark: %s" % (self.name)) - else: - print("Starting benchmark at: %s [%i]" % (inspect.stack()[1][3], inspect.stack()[1][2])) + if self.write_output: + if self.name: + print("Starting benchmark: %s" % (self.name)) + else: + print("Starting benchmark at: %s [%i]" % (inspect.stack()[1][3], inspect.stack()[1][2])) self.start_time = time.time() def __exit__(self, type, value, traceback): - if self.name: - name = self.name - else: - name = "That" - time_taken = time.time() - self.start_time - print(" %s took: %f seconds" % (name, (time_taken))) + if self.write_output: + if self.name: + name = self.name + else: + name = "That" + + print(" %s took: %f seconds" % (name, (time_taken))) if self.name: Bench.entries[self.name] = Bench.entries.get(self.name, 0) + time_taken - @classmethod def report(cls): total_time = sum((time_taken for time_taken in cls.entries.itervalues())) @@ -42,4 +45,4 @@ class Bench(object): def reset(cls): """Reset the entries""" - cls.entries = collections.OrderedDict() \ No newline at end of file + cls.entries = collections.OrderedDict() diff --git a/wqflask/utility/logger.py b/wqflask/utility/logger.py index d56b134c..d1324dd5 100644 --- a/wqflask/utility/logger.py +++ b/wqflask/utility/logger.py @@ -81,7 +81,7 @@ class GNLogger: if fun: result = fun(sqlcommand) if LOG_SQL: - self.debug(result) + self.info(result) return result def collect(self,fun,*args): diff --git a/wqflask/utility/tools.py b/wqflask/utility/tools.py index fca5917e..c1b635e0 100644 --- a/wqflask/utility/tools.py +++ b/wqflask/utility/tools.py @@ -171,6 +171,7 @@ WEBSERVER_MODE = get_setting('WEBSERVER_MODE') LOG_LEVEL = get_setting('LOG_LEVEL') LOG_LEVEL_DEBUG = get_setting_int('LOG_LEVEL_DEBUG') LOG_SQL = get_setting_bool('LOG_SQL') +LOG_BENCH = get_setting_bool('LOG_BENCH') USE_REDIS = get_setting_bool('USE_REDIS') USE_GN_SERVER = get_setting_bool('USE_GN_SERVER') -- cgit v1.2.3