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.
131 lines
2.9 KiB
Python
131 lines
2.9 KiB
Python
|
|
class ClientPacketTypes(object):
|
|
"""
|
|
Packet types that client transmits
|
|
"""
|
|
# Name, version, revision, default DB
|
|
HELLO = 0
|
|
|
|
# Query id, query settings, stage up to which the query must be executed,
|
|
# whether the compression must be used, query text
|
|
# (without data for INSERTs).
|
|
QUERY = 1
|
|
|
|
# A block of data (compressed or not).
|
|
DATA = 2
|
|
|
|
# Cancel the query execution.
|
|
CANCEL = 3
|
|
|
|
# Check that connection to the server is alive.
|
|
PING = 4
|
|
|
|
# Check status of tables on the server.
|
|
TABLES_STATUS_REQUEST = 5
|
|
|
|
_types_str = [
|
|
'Hello', 'Query', 'Data', 'Cancel', 'Ping', 'TablesStatusRequest'
|
|
]
|
|
|
|
@classmethod
|
|
def to_str(cls, packet):
|
|
try:
|
|
return cls._types_str[packet]
|
|
except IndexError:
|
|
return 'Unknown packet'
|
|
|
|
|
|
class ServerPacketTypes(object):
|
|
"""
|
|
Packet types that server transmits.
|
|
"""
|
|
# Name, version, revision.
|
|
HELLO = 0
|
|
|
|
# A block of data (compressed or not).
|
|
DATA = 1
|
|
|
|
# The exception during query execution.
|
|
EXCEPTION = 2
|
|
|
|
# Query execution progress: rows read, bytes read.
|
|
PROGRESS = 3
|
|
|
|
# Ping response
|
|
PONG = 4
|
|
|
|
# All packets were transmitted
|
|
END_OF_STREAM = 5
|
|
|
|
# Packet with profiling info.
|
|
PROFILE_INFO = 6
|
|
|
|
# A block with totals (compressed or not).
|
|
TOTALS = 7
|
|
|
|
# A block with minimums and maximums (compressed or not).
|
|
EXTREMES = 8
|
|
|
|
# A response to TablesStatus request.
|
|
TABLES_STATUS_RESPONSE = 9
|
|
|
|
# System logs of the query execution
|
|
LOG = 10
|
|
|
|
# Columns' description for default values calculation
|
|
TABLE_COLUMNS = 11
|
|
|
|
# List of unique parts ids.
|
|
PART_UUIDS = 12
|
|
|
|
# String (UUID) describes a request for which next task is needed
|
|
READ_TASK_REQUEST = 13
|
|
|
|
# Packet with profile events from server.
|
|
PROFILE_EVENTS = 14
|
|
|
|
MERGE_TREE_ALL_RANGES_ANNOUNCEMENT = 15
|
|
|
|
# Request from a MergeTree replica to a coordinator
|
|
MERGE_TREE_READ_TASK_REQUEST = 16
|
|
|
|
# Receive server's (session-wide) default timezone
|
|
TIMEZONE_UPDATE = 17
|
|
|
|
_types_str = [
|
|
'Hello', 'Data', 'Exception', 'Progress', 'Pong', 'EndOfStream',
|
|
'ProfileInfo', 'Totals', 'Extremes', 'TablesStatusResponse', 'Log',
|
|
'TableColumns', 'PartUUIDs', 'ReadTaskRequest', 'ProfileEvents',
|
|
'MergeTreeAllRangesAnnouncement', 'MergeTreeReadTaskRequest',
|
|
'TimezoneUpdate'
|
|
]
|
|
|
|
@classmethod
|
|
def to_str(cls, packet):
|
|
try:
|
|
return cls._types_str[packet]
|
|
except IndexError:
|
|
return 'Unknown packet'
|
|
|
|
@classmethod
|
|
def strings_in_message(cls, packet):
|
|
if packet == cls.TABLE_COLUMNS:
|
|
return 2
|
|
return 0
|
|
|
|
|
|
class Compression(object):
|
|
DISABLED = 0
|
|
ENABLED = 1
|
|
|
|
|
|
class CompressionMethod(object):
|
|
LZ4 = 1
|
|
LZ4HC = 2
|
|
ZSTD = 3
|
|
|
|
|
|
class CompressionMethodByte(object):
|
|
LZ4 = 0x82
|
|
ZSTD = 0x90
|