import time
import requests

def register_to_consul():
    time.sleep(3)  # 等待服务稳定再注册
    service_name = "data-service"  # 替换为 quant-service/emotion-service 等
    port = 8000  # 替换为 8001, 8002 等

    payload = {
        "Name": service_name,
        "ID": f"{service_name}-1",
        "Address": service_name,  # 容器名作为地址
        "Port": port,
        "Check": {
            "HTTP": f"http://{service_name}:{port}/health",
            "Interval": "10s"
        }
    }

    try:
        requests.put("http://consul:8500/v1/agent/service/register", json=payload)
        print(f"[{service_name}] Registered to Consul")
    except Exception as e:
        print(f"❌ Consul register failed: {e}")