Thread max_total_rows through ClusterClient + ConnectionPool

Cycle 9 — the cumulative row-cap knob added earlier only reached as
far as DqliteConnection. ClusterClient.connect() and ConnectionPool
now accept and forward max_total_rows so every layer above the raw
protocol can set it from one place. Default is still 10_000_000;
None disables the cap.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Author: antoineleclair

src/dqliteclient/cluster.py

@@ -148,15 +148,28 @@ async def _query_leader(self, address: str) -> str | None:
         finally:
             writer.close()
 
-    async def connect(self, database: str = "default") -> DqliteConnection:
+    async def connect(
+        self,
+        database: str = "default",
+        *,
+        max_total_rows: int | None = 10_000_000,
+    ) -> DqliteConnection:
         """Connect to the cluster leader.
 
-        Returns a connection to the current leader.
+        Returns a connection to the current leader. ``max_total_rows``
+        is forwarded to the underlying :class:`DqliteConnection` so
+        callers (including :class:`ConnectionPool`) can tune the
+        cumulative row cap from one place.
         """
 
         async def try_connect() -> DqliteConnection:
             leader = await self.find_leader()
-            conn = DqliteConnection(leader, database=database, timeout=self._timeout)
+            conn = DqliteConnection(
+                leader,
+                database=database,
+                timeout=self._timeout,
+                max_total_rows=max_total_rows,
+            )
             await conn.connect()
             return conn
 

src/dqliteclient/pool.py

@@ -57,6 +57,7 @@ def __init__(
         timeout: float = 10.0,
         cluster: ClusterClient | None = None,
         node_store: NodeStore | None = None,
+        max_total_rows: int | None = 10_000_000,
     ) -> None:
         """Initialize connection pool.
 
@@ -91,6 +92,7 @@ def __init__(
         self._min_size = min_size
         self._max_size = max_size
         self._timeout = timeout
+        self._max_total_rows = max_total_rows
 
         if cluster is not None:
             self._cluster = cluster
@@ -124,7 +126,9 @@ async def initialize(self) -> None:
 
     async def _create_connection(self) -> DqliteConnection:
         """Create a new connection to the leader."""
-        conn = await self._cluster.connect(database=self._database)
+        conn = await self._cluster.connect(
+            database=self._database, max_total_rows=self._max_total_rows
+        )
         self._size += 1
         return conn