tyler/proxmox-backup
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

src/server/rest.rs: avoid unwrap

Browse Source
This commit is contained in:
Dietmar Maurer
2019-07-03 12:00:43 +02:00
parent 7fb4f5642a
commit 80af046794
1 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/server/rest.rs
View File

@ -49,8 +49,14 @@ impl MakeService<&tokio_openssl::SslStream<tokio::net::TcpStream>> for RestServe
type Service = ApiService;
type Future = Box<dyn Future<Item = Self::Service, Error = Self::MakeError> + Send>;
fn make_service(&mut self, ctx: &tokio_openssl::SslStream<tokio::net::TcpStream>) -> Self::Future {
let peer = ctx.get_ref().get_ref().peer_addr().unwrap();
Box::new(future::ok(ApiService { peer, api_config: self.api_config.clone() }))
match ctx.get_ref().get_ref().peer_addr() {
Err(err) => {
Box::new(future::err(format_err!("unable to get peer address - {}", err)))
}
Ok(peer) => {
Box::new(future::ok(ApiService { peer, api_config: self.api_config.clone() }))
}
}
}
}
@ -63,8 +69,14 @@ impl MakeService<&tokio::net::TcpStream> for RestServer
type Service = ApiService;
type Future = Box<dyn Future<Item = Self::Service, Error = Self::MakeError> + Send>;
fn make_service(&mut self, ctx: &tokio::net::TcpStream) -> Self::Future {
let peer = ctx.peer_addr().unwrap();
Box::new(future::ok(ApiService { peer, api_config: self.api_config.clone() }))
match ctx.peer_addr() {
Err(err) => {
Box::new(future::err(format_err!("unable to get peer address - {}", err)))
}
Ok(peer) => {
Box::new(future::ok(ApiService { peer, api_config: self.api_config.clone() }))
}
}
}
}