mirrors/buzzrelay
1
0
Fork 0
mirror of https://github.com/astro/buzzrelay.git synced 2024-11-23 04:30:58 +00:00
Code Issues Projects Releases Packages Wiki Activity

fetch, send: run signature in spawn_blocking

Browse source
This commit is contained in:
Astro 2024-06-07 02:18:11 +02:00
parent cd36a9ecb1
commit 1e42ab1ee3
2 changed files with 22 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/fetch.rs
View file

@ -2,6 +2,7 @@ use std::time::SystemTime;
use http::StatusCode;
use serde::de::DeserializeOwned;
use sigh::{PrivateKey, SigningConfig, alg::RsaSha256};
use tokio::task::spawn_blocking;
use crate::{digest, error::Error};
pub async fn authorized_fetch<T>(
@ -26,11 +27,17 @@ where
.header("accept", "application/activity+json")
.header("digest", digest_header)
.body(vec![])?;
SigningConfig::new(RsaSha256, private_key, key_id)
.sign(&mut req)?;
let private_key = private_key.clone();
let key_id = key_id.to_string();
let req = spawn_blocking(move || {
SigningConfig::new(RsaSha256, &private_key, &key_id).sign(&mut req)?;
Ok(req)
})
.await
.map_err(|e| Error::Response(format!("{e}")))?
.map_err(|e: sigh::Error| Error::Response(format!("{e}")))?;
let req: reqwest::Request = req.try_into()?;
let res = client.execute(req)
.await?;
let res = client.execute(req).await?;
if res.status() >= StatusCode::OK && res.status() < StatusCode::MULTIPLE_CHOICES {
Ok(res.json().await?)
} else {

15
src/send.rs
View file

@ -6,6 +6,7 @@ use http::StatusCode;
use metrics::histogram;
use serde::Serialize;
use sigh::{PrivateKey, SigningConfig, alg::RsaSha256};
use tokio::task::spawn_blocking;
use crate::{digest, error::Error};
pub async fn send<T: Serialize>(
@ -42,12 +43,18 @@ pub async fn send_raw(
.header("digest", digest_header)
.body(body.as_ref().clone())?;
let t1 = Instant::now();
SigningConfig::new(RsaSha256, private_key, key_id)
.sign(&mut req)?;
let private_key = private_key.clone();
let key_id = key_id.to_string();
let req = spawn_blocking(move || {
SigningConfig::new(RsaSha256, &private_key, &key_id).sign(&mut req)?;
Ok(req)
})
.await
.map_err(|e| Error::Response(format!("{e}")))?
.map_err(|e: sigh::Error| Error::Response(format!("{e}")))?;
let t2 = Instant::now();
let req: reqwest::Request = req.try_into()?;
let res = client.execute(req)
.await?;
let res = client.execute(req).await?;
let t3 = Instant::now();
histogram!("relay_http_request_duration")
.record(t2 - t1);