From f47f2f61afb44a94bee2502045998719e569e9ff Mon Sep 17 00:00:00 2001
From: Wolfus20 <silvallywish@gmail.com>
Date: Tue, 9 Jan 2024 16:23:11 +0800
Subject: [PATCH] Add default generic parameters for trace::{ResponseBody,
 ResponseFuture}

Fixes: #233
---
 tower-http/src/trace/body.rs   | 4 ++--
 tower-http/src/trace/future.rs | 7 +++++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/tower-http/src/trace/body.rs b/tower-http/src/trace/body.rs
index 7a27cd90..d713f243 100644
--- a/tower-http/src/trace/body.rs
+++ b/tower-http/src/trace/body.rs
@@ -1,4 +1,4 @@
-use super::{OnBodyChunk, OnEos, OnFailure};
+use super::{DefaultOnBodyChunk, DefaultOnEos, DefaultOnFailure, OnBodyChunk, OnEos, OnFailure};
 use crate::classify::ClassifyEos;
 use http_body::{Body, Frame};
 use pin_project_lite::pin_project;
@@ -14,7 +14,7 @@ pin_project! {
     /// Response body for [`Trace`].
     ///
     /// [`Trace`]: super::Trace
-    pub struct ResponseBody<B, C, OnBodyChunk, OnEos, OnFailure> {
+    pub struct ResponseBody<B, C, OnBodyChunk = DefaultOnBodyChunk, OnEos = DefaultOnEos, OnFailure = DefaultOnFailure> {
         #[pin]
         pub(crate) inner: B,
         pub(crate) classify_eos: Option<C>,
diff --git a/tower-http/src/trace/future.rs b/tower-http/src/trace/future.rs
index feac51bf..e205ea32 100644
--- a/tower-http/src/trace/future.rs
+++ b/tower-http/src/trace/future.rs
@@ -1,4 +1,7 @@
-use super::{OnBodyChunk, OnEos, OnFailure, OnResponse, ResponseBody};
+use super::{
+    DefaultOnBodyChunk, DefaultOnEos, DefaultOnFailure, DefaultOnResponse, OnBodyChunk, OnEos,
+    OnFailure, OnResponse, ResponseBody,
+};
 use crate::classify::{ClassifiedResponse, ClassifyResponse};
 use http::Response;
 use http_body::Body;
@@ -15,7 +18,7 @@ pin_project! {
     /// Response future for [`Trace`].
     ///
     /// [`Trace`]: super::Trace
-    pub struct ResponseFuture<F, C, OnResponse, OnBodyChunk, OnEos, OnFailure> {
+    pub struct ResponseFuture<F, C, OnResponse = DefaultOnResponse, OnBodyChunk = DefaultOnBodyChunk, OnEos = DefaultOnEos, OnFailure = DefaultOnFailure> {
         #[pin]
         pub(crate) inner: F,
         pub(crate) span: Span,