Do Interceptors support intercepting server-streamed/bidirectional events?

[pirate]

Could we do something like:

from connectrpc.interceptor import Interceptor

class LoggingInterceptor(Interceptor):
    async def intercept(self, method, request, context, next_handler):
        response = await next_handler(request, context)
        if response.stream:
            async for event in response.stream:
                if event.event_type == 'LogEvent':
                    logging.info(event.message)
                    # (server sends LogEvents when client should print something to console)
        return response


client = HelloServiceClient(
    base_url="https://api.example.com",
    session=session,
    interceptors=[LoggingInterceptor()]
)

[anuraaga]

Hi @pirate - yes streaming RPCs can also be intercepted. To keep everything type safe, each stream type has a unique method signature

https://github.com/connectrpc/connect-python/blob/main/docs%2Finterceptors.md

An interceptor can implement both of the streaming entry points, possibly delegating to a helper function if logic is shared.

Note that streams can only be consumed once so if your interceptor needs to consume messages, it also needs to yield them, looking something like this

async for event in response:
    if event.event_type == 'LogEvent':
        logging.info(event.message)
       # (server sends LogEvents when client should 
    yield event

[anuraaga]

@pirate I'm going to go ahead and close this but let me know if you still have any issues here. Thanks.