Function get_async_clickhouse

app/api/agentops/api/db/clickhouse_client.py:83–99

FastAPI dependency to get the async ClickHouse client. This allows for proper dependency injection and easier testing. Returns: AsyncClient: The async ClickHouse client instance

()

async def get_async_clickhouse():
    """
    FastAPI dependency to get the async ClickHouse client.
    This allows for proper dependency injection and easier testing.

    Returns:
        AsyncClient: The async ClickHouse client instance
    """
    global async_clickhouse

    # Check again inside the lock to prevent race conditions
    if async_clickhouse is None:
        async with _async_clickhouse_lock:
            # Check again inside the lock to prevent race conditions
            if async_clickhouse is None:
                async_clickhouse = await get_async_client(**ConnectionConfig.to_connection_dict())
    return async_clickhouse


# Annotated dependencies for better type hinting