ic-query 0.2.17

Internet Computer query CLI for NNS, SNS, and related public network metadata
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
46
47
48
49
50
51
52
53
54
55

use super::PagedCollectionPage;
use crate::progress::ProgressLine;

pub trait PagedSnapshotRefresh {
    type Complete;
    type Error;

    fn progress_text(&self) -> String;
    fn max_pages_reached(&self) -> bool;
    fn incomplete_refresh_error(&self) -> Self::Error;
    fn fetch_next_page(&mut self) -> Result<PagedCollectionPage, Self::Error>;
    fn write_running_attempt(&self, page: &PagedCollectionPage) -> Result<(), Self::Error>;
    fn page_exhausts_collection(&self, page: &PagedCollectionPage) -> bool;
    fn into_complete(self) -> Self::Complete;
}

pub fn run_paged_snapshot_refresh<Refresh>(
    mut refresh: Refresh,
) -> Result<Refresh::Complete, Refresh::Error>
where
    Refresh: PagedSnapshotRefresh,
{
    let mut progress = ProgressLine::stderr();
    progress.update(&refresh.progress_text());

    loop {
        if refresh.max_pages_reached() {
            progress.finish(&format!(
                "{} stopped before completion",
                refresh.progress_text()
            ));
            return Err(refresh.incomplete_refresh_error());
        }

        let page = match refresh.fetch_next_page() {
            Ok(page) => page,
            Err(err) => {
                progress.finish(&format!("{} failed", refresh.progress_text()));
                return Err(err);
            }
        };
        if let Err(err) = refresh.write_running_attempt(&page) {
            progress.finish(&format!("{} failed", refresh.progress_text()));
            return Err(err);
        }
        progress.update(&refresh.progress_text());

        if refresh.page_exhausts_collection(&page) {
            break;
        }
    }

    progress.finish(&format!("{} complete", refresh.progress_text()));
    Ok(refresh.into_complete())
}