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 |