NA

Author: Sai Eswar

examples/python/getting-started/consumer.py

@@ -32,7 +32,7 @@
 PARTITION_ID = 0
 BATCHES_LIMIT = 5
 
-ArgNamespace = namedtuple("ArgNamespace", ["tcp_server_address"])
+ArgNamespace = namedtuple("ArgNamespace", ["tcp_server_address", "tls", "tls_ca_file", "username", "password"])
 
 
 class ValidateUrl(argparse.Action):
@@ -57,23 +57,71 @@ def parse_args():
         action=ValidateUrl,
         default="127.0.0.1:8090",
     )
-    return parser.parse_args()
-
+    parser.add_argument(
+        "--tls",
+        action="store_true",
+        default=False,
+        help="Enable TLS for TCP connection",
+    )
+    parser.add_argument(
+        "--tls-ca-file",
+        default="",
+        help="Path to TLS CA certificate file",
+    )
+    parser.add_argument(
+        "--username",
+        default="iggy",
+        help="Username for authentication",
+    )
+    parser.add_argument(
+        "--password",
+        default="iggy",
+        help="Password for authentication",
+    )
+    args = parser.parse_args()
+    
+    # Validate TLS requirements
+    if args.tls and not args.tls_ca_file:
+        parser.error("--tls requires --tls-ca-file")
+    
+    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 = [f"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)
+    
+    return conn_str
 
 async def main():
     args: ArgNamespace = parse_args()
-    client = IggyClient(args.tcp_server_address)
+
+    # 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)
     try:
         logger.info("Connecting to IggyClient...")
         await client.connect()
         logger.info("Connected. Logging in user...")
-        await client.login_user("iggy", "iggy")
+        await client.login_user(args.username, args.password)
         logger.info("Logged in.")
         await consume_messages(client)
     except Exception as error:
         logger.exception("Exception occurred in main function: {}", error)
 
-
+        
 async def consume_messages(client: IggyClient):
     interval = 0.5  # 500 milliseconds in seconds for asyncio.sleep
     logger.info(

examples/python/getting-started/producer.py

@@ -33,7 +33,7 @@
 PARTITION_ID = 0
 BATCHES_LIMIT = 5
 
-ArgNamespace = namedtuple("ArgNamespace", ["tcp_server_address"])
+ArgNamespace = namedtuple("ArgNamespace", ["tcp_server_address", "tls", "tls_ca_file", "username", "password"])
 
 
 class ValidateUrl(argparse.Action):
@@ -58,16 +58,65 @@ def parse_args():
         action=ValidateUrl,
         default="127.0.0.1:8090",
     )
-    return parser.parse_args()
-
+    parser.add_argument(
+        "--tls",
+        action="store_true",
+        default=False,
+        help="Enable TLS for TCP connection",
+    )
+    parser.add_argument(
+        "--tls-ca-file",
+        default="",
+        help="Path to TLS CA certificate file",
+    )
+    parser.add_argument(
+        "--username",
+        default="iggy",
+        help="Username for authentication",
+    )
+    parser.add_argument(
+        "--password",
+        default="iggy",
+        help="Password for authentication",
+    )
+    args = parser.parse_args()
+    
+    # Validate TLS requirements
+    if args.tls and not args.tls_ca_file:
+        parser.error("--tls requires --tls-ca-file")
+    
+    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 = [f"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)
+    
+    return conn_str
 
 async def main():
     args: ArgNamespace = parse_args()
-    client = IggyClient(args.tcp_server_address)
+    # Build connection string with TLS support
+    connection_string = build_connection_string(args)
+    logger.info(f"Connection string: {connection_string}")
+    logger.info(f"Connecting to {args.tcp_server_address} (TLS: {args.tls})")
+
+    
+    client = IggyClient.from_connection_string(connection_string)
     logger.info("Connecting to IggyClient")
     await client.connect()
     logger.info("Connected. Logging in user...")
-    await client.login_user("iggy", "iggy")
+    await client.login_user(args.username, args.password)
     logger.info("Logged in.")
     await init_system(client)
     await produce_messages(client)