kurrentdbclient 1.2b1__tar.gz → 1.2.2__tar.gz

{kurrentdbclient-1.2b1 → kurrentdbclient-1.2.2}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: kurrentdbclient
-Version: 1.2b1
+Version: 1.2.2
 Summary: Python gRPC Client for KurrentDB
 License: BSD-3-Clause
 License-File: LICENSE
 Author: John Bywater
 Author-email: john.bywater@appropriatesoftware.net
 Requires-Python: >=3.10
-Classifier: Development Status :: 4 - Beta
+Classifier: Development Status :: 5 - Production/Stable
 Classifier: License :: OSI Approved :: BSD License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10

{kurrentdbclient-1.2b1 → kurrentdbclient-1.2.2}/kurrentdbclient/__init__.py

@@ -26,7 +26,7 @@ from kurrentdbclient.streams import (
     ReadResponse,
 )
 
-__version__ = "1.2b1"
+__version__ = "1.2.2"
 
 __all__ = [
     "DEFAULT_EXCLUDE_FILTER",

{kurrentdbclient-1.2b1 → kurrentdbclient-1.2.2}/kurrentdbclient/asyncio_client.py

@@ -52,6 +52,7 @@ from kurrentdbclient.exceptions import (
     GrpcError,
     NodeIsNotLeaderError,
     NotFoundError,
+    ProgrammingError,
     ServiceUnavailableError,
 )
 
@@ -70,6 +71,7 @@ def autoreconnect(f: _TCallable) -> _TCallable:
     async def autoreconnect_decorator(
         client: AsyncKurrentDBClient, *args: Any, **kwargs: Any
     ) -> Any:
+        await client.connect()
         try:
             return await f(client, *args, **kwargs)
 
@@ -129,22 +131,36 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
             private_key=private_key,
             certificate_chain=certificate_chain,
         )
+        self._connection: AsyncKurrentDBConnection | None = None
         self._is_reconnection_required = Event()
-        self._reconnection_lock = Lock()
+        self._connection_lock = Lock()
+
+    async def connect(self) -> None:
+        if self._connection is None:
+            async with self._connection_lock:
+                if self._connection is None:
+                    self._connection = await self._connect()
+                else:  # for coverage with Python <= 3.11
+                    pass
 
     @property
-    def connection_target(self) -> str:
-        return self._connection.grpc_target
+    def connection(self) -> AsyncKurrentDBConnection:
+        connection = self._connection
+        if connection is None:
+            msg = "Connection was never established"
+            raise ProgrammingError(msg)
+        return connection
 
-    async def connect(self) -> None:
-        self._connection = await self._connect()
+    @property
+    def connection_target(self) -> str:
+        return self.connection.grpc_target
 
     async def reconnect(self, grpc_target: str | None = None) -> None:
         self._is_reconnection_required.set()
-        async with self._reconnection_lock:
+        async with self._connection_lock:
             if self._is_reconnection_required.is_set():
                 new = await self._connect(grpc_target)
-                old, self._connection = self._connection, new
+                old, self._connection = self.connection, new
                 await old.close()
                 self._is_reconnection_required.clear()
             else:  # pragma: no cover
@@ -283,7 +299,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         if isinstance(events, NewEvent):
             events = [events]
 
