Skip to content

maximdanilchenko/aiochclient

Repository files navigation

aiochclient

PyPI version Tests Documentation Status codecov Code style: black

An async http(s) ClickHouse client for python 3.10+ supporting type conversion in both directions, streaming, lazy decoding on select queries, and a fully typed interface.

Table of Contents

Installation

You can use it with either aiohttp or httpx http connectors.

To use with aiohttp install it with command:

> pip install aiochclient[aiohttp]

Or aiochclient[aiohttp-speedups] to install with extra speedups.

To use with httpx install it with command:

> pip install aiochclient[httpx]

Or aiochclient[httpx-speedups] to install with extra speedups.

Installing with [*-speedups] adds the following:

  • faust-cchardet for aiohttp speedup
  • aiodns for aiohttp speedup
  • ciso8601 for ultra-fast datetime parsing while decoding data from ClickHouse for aiohttp and httpx.

Additionally the installation process attempts to use Cython for a speed boost (roughly 30% faster).

Quick Start

Connecting to ClickHouse

aiochclient needs aiohttp.ClientSession or httpx.AsyncClient to connect to ClickHouse:

from aiochclient import ChClient
from aiohttp import ClientSession


async def main():
    async with ClientSession() as s:
        client = ChClient(s)
        assert await client.is_alive()  # returns True if connection is Ok

Querying the database

await client.execute(
    "CREATE TABLE t (a UInt8, b Tuple(Date, Nullable(Float32))) ENGINE = Memory"
)

For INSERT queries you can pass values as *args. Values should be iterables:

await client.execute(
    "INSERT INTO t VALUES",
    (1, (dt.date(2018, 9, 7), None)),
    (2, (dt.date(2018, 9, 8), 3.14)),
)

For fetching all rows at once use the fetch method:

all_rows = await client.fetch("SELECT * FROM t")

For fetching first row from result use the fetchrow method:

row = await client.fetchrow("SELECT * FROM t WHERE a=1")

assert row[0] == 1
assert row["b"] == (dt.date(2018, 9, 7), None)

You can also use fetchval method, which returns first value of the first row from query result:

val = await client.fetchval("SELECT b FROM t WHERE a=2")

assert val == (dt.date(2018, 9, 8), 3.14)

With async iteration on the query results stream you can fetch multiple rows without loading them all into memory at once:

async for row in client.iterate(
        "SELECT number, number*2 FROM system.numbers LIMIT 10000"
):
    assert row[0] * 2 == row[1]

Use fetch/fetchrow/fetchval/iterate for SELECT queries and execute or any of last for INSERT and all another queries.

Working with query results

All fetch queries return rows as lightweight, memory efficient objects. Before v1.0.0 rows were only returned as tuples. All rows have a full mapping interface, where you can get fields by names or indexes:

row = await client.fetchrow("SELECT a, b FROM t WHERE a=1")

assert row["a"] == 1
assert row[0] == 1
assert row[:] == (1, (dt.date(2018, 9, 8), 3.14))
assert list(row.keys()) == ["a", "b"]
assert list(row.values()) == [1, (dt.date(2018, 9, 8), 3.14)]

Binary engines: RowBinary and Native

By default results are encoded/decoded through ClickHouse's text (TSV) format. Two faster binary engines are available, each opt-in via a single flag; both use the Cython extension when it's built and fall back to pure Python otherwise.

native=True — ClickHouse's column-oriented Native format, by far the fastest of aiochclient's engines. Whole columns are decoded at once (fixed-width numerics straight through the stdlib array module, strings/dates in the compiled Cursor), which gets close to the C-extension clients while keeping aiochclient dependency-light — no numpy. INSERTs are encoded column-by-column and streamed as Native blocks, so the server inserts one block while the client encodes the next.

client = ChClient(session, native=True)
await client.execute("INSERT INTO t VALUES", (1, "a"), (2, "b"))
rows = await client.fetch("SELECT * FROM t")

binary=True — the row-oriented RowBinary format. Faster than TSV in both directions, and the right choice when streaming row-by-row with iterate.

client = ChClient(session, binary=True)

Because binary encoding is type-specific, a binary INSERT first looks up the target column types (one lightweight query) and encodes the rows to match them.

