feat(python): expose TCP client configuration

core/sdk/src/prelude.rs

@@ -51,18 +51,18 @@ pub use iggy_common::{
     Aes256GcmEncryptor, Args, ArgsOptional, AutoLogin, CacheMetrics, CacheMetricsKey, ClientError,
     ClientInfoDetails, ClusterMetadata, ClusterNode, ClusterNodeRole, ClusterNodeStatus,
     CompressionAlgorithm, Consumer, ConsumerGroup, ConsumerGroupDetails, ConsumerGroupMember,
-    ConsumerKind, EncryptorKind, GlobalPermissions, HeaderField, HeaderKey, HeaderKind,
-    HeaderValue, HttpClientConfig, HttpClientConfigBuilder, HttpMethod, IdKind, Identifier,
-    IdentityInfo, IggyByteSize, IggyDuration, IggyError, IggyExpiry, IggyIndexView, IggyMessage,
-    IggyMessageHeader, IggyMessageHeaderView, IggyMessageView, IggyMessageViewIterator,
-    IggyTimestamp, MaxTopicSize, Partition, Partitioner, Partitioning, Permissions,
-    PersonalAccessTokenExpiry, PollMessages, PolledMessages, PollingKind, PollingStrategy,
-    QuicClientConfig, QuicClientConfigBuilder, QuicClientReconnectionConfig, SendMessages,
-    SendMessagesConfirmationResponse, SendMessagesResponse, Sizeable, SnapshotCompression, Stats,
-    Stream, StreamDetails, StreamPermissions, SystemSnapshotType, TcpClientConfig,
-    TcpClientConfigBuilder, TcpClientReconnectionConfig, Topic, TopicDetails, TopicPermissions,
-    TransportEndpoints, TransportProtocol, UserId, UserInfo, UserInfoDetails, UserStatus,
-    Validatable, WebSocketClientConfig, WebSocketClientConfigBuilder,
+    ConsumerKind, Credentials, EncryptorKind, GlobalPermissions, HeaderField, HeaderKey,
+    HeaderKind, HeaderValue, HttpClientConfig, HttpClientConfigBuilder, HttpMethod, IdKind,
+    Identifier, IdentityInfo, IggyByteSize, IggyDuration, IggyError, IggyExpiry, IggyIndexView,
+    IggyMessage, IggyMessageHeader, IggyMessageHeaderView, IggyMessageView,
+    IggyMessageViewIterator, IggyTimestamp, MaxTopicSize, Partition, Partitioner, Partitioning,
+    Permissions, PersonalAccessTokenExpiry, PollMessages, PolledMessages, PollingKind,
+    PollingStrategy, QuicClientConfig, QuicClientConfigBuilder, QuicClientReconnectionConfig,
+    SendMessages, SendMessagesConfirmationResponse, SendMessagesResponse, Sizeable,
+    SnapshotCompression, Stats, Stream, StreamDetails, StreamPermissions, SystemSnapshotType,
+    TcpClientConfig, TcpClientConfigBuilder, TcpClientReconnectionConfig, Topic, TopicDetails,
+    TopicPermissions, TransportEndpoints, TransportProtocol, UserId, UserInfo, UserInfoDetails,
+    UserStatus, Validatable, WebSocketClientConfig, WebSocketClientConfigBuilder,
     WebSocketClientReconnectionConfig, defaults, locking,
 };
 pub use iggy_common::{

examples/python/getting-started/consumer.py

@@ -19,8 +19,16 @@
 import asyncio
 import typing
 import urllib.parse
-
-from apache_iggy import IggyClient, PollingStrategy, ReceiveMessage
+from datetime import timedelta
+
+from apache_iggy import (
+    AutoLogin,
+    IggyClient,
+    PollingStrategy,
+    ReceiveMessage,
+    TcpConfig,
+    TcpReconnectionConfig,
+)
 from loguru import logger
 
 STREAM_NAME = "sample-stream"
@@ -91,34 +99,34 @@ def parse_args() -> ArgNamespace:
     return ArgNamespace(**vars(args))
 
 
-def build_connection_string(args) -> str:
-    """Build a connection string with TLS support."""
-
-    conn_str = f"iggy://{args.username}:{args.password}@{args.tcp_server_address}"
-
-    if args.tls:
-        # Extract domain from server address (host:port -> host)
-        host = args.tcp_server_address.split(":")[0]
-        query_params = ["tls=true", f"tls_domain={host}"]
-
-        # Add CA file if provided
-        if args.tls_ca_file:
-            query_params.append(f"tls_ca_file={args.tls_ca_file}")
-        conn_str += "?" + "&".join(query_params)
+def build_config(args: ArgNamespace) -> TcpConfig:
+    """Build a TCP client configuration with auto-login and reconnection."""
 
-    return conn_str
+    return TcpConfig(
+        server_address=args.tcp_server_address,
+        auto_login=AutoLogin.username_password(args.username, args.password),
+        reconnection=TcpReconnectionConfig(
+            enabled=True,
+            interval=timedelta(seconds=1),
+        ),
+        tls_enabled=args.tls,
+        tls_ca_file=args.tls_ca_file or None,
+    )
 
 
 async def main():
     args: ArgNamespace = parse_args()
+    try:
+        config = build_config(args)
+    except ValueError as error:
+        logger.error(f"Invalid client configuration: {error}")
+        return
+    logger.info(f"Connecting to {args.tcp_server_address} (TLS: {args.tls})")
 
-    # Build connection string with TLS support
-    connection_string = build_connection_string(args)
-    logger.info(f"Connection string: {connection_string}")
-
-    client = IggyClient.from_connection_string(connection_string)
+    client = IggyClient(config)
     try:
         logger.info("Connecting to IggyClient...")
+        # No login_user() call: auto_login replays the credentials on every connect.
         await client.connect()
         logger.info("Connected.")
         await consume_messages(client)

examples/python/getting-started/producer.py

@@ -19,8 +19,16 @@
 import asyncio
 import typing
 import urllib.parse
-
-from apache_iggy import IggyClient, StreamDetails, TopicDetails
+from datetime import timedelta
+
+from apache_iggy import (
+    AutoLogin,
+    IggyClient,
+    StreamDetails,
+    TcpConfig,
+    TcpReconnectionConfig,
+    TopicDetails,
+)
 from apache_iggy import SendMessage as Message
 from loguru import logger
 
@@ -92,33 +100,33 @@ def parse_args() -> ArgNamespace:
     return ArgNamespace(**vars(args))
 
 
-def build_connection_string(args) -> str:
-    """Build a connection string with TLS support."""
-
-    conn_str = f"iggy://{args.username}:{args.password}@{args.tcp_server_address}"
-
-    if args.tls:
-        # Extract domain from server address (host:port -> host)
-        host = args.tcp_server_address.split(":")[0]
-        query_params = ["tls=true", f"tls_domain={host}"]
+def build_config(args: ArgNamespace) -> TcpConfig:
+    """Build a TCP client configuration with auto-login and reconnection."""
 
-        # Add CA file if provided
-        if args.tls_ca_file:
-            query_params.append(f"tls_ca_file={args.tls_ca_file}")
-        conn_str += "?" + "&".join(query_params)
-
-    return conn_str
+    return TcpConfig(
+        server_address=args.tcp_server_address,
+        auto_login=AutoLogin.username_password(args.username, args.password),
+        reconnection=TcpReconnectionConfig(
+            enabled=True,
+            interval=timedelta(seconds=1),
+        ),
+        tls_enabled=args.tls,
+        tls_ca_file=args.tls_ca_file or None,
+    )
 
 
 async def main():
     args: ArgNamespace = parse_args()
-    # Build connection string with TLS support
-    connection_string = build_connection_string(args)
-    logger.info(f"Connection string: {connection_string}")
+    try:
+        config = build_config(args)
+    except ValueError as error:
+        logger.error(f"Invalid client configuration: {error}")
+        return
     logger.info(f"Connecting to {args.tcp_server_address} (TLS: {args.tls})")
 
-    client = IggyClient.from_connection_string(connection_string)
+    client = IggyClient(config)
     logger.info("Connecting to IggyClient")
+    # No login_user() call: auto_login replays the credentials on every connect.
     await client.connect()
     logger.info("Connected.")
     await init_system(client)

foreign/python/Cargo.toml

@@ -47,4 +47,5 @@ pyo3-async-runtimes = { version = "0.29.0", features = [
     "tokio-runtime",
 ] }
 pyo3-stub-gen = "0.23.0"
+secrecy = "0.10"
 tokio = "1.53.1"

foreign/python/README.md

@@ -139,6 +139,42 @@ running prek / committing / pushing. This list is not exhaustive and other hook
    ./scripts/ci/markdownlint.sh --fix foreign/python/README.md # read the diff after applying this, sometimes it gives unwanted results, e.g. messing up enumerations
    ```
 
+## Client Configuration
+
+`IggyClient` takes either a server address or a `TcpConfig`:
+
+```python
+import asyncio
+from datetime import timedelta
+
+from apache_iggy import AutoLogin, IggyClient, TcpConfig, TcpReconnectionConfig
+
+
+async def main():
+    client = IggyClient(
+        TcpConfig(
+            server_address="127.0.0.1:8090",
+            auto_login=AutoLogin.username_password("iggy", "iggy"),
+            reconnection=TcpReconnectionConfig(
+                enabled=True,
+                max_retries=10,
+                interval=timedelta(seconds=2),
+                reestablish_after=timedelta(seconds=30),
+            ),
+            heartbeat_interval=timedelta(seconds=5),
+            # tls_enabled=True,
+            # tls_domain="localhost",
+            # tls_ca_file="../../core/certs/iggy_ca_cert.pem",
+            # tls_validate_certificate=True,
+            # nodelay=True,
+        )
+    )
+    await client.connect()
+
+
+asyncio.run(main())
+```
+
 ## Examples
 
 Refer to the [examples/python/](https://github.com/apache/iggy/tree/master/examples/python) directory for usage examples.