Implement reconnects on writing and reading frames (#45)

Author: vrslev

.github/dependabot.yml

@@ -9,8 +9,3 @@ updates:
     directory: /
     schedule:
       interval: monthly
-
-  - package-ecosystem: docker
-    directory: /
-    schedule:
-      interval: monthly

Dockerfile

@@ -1,13 +1,14 @@
 ARG PYTHON_VERSION
-FROM ghcr.io/astral-sh/uv:0.2.31 as uv
 FROM python:${PYTHON_VERSION}-slim-bullseye
 
+# hadolint ignore=DL3013,DL3042
+RUN pip install uv
+
 WORKDIR /app
 COPY pyproject.toml README.md ./
 COPY stompman/__init__.py stompman/__init__.py
 
 ENV SETUPTOOLS_SCM_PRETEND_VERSION=0
-RUN --mount=from=uv,source=/uv,target=/bin/uv \
-    --mount=type=cache,target=~/.cache/uv \
+RUN --mount=type=cache,target=~/.cache/uv \
     uv lock && uv sync
 COPY . .

README.md

@@ -105,18 +105,9 @@ stompman takes care of cleaning up resources automatically. When you leave the c
 
 ### Handling Connectivity Issues
 
-- If multiple servers were provided, stompman will attempt to connect to each one simultaneously and will use the first that succeeds.
+- If multiple servers were provided, stompman will attempt to connect to each one simultaneously and will use the first that succeeds. If all servers fail to connect, an `stompman.FailedAllConnectAttemptsError` will be raised. In normal situation it doesn't need to be handled: tune retry and timeout parameters in `stompman.Client()` to your needs.
 
-- If all servers fail to connect, an `stompman.FailedAllConnectAttemptsError` will be raised. In normal situation it doesn't need to be handled: tune retry and timeout parameters in `stompman.Client()` to your needs.
-
-- If a connection is lost, a `stompman.ConnectionLostError` will be raised. You should implement reconnect logic manually, for example, with stamina:
-
-  ```python
-  for attempt in stamina.retry_context(on=stompman.ConnectionLostError):
-      with attempt:
-          async with stompman.Client(...) as client:
-              ...
-  ```
+- When connection is lost, stompman will handle it automatically. `stompman.FailedAllConnectAttemptsError` will be raised if all connection attempts fail. `stompman.RepeatedConnectionLostError` will be raised if connection succeeds but operation (like sending a frame) leads to connection getting lost.
 
 ### ...and caveats
 

pyproject.toml

@@ -58,7 +58,7 @@ ignore = ["D1", "D203", "D213", "COM812", "ISC001", "CPY001"]
 extend-per-file-ignores = { "tests/*" = ["S101", "SLF001", "ARG"] }
 
 [tool.pytest.ini_options]
-addopts = "--cov -s"
+addopts = "--cov -s -vv"
 
 [tool.coverage.report]
 skip_covered = true

stompman/__init__.py

@@ -1,15 +1,19 @@
-from stompman.client import Client, ConnectionParameters, Heartbeat, Subscription, Transaction
+from stompman.client import Client, Subscription, Transaction
+from stompman.config import ConnectionParameters, Heartbeat
 from stompman.connection import AbstractConnection, Connection
+from stompman.connection_manager import ActiveConnectionState, ConnectionManager
 from stompman.errors import (
     ConnectionConfirmationTimeoutError,
     ConnectionLostError,
     Error,
     FailedAllConnectAttemptsError,
+    RepeatedConnectionLostError,
     UnsupportedProtocolVersionError,
 )
 from stompman.frames import (
     AbortFrame,
     AckFrame,
+    AckMode,
     AnyClientFrame,
     AnyRealServerFrame,
     AnyServerFrame,
@@ -32,6 +36,8 @@
     "AbortFrame",
     "AbstractConnection",
     "AckFrame",
+    "AckMode",
+    "ActiveConnectionState",
     "AnyClientFrame",
     "AnyRealServerFrame",
     "AnyServerFrame",
@@ -43,6 +49,7 @@
     "Connection",
     "ConnectionConfirmationTimeoutError",
     "ConnectionLostError",
+    "ConnectionManager",
     "ConnectionParameters",
     "DisconnectFrame",
     "Error",
@@ -53,6 +60,7 @@
     "MessageFrame",
     "NackFrame",
     "ReceiptFrame",
+    "RepeatedConnectionLostError",
     "SendFrame",
     "SubscribeFrame",
     "Subscription",