123COMMENT

2025-09-30 14:08:04 +05:30
parent 379fb67aae
commit a81fdb6dd4
19 changed files with 1917 additions and 0 deletions
@@ -0,0 +1,154 @@
+"""
+
+    AUTHOR:
+
+        Khushal P Soonderji
+
+    DATE:
+
+        Saturday, 28th Dec., 2024
+
+    OBJECTIVE:
+
+        To provide a quick way to test out passthrough messages over Kafka.
+
+    REFERENCES:
+
+        N/A
+
+    DOWNLOADS:
+
+        N/A
+
+"""
+
+
+#   *****************************************************************************************************************
+#  *****                                                                                                          ****
+# ***                                                   IMPORT                                                      ***
+#  *****                                                                                                          ****
+#   *****************************************************************************************************************
+
+
+# To make sibling directories accessible for imports:
+import sys
+sys.path.append(".")
+sys.path.append("..")
+
+# System-level:
+import os
+
+# Utils:
+from utils_v2.string import json
+from utils_v2.system import files
+from utils_v2.date_time import date_time
+from utils_v2.queue.async_kafka import ProducerKafka, ConsumerKafka, get_ssl_context
+
+# For async activities:
+import asyncio
+
+# Common:
+from shared import constants
+
+# For random choices:
+import random
+
+
+#   *****************************************************************************************************************
+#  *****                                                                                                          ****
+# ***                                           MACROS / ONE-TIME INIT                                              ***
+#  *****                                                                                                          ****
+#   *****************************************************************************************************************
+
+
+# --- Nothing Yet
+
+
+#   *****************************************************************************************************************
+#  *****                                                                                                          ****
+# ***                                                VARIABLES                                                      ***
+#  *****                                                                                                          ****
+#   *****************************************************************************************************************
+
+
+# --- Nothing Yet
+
+
+#   *****************************************************************************************************************
+#  *****                                                                                                          ****
+# ***                                                  FUNCTIONS                                                    ***
+#  *****                                                                                                          ****
+#   *****************************************************************************************************************
+
+
+# --- Nothing Yet
+
+
+#   *****************************************************************************************************************
+#  *****                                                                                                          ****
+# ***                                               MAIN PROGRAM                                                    ***
+#  *****                                                                                                          ****
+#   *****************************************************************************************************************
+
+
+if __name__ == "__main__":
+
+    # Define the test params:
+    TOPIC = "socket-io-bcast"
+    BOOTSTRAP_SERVERS = "del.ditscentre.in:9092"
+    cwd = files.get_cwd()
+    pdir = files.get_parent_directory(cwd, depth = 2)
+    SSL_CONTEXT = ssl_context = get_ssl_context(
+        ca_file = os.path.join(pdir, "creds", "kafka", "cert_authority.pem"),
+        cert_file = os.path.join(pdir, "creds", "kafka", "fullchain.pem"),
+        key_file = os.path.join(pdir, "creds", "kafka", "privkey.pem")
+    )
+
+    async def keep_producing():
+
+        # Create the producer:
+        my_producer = ProducerKafka(
+            topic = TOPIC,
+            bootstrap_servers = BOOTSTRAP_SERVERS,
+            security_protocol = "SSL",
+            ssl_context = SSL_CONTEXT
+        )
+
+        # Create a message for the producer to produce:
+        strategy_message = {
+            "to": "FC9O3N75Ax7B1v4NAAAD",
+            "event": "Strategy",
+            "namespace": "/finstitutions/trading",
+            "data": {
+                "message": "Your strategy ABC says buy XYZ.",
+                "playSound": True
+            }
+        }
+
+        corporate_action_message = {
+            "to": "FC9O3N75Ax7B1v4NAAAD",
+            "event": "Corporate Action",
+            "namespace": "/finstitutions/trading",
+            "data": {
+                "message": "Stock ABC has a corporate action tomorrow.",
+                "playSound": True,
+                "date": "29-Dec-2024",
+                "action": "Extraordinary Board Meeting"
+            }
+        }
+
+        # Keep sending the message in intervals:
+        while True:
+            producer_message = random.choice([
+                strategy_message,
+                corporate_action_message
+            ])
+            success = await my_producer.produce(producer_message)
+            print("PRODUCED:", success)
+            await asyncio.sleep(1.0)
+
+    async def main():
+        await asyncio.gather(*[keep_producing()])
+
+
+    asyncio.run(main())