valkey-glide 2.1.0rc2__cp313-cp313-macosx_10_7_x86_64.whl → 2.1.0rc4__cp313-cp313-macosx_10_7_x86_64.whl

glide/async_commands/standalone_commands.py

@@ -166,6 +166,28 @@ class StandaloneCommands(CoreCommands):
         """
         Change the currently selected database.
 
+        **WARNING**: This command is NOT RECOMMENDED for production use.
+        Upon reconnection, the client will revert to the database_id specified
+        in the client configuration (default: 0), NOT the database selected
+        via this command.
+
+        **RECOMMENDED APPROACH**: Use the database_id parameter in client
+        configuration instead:
+
+        ```python
+        client = await GlideClient.create_client(
+            GlideClientConfiguration(
+                addresses=[NodeAddress("localhost", 6379)],
+                database_id=5  # Recommended: persists across reconnections
+            )
+        )
+        ```
+
+        **RECONNECTION BEHAVIOR**: After any reconnection (due to network issues,
+        timeouts, etc.), the client will automatically revert to the database_id
+        specified during client creation, losing any database selection made via
+        this SELECT command.
+
         See [valkey.io](https://valkey.io/commands/select/) for details.
 
         Args:

glide_shared/config.py

@@ -249,6 +249,8 @@ class BaseClientConfiguration:
         reconnect_strategy (Optional[BackoffStrategy]): Strategy used to determine how and when to reconnect, in case of
             connection failures.
             If not set, a default backoff strategy will be used.
+        database_id (Optional[int]): Index of the logical database to connect to.
+            Must be a non-negative integer.If not set, the client will connect to database 0.
         client_name (Optional[str]): Client name to be used for the client. Will be used with CLIENT SETNAME command
             during connection establishment.
         protocol (ProtocolVersion): Serialization protocol to be used. If not set, `RESP3` will be used.
@@ -292,6 +294,7 @@ class BaseClientConfiguration:
         read_from: ReadFrom = ReadFrom.PRIMARY,
         request_timeout: Optional[int] = None,
         reconnect_strategy: Optional[BackoffStrategy] = None,
+        database_id: Optional[int] = None,
         client_name: Optional[str] = None,
         protocol: ProtocolVersion = ProtocolVersion.RESP3,
         inflight_requests_limit: Optional[int] = None,
@@ -305,6 +308,7 @@ class BaseClientConfiguration:
         self.read_from = read_from
         self.request_timeout = request_timeout
         self.reconnect_strategy = reconnect_strategy
+        self.database_id = database_id
         self.client_name = client_name
         self.protocol = protocol
         self.inflight_requests_limit = inflight_requests_limit
@@ -322,6 +326,39 @@ class BaseClientConfiguration:
                 "client_az must be set when read_from is set to AZ_AFFINITY_REPLICAS_AND_PRIMARY"
             )
 
+    def _set_addresses_in_request(self, request: ConnectionRequest) -> None:
+        """Set addresses in the protobuf request."""
+        for address in self.addresses:
+            address_info = request.addresses.add()
+            address_info.host = address.host
+            address_info.port = address.port
+
+    def _set_reconnect_strategy_in_request(self, request: ConnectionRequest) -> None:
+        """Set reconnect strategy in the protobuf request."""
+        if not self.reconnect_strategy:
+            return
+
+        request.connection_retry_strategy.number_of_retries = (
+            self.reconnect_strategy.num_of_retries
+        )
+        request.connection_retry_strategy.factor = self.reconnect_strategy.factor
+        request.connection_retry_strategy.exponent_base = (
+            self.reconnect_strategy.exponent_base
+        )
+        if self.reconnect_strategy.jitter_percent is not None:
+            request.connection_retry_strategy.jitter_percent = (
+                self.reconnect_strategy.jitter_percent
+            )
+
+    def _set_credentials_in_request(self, request: ConnectionRequest) -> None:
+        """Set credentials in the protobuf request."""
+        if not self.credentials:
+            return
+
+        if self.credentials.username:
+            request.authentication_info.username = self.credentials.username
+        request.authentication_info.password = self.credentials.password
+
     def _create_a_protobuf_conn_request(
         self, cluster_mode: bool = False
     ) -> ConnectionRequest:
@@ -335,44 +372,34 @@ class BaseClientConfiguration:
             ConnectionRequest: Protobuf ConnectionRequest.
         """
         request = ConnectionRequest()
-        for address in self.addresses:
-            address_info = request.addresses.add()
-            address_info.host = address.host
-            address_info.port = address.port
+
+        # Set basic configuration
+        self._set_addresses_in_request(request)
         request.tls_mode = TlsMode.SecureTls if self.use_tls else TlsMode.NoTls
         request.read_from = self.read_from.value
+        request.cluster_mode_enabled = cluster_mode
+        request.protocol = self.protocol.value
+
+        # Set optional configuration
         if self.request_timeout:
             request.request_timeout = self.request_timeout
