more formatting

Author: csegarragonz

faasmctl/tasks/invoke.py

@@ -87,8 +87,17 @@ def invoke(
     exec_time = "{:.2f} s".format(
         get_execution_time_from_message_results(result, unit="s")
     )
+    cold_start_time = "{:.2f} ms".format(
+        get_execution_time_from_message_results(
+            result, unit="ms", ext_start_ts=(start_ts * 1e3)
+        )
+    )
 
     if output_format is not None:
+        if output_format == "cold-start":
+            print(cold_start_time[:-3])
+            return 0
+
         if output_format == "exec-time":
             print(exec_time[:-2])
             return 0

faasmctl/util/results.py

@@ -1,4 +1,4 @@
-def get_execution_time_from_message_results(result, unit="s"):
+def get_execution_time_from_message_results(result, unit="s", ext_start_ts=None):
     valid_units = ["s", "ms", "us"]
     if unit not in valid_units:
         print(
@@ -13,6 +13,8 @@ def get_execution_time_from_message_results(result, unit="s"):
         start_ts = min([msg.startTimestamp for msg in result.messageResults])
     except AttributeError:
         start_ts = min([msg.timestamp for msg in result.messageResults])
+    if ext_start_ts is not None:
+        start_ts = ext_start_ts
     end_ts = max([msg.finishTimestamp for msg in result.messageResults])
 
     if unit == "s":