Unverified Commit 3749a6c9 authored by John DiSanti's avatar John DiSanti Committed by GitHub
Browse files

Fix response tracing for streaming response bodies (#514) 

* Fix response tracing for streaming response bodies

* Log everything except for the streaming body for streaming responses
parent 1fd6e978

rust-runtime/smithy-http/src/middleware.rs

+2 −2
Original line number Diff line number Diff line
@@ -77,11 +77,10 @@ where 
    O: ParseHttpResponse<SdkBody, Output = Result<T, E>>,

{

    if let Some(parsed_response) = handler.parse_unloaded(&mut response) {

        trace!(response = ?response);

        return sdk_result(parsed_response, response);

    }



    trace!(response = ?response);



    let (parts, body) = response.into_parts();

    let body = match read_body(body).await {

        Ok(body) => body,
@@ -94,6 +93,7 @@ where 
    };



    let response = Response::from_parts(parts, Bytes::from(body));

    trace!(response = ?response);

    let parsed = handler.parse_loaded(&response);

    sdk_result(parsed, response.map(SdkBody::from))

}