can disable user agent

i18nsite · i18nsite · commit 8d87b440c015 · 2025-07-09T10:42:53.000+08:00
diff --git a/tonic/Cargo.toml b/tonic/Cargo.toml
@@ -24,7 +24,8 @@ codegen = ["dep:async-trait"]
 gzip = ["dep:flate2"]
 deflate = ["dep:flate2"]
 zstd = ["dep:zstd"]
-default = ["router", "transport", "codegen", "prost"]
+default = ["router", "transport", "codegen", "prost", "user-agent"]
+user-agent = []
 prost = ["dep:prost"]
 _tls-any = ["dep:tokio-rustls", "dep:tokio", "tokio?/rt", "tokio?/macros"] # Internal. Please choose one of `tls-ring` or `tls-aws-lc`
 tls-ring = ["_tls-any", "tokio-rustls/ring"]
diff --git a/tonic/src/transport/channel/service/connection.rs b/tonic/src/transport/channel/service/connection.rs
@@ -1,25 +1,26 @@
-use super::{AddOrigin, Reconnect, SharedExec, UserAgent};
-use crate::{
-    body::Body,
-    transport::{channel::BoxFuture, service::GrpcTimeout, Endpoint},
-};
-use http::{Request, Response, Uri};
-use hyper::rt;
-use hyper::{client::conn::http2::Builder, rt::Executor};
-use hyper_util::rt::TokioTimer;
 use std::{
     fmt,
     task::{Context, Poll},
 };
-use tower::load::Load;
+
+use http::{Request, Response, Uri};
+use hyper::{client::conn::http2::Builder, rt, rt::Executor};
+use hyper_util::rt::TokioTimer;
 use tower::{
     layer::Layer,
     limit::{concurrency::ConcurrencyLimitLayer, rate::RateLimitLayer},
+    load::Load,
     util::BoxService,
     ServiceBuilder, ServiceExt,
 };
 use tower_service::Service;
 
+use super::{AddOrigin, Reconnect, SharedExec, UserAgent};
+use crate::{
+    body::Body,
+    transport::{channel::BoxFuture, service::GrpcTimeout, Endpoint},
+};
+
 pub(crate) struct Connection {
     inner: BoxService<Request<Body>, Response<Body>, crate::BoxError>,
 }
@@ -55,13 +56,16 @@ impl Connection {
             settings.max_header_list_size(val);
         }
 
-        let stack = ServiceBuilder::new()
-            .layer_fn(|s| {
-                let origin = endpoint.origin.as_ref().unwrap_or(endpoint.uri()).clone();
+        let stack = ServiceBuilder::new().layer_fn(|s| {
+            let origin = endpoint.origin.as_ref().unwrap_or(endpoint.uri()).clone();
+
+            AddOrigin::new(s, origin)
+        });
+
+        #[cfg(feature = "user-agent")]
+        let stack = stack.layer_fn(|s| UserAgent::new(s, endpoint.user_agent.clone()));
 
-                AddOrigin::new(s, origin)
-            })
-            .layer_fn(|s| UserAgent::new(s, endpoint.user_agent.clone()))
+        let stack = stack
             .layer_fn(|s| GrpcTimeout::new(s, endpoint.timeout))
             .option_layer(endpoint.concurrency_limit.map(ConcurrencyLimitLayer::new))
             .option_layer(endpoint.rate_limit.map(|(l, d)| RateLimitLayer::new(l, d)))
