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
46
47
48
49
50
51
52
53
54
55
56
|
use reflink_copy::ReflinkSupport;
use std::fs;
use std::io;
use std::path::Path;
use walkdir::WalkDir;
// cargo run --example copy_folder V:/1 V:/2
fn main() -> io::Result<()> {
let args: Vec<_> = std::env::args().collect();
if args.len() < 3 {
eprintln!("Usage: {} <source_folder> <target_folder>", args[0]);
return Ok(());
}
let from = Path::new(&args[1]);
let to = Path::new(&args[2]);
let reflink_support = reflink_copy::check_reflink_support(from, to)?;
println!("Reflink support: {reflink_support:?}");
let mut reflinked_count = 0u64;
let mut copied_count = 0u64;
for entry in WalkDir::new(from) {
let entry = entry?;
let relative_path = entry.path().strip_prefix(from).unwrap();
let target_path = to.join(relative_path);
if entry.file_type().is_dir() {
fs::create_dir_all(&target_path)?;
} else {
match reflink_support {
ReflinkSupport::Supported => {
reflink_copy::reflink(entry.path(), target_path)?;
reflinked_count = reflinked_count.saturating_add(1);
}
ReflinkSupport::Unknown => {
let result = reflink_copy::reflink_or_copy(entry.path(), target_path)?;
if result.is_some() {
copied_count = copied_count.saturating_add(1);
} else {
reflinked_count = reflinked_count.saturating_add(1);
}
}
ReflinkSupport::NotSupported => {
fs::copy(entry.path(), target_path)?;
copied_count = copied_count.saturating_add(1);
}
}
}
}
println!("reflinked files count: {reflinked_count}");
println!("copied files count: {copied_count}");
Ok(())
}
|
