daan/osrs-prometheus-exporter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

Use automatically generated user agent for http requests based on crate metadata

Browse source
This commit is contained in:
Daan Boerlage 2023-04-08 02:06:36 +02:00
parent af8e480ef8
commit 91a7b4d2f1
Signed by: daan
GPG key ID: FCE070E1E4956606
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/transport/middleware/headers.rs
View file

@ -2,13 +2,15 @@ use async_trait::async_trait;
use reqwest::{Request, Response}; use reqwest::{Request, Response};
use reqwest_middleware::{Middleware, Next, Result}; use reqwest_middleware::{Middleware, Next, Result};
pub const USER_AGENT: &str = "osrs-prometheus-exporter"; pub struct DefaultHeaderMiddleware {
user_agent: String,
pub struct DefaultHeaderMiddleware {} }
impl DefaultHeaderMiddleware { impl DefaultHeaderMiddleware {
pub fn new() -> Self { pub fn new() -> Self {
DefaultHeaderMiddleware {} DefaultHeaderMiddleware {
user_agent: format!("{}/{}", env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"))
}
} }
} }
@ -20,8 +22,8 @@ impl Middleware for DefaultHeaderMiddleware {
extensions: &mut task_local_extensions::Extensions, extensions: &mut task_local_extensions::Extensions,
next: Next<'_>, next: Next<'_>,
) -> Result<Response> { ) -> Result<Response> {
req.headers_mut() let ua = self.user_agent.as_str();
.insert("User-Agent", USER_AGENT.parse().unwrap()); req.headers_mut().insert("User-Agent", ua.parse().unwrap());
next.run(req, extensions).await next.run(req, extensions).await
} }