diff options
Diffstat (limited to 'tvix/castore/src/proto')
| -rw-r--r-- | tvix/castore/src/proto/grpc_directoryservice_wrapper.rs | 2 | ||||
| -rw-r--r-- | tvix/castore/src/proto/mod.rs | 12 | ||||
| -rw-r--r-- | tvix/castore/src/proto/tests/directory.rs | 20 |
3 files changed, 17 insertions, 17 deletions
diff --git a/tvix/castore/src/proto/grpc_directoryservice_wrapper.rs b/tvix/castore/src/proto/grpc_directoryservice_wrapper.rs index 5e143a7bd7..097958050e 100644 --- a/tvix/castore/src/proto/grpc_directoryservice_wrapper.rs +++ b/tvix/castore/src/proto/grpc_directoryservice_wrapper.rs @@ -96,7 +96,7 @@ impl proto::directory_service_server::DirectoryService for GRPCDirectoryServiceW // We don't need to keep the contents around, they're stored in the DB. // https://github.com/rust-lang/rust-clippy/issues/5812 #[allow(clippy::mutable_key_type)] - let mut seen_directories_sizes: HashMap<B3Digest, u32> = HashMap::new(); + let mut seen_directories_sizes: HashMap<B3Digest, u64> = HashMap::new(); let mut last_directory_dgst: Option<B3Digest> = None; // Consume directories, and insert them into the store. diff --git a/tvix/castore/src/proto/mod.rs b/tvix/castore/src/proto/mod.rs index 07892ab799..c734e8be89 100644 --- a/tvix/castore/src/proto/mod.rs +++ b/tvix/castore/src/proto/mod.rs @@ -172,23 +172,23 @@ fn insert_once<'n>( Ok(()) } -fn checked_sum(iter: impl IntoIterator<Item = u32>) -> Option<u32> { - iter.into_iter().try_fold(0u32, |acc, i| acc.checked_add(i)) +fn checked_sum(iter: impl IntoIterator<Item = u64>) -> Option<u64> { + iter.into_iter().try_fold(0u64, |acc, i| acc.checked_add(i)) } impl Directory { /// The size of a directory is the number of all regular and symlink elements, /// the number of directory elements, and their size fields. - pub fn size(&self) -> u32 { + pub fn size(&self) -> u64 { if cfg!(debug_assertions) { self.size_checked() - .expect("Directory::size exceeds u32::MAX") + .expect("Directory::size exceeds u64::MAX") } else { - self.size_checked().unwrap_or(u32::MAX) + self.size_checked().unwrap_or(u64::MAX) } } - fn size_checked(&self) -> Option<u32> { + fn size_checked(&self) -> Option<u64> { checked_sum([ self.files.len().try_into().ok()?, self.symlinks.len().try_into().ok()?, diff --git a/tvix/castore/src/proto/tests/directory.rs b/tvix/castore/src/proto/tests/directory.rs index f2295740f2..a5e4f3c9c1 100644 --- a/tvix/castore/src/proto/tests/directory.rs +++ b/tvix/castore/src/proto/tests/directory.rs @@ -60,13 +60,13 @@ fn size() { #[test] #[cfg_attr(not(debug_assertions), ignore)] -#[should_panic = "Directory::size exceeds u32::MAX"] +#[should_panic = "Directory::size exceeds u64::MAX"] fn size_unchecked_panic() { let d = Directory { directories: vec![DirectoryNode { name: "foo".into(), digest: DUMMY_DIGEST.to_vec().into(), - size: u32::MAX, + size: u64::MAX, }], ..Default::default() }; @@ -81,12 +81,12 @@ fn size_unchecked_saturate() { directories: vec![DirectoryNode { name: "foo".into(), digest: DUMMY_DIGEST.to_vec().into(), - size: u32::MAX, + size: u64::MAX, }], ..Default::default() }; - assert_eq!(d.size(), u32::MAX); + assert_eq!(d.size(), u64::MAX); } #[test] @@ -98,18 +98,18 @@ fn size_checked() { directories: vec![DirectoryNode { name: "foo".into(), digest: DUMMY_DIGEST.to_vec().into(), - size: u32::MAX - 1, + size: u64::MAX - 1, }], ..Default::default() }; - assert_eq!(d.size_checked(), Some(u32::MAX)); + assert_eq!(d.size_checked(), Some(u64::MAX)); } { let d = Directory { directories: vec![DirectoryNode { name: "foo".into(), digest: DUMMY_DIGEST.to_vec().into(), - size: u32::MAX, + size: u64::MAX, }], ..Default::default() }; @@ -121,12 +121,12 @@ fn size_checked() { DirectoryNode { name: "foo".into(), digest: DUMMY_DIGEST.to_vec().into(), - size: u32::MAX / 2, + size: u64::MAX / 2, }, DirectoryNode { name: "foo".into(), digest: DUMMY_DIGEST.to_vec().into(), - size: u32::MAX / 2, + size: u64::MAX / 2, }, ], ..Default::default() @@ -363,7 +363,7 @@ fn validate_overflow() { directories: vec![DirectoryNode { name: "foo".into(), digest: DUMMY_DIGEST.to_vec().into(), - size: u32::MAX, + size: u64::MAX, }], ..Default::default() }; |