-        if self.reconnect_strategy:
-            request.connection_retry_strategy.number_of_retries = (
-                self.reconnect_strategy.num_of_retries
-            )
-            request.connection_retry_strategy.factor = self.reconnect_strategy.factor
-            request.connection_retry_strategy.exponent_base = (
-                self.reconnect_strategy.exponent_base
-            )
-            if self.reconnect_strategy.jitter_percent is not None:
-                request.connection_retry_strategy.jitter_percent = (
-                    self.reconnect_strategy.jitter_percent
-                )
 
-        request.cluster_mode_enabled = True if cluster_mode else False
-        if self.credentials:
-            if self.credentials.username:
-                request.authentication_info.username = self.credentials.username
-            request.authentication_info.password = self.credentials.password
+        self._set_reconnect_strategy_in_request(request)
+        self._set_credentials_in_request(request)
+
         if self.client_name:
             request.client_name = self.client_name
-        request.protocol = self.protocol.value
         if self.inflight_requests_limit:
             request.inflight_requests_limit = self.inflight_requests_limit
         if self.client_az:
             request.client_az = self.client_az
+        if self.database_id is not None:
+            request.database_id = self.database_id
         if self.advanced_config:
             self.advanced_config._create_a_protobuf_conn_request(request)
-
         if self.lazy_connect is not None:
             request.lazy_connect = self.lazy_connect
+
         return request
 
     def _is_pubsub_configured(self) -> bool:
@@ -425,7 +452,7 @@ class GlideClientConfiguration(BaseClientConfiguration):
         reconnect_strategy (Optional[BackoffStrategy]): Strategy used to determine how and when to reconnect, in case of
             connection failures.
             If not set, a default backoff strategy will be used.
-        database_id (Optional[int]): index of the logical database to connect to.
+        database_id (Optional[int]): Index of the logical database to connect to.
         client_name (Optional[str]): Client name to be used for the client. Will be used with CLIENT SETNAME command during
             connection establishment.
         protocol (ProtocolVersion): The version of the RESP protocol to communicate with the server.
@@ -500,6 +527,7 @@ class GlideClientConfiguration(BaseClientConfiguration):
             read_from=read_from,
             request_timeout=request_timeout,
             reconnect_strategy=reconnect_strategy,
+            database_id=database_id,
             client_name=client_name,
             protocol=protocol,
             inflight_requests_limit=inflight_requests_limit,
@@ -507,7 +535,6 @@ class GlideClientConfiguration(BaseClientConfiguration):
             advanced_config=advanced_config,
             lazy_connect=lazy_connect,
         )
-        self.database_id = database_id
         self.pubsub_subscriptions = pubsub_subscriptions
 
     def _create_a_protobuf_conn_request(
@@ -515,8 +542,6 @@ class GlideClientConfiguration(BaseClientConfiguration):
     ) -> ConnectionRequest:
         assert cluster_mode is False
         request = super()._create_a_protobuf_conn_request(cluster_mode)
-        if self.database_id:
-            request.database_id = self.database_id
 
         if self.pubsub_subscriptions:
             if self.protocol == ProtocolVersion.RESP2:
@@ -592,6 +617,7 @@ class GlideClusterClientConfiguration(BaseClientConfiguration):
         reconnect_strategy (Optional[BackoffStrategy]): Strategy used to determine how and when to reconnect, in case of
             connection failures.
             If not set, a default backoff strategy will be used.
+        database_id (Optional[int]): Index of the logical database to connect to.
         client_name (Optional[str]): Client name to be used for the client. Will be used with CLIENT SETNAME command during
             connection establishment.
         protocol (ProtocolVersion): The version of the RESP protocol to communicate with the server.
@@ -661,6 +687,7 @@ class GlideClusterClientConfiguration(BaseClientConfiguration):
         read_from: ReadFrom = ReadFrom.PRIMARY,
         request_timeout: Optional[int] = None,
         reconnect_strategy: Optional[BackoffStrategy] = None,
