Add duration of processing request in the response

rpc_server.py

@@ -11,6 +11,7 @@
 import cjson
 import sys
 from pprint import pprint
+import time
 
 host = sys.argv[1]
 user = sys.argv[2]
@@ -34,8 +35,8 @@
 
 
 def on_request(ch, method, props, body):
-
+    st = time.time()
     # print "\nReceived '%s'" % (body,)
     response = cjson.decode(body)
     pprint(response['in'])
     execute(response['in'], response['out'])
@@ -38,7 +39,9 @@
     # print "\nReceived '%s'" % (body,)
     response = cjson.decode(body)
     pprint(response['in'])
     execute(response['in'], response['out'])
+    response['out']['duration'] = time.time() - st
+    response = cjson.encode(response['out'])
     ch.basic_publish(exchange='',
                      routing_key=props.reply_to,
                      properties=pika.BasicProperties(correlation_id=\
@@ -46,7 +49,7 @@
                                                      content_type=\
                                                          'application/json',
                                                      delivery_mode = 1),
-                     body=cjson.encode(response['out']))
+                     body=response)
     ch.basic_ack(delivery_tag=method.delivery_tag)