Skip to content

Commit

Permalink
feat(rpc): add a way to provide a custom User-Agent (#1429)
Browse files Browse the repository at this point in the history
* feat: add a way to provide a custom User-Agent

* chore: cargo fmt
  • Loading branch information
freak12techno authored May 29, 2024
1 parent 87455bc commit 2135196
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
* `[tendermint-rpc]` Add a way to specify custom User-Agent for HttpClient
([#1425](https://github.com/informalsystems/tendermint-rs/issues/1425))
10 changes: 9 additions & 1 deletion rpc/src/client/transport/http.rs
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,7 @@ pub struct Builder {
url: HttpClientUrl,
compat: CompatMode,
proxy_url: Option<HttpClientUrl>,
user_agent: Option<String>,
timeout: Duration,
}

Expand Down Expand Up @@ -93,10 +94,16 @@ impl Builder {
self
}

/// Specify the custom User-Agent header the client.
pub fn user_agent(mut self, agent: String) -> Self {
self.user_agent = Some(agent);
self
}

/// Try to create a client with the options specified for this builder.
pub fn build(self) -> Result<HttpClient, Error> {
let builder = reqwest::ClientBuilder::new()
.user_agent(USER_AGENT)
.user_agent(self.user_agent.unwrap_or(USER_AGENT.to_string()))
.timeout(self.timeout);
let inner = match self.proxy_url {
None => builder.build().map_err(Error::http)?,
Expand Down Expand Up @@ -152,6 +159,7 @@ impl HttpClient {
url,
compat: Default::default(),
proxy_url: None,
user_agent: None,
timeout: Duration::from_secs(30),
}
}
Expand Down

0 comments on commit 2135196

Please sign in to comment.