From f543c74dff4793fead5724be6a8db6683c847859 Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Wed, 12 Aug 2020 13:44:48 +0200
Subject: [PATCH] Implement simple broker debugger to console

---
 core-broker/core-broker/broker_debugger.py | 70 ++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 core-broker/core-broker/broker_debugger.py

diff --git a/core-broker/core-broker/broker_debugger.py b/core-broker/core-broker/broker_debugger.py
new file mode 100644
index 0000000..52b504b
--- /dev/null
+++ b/core-broker/core-broker/broker_debugger.py
@@ -0,0 +1,70 @@
+from datetime import datetime
+import json
+
+from pika import BlockingConnection, ConnectionParameters
+
+DEFAULT_BROKER_URL = "127.0.0.1"
+STATUS_EXCHANGE = "status"
+
+COLOR_END_MARKER = '\x1b[0m'
+
+
+def get_color_for_action(action):
+
+    color_mapping = {
+        "write_request": "\x1b[34;1m",
+        "write_start": "\x1b[1;33;1m",
+        "write_finished": "\x1b[1;32;1m"
+    }
+
+    return color_mapping.get(action, "")
+
+
+def on_status(channel, method_frame, header_frame, body):
+    header = header_frame.headers
+    request = json.loads(body.decode())
+
+    action = header["action"]
+    source = header["source"]
+
+    action_output = get_color_for_action(action) + action + COLOR_END_MARKER
+    time_output = datetime.utcnow().strftime("%Y%m%d-%H:%M:%S.%f")
+
+    print("[%s] %s %s" % (time_output, action_output, source))
+    print(request)
+
+
+def connect_to_broker(broker_url):
+    connection = BlockingConnection(ConnectionParameters(broker_url))
+    channel = connection.channel()
+
+    channel.exchange_declare(exchange=STATUS_EXCHANGE,
+                             exchange_type="fanout")
+    queue = channel.queue_declare(queue="", exclusive=True).method.queue
+    channel.queue_bind(queue=queue,
+                       exchange=STATUS_EXCHANGE)
+
+    channel.basic_consume(queue, on_status)
+
+    try:
+        channel.start_consuming()
+    except KeyboardInterrupt:
+        channel.stop_consuming()
+
+
+def main():
+    import argparse
+    parser = argparse.ArgumentParser(
+        description="Connect and listen to broker events.")
+
+    parser.add_argument('--broker_url', dest='broker_url',
+                        default=DEFAULT_BROKER_URL,
+                        help='RabbitMQ broker URL')
+
+    args = parser.parse_args()
+
+    connect_to_broker(broker_url=args.broker_url)
+
+
+if __name__ == '__main__':
+    main()