valkey-glide 1.3.5rc2__pp39-pypy39_pp73-macosx_10_7_x86_64.whl

glide/protobuf/connection_request_pb2.py

@@ -0,0 +1,52 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: protobuf/connection_request.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!protobuf/connection_request.proto\x12\x12\x63onnection_request\")\n\x0bNodeAddress\x12\x0c\n\x04host\x18\x01 \x01(\t\x12\x0c\n\x04port\x18\x02 \x01(\r\"8\n\x12\x41uthenticationInfo\x12\x10\n\x08password\x18\x01 \x01(\t\x12\x10\n\x08username\x18\x02 \x01(\t\"7\n\x1cPeriodicChecksManualInterval\x12\x17\n\x0f\x64uration_in_sec\x18\x01 \x01(\r\"\x18\n\x16PeriodicChecksDisabled\"8\n\x18PubSubChannelsOrPatterns\x12\x1c\n\x14\x63hannels_or_patterns\x18\x01 \x03(\x0c\"\xf1\x01\n\x13PubSubSubscriptions\x12k\n\x1c\x63hannels_or_patterns_by_type\x18\x01 \x03(\x0b\x32\x45.connection_request.PubSubSubscriptions.ChannelsOrPatternsByTypeEntry\x1am\n\x1d\x43hannelsOrPatternsByTypeEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12;\n\x05value\x18\x02 \x01(\x0b\x32,.connection_request.PubSubChannelsOrPatterns:\x02\x38\x01\"\xab\x06\n\x11\x43onnectionRequest\x12\x32\n\taddresses\x18\x01 \x03(\x0b\x32\x1f.connection_request.NodeAddress\x12-\n\x08tls_mode\x18\x02 \x01(\x0e\x32\x1b.connection_request.TlsMode\x12\x1c\n\x14\x63luster_mode_enabled\x18\x03 \x01(\x08\x12\x17\n\x0frequest_timeout\x18\x04 \x01(\r\x12/\n\tread_from\x18\x05 \x01(\x0e\x32\x1c.connection_request.ReadFrom\x12N\n\x19\x63onnection_retry_strategy\x18\x06 \x01(\x0b\x32+.connection_request.ConnectionRetryStrategy\x12\x43\n\x13\x61uthentication_info\x18\x07 \x01(\x0b\x32&.connection_request.AuthenticationInfo\x12\x13\n\x0b\x64\x61tabase_id\x18\x08 \x01(\r\x12\x35\n\x08protocol\x18\t \x01(\x0e\x32#.connection_request.ProtocolVersion\x12\x13\n\x0b\x63lient_name\x18\n \x01(\t\x12[\n\x1fperiodic_checks_manual_interval\x18\x0b \x01(\x0b\x32\x30.connection_request.PeriodicChecksManualIntervalH\x00\x12N\n\x18periodic_checks_disabled\x18\x0c \x01(\x0b\x32*.connection_request.PeriodicChecksDisabledH\x00\x12\x45\n\x14pubsub_subscriptions\x18\r \x01(\x0b\x32\'.connection_request.PubSubSubscriptions\x12\x1f\n\x17inflight_requests_limit\x18\x0e \x01(\r\x12\x11\n\tclient_az\x18\x0f \x01(\t\x12\x1a\n\x12\x63onnection_timeout\x18\x10 \x01(\rB\x11\n\x0fperiodic_checks\"[\n\x17\x43onnectionRetryStrategy\x12\x19\n\x11number_of_retries\x18\x01 \x01(\r\x12\x0e\n\x06\x66\x61\x63tor\x18\x02 \x01(\r\x12\x15\n\rexponent_base\x18\x03 \x01(\r*o\n\x08ReadFrom\x12\x0b\n\x07Primary\x10\x00\x12\x11\n\rPreferReplica\x10\x01\x12\x11\n\rLowestLatency\x10\x02\x12\x0e\n\nAZAffinity\x10\x03\x12 \n\x1c\x41ZAffinityReplicasAndPrimary\x10\x04*4\n\x07TlsMode\x12\t\n\x05NoTls\x10\x00\x12\r\n\tSecureTls\x10\x01\x12\x0f\n\x0bInsecureTls\x10\x02*\'\n\x0fProtocolVersion\x12\t\n\x05RESP3\x10\x00\x12\t\n\x05RESP2\x10\x01*8\n\x11PubSubChannelType\x12\t\n\x05\x45xact\x10\x00\x12\x0b\n\x07Pattern\x10\x01\x12\x0b\n\x07Sharded\x10\x02\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'protobuf.connection_request_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_PUBSUBSUBSCRIPTIONS_CHANNELSORPATTERNSBYTYPEENTRY']._options = None
+  _globals['_PUBSUBSUBSCRIPTIONS_CHANNELSORPATTERNSBYTYPEENTRY']._serialized_options = b'8\001'
+  _globals['_READFROM']._serialized_start=1450
+  _globals['_READFROM']._serialized_end=1561
+  _globals['_TLSMODE']._serialized_start=1563
+  _globals['_TLSMODE']._serialized_end=1615
+  _globals['_PROTOCOLVERSION']._serialized_start=1617
+  _globals['_PROTOCOLVERSION']._serialized_end=1656
+  _globals['_PUBSUBCHANNELTYPE']._serialized_start=1658
+  _globals['_PUBSUBCHANNELTYPE']._serialized_end=1714
+  _globals['_NODEADDRESS']._serialized_start=57
+  _globals['_NODEADDRESS']._serialized_end=98
+  _globals['_AUTHENTICATIONINFO']._serialized_start=100
+  _globals['_AUTHENTICATIONINFO']._serialized_end=156
+  _globals['_PERIODICCHECKSMANUALINTERVAL']._serialized_start=158
+  _globals['_PERIODICCHECKSMANUALINTERVAL']._serialized_end=213
+  _globals['_PERIODICCHECKSDISABLED']._serialized_start=215
+  _globals['_PERIODICCHECKSDISABLED']._serialized_end=239
+  _globals['_PUBSUBCHANNELSORPATTERNS']._serialized_start=241
+  _globals['_PUBSUBCHANNELSORPATTERNS']._serialized_end=297
+  _globals['_PUBSUBSUBSCRIPTIONS']._serialized_start=300
+  _globals['_PUBSUBSUBSCRIPTIONS']._serialized_end=541
+  _globals['_PUBSUBSUBSCRIPTIONS_CHANNELSORPATTERNSBYTYPEENTRY']._serialized_start=432
+  _globals['_PUBSUBSUBSCRIPTIONS_CHANNELSORPATTERNSBYTYPEENTRY']._serialized_end=541
+  _globals['_CONNECTIONREQUEST']._serialized_start=544
+  _globals['_CONNECTIONREQUEST']._serialized_end=1355
+  _globals['_CONNECTIONRETRYSTRATEGY']._serialized_start=1357
+  _globals['_CONNECTIONRETRYSTRATEGY']._serialized_end=1448
+# @@protoc_insertion_point(module_scope)

