mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(tvix/store/pathinfo/grpc): instrument functions

Browse source 
Change-Id: Idb7d7144be1917fbaf83e9fd76c5b2ebb3df98d2
Reviewed-on: https://cl.tvl.fyi/c/depot/+/11400
Autosubmit: flokli <flokli@flokli.de>
Reviewed-by: raitobezarius <tvl@lahfa.xyz>
Tested-by: BuildkiteCI
This commit is contained in:
Florian Klink 2024-04-12 20:19:40 +03:00 committed by clbot
parent 7bcb896e48
commit dcd68155f0
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tvix/store/src/pathinfoservice/grpc.rs
View file

@ -1,8 +1,10 @@
use super::PathInfoService; use super::PathInfoService;
use crate::proto::{self, ListPathInfoRequest, PathInfo}; use crate::proto::{self, ListPathInfoRequest, PathInfo};
use async_stream::try_stream; use async_stream::try_stream;
use data_encoding::BASE64;
use futures::stream::BoxStream; use futures::stream::BoxStream;
use tonic::{async_trait, transport::Channel, Code}; use tonic::{async_trait, transport::Channel, Code};
use tracing::instrument;
use tvix_castore::{proto as castorepb, Error}; use tvix_castore::{proto as castorepb, Error};
/// Connects to a (remote) tvix-store PathInfoService over gRPC. /// Connects to a (remote) tvix-store PathInfoService over gRPC.
@ -25,6 +27,7 @@ impl GRPCPathInfoService {
#[async_trait] #[async_trait]
impl PathInfoService for GRPCPathInfoService { impl PathInfoService for GRPCPathInfoService {
#[instrument(level = "trace", skip_all, fields(path_info.digest = BASE64.encode(&digest)))]
async fn get(&self, digest: [u8; 20]) -> Result<Option<PathInfo>, Error> { async fn get(&self, digest: [u8; 20]) -> Result<Option<PathInfo>, Error> {
let path_info = self let path_info = self
.grpc_client .grpc_client
@ -51,6 +54,7 @@ impl PathInfoService for GRPCPathInfoService {
} }
} }
#[instrument(level = "trace", skip_all, fields(path_info.root_node = ?path_info.node))]
async fn put(&self, path_info: PathInfo) -> Result<PathInfo, Error> { async fn put(&self, path_info: PathInfo) -> Result<PathInfo, Error> {
let path_info = self let path_info = self
.grpc_client .grpc_client
@ -63,6 +67,7 @@ impl PathInfoService for GRPCPathInfoService {
Ok(path_info) Ok(path_info)
} }
#[instrument(level = "trace", skip_all, fields(root_node = ?root_node))]
async fn calculate_nar( async fn calculate_nar(
&self, &self,
root_node: &castorepb::node::Node, root_node: &castorepb::node::Node,
@ -86,6 +91,7 @@ impl PathInfoService for GRPCPathInfoService {
Ok((path_info.nar_size, nar_sha256)) Ok((path_info.nar_size, nar_sha256))
} }
#[instrument(level = "trace", skip_all)]
fn list(&self) -> BoxStream<'static, Result<PathInfo, Error>> { fn list(&self) -> BoxStream<'static, Result<PathInfo, Error>> {
let mut grpc_client = self.grpc_client.clone(); let mut grpc_client = self.grpc_client.clone();