How to route Body response used for streaming

[hniksic]

I have an endpoint declared like this:

pub async fn handler(...args...) -> Result<Body, Response> {
    // ...
    Ok(Body::from_stream(tokio_util::io::ReaderStream::new(stream)))
}

This handler is not accepted by aide::axum::routing::get_with because it returns Body. (If I change Body to e.g. Vec<u8>, the call to get_with() compiles.) I need to return Body to stream the response to the client. Is there a way to pass such a handler to aide?

[hniksic]

As a workaround, I resolved this by using a newtype:

#[derive(OperationIo)]
pub struct ExportBody(Body);

impl IntoResponse for ExportBody {
    fn into_response(self) -> Response {
        self.0.into_response()
    }
}

pub async fn get_audience_export(...args...) -> Result<ExportBody, Response> {
    // ...
    let body = Body::from_stream(tokio_util::io::ReaderStream::new(stream));
    Ok(ExportBody(body))
}

This gets the handler to be accepted by aide::axum::routing::get_with, and the actual return type is documented as Bytes in the separate doc handler.