glide/protobuf/connection_request_pb2.pyi

@@ -0,0 +1,292 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _ReadFrom:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _ReadFromEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ReadFrom.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    Primary: _ReadFrom.ValueType  # 0
+    PreferReplica: _ReadFrom.ValueType  # 1
+    LowestLatency: _ReadFrom.ValueType  # 2
+    AZAffinity: _ReadFrom.ValueType  # 3
+    AZAffinityReplicasAndPrimary: _ReadFrom.ValueType  # 4
+
+class ReadFrom(_ReadFrom, metaclass=_ReadFromEnumTypeWrapper): ...
+
+Primary: ReadFrom.ValueType  # 0
+PreferReplica: ReadFrom.ValueType  # 1
+LowestLatency: ReadFrom.ValueType  # 2
+AZAffinity: ReadFrom.ValueType  # 3
+AZAffinityReplicasAndPrimary: ReadFrom.ValueType  # 4
+global___ReadFrom = ReadFrom
+
+class _TlsMode:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _TlsModeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TlsMode.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    NoTls: _TlsMode.ValueType  # 0
+    SecureTls: _TlsMode.ValueType  # 1
+    InsecureTls: _TlsMode.ValueType  # 2
+
+class TlsMode(_TlsMode, metaclass=_TlsModeEnumTypeWrapper): ...
+
+NoTls: TlsMode.ValueType  # 0
+SecureTls: TlsMode.ValueType  # 1
+InsecureTls: TlsMode.ValueType  # 2
+global___TlsMode = TlsMode
+
+class _ProtocolVersion:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _ProtocolVersionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ProtocolVersion.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    RESP3: _ProtocolVersion.ValueType  # 0
+    RESP2: _ProtocolVersion.ValueType  # 1
+
+class ProtocolVersion(_ProtocolVersion, metaclass=_ProtocolVersionEnumTypeWrapper): ...
+
+RESP3: ProtocolVersion.ValueType  # 0
+RESP2: ProtocolVersion.ValueType  # 1
+global___ProtocolVersion = ProtocolVersion
+
+class _PubSubChannelType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _PubSubChannelTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PubSubChannelType.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    Exact: _PubSubChannelType.ValueType  # 0
+    Pattern: _PubSubChannelType.ValueType  # 1
+    Sharded: _PubSubChannelType.ValueType  # 2
+
+class PubSubChannelType(_PubSubChannelType, metaclass=_PubSubChannelTypeEnumTypeWrapper): ...
+
+Exact: PubSubChannelType.ValueType  # 0
+Pattern: PubSubChannelType.ValueType  # 1
+Sharded: PubSubChannelType.ValueType  # 2
+global___PubSubChannelType = PubSubChannelType
+
+@typing.final
+class NodeAddress(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    HOST_FIELD_NUMBER: builtins.int
+    PORT_FIELD_NUMBER: builtins.int
+    host: builtins.str
+    port: builtins.int
+    def __init__(
+        self,
+        *,
+        host: builtins.str = ...,
+        port: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["host", b"host", "port", b"port"]) -> None: ...
+
+global___NodeAddress = NodeAddress
+
+@typing.final
+class AuthenticationInfo(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PASSWORD_FIELD_NUMBER: builtins.int
+    USERNAME_FIELD_NUMBER: builtins.int
+    password: builtins.str
+    username: builtins.str
+    def __init__(
+        self,
+        *,
+        password: builtins.str = ...,
+        username: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["password", b"password", "username", b"username"]) -> None: ...
+
+global___AuthenticationInfo = AuthenticationInfo
+
+@typing.final
+class PeriodicChecksManualInterval(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DURATION_IN_SEC_FIELD_NUMBER: builtins.int
+    duration_in_sec: builtins.int
+    def __init__(
+        self,
+        *,
+        duration_in_sec: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["duration_in_sec", b"duration_in_sec"]) -> None: ...
+
+global___PeriodicChecksManualInterval = PeriodicChecksManualInterval
+
+@typing.final
+class PeriodicChecksDisabled(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___PeriodicChecksDisabled = PeriodicChecksDisabled
+
+@typing.final
+class PubSubChannelsOrPatterns(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CHANNELS_OR_PATTERNS_FIELD_NUMBER: builtins.int
+    @property
+    def channels_or_patterns(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bytes]: ...
+    def __init__(
+        self,
+        *,
+        channels_or_patterns: collections.abc.Iterable[builtins.bytes] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["channels_or_patterns", b"channels_or_patterns"]) -> None: ...
+
+global___PubSubChannelsOrPatterns = PubSubChannelsOrPatterns
+
+@typing.final
+class PubSubSubscriptions(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class ChannelsOrPatternsByTypeEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.int
+        @property
+        def value(self) -> global___PubSubChannelsOrPatterns: ...
+        def __init__(
+            self,
+            *,
+            key: builtins.int = ...,
+            value: global___PubSubChannelsOrPatterns | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    CHANNELS_OR_PATTERNS_BY_TYPE_FIELD_NUMBER: builtins.int
+    @property
+    def channels_or_patterns_by_type(self) -> google.protobuf.internal.containers.MessageMap[builtins.int, global___PubSubChannelsOrPatterns]: ...
+    def __init__(
+        self,
+        *,
+        channels_or_patterns_by_type: collections.abc.Mapping[builtins.int, global___PubSubChannelsOrPatterns] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["channels_or_patterns_by_type", b"channels_or_patterns_by_type"]) -> None: ...
+
+global___PubSubSubscriptions = PubSubSubscriptions
+
+@typing.final
+class ConnectionRequest(google.protobuf.message.Message):
+    """IMPORTANT - if you add fields here, you probably need to add them also in client/mod.rs:`sanitized_request_string`."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ADDRESSES_FIELD_NUMBER: builtins.int
+    TLS_MODE_FIELD_NUMBER: builtins.int
+    CLUSTER_MODE_ENABLED_FIELD_NUMBER: builtins.int
+    REQUEST_TIMEOUT_FIELD_NUMBER: builtins.int
+    READ_FROM_FIELD_NUMBER: builtins.int
+    CONNECTION_RETRY_STRATEGY_FIELD_NUMBER: builtins.int
+    AUTHENTICATION_INFO_FIELD_NUMBER: builtins.int
+    DATABASE_ID_FIELD_NUMBER: builtins.int
+    PROTOCOL_FIELD_NUMBER: builtins.int
+    CLIENT_NAME_FIELD_NUMBER: builtins.int
+    PERIODIC_CHECKS_MANUAL_INTERVAL_FIELD_NUMBER: builtins.int
+    PERIODIC_CHECKS_DISABLED_FIELD_NUMBER: builtins.int
+    PUBSUB_SUBSCRIPTIONS_FIELD_NUMBER: builtins.int
+    INFLIGHT_REQUESTS_LIMIT_FIELD_NUMBER: builtins.int
+    CLIENT_AZ_FIELD_NUMBER: builtins.int
+    CONNECTION_TIMEOUT_FIELD_NUMBER: builtins.int
+    tls_mode: global___TlsMode.ValueType
+    cluster_mode_enabled: builtins.bool
+    request_timeout: builtins.int
+    read_from: global___ReadFrom.ValueType
+    database_id: builtins.int
+    protocol: global___ProtocolVersion.ValueType
+    client_name: builtins.str
+    inflight_requests_limit: builtins.int
+    client_az: builtins.str
+    connection_timeout: builtins.int
+    @property
+    def addresses(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___NodeAddress]: ...
+    @property
+    def connection_retry_strategy(self) -> global___ConnectionRetryStrategy: ...
+    @property
+    def authentication_info(self) -> global___AuthenticationInfo: ...
+    @property
+    def periodic_checks_manual_interval(self) -> global___PeriodicChecksManualInterval: ...
+    @property
+    def periodic_checks_disabled(self) -> global___PeriodicChecksDisabled: ...
+    @property
+    def pubsub_subscriptions(self) -> global___PubSubSubscriptions: ...
+    def __init__(
+        self,
+        *,
+        addresses: collections.abc.Iterable[global___NodeAddress] | None = ...,
+        tls_mode: global___TlsMode.ValueType = ...,
+        cluster_mode_enabled: builtins.bool = ...,
+        request_timeout: builtins.int = ...,
+        read_from: global___ReadFrom.ValueType = ...,
+        connection_retry_strategy: global___ConnectionRetryStrategy | None = ...,
+        authentication_info: global___AuthenticationInfo | None = ...,
+        database_id: builtins.int = ...,
+        protocol: global___ProtocolVersion.ValueType = ...,
+        client_name: builtins.str = ...,
+        periodic_checks_manual_interval: global___PeriodicChecksManualInterval | None = ...,
+        periodic_checks_disabled: global___PeriodicChecksDisabled | None = ...,
+        pubsub_subscriptions: global___PubSubSubscriptions | None = ...,
+        inflight_requests_limit: builtins.int = ...,
+        client_az: builtins.str = ...,
+        connection_timeout: builtins.int = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["authentication_info", b"authentication_info", "connection_retry_strategy", b"connection_retry_strategy", "periodic_checks", b"periodic_checks", "periodic_checks_disabled", b"periodic_checks_disabled", "periodic_checks_manual_interval", b"periodic_checks_manual_interval", "pubsub_subscriptions", b"pubsub_subscriptions"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["addresses", b"addresses", "authentication_info", b"authentication_info", "client_az", b"client_az", "client_name", b"client_name", "cluster_mode_enabled", b"cluster_mode_enabled", "connection_retry_strategy", b"connection_retry_strategy", "connection_timeout", b"connection_timeout", "database_id", b"database_id", "inflight_requests_limit", b"inflight_requests_limit", "periodic_checks", b"periodic_checks", "periodic_checks_disabled", b"periodic_checks_disabled", "periodic_checks_manual_interval", b"periodic_checks_manual_interval", "protocol", b"protocol", "pubsub_subscriptions", b"pubsub_subscriptions", "read_from", b"read_from", "request_timeout", b"request_timeout", "tls_mode", b"tls_mode"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["periodic_checks", b"periodic_checks"]) -> typing.Literal["periodic_checks_manual_interval", "periodic_checks_disabled"] | None: ...
+
+global___ConnectionRequest = ConnectionRequest
+
+@typing.final
+class ConnectionRetryStrategy(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NUMBER_OF_RETRIES_FIELD_NUMBER: builtins.int
+    FACTOR_FIELD_NUMBER: builtins.int
+    EXPONENT_BASE_FIELD_NUMBER: builtins.int
+    number_of_retries: builtins.int
+    factor: builtins.int
+    exponent_base: builtins.int
+    def __init__(
+        self,
+        *,
+        number_of_retries: builtins.int = ...,
+        factor: builtins.int = ...,
+        exponent_base: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["exponent_base", b"exponent_base", "factor", b"factor", "number_of_retries", b"number_of_retries"]) -> None: ...
+
+global___ConnectionRetryStrategy = ConnectionRetryStrategy

glide/protobuf/response_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: protobuf/response.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17protobuf/response.proto\x12\x08response\"I\n\x0cRequestError\x12(\n\x04type\x18\x01 \x01(\x0e\x32\x1a.response.RequestErrorType\x12\x0f\n\x07message\x18\x02 \x01(\t\"\xd5\x01\n\x08Response\x12\x14\n\x0c\x63\x61llback_idx\x18\x01 \x01(\r\x12\x16\n\x0cresp_pointer\x18\x02 \x01(\x04H\x00\x12\x37\n\x11\x63onstant_response\x18\x03 \x01(\x0e\x32\x1a.response.ConstantResponseH\x00\x12/\n\rrequest_error\x18\x04 \x01(\x0b\x32\x16.response.RequestErrorH\x00\x12\x17\n\rclosing_error\x18\x05 \x01(\tH\x00\x12\x0f\n\x07is_push\x18\x06 \x01(\x08\x42\x07\n\x05value*O\n\x10RequestErrorType\x12\x0f\n\x0bUnspecified\x10\x00\x12\r\n\tExecAbort\x10\x01\x12\x0b\n\x07Timeout\x10\x02\x12\x0e\n\nDisconnect\x10\x03*\x1a\n\x10\x43onstantResponse\x12\x06\n\x02OK\x10\x00\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'protobuf.response_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_REQUESTERRORTYPE']._serialized_start=328
+  _globals['_REQUESTERRORTYPE']._serialized_end=407
+  _globals['_CONSTANTRESPONSE']._serialized_start=409
+  _globals['_CONSTANTRESPONSE']._serialized_end=435
+  _globals['_REQUESTERROR']._serialized_start=37
+  _globals['_REQUESTERROR']._serialized_end=110
+  _globals['_RESPONSE']._serialized_start=113
+  _globals['_RESPONSE']._serialized_end=326
+# @@protoc_insertion_point(module_scope)

glide/protobuf/response_pb2.pyi

@@ -0,0 +1,101 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _RequestErrorType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _RequestErrorTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RequestErrorType.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    Unspecified: _RequestErrorType.ValueType  # 0
+    ExecAbort: _RequestErrorType.ValueType  # 1
+    Timeout: _RequestErrorType.ValueType  # 2
+    Disconnect: _RequestErrorType.ValueType  # 3
+
+class RequestErrorType(_RequestErrorType, metaclass=_RequestErrorTypeEnumTypeWrapper): ...
+
+Unspecified: RequestErrorType.ValueType  # 0
+ExecAbort: RequestErrorType.ValueType  # 1
+Timeout: RequestErrorType.ValueType  # 2
+Disconnect: RequestErrorType.ValueType  # 3
+global___RequestErrorType = RequestErrorType
+
+class _ConstantResponse:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _ConstantResponseEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ConstantResponse.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    OK: _ConstantResponse.ValueType  # 0
+
+class ConstantResponse(_ConstantResponse, metaclass=_ConstantResponseEnumTypeWrapper): ...
+
+OK: ConstantResponse.ValueType  # 0
+global___ConstantResponse = ConstantResponse
+
+@typing.final
+class RequestError(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TYPE_FIELD_NUMBER: builtins.int
+    MESSAGE_FIELD_NUMBER: builtins.int
+    type: global___RequestErrorType.ValueType
+    message: builtins.str
+    def __init__(
+        self,
+        *,
+        type: global___RequestErrorType.ValueType = ...,
+        message: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["message", b"message", "type", b"type"]) -> None: ...
+
+global___RequestError = RequestError
+
+@typing.final
+class Response(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CALLBACK_IDX_FIELD_NUMBER: builtins.int
+    RESP_POINTER_FIELD_NUMBER: builtins.int
+    CONSTANT_RESPONSE_FIELD_NUMBER: builtins.int
+    REQUEST_ERROR_FIELD_NUMBER: builtins.int
+    CLOSING_ERROR_FIELD_NUMBER: builtins.int
+    IS_PUSH_FIELD_NUMBER: builtins.int
+    callback_idx: builtins.int
+    resp_pointer: builtins.int
+    constant_response: global___ConstantResponse.ValueType
+    closing_error: builtins.str
+    is_push: builtins.bool
+    @property
+    def request_error(self) -> global___RequestError: ...
+    def __init__(
+        self,
+        *,
+        callback_idx: builtins.int = ...,
+        resp_pointer: builtins.int = ...,
+        constant_response: global___ConstantResponse.ValueType = ...,
+        request_error: global___RequestError | None = ...,
+        closing_error: builtins.str = ...,
+        is_push: builtins.bool = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["closing_error", b"closing_error", "constant_response", b"constant_response", "request_error", b"request_error", "resp_pointer", b"resp_pointer", "value", b"value"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["callback_idx", b"callback_idx", "closing_error", b"closing_error", "constant_response", b"constant_response", "is_push", b"is_push", "request_error", b"request_error", "resp_pointer", b"resp_pointer", "value", b"value"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["value", b"value"]) -> typing.Literal["resp_pointer", "constant_response", "request_error", "closing_error"] | None: ...
+
+global___Response = Response

glide/protobuf_codec.py

@@ -0,0 +1,109 @@
+# Copyright Valkey GLIDE Project Contributors - SPDX Identifier: Apache-2.0
+
+import struct
+from typing import List, Tuple, Type
+
+from google.protobuf import message
+
+"""
+Codec for protobuf messages.
+All of the Varint functions were copied from https://github.com/protocolbuffers/protobuf
+"""
+
+
+class ProtobufCodec:
+    @classmethod
+    def _decode_varint_32(cls, buffer, pos):
+        decoder_func = cls._varint_decoder((1 << 32) - 1, int)
+        return decoder_func(buffer, pos)
+
+    @staticmethod
+    def _varint_decoder(mask, result_type):
+        """Return an encoder for a basic varint value (does not include tag).
+
+        Decoded values will be bitwise-anded with the given mask before being
+        returned, e.g. to limit them to 32 bits.  The returned decoder does not
+        take the usual "end" parameter -- the caller is expected to do bounds checking
+        after the fact (often the caller can defer such checking until later).  The
+        decoder returns a (value, new_pos) pair.
+        """
+
+        def decode_varint(buffer, pos):
+            result = 0
+            shift = 0
+            while 1:
+                b = buffer[pos]
+                result |= (b & 0x7F) << shift
+                pos += 1
+                if not (b & 0x80):
+                    result &= mask
+                    result = result_type(result)
+                    return (result, pos)
+                shift += 7
+                if shift >= 64:
+                    raise message.DecodeError("Too many bytes when decoding varint.")
+
+        return decode_varint
+
+    @staticmethod
+    def _varint_encoder():
+        """Return an encoder for a basic varint value (does not include tag)."""
+
+        local_int2byte = struct.Struct(">B").pack
+
+        def encode_varint(write, value):
+            bits = value & 0x7F
+            value >>= 7
+            while value:
+                write(local_int2byte(0x80 | bits))
+                bits = value & 0x7F
+                value >>= 7
+            return write(local_int2byte(bits))
+
+        return encode_varint
+
+    @classmethod
+    def _varint_bytes(cls, value: int) -> bytes:
+        """Encode the given integer as a varint and return the bytes."""
+
+        pieces: List[bytes] = []
+        func = cls._varint_encoder()
+        func(pieces.append, value)
+        return b"".join(pieces)
+
+    @classmethod
+    def decode_delimited(
+        cls,
+        read_bytes: bytearray,
+        read_bytes_view: memoryview,
+        offset: int,
+        message_class: Type[message.Message],
+    ) -> Tuple[message.Message, int]:
+        try:
+            msg_len, new_pos = cls._decode_varint_32(read_bytes_view, offset)
+        except IndexError:
+            # Didn't read enough bytes to decode the varint
+            raise PartialMessageException(
+                "Didn't read enough bytes to decode the varint"
+            )
+        required_read_size = new_pos + msg_len
+        if required_read_size > len(read_bytes):
+            # Recieved only partial response
+            raise PartialMessageException("Recieved only a partial response")
+        offset = new_pos
+        msg_buf = read_bytes_view[offset : offset + msg_len]
+        offset += msg_len
+        message = message_class()
+        message.ParseFromString(msg_buf)
+        return (message, offset)
+
+    @classmethod
+    def encode_delimited(cls, b_arr: bytearray, message: message.Message) -> None:
+        bytes_request = message.SerializeToString()
+        varint = cls._varint_bytes(len(bytes_request))
+        b_arr.extend(varint)
+        b_arr.extend(bytes_request)
+
+
+class PartialMessageException(Exception):
+    pass