added execution time

nficano · nficano · commit 1e8bb68d13c8 · 2016-04-04T18:34:55.000-04:00
diff --git a/README.rst b/README.rst
@@ -67,7 +67,6 @@ Next let's open ``service.py``, in here you'll find the following function:
         # You code goes here!
         e = event.get('e')
         pi = event.get('pi')
-        print "your test handler was successfully invoked!"
         return e + pi
 
 
@@ -82,10 +81,10 @@ If you now try and run:
 .. code:: bash
 
     (lambduh) $ lambda invoke
-    # "your test handler was successfully invoked!"
-    # below shows the result returned by your function execution:
     # 5.858
 
+    # your code completed in: 0.00000310s
+
 As you probably put together, the ``lambda invoke`` command grabs the values stored in the ``event.json`` file and passes them to your function.
 
 The ``event.json`` file should help you develop your Lambda service locally. You can specify an alternate ``event.json`` file by passing the ``--event-file=<filename>.json`` argument to ``lambda invoke``.
diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
@@ -1,8 +1,9 @@
 # -*- coding: utf-8 -*-
 from __future__ import print_function
 import json
-import os
 import logging
+import os
+import time
 from imp import load_source
 from shutil import copy, copyfile
 from tempfile import mkdtemp
@@ -77,9 +78,12 @@ def invoke(src, alt_event=None):
 
     # TODO: look into mocking the ``context`` variable, currently being passed
     # as None.
+    start = time.time()
     results = fn(event, None)
-    print("below shows the result returned by your function "
-          "execution:\n{0}".format(results))
+    end = time.time()
+    elapsed = end - start
+    print("{0}".format(results))
+    print("\nyour code completed in: {:.8f}s".format(elapsed))
 
 
 def init(src, minimal=False):
diff --git a/aws_lambda/project_template/service.py b/aws_lambda/project_template/service.py
@@ -5,5 +5,4 @@ def handler(event, context):
     # You code goes here!
     e = event.get('e')
     pi = event.get('pi')
-    print "your test handler was successfully invoked!"
     return e + pi
