hyper: use new hyper::upgrade

the old Body::on_upgrade method is no more

Signed-off-by: Fabian Grünbichler <f.gruenbichler@proxmox.com>
This commit is contained in:
Fabian Grünbichler 2020-12-04 09:11:29 +01:00
parent b5a202acb6
commit 89e9134a3f
4 changed files with 8 additions and 12 deletions

View File

@ -2,7 +2,7 @@ use anyhow::{bail, format_err, Error};
use futures::*; use futures::*;
use hyper::header::{HeaderValue, UPGRADE}; use hyper::header::{HeaderValue, UPGRADE};
use hyper::http::request::Parts; use hyper::http::request::Parts;
use hyper::{Body, Response, StatusCode}; use hyper::{Body, Response, Request, StatusCode};
use serde_json::{json, Value}; use serde_json::{json, Value};
use proxmox::{sortable, identity, list_subdirs_api_method}; use proxmox::{sortable, identity, list_subdirs_api_method};
@ -171,8 +171,7 @@ async move {
let env2 = env.clone(); let env2 = env.clone();
let mut req_fut = req_body let mut req_fut = hyper::upgrade::on(Request::from_parts(parts, req_body))
.on_upgrade()
.map_err(Error::from) .map_err(Error::from)
.and_then(move |conn| { .and_then(move |conn| {
env2.debug("protocol upgrade done"); env2.debug("protocol upgrade done");

View File

@ -6,6 +6,7 @@ use futures::future::{FutureExt, TryFutureExt};
use hyper::body::Body; use hyper::body::Body;
use hyper::http::request::Parts; use hyper::http::request::Parts;
use hyper::upgrade::Upgraded; use hyper::upgrade::Upgraded;
use hyper::Request;
use serde_json::{json, Value}; use serde_json::{json, Value};
use tokio::io::{AsyncBufReadExt, BufReader}; use tokio::io::{AsyncBufReadExt, BufReader};
@ -292,10 +293,10 @@ fn upgrade_to_websocket(
Some(&ticket::term_aad(&userid, "/system", port)), Some(&ticket::term_aad(&userid, "/system", port)),
)?; )?;
let (ws, response) = WebSocket::new(parts.headers)?; let (ws, response) = WebSocket::new(parts.headers.clone())?;
crate::server::spawn_internal_task(async move { crate::server::spawn_internal_task(async move {
let conn: Upgraded = match req_body.on_upgrade().map_err(Error::from).await { let conn: Upgraded = match hyper::upgrade::on(Request::from_parts(parts, req_body)).map_err(Error::from).await {
Ok(upgraded) => upgraded, Ok(upgraded) => upgraded,
_ => bail!("error"), _ => bail!("error"),
}; };

View File

@ -2,7 +2,7 @@ use anyhow::{bail, format_err, Error};
use futures::*; use futures::*;
use hyper::header::{self, HeaderValue, UPGRADE}; use hyper::header::{self, HeaderValue, UPGRADE};
use hyper::http::request::Parts; use hyper::http::request::Parts;
use hyper::{Body, Response, StatusCode}; use hyper::{Body, Response, Request, StatusCode};
use serde_json::Value; use serde_json::Value;
use proxmox::{sortable, identity}; use proxmox::{sortable, identity};
@ -130,8 +130,7 @@ fn upgrade_to_backup_reader_protocol(
let abort_future = worker.abort_future(); let abort_future = worker.abort_future();
let req_fut = req_body let req_fut = hyper::upgrade::on(Request::from_parts(parts, req_body))
.on_upgrade()
.map_err(Error::from) .map_err(Error::from)
.and_then({ .and_then({
let env = env.clone(); let env = env.clone();

View File

@ -646,10 +646,7 @@ impl HttpClient {
bail!("unknown error"); bail!("unknown error");
} }
let upgraded = resp let upgraded = hyper::upgrade::on(resp).await?;
.into_body()
.on_upgrade()
.await?;
let max_window_size = (1 << 31) - 2; let max_window_size = (1 << 31) - 2;