From a4c63accf1645bea1dfb591700f9daee970fcea8 Mon Sep 17 00:00:00 2001
From: Victor Chang <mocsharp@users.noreply.github.com>
Date: Thu, 5 Oct 2023 01:49:03 -0700
Subject: [PATCH] Print duration of callback app (#885)

Signed-off-by: Victor Chang <vicchang@nvidia.com>
---
 src/TaskManager/CallbackApp/app.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/TaskManager/CallbackApp/app.py b/src/TaskManager/CallbackApp/app.py
index ec1806b63..1455dd453 100644
--- a/src/TaskManager/CallbackApp/app.py
+++ b/src/TaskManager/CallbackApp/app.py
@@ -15,6 +15,7 @@
 import pika
 import uuid
 
+from time import perf_counter as pc
 
 def main():
     parser = argparse.ArgumentParser()
@@ -50,4 +51,9 @@ def main():
     print('Message sent.')
 
 if __name__ == "__main__":
-    main()
+    t0 = pc()
+    try:
+        main()
+    except Exception as e:
+        print(e)
+    print(f"Duration: {pc()-t0}")