How to Build a Health Check Service Registry in Python

import random
import time


class ServiceRegistry:
    def __init__(self):
        self.services = {}

    def register(self, name, address):
        self.services[name] = {
            "address": address,
            "status": "healthy",
            "registered_at": time.time(),
            "checks": 0
        }
        return f"Registered {name} at {address}"

    def deregister(self, name):
        if name in self.services:
            del self.services[name]
            return f"Deregistered {name}"
        return f"Service {name} not found"

    def health_check(self, name):
        if name not in self.services:
            return f"Service {name} not registered"

        service = self.services[name]
        service["checks"] += 1
        is_healthy = random.random() > 0.3
        service["status"] = "healthy" if is_healthy else "unhealthy"
        return f"{name} status: {service['status']} (check #{service['checks']})"

    def list_services(self):
        return list(self.services.keys())


if __name__ == "__main__":
    registry = ServiceRegistry()

    print(registry.register("api-gateway", "http://localhost:8080"))
    print(registry.register("auth-service", "http://localhost:8081"))
    print(registry.register("user-service", "http://localhost:8082"))

    print("\nRegistered services:", registry.list_services())
    print("\nFirst health check:")
    print(registry.health_check("api-gateway"))
    print(registry.health_check("auth-service"))

    print("\nSecond health check:")
    print(registry.health_check("api-gateway"))
    print(registry.health_check("api-gateway"))

    print("\n" + registry.deregister("user-service"))
    print("Remaining services:", registry.list_services())

    print("\n" + registry.health_check("user-service"))