mirror of
https://github.com/hanzo-ds/driver.git
synced 2026-10-30 20:32:40 +00:00
zeekay
eb74c996a6
Imports read `datastore_driver`, distribution `datastore-driver`, so nothing has to import another product's name to talk to Hanzo Datastore. Cut from upstream 0.2.9 — the version this stack already ran — rather than from upstream HEAD, so it is a PURE RENAME with no behaviour change. Forking from HEAD would have dragged versions along with the name (connect alone jumped 0.8.17 -> 1.6.0, a major), which is a different and much riskier change wearing a rename's clothes. X-ClickHouse-* headers are untouched: they are wire protocol the server sends and expects, and a different token shape from the module name.
70 lines
1.3 KiB
Python
70 lines
1.3 KiB
Python
from struct import Struct
|
|
|
|
from .varint import read_varint
|
|
|
|
|
|
def read_binary_str(buf):
|
|
length = read_varint(buf)
|
|
return read_binary_str_fixed_len(buf, length)
|
|
|
|
|
|
def read_binary_bytes(buf):
|
|
length = read_varint(buf)
|
|
return read_binary_bytes_fixed_len(buf, length)
|
|
|
|
|
|
def read_binary_str_fixed_len(buf, length):
|
|
return read_binary_bytes_fixed_len(buf, length).decode('utf-8')
|
|
|
|
|
|
def read_binary_bytes_fixed_len(buf, length):
|
|
return buf.read(length)
|
|
|
|
|
|
def read_binary_int(buf, fmt):
|
|
"""
|
|
Reads int from buffer with provided format.
|
|
"""
|
|
# Little endian.
|
|
s = Struct('<' + fmt)
|
|
return s.unpack(buf.read(s.size))[0]
|
|
|
|
|
|
def read_binary_int8(buf):
|
|
return read_binary_int(buf, 'b')
|
|
|
|
|
|
def read_binary_int16(buf):
|
|
return read_binary_int(buf, 'h')
|
|
|
|
|
|
def read_binary_int32(buf):
|
|
return read_binary_int(buf, 'i')
|
|
|
|
|
|
def read_binary_int64(buf):
|
|
return read_binary_int(buf, 'q')
|
|
|
|
|
|
def read_binary_uint8(buf):
|
|
return read_binary_int(buf, 'B')
|
|
|
|
|
|
def read_binary_uint16(buf):
|
|
return read_binary_int(buf, 'H')
|
|
|
|
|
|
def read_binary_uint32(buf):
|
|
return read_binary_int(buf, 'I')
|
|
|
|
|
|
def read_binary_uint64(buf):
|
|
return read_binary_int(buf, 'Q')
|
|
|
|
|
|
def read_binary_uint128(buf):
|
|
hi = read_binary_int(buf, 'Q')
|
|
lo = read_binary_int(buf, 'Q')
|
|
|
|
return (hi << 64) + lo
|