Raw API
Client raw_query メソッド
Client.raw_query メソッドを使用すると、クライアント接続を通じて ClickHouse の HTTP クエリインターフェイスを直接利用できます。戻り値は未処理の bytes オブジェクトです。このメソッドは、パラメータバインディング、エラーハンドリング、再試行、設定管理を最小限のインターフェイスで扱える便利なラッパーを提供します。
結果として返される
bytes オブジェクトの処理は呼び出し元の責任です。なお、Client.query_arrow は、このメソッドを ClickHouse の Arrow 出力フォーマットで利用するごく薄いラッパーにすぎません。
Client raw_stream メソッド
Client.raw_stream メソッドは raw_query と同じ API ですが、バイト chunk の io.IOBase ストリームを返します。処理が完了したら、ストリームを閉じてください。AsyncClient.raw_stream は await して使用し、async with と async for で使用する async StreamContext を返します。
Client raw_insert メソッド
Client.raw_insert メソッドを使うと、クライアント接続を介して bytes オブジェクトまたは bytes オブジェクトを生成するジェネレーターを直接 insert できます。insert payload の処理を行わないため、非常に高いパフォーマンスを発揮します。このメソッドには、settings と insert format を指定するオプションがあります。
insert_block が指定されたフォーマットで、指定された圧縮 method を使用していることを保証する責任は呼び出し元にあります。ClickHouse Connect は、ファイルのアップロードや PyArrow Tables に対してこれらの raw insert を使用し、パースは ClickHouse server に委ねます。
クエリ結果をファイルとして保存する
raw_stream メソッドを使うと、ClickHouse からローカルファイルシステムへファイルを直接ストリーミングできます。たとえば、クエリ結果を CSV ファイルとして保存するには、次のコードスニペットを使用します。
output.csvファイルが生成されます。
マルチスレッド、マルチプロセス、非同期/イベント駆動のユースケース
QueryContext または InsertContext オブジェクトに状態を保持するため、これらのヘルパーオブジェクトはスレッドセーフではなく、複数の処理ストリーム間で共有すべきではありません。コンテキストオブジェクトの詳細については、QueryContexts および InsertContexts の各セクションを参照してください。
さらに、同時に 2 つ以上のクエリや insert が「進行中」のアプリケーションでは、もう 2 つ注意すべき点があります。1 つ目はクエリ/insert に関連付けられる ClickHouse の「session」で、2 つ目は ClickHouse Connect Client のインスタンスで使用される HTTP 接続プールです。
AsyncClient
asyncio アプリケーション向けに aiohttp ベースのネイティブクライアントを提供しています。使用する前に、オプションの依存関係をインストールしてください。
get_async_client を await して、クライアントを作成・初期化します。query、command、insert などの I/Oメソッドはコルーチンです。
get_async_client では、同時実行するコルーチン間でクライアントを共有できるよう、デフォルトで session ID の自動生成が無効になっています。明示的な session_id または autogenerate_session_id=True を渡すのは、session 状態が必要で、かつその session 内で同時実行クエリを行わない場合に限ってください。
ClickHouse session ID の管理
- 複数のクエリに特定の ClickHouse settings を関連付けるため (ユーザー設定を参照) 。ClickHouse の
SETコマンドは、ユーザーsessionの範囲で設定を変更するために使用されます。 - 一時テーブルを追跡するため
Client は生成された session ID を使用します。そのため、SET ステートメントと一時テーブルは、そのクライアントからのリクエスト間で保持されます。async ファクトリーは、デフォルトでは session ID を生成しません。ClickHouse では同じ session 内でクエリを同時実行できず、これを試みるとクライアントは ProgrammingError を送出するため、次のいずれかのパターンを使用してください。
- session の分離が必要な各スレッド / プロセス / イベントハンドラーごとに、個別の
Clientインスタンスを作成します。これにより、クライアントごとの session 状態 (一時テーブルとSET値) が維持されます。 - 共有session状態が不要な場合は、
query、command、またはinsertの呼び出し時にsettings引数を使用して、各クエリに一意のsession_idを指定します。 - 共有クライアントでsessionを無効にするには、クライアントを作成する前に
autogenerate_session_id=Falseを設定します (または、これを直接get_clientに渡します) 。
autogenerate_session_id=False を get_client(...) に直接渡します。
この場合、ClickHouse Connect は session_id を送信しないため、server は個々のリクエストを同じ session に属するものとして扱いません。一時テーブルや session レベルの設定は、リクエストをまたいで保持されません。
HTTP接続プールのカスタマイズ
urllib3 の接続プールを使用します。デフォルトでは、すべてのクライアントインスタンスが同じ接続プールを共有しており、ほとんどのユースケースではこれで十分です。このデフォルトのプールでは、アプリケーションで使用される各 ClickHouse サーバーに対して、最大 8 本の HTTP Keep-Alive 接続が維持されます。
大規模なマルチスレッドアプリケーションでは、接続プールを分けたほうが適切な場合があります。カスタマイズした接続プールは、メインの clickhouse_connect.get_client 関数に pool_mgr キーワード引数として指定できます:
urllib3 PoolManager documentationを参照してください。
async クライアントは urllib3 を使用せず、aiohttp のプールを使用します。設定は、get_async_client の connector_limit、connector_limit_per_host、keepalive_timeout で行います。await async_client.close_connections() を呼び出すと、進行中のリクエストを中断することなくプールが入れ替わります。