feat: more instrumentation

Author: johnbchron

crates/auth-domain/src/lib.rs

@@ -50,6 +50,7 @@ impl AuthDomainService {
   }
 
   /// Sign up a [`User`].
+  #[tracing::instrument(skip(self))]
   pub async fn user_signup(
     &self,
     name: HumanName,
@@ -129,6 +130,7 @@ impl AuthDomainService {
   }
 
   /// Authenticate a [`User`].
+  #[tracing::instrument(skip(self))]
   pub async fn user_authenticate(
     &self,
     email: EmailAddress,
@@ -181,6 +183,7 @@ impl AuthnBackend for AuthDomainService {
   type Error = errors::AuthenticationError;
   type User = AuthUser;
 
+  #[tracing::instrument(skip(self))]
   async fn authenticate(
     &self,
     creds: Self::Credentials,
@@ -191,6 +194,7 @@ impl AuthnBackend for AuthDomainService {
       .map(|u| u.map(Into::into))
   }
 
+  #[tracing::instrument(skip(self))]
   async fn get_user(
     &self,
     id: &<Self::User as AxumLoginAuthUser>::Id,

crates/domain/src/upload/execute.rs

@@ -8,6 +8,7 @@ use models::{
   model::Model,
 };
 use serde::{Deserialize, Serialize};
+use tracing::{Instrument, info_span};
 
 use super::plan::UploadPlan;
 use crate::DomainService;
@@ -52,6 +53,7 @@ impl DomainService {
     let big_terrible_buffer = plan
       .nar_contents
       .collect()
+      .instrument(info_span!("collect_big_terrible_buffer"))
       .await
       .map_err(UploadExecutionError::InputDataError)?;
 

crates/grid/src/endpoints/upload.rs

@@ -50,7 +50,8 @@ pub async fn upload(
     body
       .into_data_stream()
       .map(|res| res.map_err(|e| io::Error::other(e.to_string()))),
-    Some(belt::DEFAULT_CHUNK_SIZE),
+    // Some(belt::DEFAULT_CHUNK_SIZE),
+    None,
   );
 
   let upload_req = UploadRequest {