mirrors/relay
1
0
Fork 1
mirror of https://git.asonix.dog/asonix/relay.git synced 2024-09-29 23:32:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

Clean tracing a bit more

Browse source
This commit is contained in:
asonix 2022-11-15 19:56:13 -06:00
parent 2df34c9e55
commit 5011e05c3d
11 changed files with 56 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/data/actor.rs
View file

@ -37,7 +37,7 @@ impl ActorCache {
ActorCache { db }
}
#[tracing::instrument(name = "Get Actor", skip_all, fields(id = id.to_string().as_str(), requests))]
#[tracing::instrument(level = "debug" name = "Get Actor", skip_all, fields(id = id.to_string().as_str(), requests))]
pub(crate) async fn get(
&self,
id: &IriString,
@ -54,7 +54,7 @@ impl ActorCache {
.map(MaybeCached::Fetched)
}
#[tracing::instrument(name = "Add Connection", skip(self))]
#[tracing::instrument(level = "debug", name = "Add Connection", skip(self))]
pub(crate) async fn add_connection(&self, actor: Actor) -> Result<(), Error> {
let add_connection = self.db.add_connection(actor.id.clone());
let save_actor = self.db.save_actor(actor);
@ -64,12 +64,12 @@ impl ActorCache {
Ok(())
}
#[tracing::instrument(name = "Remove Connection", skip(self))]
#[tracing::instrument(level = "debug", name = "Remove Connection", skip(self))]
pub(crate) async fn remove_connection(&self, actor: &Actor) -> Result<(), Error> {
self.db.remove_connection(actor.id.clone()).await
}
#[tracing::instrument(name = "Fetch remote actor", skip_all, fields(id = id.to_string().as_str(), requests))]
#[tracing::instrument(level = "debug", name = "Fetch remote actor", skip_all, fields(id = id.to_string().as_str(), requests))]
pub(crate) async fn get_no_cache(
&self,
id: &IriString,

8
src/data/media.rs
View file

@ -19,17 +19,17 @@ impl MediaCache {
MediaCache { db }
}
#[tracing::instrument(name = "Get media uuid", skip_all, fields(url = url.to_string().as_str()))]
#[tracing::instrument(level = "debug", name = "Get media uuid", skip_all, fields(url = url.to_string().as_str()))]
pub(crate) async fn get_uuid(&self, url: IriString) -> Result<Option<Uuid>, Error> {
self.db.media_id(url).await
}
#[tracing::instrument(name = "Get media url", skip(self))]
#[tracing::instrument(level = "debug", name = "Get media url", skip(self))]
pub(crate) async fn get_url(&self, uuid: Uuid) -> Result<Option<IriString>, Error> {
self.db.media_url(uuid).await
}
#[tracing::instrument(name = "Is media outdated", skip(self))]
#[tracing::instrument(level = "debug", name = "Is media outdated", skip(self))]
pub(crate) async fn is_outdated(&self, uuid: Uuid) -> Result<bool, Error> {
if let Some(meta) = self.db.media_meta(uuid).await? {
if meta.saved_at + MEDIA_DURATION > SystemTime::now() {
@ -40,7 +40,7 @@ impl MediaCache {
Ok(true)
}
#[tracing::instrument(name = "Get media bytes", skip(self))]
#[tracing::instrument(level = "debug", name = "Get media bytes", skip(self))]
pub(crate) async fn get_bytes(&self, uuid: Uuid) -> Result<Option<(String, Bytes)>, Error> {
if let Some(meta) = self.db.media_meta(uuid).await? {
if meta.saved_at + MEDIA_DURATION > SystemTime::now() {

12
src/data/node.rs
View file

@ -34,7 +34,7 @@ impl NodeCache {
NodeCache { db }
}
#[tracing::instrument(name = "Get nodes", skip(self))]
#[tracing::instrument(level = "debug", name = "Get nodes", skip(self))]
pub(crate) async fn nodes(&self) -> Result<Vec<Node>, Error> {
let infos = self.db.connected_info();
let instances = self.db.connected_instance();
@ -59,7 +59,7 @@ impl NodeCache {
Ok(vec)
}
#[tracing::instrument(name = "Is NodeInfo Outdated", skip_all, fields(actor_id = actor_id.to_string().as_str()))]
#[tracing::instrument(level = "debug", name = "Is NodeInfo Outdated", skip_all, fields(actor_id = actor_id.to_string().as_str()))]
pub(crate) async fn is_nodeinfo_outdated(&self, actor_id: IriString) -> bool {
self.db
.info(actor_id)
@ -68,7 +68,7 @@ impl NodeCache {
.unwrap_or(true)
}
#[tracing::instrument(name = "Is Contact Outdated", skip_all, fields(actor_id = actor_id.to_string().as_str()))]
#[tracing::instrument(level = "debug", name = "Is Contact Outdated", skip_all, fields(actor_id = actor_id.to_string().as_str()))]
pub(crate) async fn is_contact_outdated(&self, actor_id: IriString) -> bool {
self.db
.contact(actor_id)
@ -77,7 +77,7 @@ impl NodeCache {
.unwrap_or(true)
}
#[tracing::instrument(name = "Is Instance Outdated", skip_all, fields(actor_id = actor_id.to_string().as_str()))]
#[tracing::instrument(level = "debug", name = "Is Instance Outdated", skip_all, fields(actor_id = actor_id.to_string().as_str()))]
pub(crate) async fn is_instance_outdated(&self, actor_id: IriString) -> bool {
self.db
.instance(actor_id)
@ -86,7 +86,7 @@ impl NodeCache {
.unwrap_or(true)
}
#[tracing::instrument(name = "Save node info", skip_all, fields(actor_id = actor_id.to_string().as_str(), software, version, reg))]
#[tracing::instrument(level = "debug", name = "Save node info", skip_all, fields(actor_id = actor_id.to_string().as_str(), software, version, reg))]
pub(crate) async fn set_info(
&self,
actor_id: IriString,
@ -108,6 +108,7 @@ impl NodeCache {
}
#[tracing::instrument(
level = "debug",
name = "Save instance info",
skip_all,
fields(
@ -144,6 +145,7 @@ impl NodeCache {
}
#[tracing::instrument(
level = "debug",
name = "Save contact info",
skip_all,
fields(

5
src/data/state.rs
View file

@ -48,6 +48,7 @@ impl State {
}
#[tracing::instrument(
level = "debug",
name = "Get inboxes for other domains",
skip_all,
fields(
@ -85,10 +86,10 @@ impl State {
self.object_cache.write().await.put(object_id, actor_id);
}
#[tracing::instrument(name = "Building state", skip_all)]
#[tracing::instrument(level = "debug", name = "Building state", skip_all)]
pub(crate) async fn build(db: Db) -> Result<Self, Error> {
let private_key = if let Ok(Some(key)) = db.private_key().await {
tracing::info!("Using existing key");
tracing::debug!("Using existing key");
key
} else {
tracing::info!("Generating new keys");

7
src/error.rs
View file

@ -7,7 +7,6 @@ use actix_web::{
};
use http_signature_normalization_actix::PrepareSignError;
use std::{convert::Infallible, fmt::Debug, io};
use tracing::error;
use tracing_error::SpanTrace;
pub(crate) struct Error {
@ -15,6 +14,12 @@ pub(crate) struct Error {
kind: ErrorKind,
}
impl Error {
pub(crate) fn is_breaker(&self) -> bool {
matches!(self.kind, ErrorKind::Breaker)
}
}
impl std::fmt::Debug for Error {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
writeln!(f, "{:?}", self.kind)

8
src/jobs/deliver.rs
View file

@ -35,7 +35,13 @@ impl Deliver {
#[tracing::instrument(name = "Deliver", skip(state))]
async fn permform(self, state: JobState) -> Result<(), Error> {
state.requests.deliver(self.to, &self.data).await?;
if let Err(e) = state.requests.deliver(self.to, &self.data).await {
if e.is_breaker() {
tracing::debug!("Not trying due to failed breaker");
return Ok(());
}
return Err(e);
}
Ok(())
}
}

12
src/jobs/instance.rs
View file

@ -47,10 +47,18 @@ impl QueryInstance {
let scheme = self.actor_id.scheme_str();
let instance_uri = iri!(format!("{}://{}/api/v1/instance", scheme, authority));
let instance = state
let instance = match state
.requests
.fetch_json::<Instance>(instance_uri.as_str())
.await?;
.await
{
Ok(instance) => instance,
Err(e) if e.is_breaker() => {
tracing::debug!("Not retrying due to failed breaker");
return Ok(());
}
Err(e) => return Err(e),
};
let description = instance.short_description.unwrap_or(instance.description);

9
src/jobs/nodeinfo.rs
View file

@ -52,7 +52,14 @@ impl QueryNodeinfo {
return Ok(());
};
let nodeinfo = state.requests.fetch_json::<Nodeinfo>(&href).await?;
let nodeinfo = match state.requests.fetch_json::<Nodeinfo>(&href).await {
Ok(nodeinfo) => nodeinfo,
Err(e) if e.is_breaker() => {
tracing::debug!("Not retrying due to failed breaker");
return Ok(());
}
Err(e) => return Err(e),
};
state
.node_cache

2
src/middleware/verifier.rs
View file

@ -16,7 +16,7 @@ use std::{future::Future, pin::Pin};
pub(crate) struct MyVerify(pub Requests, pub ActorCache, pub State);
impl MyVerify {
#[tracing::instrument("Verify signature", skip(self))]
#[tracing::instrument("Verify signature", skip(self, signature))]
async fn verify(
&self,
algorithm: Option<Algorithm>,

11
src/requests.rs
View file

@ -20,7 +20,6 @@ use std::{
},
time::{Duration, SystemTime},
};
use tracing::{debug, info, warn};
use tracing_awc::Tracing;
const ONE_SECOND: u64 = 1;
@ -193,7 +192,7 @@ impl Requests {
fn count_err(&self) {
let count = self.consecutive_errors.fetch_add(1, Ordering::Relaxed);
if count + 1 >= self.error_limit {
warn!("{} consecutive errors, rebuilding http client", count);
tracing::warn!("{} consecutive errors, rebuilding http client", count);
*self.client.borrow_mut() = Client::builder()
.wrap(Tracing)
.add_default_header(("User-Agent", self.user_agent.clone()))
@ -261,7 +260,7 @@ impl Requests {
if let Ok(bytes) = res.body().await {
if let Ok(s) = String::from_utf8(bytes.as_ref().to_vec()) {
if !s.is_empty() {
debug!("Response from {}, {}", url, s);
tracing::warn!("Response from {}, {}", url, s);
}
}
}
@ -289,7 +288,7 @@ impl Requests {
return Err(ErrorKind::Breaker.into());
}
info!("Fetching bytes for {}", url);
tracing::info!("Fetching bytes for {}", url);
let signer = self.signer();
let client: Client = self.client.borrow().clone();
@ -329,7 +328,7 @@ impl Requests {
if let Ok(bytes) = res.body().await {
if let Ok(s) = String::from_utf8(bytes.as_ref().to_vec()) {
if !s.is_empty() {
debug!("Response from {}, {}", url, s);
tracing::warn!("Response from {}, {}", url, s);
}
}
}
@ -400,7 +399,7 @@ impl Requests {
if let Ok(bytes) = res.body().await {
if let Ok(s) = String::from_utf8(bytes.as_ref().to_vec()) {
if !s.is_empty() {
warn!("Response from {}, {}", inbox.as_str(), s);
tracing::warn!("Response from {}, {}", inbox.as_str(), s);
}
}
}

2
src/routes/inbox.rs
View file

@ -16,7 +16,7 @@ use activitystreams::{
use actix_web::{web, HttpResponse};
use http_signature_normalization_actix::prelude::{DigestVerified, SignatureVerified};
#[tracing::instrument(name = "Inbox", skip(actors, client, jobs, config, state))]
#[tracing::instrument(name = "Inbox", skip(actors, client, jobs, config, state, verified))]
pub(crate) async fn route(
state: web::Data<State>,
actors: web::Data<ActorCache>,