konevsa/proxmox.newest
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

proxmox-rest-server: make get_index async

Browse Source
This commit is contained in:
Dietmar Maurer 2021-10-01 09:38:10 +02:00
parent 58a6e5f512
commit 89766c4f95
3 changed files with 17 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
proxmox-rest-server/examples/minimal-rest-server.rs
View File

@ -45,16 +45,18 @@ impl ApiAuth for DummyAuth {
// this should return the index page of the webserver // this should return the index page of the webserver
// iow. what the user browses to // iow. what the user browses to
fn get_index( fn get_index<'a>(
_auth_id: Option<String>, _auth_id: Option<String>,
_language: Option<String>, _language: Option<String>,
_api: &ApiConfig, _api: &'a ApiConfig,
_parts: http::request::Parts, _parts: http::request::Parts,
) -> http::Response<hyper::Body> { ) -> Pin<Box<dyn Future<Output = http::Response<hyper::Body>> + Send + 'a>> {
// build an index page Box::pin(async move {
http::Response::builder() // build an index page
.body("hello world".into()) http::Response::builder()
.unwrap() .body("hello world".into())
.unwrap()
})
} }
// a few examples on how to do api calls with the Router // a few examples on how to do api calls with the Router
@ -191,7 +193,7 @@ async fn run() -> Result<(), Error> {
&ROUTER, &ROUTER,
RpcEnvironmentType::PUBLIC, RpcEnvironmentType::PUBLIC,
Arc::new(DummyAuth {}), Arc::new(DummyAuth {}),
get_index, &get_index,
)?; )?;
let rest_server = RestServer::new(config); let rest_server = RestServer::new(config);

8
proxmox-rest-server/src/api_config.rs
View File

@ -3,6 +3,8 @@ use std::path::PathBuf;
use std::time::SystemTime; use std::time::SystemTime;
use std::fs::metadata; use std::fs::metadata;
use std::sync::{Arc, Mutex, RwLock}; use std::sync::{Arc, Mutex, RwLock};
use std::future::Future;
use std::pin::Pin;
use anyhow::{bail, Error, format_err}; use anyhow::{bail, Error, format_err};
use hyper::{Method, Body, Response}; use hyper::{Method, Body, Response};
@ -16,7 +18,7 @@ use proxmox::tools::fs::{create_path, CreateOptions};
use crate::{ApiAuth, FileLogger, FileLogOptions, CommandSocket}; use crate::{ApiAuth, FileLogger, FileLogOptions, CommandSocket};
pub type GetIndexFn = fn(Option<String>, Option<String>, &ApiConfig, Parts) -> Response<Body>; pub type GetIndexFn = &'static (dyn for<'a> Fn(Option<String>, Option<String>, &'a ApiConfig, Parts) -> Pin<Box<dyn Future<Output = Response<Body>> + Send + 'a>> + Send + Sync);
/// REST server configuration /// REST server configuration
pub struct ApiConfig { pub struct ApiConfig {
@ -68,13 +70,13 @@ impl ApiConfig {
}) })
} }
pub(crate) fn get_index( pub(crate) async fn get_index(
&self, &self,
auth_id: Option<String>, auth_id: Option<String>,
language: Option<String>, language: Option<String>,
parts: Parts, parts: Parts,
) -> Response<Body> { ) -> Response<Body> {
(self.get_index_fn)(auth_id, language, self, parts) (self.get_index_fn)(auth_id, language, self, parts).await
} }
pub(crate) fn find_method( pub(crate) fn find_method(

4
proxmox-rest-server/src/rest.rs
View File

@ -732,14 +732,14 @@ async fn handle_request(
let language = extract_lang_header(&parts.headers); let language = extract_lang_header(&parts.headers);
match auth.check_auth(&parts.headers, &method).await { match auth.check_auth(&parts.headers, &method).await {
Ok((auth_id, _user_info)) => { Ok((auth_id, _user_info)) => {
return Ok(api.get_index(Some(auth_id), language, parts)); return Ok(api.get_index(Some(auth_id), language, parts).await);
} }
Err(AuthError::Generic(_)) => { Err(AuthError::Generic(_)) => {
tokio::time::sleep_until(Instant::from_std(delay_unauth_time)).await; tokio::time::sleep_until(Instant::from_std(delay_unauth_time)).await;
} }
Err(AuthError::NoData) => {} Err(AuthError::NoData) => {}
} }
return Ok(api.get_index(None, language, parts)); return Ok(api.get_index(None, language, parts).await);
} else { } else {
let filename = api.find_alias(&components); let filename = api.find_alias(&components);
let compression = extract_compression_method(&parts.headers); let compression = extract_compression_method(&parts.headers);