ifs 0.1.34

util for write / read file
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

use std::path::Path;

#[macro_export]
macro_rules! hash {
  ($($await:ident)? : $Hasher:ty, $reader:expr $(,$write:ident)?) => {{
    let mut hasher = <$Hasher>::new();
    let mut reader = $reader;
/*
Hasher 实现 std::io::Write ，因此可以使用 std::io::copy 它从任何读取器更新 A Hasher 。

不幸的是，这种标准方法可能会限制性能，因为 copy 目前使用的内部 8 KiB 缓冲区不够大，无法利用所有 SIMD 指令集。

特别是，AVX-512 需要 16 KiB 缓冲区

https://docs.rs/blake3/latest/blake3/struct.Hasher.html
*/
    let mut buf = [0; 16384];
    let mut len:usize = 0;

    loop {
      let n = reader.read(&mut buf)$(.$await)??;
      if n == 0 {
        break;
      }
      len+=n;
      let bin = &buf[..n];
      hasher.update(bin);
      $(let _ = $write.write(bin)?;)?
    }
    (hasher,len)
  }};
}

pub async fn hash<H: digest::Digest>(path: impl AsRef<Path>) -> Result<H, std::io::Error> {
  use tokio::{fs::File, io::AsyncReadExt};
  let file = File::open(path).await?;
  Ok(hash!(await: H, file).0)
}

pub fn hash_len(path: impl AsRef<Path>) -> Result<([u8; 32], usize), std::io::Error> {
  use std::io::Read;
  let file = std::fs::File::open(path)?;
  let (hash, len) = hash!(: blake3::Hasher, file);
  Ok((*hash.finalize().as_bytes(), len))
}