Apply response_body_filter when serving from cache

Previously the response header filter was applied when serving from
cache (using `proxy_cache_hit`), but not the response body filter.

1 files changed, 17 insertions, 1 deletions

diff --git a/pingora-proxy/src/proxy_cache.rs b/pingora-proxy/src/proxy_cache.rs
index 77b7384..e9914cf 100644
--- a/pingora-proxy/src/proxy_cache.rs
+++ b/pingora-proxy/src/proxy_cache.rs
@@ -309,7 +309,23 @@ impl<SV> HttpProxy<SV> {
             }
             loop {
                 match session.cache.hit_handler().read_body().await {
-                    Ok(body) => {
+                    Ok(mut body) => {
+                        let end = body.is_none();
+                        match self
+                            .inner
+                            .response_body_filter(session, &mut body, end, ctx)
+                        {
+                            Ok(Some(duration)) => {
+                                trace!("delaying response for {duration:?}");
+                                time::sleep(duration).await;
+                            }
+                            Ok(None) => { /* continue */ }
+                            Err(e) => {
+                                // body is being sent, don't treat downstream as reusable
+                                return (false, Some(e));
+                            }
+                        }
+
                         if let Some(b) = body {
                             // write to downstream
                             if let Err(e) = session