Executing code after a stream is finished.

[frederikbosch]

Hello,

I have got the following proto

service ProjectDeployment {
  rpc InitializeDeployments(ClientInitializationRequest) returns (stream Deployment) {}
}

The Tonic service looks as follows.

#[tonic::async_trait]
impl ProjectDeployment for ProjectDeploymentService {
    type InitializeDeploymentsStream =
        Pin<Box<dyn Stream<Item = Result<RpcDeployment, Status>> + Send + Sync + 'static>>;

    async fn initialize_deployments(
        &self,
        request: Request<ClientInitializationRequest>,
    ) -> Result<Response<Self::InitializeDeploymentsStream>, Status> {
        let output = async_stream::try_stream! {
            // some code here
        }

        Ok(Response::new(Box::pin(output)
            as Pin<
                Box<dyn Stream<Item = Result<RpcDeployment, Status>> + Send + Sync + 'static>,
            >))
    }
}

Want I want is execute some code after all incoming streamed messages have been converted into a response per item. If I put it between letand Ok() it is not executed after the stream is finished. How should I achieve this?