chore: add more spans

Author: Randoooom

src/services/v1/field.rs

@@ -30,6 +30,7 @@ use feedback_fusion_common::proto::{
     UpdateFieldRequest,
 };
 
+#[instrument(skip_all)]
 pub async fn create_field(
     context: &FeedbackFusionV1Context,
     request: Request<CreateFieldRequest>,
@@ -51,6 +52,7 @@ pub async fn create_field(
     Ok(Response::new(field.into()))
 }
 
+#[instrument(skip_all)]
 pub async fn get_active_fields(
     context: &PublicFeedbackFusionV1Context,
     request: Request<GetFieldsRequest>,
@@ -88,6 +90,7 @@ pub async fn get_active_fields(
     }))
 }
 
+#[instrument(skip_all)]
 pub async fn get_fields(
     context: &FeedbackFusionV1Context,
     request: Request<GetFieldsRequest>,
@@ -117,6 +120,7 @@ pub async fn get_fields(
     }))
 }
 
+#[instrument(skip_all)]
 pub async fn update_field(
     context: &FeedbackFusionV1Context,
     request: Request<UpdateFieldRequest>,
@@ -144,6 +148,7 @@ pub async fn update_field(
     Ok(Response::new(field.into()))
 }
 
+#[instrument(skip_all)]
 pub async fn delete_field(
     context: &FeedbackFusionV1Context,
     request: Request<DeleteFieldRequest>,

src/services/v1/mod.rs

@@ -85,7 +85,7 @@ macro_rules! handler {
                 )*
             ];
 
-             match policy.evaluate(
+            match policy.evaluate(
                 $request
                     .extensions()
                     .get::<OIDCClaims>()
@@ -96,7 +96,10 @@ macro_rules! handler {
                     handler!($handler, $self, $request)
                 },
                 Err(_) => {
-                    let target = async $target.await;
+                    let target = info_span!("Extracting referenced target")
+                                    .in_scope(|| async {
+                                        async $target.await
+                                    }).await;
 
                     match target {
                         Ok(target) => {
@@ -107,7 +110,7 @@ macro_rules! handler {
                                      )*
                                 ];
 
-                                 policy
+                                policy
                                     .evaluate(
                                         $request
                                             .extensions()
@@ -139,7 +142,7 @@ macro_rules! handler {
                     }
 
                 }
-            } 
+            }
         }
     }};
     ($handler:path, $self:ident, $request:ident, $($scope:literal $(,)?)*) => {{
@@ -150,7 +153,7 @@ macro_rules! handler {
             )*
         ];
 
-         policy
+        policy
             .evaluate(
                 $request
                     .extensions()

src/services/v1/prompt.rs

@@ -27,6 +27,7 @@ use feedback_fusion_common::proto::{
 };
 use validator::Validate;
 
+#[instrument(skip_all)]
 pub async fn create_prompt(
     context: &FeedbackFusionV1Context,
     request: Request<CreatePromptRequest>,
@@ -47,6 +48,7 @@ pub async fn create_prompt(
     Ok(Response::new(prompt.into()))
 }
 
+#[instrument(skip_all)]
 pub async fn get_prompt(
     context: &PublicFeedbackFusionV1Context,
     request: Request<GetPromptRequest>,
@@ -65,6 +67,7 @@ pub async fn get_prompt(
     }
 }
 
+#[instrument(skip_all)]
 pub async fn get_prompts(
     context: &FeedbackFusionV1Context,
     request: Request<GetPromptsRequest>,
@@ -91,6 +94,7 @@ pub async fn get_prompts(
     }))
 }
 
+#[instrument(skip_all)]
 pub async fn update_prompt(
     context: &FeedbackFusionV1Context,
     request: Request<UpdatePromptRequest>,
@@ -112,6 +116,7 @@ pub async fn update_prompt(
     Ok(Response::new(prompt.into()))
 }
 
+#[instrument(skip_all)]
 pub async fn delete_prompt(
     context: &FeedbackFusionV1Context,
     request: Request<DeletePromptRequest>,

src/services/v1/response.rs

@@ -32,6 +32,7 @@ use feedback_fusion_common::proto::{
 use rbatis::rbatis_codegen::IntoSql;
 use std::collections::HashMap;
 
+#[instrument(skip_all)]
 pub async fn create_responses(
     context: &PublicFeedbackFusionV1Context,
     request: Request<CreateResponsesRequest>,
@@ -107,6 +108,7 @@ async fn field_responses(
     impled!()
 }
 
+#[instrument(skip_all)]
 pub async fn get_responses(
     context: &FeedbackFusionV1Context,
     request: Request<GetResponsesRequest>,