about summary refs log tree commit diff
path: root/tvix/castore/src/directoryservice/grpc.rs
diff options
context:
space:
mode:
authorFlorian Klink <flokli@flokli.de>2024-03-20T15·31+0200
committerflokli <flokli@flokli.de>2024-03-20T21·02+0000
commit345a639e799bed7a11ab75e3e90b4246e64269df (patch)
treeeb2fd33d40ebae07d79bfea80a05f47e1ac07ba8 /tvix/castore/src/directoryservice/grpc.rs
parent60b47b336b2c1bc2d39dbe5b6360cab1cea9a5eb (diff)
refactor(tvix/castore): instrument DirectoryPutter impls consistently r/7756
Log the entire span with "trace" level, not just its `ret` level.

The level of the error value event defaults to ERROR, so we don't loose
these.

B3Digest implements Debug and Display the same way, so we can omit the
`(Display)` part in `ret(Display)` for them.

Change-Id: Id00d123a5798e5bdc9820dd97ae2b4d4eb5455f0
Reviewed-on: https://cl.tvl.fyi/c/depot/+/11218
Tested-by: BuildkiteCI
Reviewed-by: Connor Brewster <cbrewster@hey.com>
Diffstat (limited to 'tvix/castore/src/directoryservice/grpc.rs')
-rw-r--r--tvix/castore/src/directoryservice/grpc.rs4
1 files changed, 3 insertions, 1 deletions
diff --git a/tvix/castore/src/directoryservice/grpc.rs b/tvix/castore/src/directoryservice/grpc.rs
index fbb9cce64e88..fe410a38257d 100644
--- a/tvix/castore/src/directoryservice/grpc.rs
+++ b/tvix/castore/src/directoryservice/grpc.rs
@@ -234,6 +234,7 @@ impl GRPCPutter {
 
 #[async_trait]
 impl DirectoryPutter for GRPCPutter {
+    #[instrument(level = "trace", skip_all, fields(directory.digest=%directory.digest()), err)]
     async fn put(&mut self, directory: proto::Directory) -> Result<(), crate::Error> {
         match self.rq {
             // If we're not already closed, send the directory to directory_sender.
@@ -253,7 +254,8 @@ impl DirectoryPutter for GRPCPutter {
         }
     }
 
-    /// Closes the stream for sending, and returns the value
+    /// Closes the stream for sending, and returns the value.
+    #[instrument(level = "trace", skip_all, ret, err)]
     async fn close(&mut self) -> Result<B3Digest, crate::Error> {
         // get self.rq, and replace it with None.
         // This ensures we can only close it once.