Handle errors on sending messages on the controller

Author: jimi-c

receptor/entrypoints.py

@@ -28,7 +28,7 @@ def run_as_ping(config):
             response = controller.send_directive('receptor:ping', config.ping_recipient, now.isoformat(), sock)
             resp_json = json.loads(response)
             if 'code' in resp_json and resp_json['code'] != 0:
-                sys.stdout.buffer.write(b"Failed to ping node.\n")
+                sys.stdout.buffer.write(b"Failed to ping node: %b\n" % (resp_json['payload'].encode('utf-8'),))
             else:
                 sys.stdout.buffer.write(response + b"\n")
             sys.stdout.flush()

receptor/protocol.py

@@ -1,9 +1,11 @@
-import datetime
 import asyncio
+import datetime
+import functools
+import json
 import logging
 import time
-import json
 import uuid
+
 from collections import deque
 from .messages import envelope
 from .exceptions import ReceptorBufferError
@@ -188,5 +190,28 @@ def data_received(self, data):
             directive=directive,
         )
         # TODO: Persistent registry?
-        self.loop.create_task(self.receptor.router.send(inner_env,
-                                                        expected_response=True))
+        send_task = self.loop.create_task(
+            self.receptor.router.send(
+                inner_env,
+                expected_response=True
+            )
+        )
+        send_task.add_done_callback(
+            functools.partial(self.data_received_callback, inner_env)
+        )
+
+    def data_received_callback(self, inner_env, fut):
+        try:
+            fut.result()
+        except Exception as e:
+            self.transport.write(
+                json.dumps(
+                    dict(
+                        timestamp=datetime.datetime.utcnow().isoformat(),
+                        in_response_to=inner_env.message_id,
+                        payload=str(e),
+                        code=1,
+                    )
+                ).encode()
+            )
+