From 7bcb896e48cee8b40ecb846cfaf255091561395e Mon Sep 17 00:00:00 2001
From: Florian Klink <flokli@flokli.de>
Date: Fri, 12 Apr 2024 20:19:06 +0300
Subject: feat(tvix/castore/directory/grpc): instrument functions

Change-Id: I9cc0a6a32184773597556ab5f9250257aa18ca4e
Reviewed-on: https://cl.tvl.fyi/c/depot/+/11399
Autosubmit: flokli <flokli@flokli.de>
Reviewed-by: raitobezarius <tvl@lahfa.xyz>
Tested-by: BuildkiteCI
---
 tvix/castore/src/directoryservice/grpc.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'tvix/castore/src/directoryservice')

diff --git a/tvix/castore/src/directoryservice/grpc.rs b/tvix/castore/src/directoryservice/grpc.rs
index 84cf01e1679e..7402fe1b5659 100644
--- a/tvix/castore/src/directoryservice/grpc.rs
+++ b/tvix/castore/src/directoryservice/grpc.rs
@@ -34,6 +34,7 @@ impl GRPCDirectoryService {
 
 #[async_trait]
 impl DirectoryService for GRPCDirectoryService {
+    #[instrument(level = "trace", skip_all, fields(directory.digest = %digest))]
     async fn get(
         &self,
         digest: &B3Digest,
@@ -82,6 +83,7 @@ impl DirectoryService for GRPCDirectoryService {
         }
     }
 
+    #[instrument(level = "trace", skip_all, fields(directory.digest = %directory.digest()))]
     async fn put(&self, directory: crate::proto::Directory) -> Result<B3Digest, crate::Error> {
         let resp = self
             .grpc_client
@@ -101,7 +103,7 @@ impl DirectoryService for GRPCDirectoryService {
         }
     }
 
-    #[instrument(skip_all, fields(directory.digest = %root_directory_digest))]
+    #[instrument(level = "trace", skip_all, fields(directory.digest = %root_directory_digest))]
     fn get_recursive(
         &self,
         root_directory_digest: &B3Digest,
-- 
cgit 1.4.1