diff --git a/src/main.rs b/src/main.rs index bc8b6bf..d2938e3 100644 --- a/src/main.rs +++ b/src/main.rs @@ -14,18 +14,21 @@ impl ServiceTrait> for Service { type Future = Pin> + Send>>; fn call(&self, req: Request) -> Self::Future { - let mut remote_ip = self.remote_ip; + let mut remote_ip = self.remote_ip.map(|x| x.ip()); if let Some(ip) = req.headers().get("X-Forwarded-For") { if let Ok(str_ip) = ip.to_str() { if let Ok(ip) = str_ip.parse::() { + remote_ip = Some(ip.ip()); + } + if let Ok(ip) = str_ip.parse::() { remote_ip = Some(ip); } } } - + if let Some(ip) = remote_ip { - let mut res = Response::new(ip.ip().to_string()); + let mut res = Response::new(ip.to_string()); res.headers_mut().append("Content-Type", HeaderValue::from_static("text/plain")); Box::pin(async { Ok(res) }) } else {