Merge pull request #1 from Yashas-naidu/yash

(Week-1 deliverables) Created 2 code files to collect logs,heartbeat and registration from 3 microservices

Author: dheerajcl

node.py

@@ -0,0 +1,109 @@
+# node.py
+import uuid
+import time
+import threading
+import random
+import json
+from datetime import datetime
+from colorama import Fore, Style, init
+from threading import Lock
+
+# Initialize colorama for colored terminal output
+init(autoreset=True)
+
+# Global lock for synchronized printing
+print_lock = Lock()
+
+class Node:
+    # Colors for message types
+    message_colors = {
+        "registration": Fore.CYAN,
+        "heartbeat": Fore.RED,
+        "Log": Fore.GREEN
+    }
+    # Colors for keys and values within messages
+    key_color = Fore.LIGHTMAGENTA_EX
+    value_color = Fore.LIGHTYELLOW_EX
+
+    def __init__(self, service_name):
+        self.node_id = str(uuid.uuid4())
+        self.service_name = service_name
+        self.status = "UP"
+        self.register_node()
+    
+    def format_message(self, message):
+        """
+        Format JSON message to color keys and values distinctly.
+        """
+        formatted_message = ""
+        for key, value in message.items():
+            # Apply color for keys and values
+            formatted_message += f"{Node.key_color}{key}{Style.RESET_ALL}: {Node.value_color}{value}{Style.RESET_ALL}, "
+        return "{" + formatted_message.rstrip(", ") + "}"
+
+    def print_message(self, message_type, message_content):
+        with print_lock:
+            color = Node.message_colors[message_type]
+            formatted_content = self.format_message(json.loads(message_content))
+            print(color + f"{message_type.capitalize()}:" + Style.RESET_ALL, formatted_content)
+    
+    def register_node(self):
+        registration_message = {
+            "node_id": self.node_id,
+            "message_type": "REGISTRATION",
+            "service_name": self.service_name,
+            "timestamp": datetime.now().isoformat()
+        }
+        self.print_message("registration", json.dumps(registration_message))
+    
+    def generate_log(self, log_level, message, extra_data=None):
+        log_message = {
+            "log_id": str(uuid.uuid4()),
+            "node_id": self.node_id,
+            "log_level": log_level,
+            "message_type": "LOG",
+            "message": message,
+            "service_name": self.service_name,
+            "timestamp": datetime.now().isoformat()
+        }
+        if extra_data:
+            log_message.update(extra_data)
+        self.print_message("Log", json.dumps(log_message))
+
+    def send_heartbeat(self):
+        heartbeat_message = {
+            "node_id": self.node_id,
+            "message_type": "HEARTBEAT",
+            "status": self.status,
+            "timestamp": datetime.now().isoformat()
+        }
+        self.print_message("heartbeat", json.dumps(heartbeat_message))
+
+    def start_heartbeat(self, interval=5):
+        def heartbeat():
+            while self.status == "UP":
+                self.send_heartbeat()
+                time.sleep(interval)
+        threading.Thread(target=heartbeat).start()
+    
+    def start_log_generation(self, interval=3):
+        def generate_logs():
+            log_levels = ["INFO", "WARN", "ERROR"]
+            while self.status == "UP":
+                log_level = random.choice(log_levels)
+                if log_level == "INFO":
+                    self.generate_log("INFO", "This is an info log.")
+                elif log_level == "WARN":
+                    self.generate_log("WARN", "This is a warning log.", {
+                        "response_time_ms": random.randint(100, 500),
+                        "threshold_limit_ms": 300
+                    })
+                elif log_level == "ERROR":
+                    self.generate_log("ERROR", "This is an error log.", {
+                        "error_details": {
+                            "error_code": "500",
+                            "error_message": "Internal Server Error"
+                        }
+                    })
+                time.sleep(interval)
+        threading.Thread(target=generate_logs).start()

run_nodes.py

@@ -0,0 +1,11 @@
+# run_nodes.py
+from node import Node
+
+if __name__ == "__main__":
+    services = ["PaymentService", "OrderService", "InventoryService"]
+    nodes = [Node(service) for service in services]
+    
+    # Start log generation and heartbeat for each node
+    for node in nodes:
+        node.start_heartbeat()
+        node.start_log_generation()