| Index: common/py_trace_event/py_trace_event/trace_event_impl/decorators.py
|
| diff --git a/common/py_trace_event/py_trace_event/trace_event_impl/decorators.py b/common/py_trace_event/py_trace_event/trace_event_impl/decorators.py
|
| index ab1463d50581d2ea4c4c42573f153b368349307e..dc753f1f61bb0f39fd01eb168b47d707c36441d4 100644
|
| --- a/common/py_trace_event/py_trace_event/trace_event_impl/decorators.py
|
| +++ b/common/py_trace_event/py_trace_event/trace_event_impl/decorators.py
|
| @@ -7,18 +7,19 @@ import time
|
| import functools
|
|
|
| import log
|
| +from py_trace_event import trace_time
|
|
|
|
|
| @contextlib.contextmanager
|
| def trace(name, **kwargs):
|
| category = "python"
|
| - start = time.time()
|
| + start = trace_time.Now()
|
| args_to_log = {key: repr(value) for key, value in kwargs.iteritems()}
|
| log.add_trace_event("B", start, category, name, args_to_log)
|
| try:
|
| yield
|
| finally:
|
| - end = time.time()
|
| + end = trace_time.Now()
|
| log.add_trace_event("E", end, category, name)
|
|
|
| def traced(*args):
|
| @@ -68,12 +69,12 @@ def traced(*args):
|
| name: repr(get_arg_value(name, index, default))
|
| for name, index, default in args_to_log}
|
|
|
| - start = time.time()
|
| + start = trace_time.Now()
|
| log.add_trace_event("B", start, category, name, arg_values)
|
| try:
|
| return func(*args, **kwargs)
|
| finally:
|
| - end = time.time()
|
| + end = trace_time.Now()
|
| log.add_trace_event("E", end, category, name)
|
| return traced_function
|
|
|
|
|