-        return await self._connection.streams.batch_append(
+        return await self.connection.streams.batch_append(
             stream_name=stream_name,
             current_version=current_version,
             events=events,
@@ -308,7 +324,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         timeout = timeout if timeout is not None else self._default_deadline
         if isinstance(events, NewEvents):
             events = [events]
-        return await self._connection.v2streams.multi_append(
+        return await self.connection.v2streams.multi_append(
             events=events,
             timeout=timeout,
             metadata=self._call_metadata,
@@ -334,7 +350,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         Reads recorded events in "all streams" in the database.
         """
-        return await self._connection.streams.read(
+        return await self.connection.streams.read(
             commit_position=commit_position,
             backwards=backwards,
             resolve_links=resolve_links,
@@ -442,7 +458,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         Reads recorded events from the named stream.
         """
-        return await self._connection.streams.read(
+        return await self.connection.streams.read(
             stream_name=stream_name,
             stream_position=stream_position,
             backwards=backwards,
@@ -531,7 +547,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         Starts a catch-up subscription, from which all
         recorded events in the database can be received.
         """
-        return await self._connection.streams.read(
+        return await self.connection.streams.read(
             commit_position=commit_position,
             from_end=from_end,
             resolve_links=resolve_links,
@@ -637,7 +653,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         Starts a catch-up subscription from which
         recorded events in a stream can be received.
         """
-        return await self._connection.streams.read(
+        return await self.connection.streams.read(
             stream_name=stream_name,
             stream_position=stream_position,
             from_end=from_end,
@@ -662,7 +678,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
     ) -> None:
         # Todo: Reconsider using current_version=None to indicate "stream exists"?
         timeout = timeout if timeout is not None else self._default_deadline
-        await self._connection.streams.delete(
+        await self.connection.streams.delete(
             stream_name=stream_name,
             current_version=current_version,
             timeout=timeout,
@@ -681,7 +697,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         credentials: grpc.CallCredentials | None = None,
     ) -> None:
         timeout = timeout if timeout is not None else self._default_deadline
-        await self._connection.streams.tombstone(
+        await self.connection.streams.tombstone(
             stream_name=stream_name,
             current_version=current_version,
             timeout=timeout,
@@ -699,12 +715,13 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         credentials: grpc.CallCredentials | None = None,
     ) -> int | Literal[StreamState.NO_STREAM]:
         """
-        Returns the current position of the end of a stream.
+        Returns the current position of the end of a stream,
+        or StreamState.NO_STREAM if the stream is not found.
         """
         try:
             events = [
                 e
-                async for e in await self._connection.streams.read(
+                async for e in await self.connection.streams.read(
                     stream_name=stream_name,
                     backwards=True,
                     limit=1,
@@ -841,7 +858,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.persistent_subscriptions.create(
+        await self.connection.persistent_subscriptions.create(
             group_name=group_name,
             from_end=from_end,
             commit_position=commit_position,
@@ -973,7 +990,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.persistent_subscriptions.create(
+        await self.connection.persistent_subscriptions.create(
             group_name=group_name,
             stream_name=stream_name,
             from_end=from_end,
@@ -1011,7 +1028,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         Reads a persistent subscription on all streams.
         """
-        return await self._connection.persistent_subscriptions.read(
+        return await self.connection.persistent_subscriptions.read(
             group_name=group_name,
             event_buffer_size=event_buffer_size,
             max_ack_batch_size=max_ack_batch_size,
@@ -1039,7 +1056,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         Reads a persistent subscription on one stream.
         """
-        return await self._connection.persistent_subscriptions.read(
+        return await self.connection.persistent_subscriptions.read(
             group_name=group_name,
             stream_name=stream_name,
             event_buffer_size=event_buffer_size,
@@ -1064,7 +1081,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         Gets info for a persistent subscription.
         """
-        return await self._connection.persistent_subscriptions.get_info(
+        return await self.connection.persistent_subscriptions.get_info(
             group_name=group_name,
             stream_name=stream_name,
             timeout=timeout,
@@ -1083,7 +1100,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         Lists all persistent subscriptions.
         """
-        return await self._connection.persistent_subscriptions.list(
+        return await self.connection.persistent_subscriptions.list(
             timeout=timeout,
             metadata=self._call_metadata,
             credentials=credentials or self._call_credentials,
@@ -1101,7 +1118,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         Lists persistent stream subscriptions.
         """
-        return await self._connection.persistent_subscriptions.list(
+        return await self.connection.persistent_subscriptions.list(
             stream_name=stream_name,
             timeout=timeout,
             metadata=self._call_metadata,
@@ -1262,7 +1279,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
             extra_statistics=extra_statistics,
         )
 
-        await self._connection.persistent_subscriptions.update(
+        await self.connection.persistent_subscriptions.update(
             group_name=group_name,
             stream_name=stream_name,
             **kwargs,
@@ -1417,7 +1434,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
             extra_statistics=extra_statistics,
         )
 
-        await self._connection.persistent_subscriptions.update(
+        await self.connection.persistent_subscriptions.update(
             group_name=group_name,
             **kwargs,
             timeout=timeout if timeout is not None else self._default_deadline,
@@ -1437,7 +1454,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
     ) -> None:
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.persistent_subscriptions.replay_parked(
+        await self.connection.persistent_subscriptions.replay_parked(
             group_name=group_name,
             stream_name=stream_name,
             timeout=timeout,
@@ -1460,7 +1477,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.persistent_subscriptions.delete(
+        await self.connection.persistent_subscriptions.delete(
             group_name=group_name,
             stream_name=stream_name,
             timeout=timeout,
@@ -1485,7 +1502,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.projections.create(
+        await self.connection.projections.create(
             query=query,
             name=name,
             emit_enabled=emit_enabled,
@@ -1511,7 +1528,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.projections.update(
+        await self.connection.projections.update(
             name=name,
             query=query,
             emit_enabled=emit_enabled,
@@ -1537,7 +1554,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.projections.delete(
+        await self.connection.projections.delete(
             name=name,
             delete_emitted_streams=delete_emitted_streams,
             delete_state_stream=delete_state_stream,
@@ -1561,7 +1578,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        return await self._connection.projections.get_statistics(
+        return await self.connection.projections.get_statistics(
             name=name,
             timeout=timeout,
             metadata=self._call_metadata,
@@ -1581,7 +1598,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        return await self._connection.projections.list_statistics(
+        return await self.connection.projections.list_statistics(
             timeout=timeout,
             metadata=self._call_metadata,
             credentials=credentials or self._call_credentials,
@@ -1600,7 +1617,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        return await self._connection.projections.list_statistics(
+        return await self.connection.projections.list_statistics(
             all=True,
             timeout=timeout,
             metadata=self._call_metadata,
@@ -1621,7 +1638,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.projections.disable(
+        await self.connection.projections.disable(
             name=name,
             write_checkpoint=True,
             timeout=timeout,
@@ -1643,7 +1660,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.projections.disable(
+        await self.connection.projections.disable(
             name=name,
             write_checkpoint=False,
             timeout=timeout,
@@ -1665,7 +1682,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.projections.enable(
+        await self.connection.projections.enable(
             name=name,
             timeout=timeout,
             metadata=self._call_metadata,
@@ -1686,7 +1703,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        await self._connection.projections.reset(
+        await self.connection.projections.reset(
             name=name,
             write_checkpoint=True,
             timeout=timeout,
@@ -1709,7 +1726,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        return await self._connection.projections.get_state(
+        return await self.connection.projections.get_state(
             name=name,
             partition=partition,
             timeout=timeout,
@@ -1752,7 +1769,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
         """
         timeout = timeout if timeout is not None else self._default_deadline
 
-        return await self._connection.projections.restart_subsystem(
+        return await self.connection.projections.restart_subsystem(
             timeout=timeout,
             metadata=self._call_metadata,
             credentials=credentials or self._call_credentials,
@@ -1761,7 +1778,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
     async def close(self) -> None:
         if not self._is_closed:
             try:
-                esdb_connection = self._connection
+                esdb_connection = self.connection
                 del self._connection
             except AttributeError:  # pragma: no cover
                 pass
@@ -1770,6 +1787,7 @@ class AsyncKurrentDBClient(BaseKurrentDBClient):
                 self._is_closed = True
 
     async def __aenter__(self) -> Self:
+        await self.connect()
         return self
 
     async def __aexit__(self, *args: object, **kwargs: Any) -> None:

{kurrentdbclient-1.2b1 → kurrentdbclient-1.2.2}/kurrentdbclient/client.py

@@ -733,7 +733,8 @@ class KurrentDBClient(BaseKurrentDBClient):
         credentials: grpc.CallCredentials | None = None,
     ) -> int | Literal[StreamState.NO_STREAM]:
         """
-        Returns the current position of the end of a stream.
+        Returns the current position of the end of a stream,
+        or StreamState.NO_STREAM if the stream is not found.
         """
         try:
             last_event = next(

{kurrentdbclient-1.2b1 → kurrentdbclient-1.2.2}/kurrentdbclient/common.py

@@ -37,6 +37,7 @@ from kurrentdbclient.exceptions import (
     GrpcDeadlineExceededError,
     GrpcError,
     InternalError,
+    InvalidCommitPositionError,
     KurrentDBClientError,
     MaximumSubscriptionsReachedError,
     MultiAppendToSameStreamError,
@@ -319,6 +320,11 @@ def handle_rpc_error(e: grpc.RpcError) -> KurrentDBClientError:  # noqa: PLR0911
             ):
                 # Projections.Delete does this....
                 return OperationFailedError(details_str)
+            # no cover <25.1: start
+            if "Unexpected FilteredReadAllResult: InvalidPosition" in details_str:
+                # Commit position does not exist.
+                return InvalidCommitPositionError(details_str)
+            # no cover <25.1: stop
             return UnknownError(details_str)  # pragma: no cover
 
         if e.code() == grpc.StatusCode.ABORTED:

{kurrentdbclient-1.2b1 → kurrentdbclient-1.2.2}/kurrentdbclient/exceptions.py

@@ -322,3 +322,9 @@ class InternalError(GrpcError):
     """
     Raised when a grpc INTERNAL error is encountered.
     """
+
+
+class InvalidCommitPositionError(KurrentDBClientError):
+    """
+    Raised when reading from an commit position that does not exist.
+    """

{kurrentdbclient-1.2b1 → kurrentdbclient-1.2.2}/kurrentdbclient/instrumentation/opentelemetry/__init__.py

@@ -20,6 +20,7 @@ from kurrentdbclient.instrumentation.opentelemetry.spanners import (
     span_append_to_stream,
     span_catchup_subscription,
     span_get_stream,
+    span_multi_append_to_stream,
     span_persistent_subscription,
     span_read_stream,
 )
@@ -68,6 +69,12 @@ class KurrentDBClientInstrumentor(_BaseInstrumentor):
             spanner_func=span_append_to_stream,
             tracer=tracer,
         )
+        apply_spanner(
+            patched_class=KurrentDBClient,
+            spanned_func=KurrentDBClient.multi_append_to_stream,
+            spanner_func=span_multi_append_to_stream,
+            tracer=tracer,
+        )
         apply_spanner(
             patched_class=KurrentDBClient,
             spanned_func=KurrentDBClient.subscribe_to_stream,
@@ -112,6 +119,7 @@ class KurrentDBClientInstrumentor(_BaseInstrumentor):
 
     def _uninstrument(self, **kwargs: Any) -> None:
         unwrap(KurrentDBClient, "append_to_stream")
+        unwrap(KurrentDBClient, "multi_append_to_stream")
         unwrap(KurrentDBClient, "subscribe_to_stream")
         unwrap(KurrentDBClient, "subscribe_to_all")
         unwrap(KurrentDBClient, "read_subscription_to_stream")
@@ -136,6 +144,12 @@ class AsyncKurrentDBClientInstrumentor(_BaseInstrumentor):
             spanner_func=span_append_to_stream,
             tracer=tracer,
         )
+        apply_spanner(
+            patched_class=AsyncKurrentDBClient,
+            spanned_func=AsyncKurrentDBClient.multi_append_to_stream,
+            spanner_func=span_multi_append_to_stream,
+            tracer=tracer,
+        )
         apply_spanner(
             patched_class=AsyncKurrentDBClient,
             spanned_func=AsyncKurrentDBClient.subscribe_to_stream,
@@ -176,6 +190,7 @@ class AsyncKurrentDBClientInstrumentor(_BaseInstrumentor):
 
     def _uninstrument(self, **kwargs: Any) -> None:
         unwrap(AsyncKurrentDBClient, "append_to_stream")
+        unwrap(AsyncKurrentDBClient, "multi_append_to_stream")
         unwrap(AsyncKurrentDBClient, "subscribe_to_stream")
         unwrap(AsyncKurrentDBClient, "subscribe_to_all")
         unwrap(AsyncKurrentDBClient, "read_subscription_to_stream")

{kurrentdbclient-1.2b1 → kurrentdbclient-1.2.2}/kurrentdbclient/instrumentation/opentelemetry/spanners.py

@@ -32,6 +32,7 @@ from kurrentdbclient import (
     AsyncReadResponse,
     KurrentDBClient,
     NewEvent,
+    NewEvents,
     ReadResponse,
     RecordedEvent,
     StreamState,
@@ -70,14 +71,17 @@ if TYPE_CHECKING:
 
 
 STREAMS_APPEND = "streams.append"
+STREAMS_MULTI_APPEND = "streams.multi-append"
 STREAMS_SUBSCRIBE = "streams.subscribe"
 SPAN_NAMES_BY_CLIENT_METHOD = {
     KurrentDBClient.append_to_stream.__qualname__: STREAMS_APPEND,
+    KurrentDBClient.multi_append_to_stream.__qualname__: STREAMS_MULTI_APPEND,
     KurrentDBClient.subscribe_to_all.__qualname__: STREAMS_SUBSCRIBE,
     KurrentDBClient.subscribe_to_stream.__qualname__: STREAMS_SUBSCRIBE,
     KurrentDBClient.read_subscription_to_all.__qualname__: STREAMS_SUBSCRIBE,
     KurrentDBClient.read_subscription_to_stream.__qualname__: STREAMS_SUBSCRIBE,
     AsyncKurrentDBClient.append_to_stream.__qualname__: STREAMS_APPEND,
+    AsyncKurrentDBClient.multi_append_to_stream.__qualname__: STREAMS_MULTI_APPEND,
     AsyncKurrentDBClient.subscribe_to_all.__qualname__: STREAMS_SUBSCRIBE,
     AsyncKurrentDBClient.subscribe_to_stream.__qualname__: STREAMS_SUBSCRIBE,
     AsyncKurrentDBClient.read_subscription_to_all.__qualname__: STREAMS_SUBSCRIBE,
@@ -85,11 +89,13 @@ SPAN_NAMES_BY_CLIENT_METHOD = {
 }
 SPAN_KINDS_BY_CLIENT_METHOD = {
     KurrentDBClient.append_to_stream.__qualname__: SpanKind.PRODUCER,
+    KurrentDBClient.multi_append_to_stream.__qualname__: SpanKind.PRODUCER,
     KurrentDBClient.subscribe_to_all.__qualname__: SpanKind.CONSUMER,
     KurrentDBClient.subscribe_to_stream.__qualname__: SpanKind.CONSUMER,
     KurrentDBClient.read_subscription_to_all.__qualname__: SpanKind.CONSUMER,
     KurrentDBClient.read_subscription_to_stream.__qualname__: SpanKind.CONSUMER,
     AsyncKurrentDBClient.append_to_stream.__qualname__: SpanKind.PRODUCER,
+    AsyncKurrentDBClient.multi_append_to_stream.__qualname__: SpanKind.PRODUCER,
     AsyncKurrentDBClient.subscribe_to_all.__qualname__: SpanKind.CONSUMER,
     AsyncKurrentDBClient.subscribe_to_stream.__qualname__: SpanKind.CONSUMER,
     AsyncKurrentDBClient.read_subscription_to_all.__qualname__: SpanKind.CONSUMER,
@@ -346,6 +352,30 @@ class AsyncAppendToStreamMethod(Protocol):
         pass  # pragma: no cover
 
 
+class MultiAppendToStreamMethod(Protocol):
+    def __call__(
+        self,
+        /,
+        events: NewEvents | Iterable[NewEvents],
+        *,
+        timeout: float | None = None,
+        credentials: grpc.CallCredentials | None = None,
+    ) -> int:
+        pass  # pragma: no cover
+
+
+class AsyncMultiAppendToStreamMethod(Protocol):
+    async def __call__(
+        self,
+        /,
+        events: NewEvents | Iterable[NewEvents],
+        *,
+        timeout: float | None = None,
+        credentials: grpc.CallCredentials | None = None,
+    ) -> int:
+        pass  # pragma: no cover
+
+
 @overload
 def span_append_to_stream(
     tracer: Tracer,
@@ -401,7 +431,7 @@ def span_append_to_stream(
                 db_operation_name=span_name,
                 stream_name=stream_name,
             )
-            events = _set_context_in_events(span.get_span_context(), events)
+            events = _set_context_in_new_event_objects(span.get_span_context(), events)
             yield spanned_func(
                 stream_name,
                 current_version=current_version,
@@ -416,6 +446,67 @@ def span_append_to_stream(
             _set_span_ok(span)
 
 
+@overload
+def span_multi_append_to_stream(
+    tracer: Tracer,
+    instance: BaseKurrentDBClient,
+    spanned_func: AsyncMultiAppendToStreamMethod,
+    /,
+    events: NewEvents | Iterable[NewEvents],
+    *,
+    timeout: float | None = None,
+    credentials: grpc.CallCredentials | None = None,
+) -> AsyncSpannerResponse[int]:
+    pass  # pragma: no cover
+
+
+@overload
+def span_multi_append_to_stream(
+    tracer: Tracer,
+    instance: BaseKurrentDBClient,
+    spanned_func: MultiAppendToStreamMethod,
+    /,
+    events: NewEvents | Iterable[NewEvents],
+    *,
+    timeout: float | None = None,
+    credentials: grpc.CallCredentials | None = None,
+) -> SpannerResponse[int]:
+    pass  # pragma: no cover
+
+
+def span_multi_append_to_stream(
+    tracer: Tracer,
+    instance: BaseKurrentDBClient,
+    spanned_func: MultiAppendToStreamMethod | AsyncMultiAppendToStreamMethod,
+    /,
+    events: NewEvents | Iterable[NewEvents],
+    *,
+    timeout: float | None = None,
+    credentials: grpc.CallCredentials | None = None,
+) -> OverloadedSpannerResponse[int, int]:  # pragma: <25.1 no cover
+
+    span_name, span_kind = _get_span_name_and_kind(spanned_func)
+
+    with _start_span(tracer, span_name, span_kind) as span:
+        try:
+            _enrich_span(
+                span=span,
+                client=instance,
+                db_operation_name=span_name,
+            )
+            events = _set_context_in_new_events_objects(span.get_span_context(), events)
+            yield spanned_func(
+                events=events,
+                timeout=timeout,
+                credentials=credentials,
+            )
+        except Exception as e:
+            _set_span_error(span, e)
+            raise
+        else:
+            _set_span_ok(span)
+
+
 class CatchupSubscriptionMethod(Protocol):
     def __call__(
         self,
@@ -1035,7 +1126,7 @@ METADATA_TRACE_ID = "$traceId"
 METADATA_SPAN_ID = "$spanId"
 
 
-def _set_context_in_events(
+def _set_context_in_new_event_objects(
     context: SpanContext, events: NewEvent | Iterable[NewEvent]
 ) -> Sequence[NewEvent]:
     # Kind of propagate OpenTelemetry context in "standard KurrentDB" style.
@@ -1063,6 +1154,25 @@ def _set_context_in_events(
     return reconstructed_events
 
 
+def _set_context_in_new_events_objects(
+    context: SpanContext, events: NewEvents | Iterable[NewEvents]
+) -> Sequence[NewEvents]:  # pragma: <25.1 no cover
+    # Kind of propagate OpenTelemetry context in "standard KurrentDB" style.
+    if isinstance(events, NewEvents):
+        events = [events]
+    return [
+        NewEvents(
+            new_events_obj.stream_name,
+            _set_context_in_new_event_objects(
+                context,
+                new_events_obj.events,
+            ),
+            current_version=new_events_obj.current_version,
+        )
+        for new_events_obj in events
+    ]
+
+
 def _extract_context_from_event(
     recorded_event: NewEvent | RecordedEvent,
 ) -> Context | None:

{kurrentdbclient-1.2b1 → kurrentdbclient-1.2.2}/kurrentdbclient/instrumentation/opentelemetry/utils.py

@@ -18,6 +18,7 @@ import wrapt
 from opentelemetry.trace import Span, SpanKind, Status, StatusCode, Tracer
 from typing_extensions import ParamSpec
 
+from kurrentdbclient import AsyncKurrentDBClient
 from kurrentdbclient.instrumentation.opentelemetry.attributes import Attributes
 
 if TYPE_CHECKING:
@@ -76,6 +77,8 @@ def apply_spanner(
             args: Any,
             kwargs: Any,
         ) -> R:
+            if isinstance(instance, AsyncKurrentDBClient):
+                await instance.connect()
             # Use the spanner function to execute the spanned function.
             with async_spanner(tracer, instance, original, *args, **kwargs) as result:
                 return await result

{kurrentdbclient-1.2b1 → kurrentdbclient-1.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "kurrentdbclient"
-version = "1.2b1"
+version = "1.2.2"
 description = "Python gRPC Client for KurrentDB"
 license = { text = "BSD-3-Clause" }
 readme = "README.md"
@@ -14,8 +14,8 @@ authors = [
 ]
 classifiers = [
 #    "Development Status :: 3 - Alpha",
-    "Development Status :: 4 - Beta",
-#    "Development Status :: 5 - Production/Stable",
+#    "Development Status :: 4 - Beta",
+    "Development Status :: 5 - Production/Stable",
     "License :: OSI Approved :: BSD License",
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3.10",
@@ -80,6 +80,7 @@ exclude = '''
   | build
   | dist
   | kurrentdbclient/protos
+  | docs/node_modules
 )/
 '''