streaming

frequenz.client.base.streaming

Implementation of the grpc streaming helper.

Attributes

frequenz.client.base.streaming.InputT module-attribute

InputT = TypeVar('InputT')

The input type of the stream.

frequenz.client.base.streaming.OutputT module-attribute

OutputT = TypeVar('OutputT')

The output type of the stream.

Classes

frequenz.client.base.streaming.GrpcStreamBroadcaster

Bases: Generic[InputT, OutputT]

Helper class to handle grpc streaming methods.

Source code in frequenz/client/base/streaming.py

class GrpcStreamBroadcaster(Generic[InputT, OutputT]):
    """Helper class to handle grpc streaming methods."""

    def __init__(
        self,
        stream_name: str,
        stream_method: Callable[[], grpc.aio.UnaryStreamCall[Any, InputT]],
        transform: Callable[[InputT], OutputT],
        retry_strategy: retry.Strategy | None = None,
    ):
        """Initialize the streaming helper.

        Args:
            stream_name: A name to identify the stream in the logs.
            stream_method: A function that returns the grpc stream. This function is
                called everytime the connection is lost and we want to retry.
            transform: A function to transform the input type to the output type.
            retry_strategy: The retry strategy to use, when the connection is lost. Defaults
                to retries every 3 seconds, with a jitter of 1 second, indefinitely.
        """
        self._stream_name = stream_name
        self._stream_method = stream_method
        self._transform = transform
        self._retry_strategy = (
            retry.LinearBackoff() if retry_strategy is None else retry_strategy.copy()
        )

        self._channel: channels.Broadcast[OutputT] = channels.Broadcast(
            f"GrpcStreamBroadcaster-{stream_name}"
        )
        self._task = asyncio.create_task(self._run())

    def new_receiver(self, maxsize: int = 50) -> channels.Receiver[OutputT]:
        """Create a new receiver for the stream.

        Args:
            maxsize: The maximum number of messages to buffer.

        Returns:
            A new receiver.
        """
        return self._channel.new_receiver(maxsize=maxsize)

    async def stop(self) -> None:
        """Stop the streaming helper."""
        if self._task.done():
            return
        self._task.cancel()
        try:
            await self._task
        except asyncio.CancelledError:
            pass
        await self._channel.close()

    async def _run(self) -> None:
        """Run the streaming helper."""
        sender = self._channel.new_sender()

        while True:
            _logger.debug("Making call to grpc streaming method: %s", self._stream_name)

            try:
                call = self._stream_method()
                async for msg in call:
                    await sender.send(self._transform(msg))
            except grpc.aio.AioRpcError:
                _logger.exception(
                    "Error in grpc streaming method: %s", self._stream_name
                )
            if interval := self._retry_strategy.next_interval():
                _logger.warning(
                    "`%s`, connection ended, retrying %s in %0.3f seconds.",
                    self._stream_name,
                    self._retry_strategy.get_progress(),
                    interval,
                )
                await asyncio.sleep(interval)
            else:
                _logger.warning(
                    "`%s`, connection ended, retry limit exceeded %s.",
                    self._stream_name,
                    self._retry_strategy.get_progress(),
                )
                await self._channel.close()
                break

Functions

__init__

__init__(
    stream_name: str,
    stream_method: Callable[
        [], UnaryStreamCall[Any, InputT]
    ],
    transform: Callable[[InputT], OutputT],
    retry_strategy: Strategy | None = None,
)

Initialize the streaming helper.

PARAMETER	DESCRIPTION
stream_name	A name to identify the stream in the logs. TYPE: str
stream_method	A function that returns the grpc stream. This function is called everytime the connection is lost and we want to retry. TYPE: Callable[[], UnaryStreamCall[Any, InputT]]
transform	A function to transform the input type to the output type. TYPE: Callable[[InputT], OutputT]
retry_strategy	The retry strategy to use, when the connection is lost. Defaults to retries every 3 seconds, with a jitter of 1 second, indefinitely. TYPE: Strategy | None DEFAULT: None

Source code in frequenz/client/base/streaming.py

def __init__(
    self,
    stream_name: str,
    stream_method: Callable[[], grpc.aio.UnaryStreamCall[Any, InputT]],
    transform: Callable[[InputT], OutputT],
    retry_strategy: retry.Strategy | None = None,
):
    """Initialize the streaming helper.

    Args:
        stream_name: A name to identify the stream in the logs.
        stream_method: A function that returns the grpc stream. This function is
            called everytime the connection is lost and we want to retry.
        transform: A function to transform the input type to the output type.
        retry_strategy: The retry strategy to use, when the connection is lost. Defaults
            to retries every 3 seconds, with a jitter of 1 second, indefinitely.
    """
    self._stream_name = stream_name
    self._stream_method = stream_method
    self._transform = transform
    self._retry_strategy = (
        retry.LinearBackoff() if retry_strategy is None else retry_strategy.copy()
    )

    self._channel: channels.Broadcast[OutputT] = channels.Broadcast(
        f"GrpcStreamBroadcaster-{stream_name}"
    )
    self._task = asyncio.create_task(self._run())

new_receiver

new_receiver(maxsize: int = 50) -> Receiver[OutputT]

Create a new receiver for the stream.

PARAMETER	DESCRIPTION
maxsize	The maximum number of messages to buffer. TYPE: int DEFAULT: 50

RETURNS	DESCRIPTION
Receiver[OutputT]	A new receiver.

Source code in frequenz/client/base/streaming.py

def new_receiver(self, maxsize: int = 50) -> channels.Receiver[OutputT]:
    """Create a new receiver for the stream.

    Args:
        maxsize: The maximum number of messages to buffer.

    Returns:
        A new receiver.
    """
    return self._channel.new_receiver(maxsize=maxsize)

stop async

stop() -> None

Stop the streaming helper.

Source code in frequenz/client/base/streaming.py

async def stop(self) -> None:
    """Stop the streaming helper."""
    if self._task.done():
        return
    self._task.cancel()
    try:
        await self._task
    except asyncio.CancelledError:
        pass
    await self._channel.close()