diff options
| author | Bryan Newbold <bnewbold@archive.org> | 2020-08-26 18:48:40 -0700 |
|---|---|---|
| committer | Bryan Newbold <bnewbold@archive.org> | 2020-08-26 18:48:40 -0700 |
| commit | 7e4544158d16f65acb4c69a0d4566008f2030a43 (patch) | |
| tree | 7861332587c9ce350195103a12a0bda504fbdd04 /src/main.rs | |
| parent | 13220ca46bbc9fd0001c1c942c3b7238e0f596ee (diff) | |
| download | es-public-proxy-7e4544158d16f65acb4c69a0d4566008f2030a43.tar.gz es-public-proxy-7e4544158d16f65acb4c69a0d4566008f2030a43.zip | |
improve error handling/responses/logging
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 32 |
1 files changed, 22 insertions, 10 deletions
diff --git a/src/main.rs b/src/main.rs index b55999b..1855ab5 100644 --- a/src/main.rs +++ b/src/main.rs @@ -7,7 +7,7 @@ use toml; #[macro_use] extern crate log; -use es_public_proxy::{filter_request, ProxyConfig}; +use es_public_proxy::{filter_request, ProxyConfig, ProxyError}; const CARGO_VERSION: Option<&'static str> = option_env!("CARGO_PKG_VERSION"); @@ -16,22 +16,35 @@ async fn upstream_req( config: ProxyConfig, ) -> Result<Response<Body>, hyper::Error> { info!("request: {} {}", req.method(), req.uri()); - let parsed = filter_request(req, &config).await; - let mut resp = match parsed { - Ok(upstream_req) => { - debug!("sending request..."); - Client::new().request(upstream_req).await? - } + let upstream_resp: Result<Response<Body>, ProxyError> = match filter_request(req, &config).await { + Ok(parsed_req) => { + debug!("forwarding request upstream"); + Client::new().request(parsed_req) + .await + .map_err(|e| { + warn!("upstream error: {}", e); + ProxyError::UpstreamError(e.to_string()) + }) + }, + Err(e) => { + warn!("blocked: {:?}", e); + Err(e) + }, + }; + let mut resp = match upstream_resp { + Ok(resp) => resp, Err(other) => Response::builder() .status(other.http_status_code()) .header("Content-Type", "application/json; charset=UTF-8") .body( serde_json::to_string(&other.to_json_value()) - .unwrap() + .expect("trivial JSON value serialization failed") .into(), ) - .unwrap(), + .expect("trivial HTTP response construction failed"), }; + + // regardless of the response type (error or upstream response), add HTTP headers resp.headers_mut() .insert("Via", HeaderValue::from_static("1.1 es-public-proxy")); if config.enable_cors == Some(true) { @@ -46,7 +59,6 @@ async fn upstream_req( HeaderValue::from_static("DNT,User-Agent,Content-Type"), ); } - debug!("resp!"); Ok(resp) } |