"""Temporal client 辅助函数。

API 和 worker 都需要连接 Temporal Server。
这里做了一个简单的单例缓存，避免重复建立连接。
"""

import asyncio

from temporalio.client import Client

from app.config.settings import get_settings

_client: Client | None = None
_client_lock = asyncio.Lock()


async def get_temporal_client() -> Client:
    """返回缓存后的 Temporal Client。

    第一次调用时才真正连接 Temporal；后续复用同一个 client。
    """

    global _client
    if _client is not None:
        return _client

    async with _client_lock:
        # 双重检查，避免并发场景下重复 connect。
        if _client is None:
            settings = get_settings()
            _client = await Client.connect(
                settings.temporal_address,
                namespace=settings.temporal_namespace,
            )
    return _client


def set_temporal_client(client: Client | None) -> None:
    """覆盖缓存的 Temporal Client。

    主要用于测试场景，把真实连接替换成 Temporal 测试环境里的 client。
    """

    global _client
    _client = client