mirror/trailbase
1
0
Fork 0
mirror of https://github.com/trailbaseio/trailbase.git synced 2026-02-04 16:19:04 -06:00
Code Issues Packages Projects Releases Wiki Activity

Stream WASM http response body.

Browse Source
This commit is contained in:
Sebastian Jeltsch
2026-01-09 22:24:48 +01:00
parent 3e9997c3d3
commit 5baea96182
1 changed files with 32 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
crates/core/src/wasm/mod.rs
View File

@@ -171,37 +171,41 @@ pub(crate) async fn install_routes_and_jobs(
async move |params: RawPathParams, user: Option<User>, req: Request| -> Response {
// Construct WASI request form hyper/axum request.
let (mut parts, body) = req.into_parts();
let request = hyper::Request::from_parts(
{
let Ok(header_value) = to_header_value(&HttpContext {
kind: HttpContextKind::Http,
registered_path,
path_params: params
.iter()
.map(|(name, value)| (name.to_string(), value.to_string()))
.collect(),
user: user.map(|u| HttpContextUser {
id: u.id,
email: u.email,
csrf_token: u.csrf_token,
}),
}) else {
return Response::builder()
.status(StatusCode::BAD_REQUEST)
.body("header encoding failed".into())
.unwrap_or_default();
};
parts.headers.insert("__context", header_value);
parts
},
let Ok(header_value) = to_header_value(&HttpContext {
kind: HttpContextKind::Http,
registered_path,
path_params: params
.iter()
.map(|(name, value)| (name.to_string(), value.to_string()))
.collect(),
user: user.map(|u| HttpContextUser {
id: u.id,
email: u.email,
csrf_token: u.csrf_token,
}),
}) else {
return Response::builder()
.status(StatusCode::BAD_REQUEST)
.body("header encoding failed".into())
.unwrap_or_default();
};
parts.headers.insert("__context", header_value);
let request = hyper::Request::from_parts(
parts,
UnsyncBoxBody::new(
// NOTE: Ideally we'd stream the request body, however there's no way for us to re-map
// axum::Error to hyper::Error. All hyper::Error's constructors are private. This is
// likely an oversight in wasi_http.
http_body_util::Full::new({
let Ok(body) = body.collect().await else {
return internal("request buffering failed");
};
body.to_bytes()
})
// Remapping Body impl's error type from infallible to hyper::Error.
.map_err(|_| unreachable!()),
),
);
@@ -216,13 +220,11 @@ pub(crate) async fn install_routes_and_jobs(
Ok(response) => {
// Construct hyper/axum response from WASI response.
let (parts, body) = response.into_parts();
Response::from_parts(parts, {
// TODO: allow streaming responses, i.e. plumb WASI stream into hyper stream.
let Ok(body) = body.collect().await else {
return internal("response");
};
body.to_bytes().into()
})
Response::from_parts(
parts,
axum::body::Body::from_stream(body.into_data_stream()),
)
}
Err(err) => {
debug!("`call_incoming_http_handler` returned: {err}");