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.
69 lines
1.7 KiB
Cython
69 lines
1.7 KiB
Cython
from cpython cimport PyBytes_FromStringAndSize
|
|
|
|
|
|
def make_varint(unsigned long long number):
|
|
"""
|
|
Writes integer of variable length using LEB128.
|
|
"""
|
|
cdef unsigned char to_write, i = 0
|
|
# unsigned PY_LONG_LONG checks integer on function call and
|
|
# raises OverflowError if integer overflows unsigned PY_LONG_LONG.
|
|
# Long enough for handling unsigned PY_LONG_LONG.
|
|
cdef unsigned char num_buf[32]
|
|
|
|
while True:
|
|
to_write = number & 0x7f
|
|
number >>= 7
|
|
if number:
|
|
num_buf[i] = to_write | 0x80
|
|
i += 1
|
|
else:
|
|
num_buf[i] = to_write
|
|
i += 1
|
|
break
|
|
|
|
return PyBytes_FromStringAndSize(<char *>num_buf, i)
|
|
|
|
|
|
def write_varint(unsigned long long number, buf):
|
|
"""
|
|
Writes integer of variable length using LEB128.
|
|
"""
|
|
cdef unsigned char to_write, i = 0
|
|
# unsigned PY_LONG_LONG checks integer on function call and
|
|
# raises OverflowError if integer overflows unsigned PY_LONG_LONG.
|
|
# Long enough for handling unsigned PY_LONG_LONG.
|
|
cdef unsigned char num_buf[32]
|
|
|
|
while True:
|
|
to_write = number & 0x7f
|
|
number >>= 7
|
|
if number:
|
|
num_buf[i] = to_write | 0x80
|
|
i += 1
|
|
else:
|
|
num_buf[i] = to_write
|
|
i += 1
|
|
break
|
|
|
|
buf.write(PyBytes_FromStringAndSize(<char *>num_buf, i))
|
|
|
|
|
|
def read_varint(f):
|
|
"""
|
|
Reads integer of variable length using LEB128.
|
|
"""
|
|
cdef unsigned char shift = 0
|
|
cdef unsigned long long i, result = 0
|
|
|
|
read_one = f.read_one
|
|
|
|
while True:
|
|
i = read_one()
|
|
result |= (i & 0x7f) << shift
|
|
shift += 7
|
|
if i < 0x80:
|
|
break
|
|
|
|
return result
|