feat(volo-http): add layer for RequestBuilder (#633)

* feat(volo-http): add `layer` for `RequestBuilder`

Note that the layer should generate a `OneShotService`

* chore(volo-http): bump volo-http to 0.5.1

---------

Signed-off-by: Yu Li <liyu.yukiteru@bytedance.com>

Author: yukiiiteru

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "volo-http"
-version = "0.5.0"
+version = "0.5.1"
 edition.workspace = true
 homepage.workspace = true
 repository.workspace = true
@@ -138,4 +138,4 @@ native-tls-vendored = ["native-tls", "volo/native-tls-vendored"]
 
 [package.metadata.docs.rs]
 all-features = true
-rustdoc-args = ["--cfg", "docsrs"]
+rustdoc-args = ["--cfg", "docsrs"]

volo-http/Cargo.toml

@@ -11,6 +11,7 @@ use http::{
     uri::{PathAndQuery, Scheme, Uri},
     version::Version,
 };
+use motore::layer::Layer;
 use volo::{
     client::{Apply, OneShotService, WithOptService},
     net::Address,
@@ -345,17 +346,27 @@ impl<S, B> RequestBuilder<S, B> {
         self.request.body()
     }
 
-    /// Apply a [`CallOpt`] to the request.
-    pub fn with_callopt(self, callopt: CallOpt) -> RequestBuilder<WithOptService<S, CallOpt>, B> {
+    /// Add a new [`Layer`] to the front of request builder.
+    ///
+    /// Note that the [`Layer`] generated `Service` should be a [`OneShotService`].
+    pub fn layer<L>(self, layer: L) -> RequestBuilder<L::Service, B>
+    where
+        L: Layer<S>,
+    {
         RequestBuilder {
-            inner: WithOptService::new(self.inner, callopt),
+            inner: layer.layer(self.inner),
             target: self.target,
             version: self.version,
             request: self.request,
             status: self.status,
         }
     }
 
+    /// Apply a [`CallOpt`] to the request.
+    pub fn with_callopt(self, callopt: CallOpt) -> RequestBuilder<WithOptService<S, CallOpt>, B> {
+        self.layer(WithOptLayer::new(callopt))
+    }
+
     fn set_version(&mut self) {
         let ver = match self.version {
             Some(ver) => ver,
@@ -392,3 +403,21 @@ impl<S, B> RequestBuilder<S, B> {
         self.inner.call(&mut cx, self.request).await
     }
 }
+
+struct WithOptLayer {
+    opt: CallOpt,
+}
+
+impl WithOptLayer {
+    const fn new(opt: CallOpt) -> Self {
+        Self { opt }
+    }
+}
+
+impl<S> Layer<S> for WithOptLayer {
+    type Service = WithOptService<S, CallOpt>;
+
+    fn layer(self, inner: S) -> Self::Service {
+        WithOptService::new(inner, self.opt)
+    }
+}