Log starts and completions

datahub-project · Feb 4, 2022 · cba2eba · cba2eba
1 parent c8922b3
commit cba2eba
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/metadata-ingestion/src/datahub/telemetry/telemetry.py b/metadata-ingestion/src/datahub/telemetry/telemetry.py
@@ -171,8 +171,9 @@ def ping(
 def with_telemetry(func: Callable[..., T]) -> Callable[..., T]:
     @wraps(func)
     def wrapper(*args: Any, **kwargs: Any) -> Any:
+        telemetry_instance.ping(func.__module__, func.__name__, "started")
         res = func(*args, **kwargs)
-        telemetry_instance.ping(func.__module__, func.__name__)
+        telemetry_instance.ping(func.__module__, func.__name__, "completed")
         return res
 
     return wrapper