httm 0.16.5

A CLI tool for viewing snapshot file versions on ZFS and btrfs datasets
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

//       ___           ___           ___           ___
//      /\__\         /\  \         /\  \         /\__\
//     /:/  /         \:\  \        \:\  \       /::|  |
//    /:/__/           \:\  \        \:\  \     /:|:|  |
//   /::\  \ ___       /::\  \       /::\  \   /:/|:|__|__
//  /:/\:\  /\__\     /:/\:\__\     /:/\:\__\ /:/ |::::\__\
//  \/__\:\/:/  /    /:/  \/__/    /:/  \/__/ \/__/~~/:/  /
//       \::/  /    /:/  /        /:/  /            /:/  /
//       /:/  /     \/__/         \/__/            /:/  /
//      /:/  /                                    /:/  /
//      \/__/                                     \/__/
//
// (c) Robert Swinford <robert.swinford<...at...>gmail.com>
//
// For the full copyright and license information, please view the LICENSE file
// that was distributed with this source code.

use std::{path::Path, path::PathBuf};

use rayon::prelude::*;

use crate::data::filesystem_map::{MapOfAlts, MostProximateAndOptAlts};
use crate::library::results::{HttmError, HttmResult};
use crate::MapOfDatasets;

// instead of looking up, precompute possible alt replicated mounts before exec
pub fn precompute_alt_replicated(map_of_datasets: &MapOfDatasets) -> MapOfAlts {
    map_of_datasets
        .par_iter()
        .flat_map(|(mount, _dataset_info)| {
            get_alt_replicated_datasets(mount, map_of_datasets)
                .map(|datasets| (mount.to_path_buf(), datasets))
        })
        .collect()
}

fn get_alt_replicated_datasets(
    proximate_dataset_mount: &Path,
    map_of_datasets: &MapOfDatasets,
) -> HttmResult<MostProximateAndOptAlts> {
    let proximate_dataset_fs_name = match &map_of_datasets.get(proximate_dataset_mount) {
        Some(dataset_info) => dataset_info.name.clone(),
        None => {
            return Err(HttmError::new("httm was unable to detect an alternate replicated mount point.  Perhaps the replicated filesystem is not mounted?").into());
        }
    };

    // find a filesystem that ends with our most local filesystem name
    // but which has a prefix, like a different pool name: rpool might be
    // replicated to tank/rpool
    let mut alt_replicated_mounts: Vec<PathBuf> = map_of_datasets
        .par_iter()
        .filter(|(_mount, dataset_info)| {
            dataset_info.name != proximate_dataset_fs_name
                && dataset_info
                    .name
                    .ends_with(proximate_dataset_fs_name.as_str())
        })
        .map(|(mount, _fsname)| mount)
        .cloned()
        .collect();

    if alt_replicated_mounts.is_empty() {
        // could not find the any replicated mounts
        Err(HttmError::new("httm was unable to detect an alternate replicated mount point.  Perhaps the replicated filesystem is not mounted?").into())
    } else {
        alt_replicated_mounts.sort_unstable_by_key(|path| path.as_os_str().len());
        Ok(MostProximateAndOptAlts {
            proximate_dataset_mount: proximate_dataset_mount.to_path_buf(),
            opt_datasets_of_interest: Some(alt_replicated_mounts),
        })
    }
}