A couple of native=True caveats: decode=False (raw bytes) is a TSV-only feature, and the Native format does not carry a column's timezone, so a tz-aware DateTime/DateTime64 comes back as a naive UTC datetime (TSV and RowBinary apply the timezone).

A native INSERT streams its body as several Native blocks so the server can insert one while the client encodes the next (insert_block_size rows per block, default 8192). The flip side is that a multi-block insert is not atomic on a client-side encoding error — if a value somewhere in the rows fails to encode, the blocks already sent are committed. Inserts that fit in one block stay all-or-nothing; pass insert_block_size=0 to always send a single atomic block (no overlap):

client = ChClient(session, native=True, insert_block_size=0)  # atomic inserts

Speed

Engine comparison on mixed-type rows, fully decoded, with the Cython extension built — Native is 3–4× faster than the row-oriented engines:

Engine SELECT (decode) INSERT
TSV ~150k rows/sec ~165k rows/sec
RowBinary ~170k rows/sec ~170k rows/sec
Native ~565k rows/sec ~415k rows/sec

Against the popular Python ClickHouse clients on the same workload: on INSERT the Native engine matches clickhouse-connect and leads the rest; on SELECT it is competitive but clickhouse-connect is typically faster — its C+numpy columnar decode is less CPU-bound than aiochclient's per-row Python objects (the gap narrows on a fast machine and widens on a slow one).

Client SELECT INSERT
clickhouse-connect (HTTP, async) ~750k rows/sec ~415k rows/sec
aiochclient — Native (HTTP, async) ~565k rows/sec ~415k rows/sec
clickhouse-driver (native, sync) ~330k rows/sec ~290k rows/sec
asynch (native, async) ~62k rows/sec ~95k rows/sec

Indicative best-of-8 figures from the GitHub-hosted CI runner (a modest shared VM; a dev machine is several times faster across the board, which also narrows the SELECT gap). They vary run to run, with the data, and with CPU load. Full methodology lives in BENCHMARK_RESULTS.md; reproduce with benchmarks_vs_libs.py (clients) and benchmarks.py (engines).

Documentation

To check out the api docs, visit the readthedocs site..

Type Conversion

aiochclient automatically converts types from ClickHouse to python types and vice-versa.

ClickHouse type Python type
Bool bool
UInt8 int
UInt16 int
UInt32 int
UInt64 int
UInt128 int
UInt256 int
Int8 int
Int16 int
Int32 int
Int64 int
Int128 int
Int256 int
Float32 float
Float64 float
String str
FixedString str
Enum8 str
Enum16 str
Date datetime.date
DateTime datetime.datetime
DateTime64 datetime.datetime
Decimal decimal.Decimal
Decimal32 decimal.Decimal
Decimal64 decimal.Decimal
Decimal128 decimal.Decimal
IPv4 ipaddress.IPv4Address
IPv6 ipaddress.IPv6Address
UUID uuid.UUID
Nothing None
Tuple(T1, T2, ...) Tuple[T1, T2, ...]
Array(T) List[T]
Nullable(T) None or T
LowCardinality(T) T
Map(T1, T2) Dict[T1, T2]
Nested(T1, T2, ...) List[Tuple[T1, T2, ...], Tuple[T1, T2, ...]]

Connection Pool Settings

aiochclient uses the aiohttp.TCPConnector to determine pool size. By default, the pool limit is 100 open connections.

Notes on Speed

For the best throughput:

  • Pick a binary engine. native=True is the fastest for bulk SELECT/INSERT and binary=True (RowBinary) is best for row-by-row streaming via iterate — both are several times faster than the default TSV format. See Binary engines: RowBinary and Native for the numbers and trade-offs.
  • Build the Cython extension (used automatically when present; the binary engines rely on it) and install the *-speedups extra — ciso8601 for fast datetime parsing, plus aiodns + faust-cchardet for aiohttp.
  • Use uvloop.

Throughput is system-dependent (CPU speed and load especially); full per-engine and cross-client figures are in BENCHMARK_RESULTS.md.

About

Lightweight async http(s) ClickHouse client for python 3.10+ with types converting

Topics

Resources

License

Stars

Watchers

Forks

Packages

 
 
 

Contributors