dqlite-client 0.1.0__py3-none-any.whl

dqlite_client-0.1.0.dist-info/METADATA

@@ -0,0 +1,78 @@
+Metadata-Version: 2.4
+Name: dqlite-client
+Version: 0.1.0
+Summary: Async Python client for dqlite with connection pooling and leader detection
+Project-URL: Homepage, https://github.com/letsdiscodev/python-dqlite-client
+Project-URL: Repository, https://github.com/letsdiscodev/python-dqlite-client
+Project-URL: Issues, https://github.com/letsdiscodev/python-dqlite-client/issues
+Author-email: Antoine Leclair <antoineleclair@gmail.com>
+License-Expression: MIT
+Keywords: async,asyncio,database,distributed,dqlite,sqlite
+Classifier: Development Status :: 3 - Alpha
+Classifier: Framework :: AsyncIO
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Database
+Classifier: Topic :: Database :: Database Engines/Servers
+Classifier: Typing :: Typed
+Requires-Python: >=3.13
+Requires-Dist: dqlite-wire>=0.1.0
+Provides-Extra: dev
+Requires-Dist: mypy>=1.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23; extra == 'dev'
+Requires-Dist: pytest-cov>=4.0; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: ruff>=0.4; extra == 'dev'
+Provides-Extra: yaml
+Requires-Dist: pyyaml>=6.0; extra == 'yaml'
+Description-Content-Type: text/markdown
+
+# dqlite-client
+
+Async Python client for [dqlite](https://dqlite.io/), following asyncpg patterns.
+
+## Installation
+
+```bash
+pip install dqlite-client
+```
+
+## Usage
+
+```python
+import asyncio
+from dqliteclient import connect
+
+async def main():
+    conn = await connect("localhost:9001")
+    async with conn.transaction():
+        await conn.execute("CREATE TABLE IF NOT EXISTS test (id INTEGER PRIMARY KEY, name TEXT)")
+        await conn.execute("INSERT INTO test (name) VALUES (?)", ["hello"])
+        rows = await conn.fetch("SELECT * FROM test")
+        for row in rows:
+            print(row)
+    await conn.close()
+
+asyncio.run(main())
+```
+
+## Connection Pooling
+
+```python
+from dqliteclient import create_pool
+
+pool = await create_pool(["localhost:9001", "localhost:9002", "localhost:9003"])
+async with pool.acquire() as conn:
+    rows = await conn.fetch("SELECT 1")
+```
+
+## Development
+
+See [DEVELOPMENT.md](DEVELOPMENT.md) for setup and contribution guidelines.
+
+## License
+
+MIT

dqlite_client-0.1.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+dqliteclient/__init__.py,sha256=28ef1wvsEvxLr8Q8qQ4PbpQGdy5pq6AzvgdrLuxjZHU,1934
+dqliteclient/cluster.py,sha256=44FCWwnFsvMrPZetoYShES6Jsza_QRHZoZ468xS_LE4,3361
+dqliteclient/connection.py,sha256=-dIk_BS8WOUKpm5XXBBpEj9XuiVdITZOy5kQl_GsiRU,4971
+dqliteclient/exceptions.py,sha256=eq1t7WEyWg33N4rGaEyW69kouduIjs7bcEThhlGPvZo,682
+dqliteclient/node_store.py,sha256=OnhTneRakDeMrvzCTdodtmGLvXsAQZd8BEO1rdu3io8,1236
+dqliteclient/pool.py,sha256=2TJ5PRy0TJky3LzlE0MNEksH8jyMdtWt9q6AM5D5GRw,4125
+dqliteclient/protocol.py,sha256=0iIlkdhIrQtvrMF0DJFTyPr_qLnV-qUaNdFJnWUuaFw,6991
+dqliteclient/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dqliteclient/retry.py,sha256=wYtat__HP5jc3MWcimP0BNUqZfGc1FiM3qDrpTRpxqs,1327
+dqlite_client-0.1.0.dist-info/METADATA,sha256=t2LfkDweHFimfNCKfkoahG1MhVunOeBqlBba18DSH5E,2346
+dqlite_client-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+dqlite_client-0.1.0.dist-info/RECORD,,

dqlite_client-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

dqliteclient/__init__.py

@@ -0,0 +1,82 @@
+"""Async Python client for dqlite."""
+
+from dqliteclient.cluster import ClusterClient
+from dqliteclient.connection import DqliteConnection
+from dqliteclient.exceptions import (
+    ClusterError,
+    ConnectionError,
+    DqliteError,
+    OperationalError,
+    ProtocolError,
+)
+from dqliteclient.node_store import MemoryNodeStore, NodeStore
+from dqliteclient.pool import ConnectionPool
+
+__all__ = [
+    "connect",
+    "create_pool",
+    "DqliteConnection",
+    "ConnectionPool",
+    "ClusterClient",
+    "NodeStore",
+    "MemoryNodeStore",
+    "DqliteError",
+    "ConnectionError",
+    "ProtocolError",
+    "ClusterError",
+    "OperationalError",
+]
+
+__version__ = "0.1.0"
+
+
+async def connect(
+    address: str,
+    *,
+    database: str = "default",
+    timeout: float = 10.0,
+) -> DqliteConnection:
+    """Connect to a dqlite node.
+
+    Args:
+        address: Node address in "host:port" format
+        database: Database name to open
+        timeout: Connection timeout in seconds
+
+    Returns:
+        A connected DqliteConnection
+    """
+    conn = DqliteConnection(address, database=database, timeout=timeout)
+    await conn.connect()
+    return conn
+
+
+async def create_pool(
+    addresses: list[str],
+    *,
+    database: str = "default",
+    min_size: int = 1,
+    max_size: int = 10,
+    timeout: float = 10.0,
+) -> ConnectionPool:
+    """Create a connection pool with automatic leader detection.
+
+    Args:
+        addresses: List of node addresses in "host:port" format
+        database: Database name to open
+        min_size: Minimum number of connections to maintain
+        max_size: Maximum number of connections
+        timeout: Connection timeout in seconds
+
+    Returns:
+        An initialized ConnectionPool
+    """
+    pool = ConnectionPool(
+        addresses,
+        database=database,
+        min_size=min_size,
+        max_size=max_size,
+        timeout=timeout,
+    )
+    await pool.initialize()
+    return pool

dqliteclient/cluster.py

@@ -0,0 +1,105 @@
+"""Cluster management and leader detection for dqlite."""
+
+import asyncio
+
+from dqliteclient.connection import DqliteConnection
+from dqliteclient.exceptions import ClusterError
+from dqliteclient.node_store import MemoryNodeStore, NodeInfo, NodeStore
+from dqliteclient.protocol import DqliteProtocol
+from dqliteclient.retry import retry_with_backoff
+
+
+class ClusterClient:
+    """Client with automatic leader detection and failover."""
+
+    def __init__(
+        self,
+        node_store: NodeStore,
+        *,
+        timeout: float = 10.0,
+    ) -> None:
+        """Initialize cluster client.
+
+        Args:
+            node_store: Store for cluster node information
+            timeout: Connection timeout in seconds
+        """
+        self._node_store = node_store
+        self._timeout = timeout
+        self._leader_address: str | None = None
+
+    @classmethod
+    def from_addresses(cls, addresses: list[str], timeout: float = 10.0) -> "ClusterClient":
+        """Create cluster client from list of addresses."""
+        store = MemoryNodeStore(addresses)
+        return cls(store, timeout=timeout)
+
+    async def find_leader(self) -> str:
+        """Find the current cluster leader.
+
+        Returns the leader address.
+        """
+        nodes = await self._node_store.get_nodes()
+
+        if not nodes:
+            raise ClusterError("No nodes configured")
+
+        errors: list[str] = []
+
+        for node in nodes:
+            try:
+                leader_address = await self._query_leader(node.address)
+                if leader_address:
+                    self._leader_address = leader_address
+                    return leader_address
+            except Exception as e:
+                errors.append(f"{node.address}: {e}")
+                continue
+
+        raise ClusterError(f"Could not find leader. Errors: {'; '.join(errors)}")
+
+    async def _query_leader(self, address: str) -> str | None:
+        """Query a node for the current leader."""
+        host, port_str = address.rsplit(":", 1)
+        port = int(port_str)
+
+        try:
+            reader, writer = await asyncio.wait_for(
+                asyncio.open_connection(host, port),
+                timeout=self._timeout,
+            )
+        except (TimeoutError, OSError):
+            return None
+
+        protocol = DqliteProtocol(reader, writer)
+
+        try:
+            await protocol.handshake()
+            node_id, leader_addr = await protocol.get_leader()
+
+            # If address is empty, this node is the leader
+            if not leader_addr:
+                return address
+
+            return leader_addr
+        finally:
+            protocol.close()
+            await protocol.wait_closed()
+
+    async def connect(self, database: str = "default") -> DqliteConnection:
+        """Connect to the cluster leader.
+
+        Returns a connection to the current leader.
+        """
+
+        async def try_connect() -> DqliteConnection:
+            leader = await self.find_leader()
+            conn = DqliteConnection(leader, database=database, timeout=self._timeout)
+            await conn.connect()
+            return conn
+
+        return await retry_with_backoff(try_connect, max_attempts=5)
+
+    async def update_nodes(self, nodes: list[NodeInfo]) -> None:
+        """Update the node store with new node information."""
+        await self._node_store.set_nodes(nodes)

dqliteclient/connection.py

@@ -0,0 +1,146 @@
+"""High-level connection interface for dqlite."""
+
+import asyncio
+from collections.abc import AsyncIterator
+from contextlib import asynccontextmanager
+from typing import Any
+
+from dqliteclient.exceptions import ConnectionError
+from dqliteclient.protocol import DqliteProtocol
+
+
+class DqliteConnection:
+    """High-level async connection to a dqlite database."""
+
+    def __init__(
+        self,
+        address: str,
+        *,
+        database: str = "default",
+        timeout: float = 10.0,
+    ) -> None:
+        """Initialize connection (does not connect yet).
+
+        Args:
+            address: Node address in "host:port" format
+            database: Database name to open
+            timeout: Connection timeout in seconds
+        """
+        self._address = address
+        self._database = database
+        self._timeout = timeout
+        self._protocol: DqliteProtocol | None = None
+        self._db_id: int | None = None
+        self._in_transaction = False
+
+    @property
+    def address(self) -> str:
+        """Get the connection address."""
+        return self._address
+
+    @property
+    def is_connected(self) -> bool:
+        """Check if connected."""
+        return self._protocol is not None
+
+    async def connect(self) -> None:
+        """Establish connection to the database."""
+        if self._protocol is not None:
+            return
+
+        host, port_str = self._address.rsplit(":", 1)
+        port = int(port_str)
+
+        try:
+            reader, writer = await asyncio.wait_for(
+                asyncio.open_connection(host, port),
+                timeout=self._timeout,
+            )
+        except TimeoutError as e:
+            raise ConnectionError(f"Connection to {self._address} timed out") from e
+        except OSError as e:
+            raise ConnectionError(f"Failed to connect to {self._address}: {e}") from e
+
+        self._protocol = DqliteProtocol(reader, writer)
+
+        try:
+            await self._protocol.handshake()
+            self._db_id = await self._protocol.open_database(self._database)
+        except Exception:
+            self._protocol.close()
+            self._protocol = None
+            raise
+
+    async def close(self) -> None:
+        """Close the connection."""
+        if self._protocol is not None:
+            self._protocol.close()
+            await self._protocol.wait_closed()
+            self._protocol = None
+            self._db_id = None
+
+    async def __aenter__(self) -> "DqliteConnection":
+        await self.connect()
+        return self
+
+    async def __aexit__(self, *args: Any) -> None:
+        await self.close()
+
+    def _ensure_connected(self) -> tuple[DqliteProtocol, int]:
+        """Ensure we're connected and return protocol and db_id."""
+        if self._protocol is None or self._db_id is None:
+            raise ConnectionError("Not connected")
+        return self._protocol, self._db_id
+
+    async def execute(self, sql: str, params: list[Any] | None = None) -> tuple[int, int]:
+        """Execute a SQL statement.
+
+        Returns (last_insert_id, rows_affected).
+        """
+        protocol, db_id = self._ensure_connected()
+        return await protocol.exec_sql(db_id, sql, params)
+
+    async def fetch(self, sql: str, params: list[Any] | None = None) -> list[dict[str, Any]]:
+        """Execute a query and return results as list of dicts."""
+        protocol, db_id = self._ensure_connected()
+        columns, rows = await protocol.query_sql(db_id, sql, params)
+        return [dict(zip(columns, row, strict=True)) for row in rows]
+
+    async def fetchall(self, sql: str, params: list[Any] | None = None) -> list[list[Any]]:
+        """Execute a query and return results as list of lists."""
+        protocol, db_id = self._ensure_connected()
+        _, rows = await protocol.query_sql(db_id, sql, params)
+        return rows
+
+    async def fetchone(self, sql: str, params: list[Any] | None = None) -> dict[str, Any] | None:
+        """Execute a query and return the first result."""
+        results = await self.fetch(sql, params)
+        return results[0] if results else None
+
+    async def fetchval(self, sql: str, params: list[Any] | None = None) -> Any:
+        """Execute a query and return the first column of the first row."""
+        protocol, db_id = self._ensure_connected()
+        _, rows = await protocol.query_sql(db_id, sql, params)
+        if rows and rows[0]:
+            return rows[0][0]
+        return None
+
+    @asynccontextmanager
+    async def transaction(self) -> AsyncIterator[None]:
+        """Context manager for transactions."""
+        if self._in_transaction:
+            # Nested transaction - just yield
+            yield
+            return
+
+        await self.execute("BEGIN")
+        self._in_transaction = True
+
+        try:
+            yield
+            await self.execute("COMMIT")
+        except Exception:
+            await self.execute("ROLLBACK")
+            raise
+        finally:
+            self._in_transaction = False

dqliteclient/exceptions.py

@@ -0,0 +1,37 @@
+"""Exceptions for dqlite client."""
+
+
+class DqliteError(Exception):
+    """Base exception for dqlite client errors."""
+
+    pass
+
+
+class ConnectionError(DqliteError):
+    """Error establishing or maintaining connection."""
+
+    pass
+
+
+class ProtocolError(DqliteError):
+    """Protocol-level error."""
+
+    pass
+
+
+class ClusterError(DqliteError):
+    """Cluster-related error (leader not found, etc)."""
+
+    pass
+
+
+class OperationalError(DqliteError):
+    """Database operation error."""
+
+    code: int
+    message: str
+
+    def __init__(self, code: int, message: str) -> None:
+        self.code = code
+        self.message = message
+        super().__init__(f"[{code}] {message}")

dqliteclient/node_store.py

@@ -0,0 +1,45 @@
+"""Node store interfaces for cluster discovery."""
+
+from abc import ABC, abstractmethod
+from dataclasses import dataclass
+
+
+@dataclass
+class NodeInfo:
+    """Information about a cluster node."""
+
+    node_id: int
+    address: str
+    role: int  # 0=spare, 1=voter, 2=standby
+
+
+class NodeStore(ABC):
+    """Abstract interface for storing cluster node information."""
+
+    @abstractmethod
+    async def get_nodes(self) -> list[NodeInfo]:
+        """Get list of known nodes."""
+        ...
+
+    @abstractmethod
+    async def set_nodes(self, nodes: list[NodeInfo]) -> None:
+        """Update list of known nodes."""
+        ...
+
+
+class MemoryNodeStore(NodeStore):
+    """In-memory node store."""
+
+    def __init__(self, initial_addresses: list[str] | None = None) -> None:
+        self._nodes: list[NodeInfo] = []
+        if initial_addresses:
+            for i, addr in enumerate(initial_addresses):
+                self._nodes.append(NodeInfo(node_id=i, address=addr, role=1))
+
+    async def get_nodes(self) -> list[NodeInfo]:
+        """Get list of known nodes."""
+        return list(self._nodes)
+
+    async def set_nodes(self, nodes: list[NodeInfo]) -> None:
+        """Update list of known nodes."""
+        self._nodes = list(nodes)

dqliteclient/pool.py

@@ -0,0 +1,130 @@
+"""Connection pooling for dqlite."""
+
+import asyncio
+from collections.abc import AsyncIterator
+from contextlib import asynccontextmanager
+from typing import Any
+
+from dqliteclient.cluster import ClusterClient
+from dqliteclient.connection import DqliteConnection
+from dqliteclient.exceptions import ConnectionError
+
+
+class ConnectionPool:
+    """Connection pool with automatic leader detection."""
+
+    def __init__(
+        self,
+        addresses: list[str],
+        *,
+        database: str = "default",
+        min_size: int = 1,
+        max_size: int = 10,
+        timeout: float = 10.0,
+    ) -> None:
+        """Initialize connection pool.
+
+        Args:
+            addresses: List of node addresses
+            database: Database name
+            min_size: Minimum connections to maintain
+            max_size: Maximum connections allowed
+            timeout: Connection timeout
+        """
+        self._addresses = addresses
+        self._database = database
+        self._min_size = min_size
+        self._max_size = max_size
+        self._timeout = timeout
+
+        self._cluster = ClusterClient.from_addresses(addresses, timeout=timeout)
+        self._pool: asyncio.Queue[DqliteConnection] = asyncio.Queue(maxsize=max_size)
+        self._size = 0
+        self._lock = asyncio.Lock()
+        self._closed = False
+
+    async def initialize(self) -> None:
+        """Initialize the pool with minimum connections."""
+        for _ in range(self._min_size):
+            conn = await self._create_connection()
+            await self._pool.put(conn)
+
+    async def _create_connection(self) -> DqliteConnection:
+        """Create a new connection to the leader."""
+        conn = await self._cluster.connect(database=self._database)
+        self._size += 1
+        return conn
+
+    @asynccontextmanager
+    async def acquire(self) -> AsyncIterator[DqliteConnection]:
+        """Acquire a connection from the pool."""
+        if self._closed:
+            raise ConnectionError("Pool is closed")
+
+        conn: DqliteConnection | None = None
+
+        # Try to get from pool
+        try:
+            conn = self._pool.get_nowait()
+        except asyncio.QueueEmpty:
+            # Create new if under max
+            async with self._lock:
+                if self._size < self._max_size:
+                    conn = await self._create_connection()
+
+        # Wait for one if at max
+        if conn is None:
+            conn = await self._pool.get()
+
+        try:
+            # Verify connection is still good
+            if not conn.is_connected:
+                await conn.connect()
+
+            yield conn
+        except Exception:
+            # On error, close connection and create new one
+            import contextlib
+
+            with contextlib.suppress(Exception):
+                await conn.close()
+            self._size -= 1
+            raise
+        else:
+            # Return to pool
+            try:
+                self._pool.put_nowait(conn)
+            except asyncio.QueueFull:
+                # Pool full, close connection
+                await conn.close()
+                self._size -= 1
+
+    async def execute(self, sql: str, params: list[Any] | None = None) -> tuple[int, int]:
+        """Execute a SQL statement using a pooled connection."""
+        async with self.acquire() as conn:
+            return await conn.execute(sql, params)
+
+    async def fetch(self, sql: str, params: list[Any] | None = None) -> list[dict[str, Any]]:
+        """Execute a query using a pooled connection."""
+        async with self.acquire() as conn:
+            return await conn.fetch(sql, params)
+
+    async def close(self) -> None:
+        """Close all connections in the pool."""
+        self._closed = True
+
+        while not self._pool.empty():
+            try:
+                conn = self._pool.get_nowait()
+                await conn.close()
+            except asyncio.QueueEmpty:
+                break
+
+        self._size = 0
+
+    async def __aenter__(self) -> "ConnectionPool":
+        await self.initialize()
+        return self
+
+    async def __aexit__(self, *args: Any) -> None:
+        await self.close()

dqliteclient/protocol.py

@@ -0,0 +1,213 @@
+"""Low-level protocol handler for dqlite."""
+
+import asyncio
+from typing import Any
+
+from dqliteclient.exceptions import ConnectionError, OperationalError, ProtocolError
+from dqlitewire import MessageDecoder, MessageEncoder, ReadBuffer
+from dqlitewire.messages import (
+    ClientRequest,
+    DbResponse,
+    ExecSqlRequest,
+    FailureResponse,
+    FinalizeRequest,
+    LeaderRequest,
+    LeaderResponse,
+    OpenRequest,
+    PrepareRequest,
+    QuerySqlRequest,
+    ResultResponse,
+    RowsResponse,
+    StmtResponse,
+    WelcomeResponse,
+)
+from dqlitewire.messages.base import Message
+
+
+class DqliteProtocol:
+    """Low-level protocol handler for a single dqlite connection."""
+
+    def __init__(
+        self,
+        reader: asyncio.StreamReader,
+        writer: asyncio.StreamWriter,
+    ) -> None:
+        self._reader = reader
+        self._writer = writer
+        self._encoder = MessageEncoder()
+        self._decoder = MessageDecoder(is_request=False)
+        self._buffer = ReadBuffer()
+        self._client_id = 0
+        self._heartbeat_timeout = 0
+
+    async def handshake(self, client_id: int = 0) -> int:
+        """Perform protocol handshake.
+
+        Returns the heartbeat timeout from server.
+        """
+        # Send protocol version
+        self._writer.write(self._encoder.encode_handshake())
+        await self._writer.drain()
+
+        # Send client registration
+        request = ClientRequest(client_id=client_id)
+        self._writer.write(request.encode())
+        await self._writer.drain()
+
+        # Read welcome response
+        response = await self._read_response()
+
+        if isinstance(response, FailureResponse):
+            raise ProtocolError(f"Handshake failed: {response.message}")
+
+        if not isinstance(response, WelcomeResponse):
+            raise ProtocolError(f"Expected WelcomeResponse, got {type(response).__name__}")
+
+        self._client_id = client_id
+        self._heartbeat_timeout = response.heartbeat_timeout
+        return response.heartbeat_timeout
+
+    async def get_leader(self) -> tuple[int, str]:
+        """Request leader information.
+
+        Returns (node_id, address).
+        """
+        request = LeaderRequest()
+        self._writer.write(request.encode())
+        await self._writer.drain()
+
+        response = await self._read_response()
+
+        if isinstance(response, FailureResponse):
+            raise OperationalError(response.code, response.message)
+
+        if not isinstance(response, LeaderResponse):
+            raise ProtocolError(f"Expected LeaderResponse, got {type(response).__name__}")
+
+        return response.node_id, response.address
+
+    async def open_database(self, name: str, flags: int = 0, vfs: str = "") -> int:
+        """Open a database.
+
+        Returns the database ID.
+        """
+        request = OpenRequest(name=name, flags=flags, vfs=vfs)
+        self._writer.write(request.encode())
+        await self._writer.drain()
+
+        response = await self._read_response()
+
+        if isinstance(response, FailureResponse):
+            raise OperationalError(response.code, response.message)
+
+        if not isinstance(response, DbResponse):
+            raise ProtocolError(f"Expected DbResponse, got {type(response).__name__}")
+
+        return response.db_id
+
+    async def prepare(self, db_id: int, sql: str) -> tuple[int, int]:
+        """Prepare a SQL statement.
+
+        Returns (stmt_id, num_params).
+        """
+        request = PrepareRequest(db_id=db_id, sql=sql)
+        self._writer.write(request.encode())
+        await self._writer.drain()
+
+        response = await self._read_response()
+
+        if isinstance(response, FailureResponse):
+            raise OperationalError(response.code, response.message)
+
+        if not isinstance(response, StmtResponse):
+            raise ProtocolError(f"Expected StmtResponse, got {type(response).__name__}")
+
+        return response.stmt_id, response.num_params
+
+    async def finalize(self, db_id: int, stmt_id: int) -> None:
+        """Finalize (close) a prepared statement."""
+        request = FinalizeRequest(db_id=db_id, stmt_id=stmt_id)
+        self._writer.write(request.encode())
+        await self._writer.drain()
+
+        response = await self._read_response()
+
+        if isinstance(response, FailureResponse):
+            raise OperationalError(response.code, response.message)
+
+    async def exec_sql(
+        self, db_id: int, sql: str, params: list[Any] | None = None
+    ) -> tuple[int, int]:
+        """Execute SQL directly.
+
+        Returns (last_insert_id, rows_affected).
+        """
+        request = ExecSqlRequest(db_id=db_id, sql=sql, params=params or [])
+        self._writer.write(request.encode())
+        await self._writer.drain()
+
+        response = await self._read_response()
+
+        if isinstance(response, FailureResponse):
+            raise OperationalError(response.code, response.message)
+
+        if not isinstance(response, ResultResponse):
+            raise ProtocolError(f"Expected ResultResponse, got {type(response).__name__}")
+
+        return response.last_insert_id, response.rows_affected
+
+    async def query_sql(
+        self, db_id: int, sql: str, params: list[Any] | None = None
+    ) -> tuple[list[str], list[list[Any]]]:
+        """Execute a query directly.
+
+        Returns (column_names, rows).
+        """
+        request = QuerySqlRequest(db_id=db_id, sql=sql, params=params or [])
+        self._writer.write(request.encode())
+        await self._writer.drain()
+
+        response = await self._read_response()
+
+        if isinstance(response, FailureResponse):
+            raise OperationalError(response.code, response.message)
+
+        if not isinstance(response, RowsResponse):
+            raise ProtocolError(f"Expected RowsResponse, got {type(response).__name__}")
+
+        # Store column names from first response
+        column_names = response.column_names
+
+        # Handle multi-part responses
+        all_rows = list(response.rows)
+        while response.has_more:
+            next_response = await self._read_response()
+            if isinstance(next_response, RowsResponse):
+                all_rows.extend(next_response.rows)
+                response = next_response
+            else:
+                break
+
+        return column_names, all_rows
+
+    async def _read_response(self) -> Message:
+        """Read and decode the next response message."""
+        while not self._decoder.has_message():
+            data = await self._reader.read(4096)
+            if not data:
+                raise ConnectionError("Connection closed by server")
+            self._decoder.feed(data)
+
+        message = self._decoder.decode()
+        if message is None:
+            raise ProtocolError("Failed to decode message")
+
+        return message
+
+    def close(self) -> None:
+        """Close the connection."""
+        self._writer.close()
+
+    async def wait_closed(self) -> None:
+        """Wait for the connection to close."""
+        await self._writer.wait_closed()

dqliteclient/retry.py

@@ -0,0 +1,51 @@
+"""Retry utilities with exponential backoff."""
+
+import asyncio
+import random
+from collections.abc import Awaitable, Callable
+
+
+async def retry_with_backoff[T](
+    func: Callable[[], Awaitable[T]],
+    max_attempts: int = 5,
+    base_delay: float = 0.1,
+    max_delay: float = 10.0,
+    jitter: float = 0.1,
+) -> T:
+    """Retry an async function with exponential backoff.
+
+    Args:
+        func: Async function to retry
+        max_attempts: Maximum number of attempts
+        base_delay: Initial delay between retries in seconds
+        max_delay: Maximum delay between retries
+        jitter: Random jitter factor (0-1)
+
+    Returns:
+        Result of the function
+
+    Raises:
+        The last exception if all attempts fail
+    """
+    last_error: Exception | None = None
+
+    for attempt in range(max_attempts):
+        try:
+            return await func()
+        except Exception as e:
+            last_error = e
+
+            if attempt == max_attempts - 1:
+                break
+
+            # Calculate delay with exponential backoff
+            delay = min(base_delay * (2**attempt), max_delay)
+
+            # Add jitter
+            if jitter > 0:
+                delay = delay * (1 + random.uniform(-jitter, jitter))
+
+            await asyncio.sleep(delay)
+
+    assert last_error is not None
+    raise last_error