+        database_id: Optional[int] = None,
         client_name: Optional[str] = None,
         protocol: ProtocolVersion = ProtocolVersion.RESP3,
         periodic_checks: Union[
@@ -679,6 +706,7 @@ class GlideClusterClientConfiguration(BaseClientConfiguration):
             read_from=read_from,
             request_timeout=request_timeout,
             reconnect_strategy=reconnect_strategy,
+            database_id=database_id,
             client_name=client_name,
             protocol=protocol,
             inflight_requests_limit=inflight_requests_limit,

{valkey_glide-2.1.0rc2.dist-info → valkey_glide-2.1.0rc4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: valkey-glide
-Version: 2.1.0rc2
+Version: 2.1.0rc4
 Classifier: Topic :: Database
 Classifier: Topic :: Utilities
 Classifier: License :: OSI Approved :: Apache Software License

{valkey_glide-2.1.0rc2.dist-info → valkey_glide-2.1.0rc4.dist-info}/RECORD

@@ -1,10 +1,10 @@
-valkey_glide-2.1.0rc2.dist-info/METADATA,sha256=L-6ijxqrcX8M-7wPnlDsySJuTqq7QGhovD2ZyY-oafo,6978
-valkey_glide-2.1.0rc2.dist-info/WHEEL,sha256=WKnnnV-Uggn89gY5LlN93bk9B3X9JgieltY0v0WQhvw,107
+valkey_glide-2.1.0rc4.dist-info/METADATA,sha256=e8NJtDzIcPdqAsWJatuyesUpHnnsiK2EUl81jDqQWzY,6978
+valkey_glide-2.1.0rc4.dist-info/WHEEL,sha256=WKnnnV-Uggn89gY5LlN93bk9B3X9JgieltY0v0WQhvw,107
 glide/async_commands/glide_json.py,sha256=H5dnB4vJkGPpNXJZhhVyPQoTSbkN_chq5Xw6tkP0wMQ,60410
 glide/async_commands/ft.py,sha256=1v96sBCgiXtPbj85ZurY3ruLCmok3cPoaIXSiJBEv9Y,16849
 glide/async_commands/__init__.py,sha256=_tbTAFATlzp4L2qe-H77PpAQK-16VsV-y7uKNUKLC_o,136
 glide/async_commands/core.py,sha256=RdfOTxpvgY4sQsSL3PqHi-05CV_AmDhsZad-ALg4EHU,328057
-glide/async_commands/standalone_commands.py,sha256=y6hH6iE4ngAJlB9JTa1y_ZHRmM1MDMvWQKOHghyN73E,39450
+glide/async_commands/standalone_commands.py,sha256=6oPesnC8WK816JbBySUZ9dSXPbdtN0dvd8rRde646NQ,40364
 glide/async_commands/cluster_commands.py,sha256=Lji6eo2LaoXwy4Ha3e6bcGOHyMQHdDBiSQLt1kf9gGo,60129
 glide/__init__.py,sha256=r0VyTcuIdVzXTTZW6TmWwCsTCIg54ciyba1iDM8PFVQ,7055
 glide/glide.pyi,sha256=6JIpAeADQ-1fU1Mp99pyh7aJORDxAAdtQebV5aByIVI,2093
@@ -27,7 +27,7 @@ glide_shared/commands/server_modules/json_batch.py,sha256=uqkLXPCOkzp53GfffnII1x
 glide_shared/commands/server_modules/json_options.py,sha256=0dFk3EEtcERpmkNoO8930MqfEHLVaQRBRcg3CpL7nmE,3042
 glide_shared/commands/sorted_set.py,sha256=hnXud0Ewn1uBdM6T2xzfCneZWfTeP-2U0XfEDGNCqlI,11466
 glide_shared/commands/stream.py,sha256=MlllDP3teVX8BOB4ANb0JFC4RkUp8SQLCGlGMSPU_s4,15589
-glide_shared/config.py,sha256=UvbRFMSVC8rVux895RG9mLsiIC_8waavc3OfM5bnUQs,33108
+glide_shared/config.py,sha256=FUm6msEcJZTPqHaiN5e1g-7IjD2QHcOOBCYN-VdJFHk,34155
 glide_shared/constants.py,sha256=pS7Xbjvq_qG3mlolVf51cCAGzXRhGdhvGIs2QK3R_b4,4375
 glide_shared/exceptions.py,sha256=Z_szeAE-m1gsoBr5bPQymF6W6wiaunoHvBrt1dHbiYw,1852
 glide_shared/protobuf/command_request_pb2.py,sha256=AHS04ukttbfhXQtEsW5_wAPUS6TzGMd5uqcrSt7lWJo,19320
@@ -35,5 +35,5 @@ glide_shared/protobuf/connection_request_pb2.py,sha256=b3jDmMetSHUrZGmQd8b5DooZZ
 glide_shared/protobuf/response_pb2.py,sha256=oT2GHUwjrxvBVdINRKCvb3_BFN0Fq2wqwHbTj7KAX2E,2110
 glide_shared/protobuf_codec.py,sha256=xwt4-D4WbvNIY_vjOd-00c73HOyNjWq7nN-Z718PBVA,3667
 glide_shared/routes.py,sha256=HFccxCzXQXSi6Y1HFsRUm9ZcbJgCrar6y6uOWpGJWe0,4746
-glide/glide.cpython-313-darwin.so,sha256=6SmuEaig2dG5mXtwWeDaRh15MdckH0qKG3O8mWY3W8k,11432480
-valkey_glide-2.1.0rc2.dist-info/RECORD,,
+glide/glide.cpython-313-darwin.so,sha256=7vJIxawCgOUpe7hV1Bs63BI0ZKzpHT4t-DE-MNJz40A,11416096
+valkey_glide-2.1.0rc4.dist-info/RECORD,,