ic-query 0.59.0

Internet Computer query library for CloudEngine, NNS, SNS, ICRC, system canisters, and 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
56
57
58
59
60
61
62
63
64
65
66
67
68
69

//! Module: sns::report::neurons_cache::refresh::run
//!
//! Responsibility: expose complete SNS neuron snapshot refresh entry points.
//! Does not own: shared locking and attempt mechanics, cache publication, or text rendering.
//! Boundary: adapts neuron collection and publication to the shared refresh lifecycle.

use crate::{
    QueryProgress,
    progress::IgnoreQueryProgress,
    sns::report::{
        SnsHostError, SnsSnapshotRefreshReport, SnsSnapshotRefreshRequest,
        cache_refresh::{publish_complete_sns_snapshot, run_resolved_sns_snapshot_refresh},
        live::LiveSnsSource,
        neurons_cache::{collection::fetch_complete_sns_neurons, paths::SnsNeuronsCacheCollection},
        source::SnsNeuronsSource,
    },
};

pub const DEFAULT_SNS_NEURONS_REFRESH_LOCK_STALE_SECONDS: u64 = 30 * 60;

pub async fn refresh_sns_neurons_cache(
    request: &SnsSnapshotRefreshRequest,
) -> Result<SnsSnapshotRefreshReport, SnsHostError> {
    refresh_sns_neurons_cache_with_source(request, &LiveSnsSource::default()).await
}

/// Refresh a complete SNS neuron snapshot and emit structured progress events.
pub async fn refresh_sns_neurons_cache_with_progress(
    request: &SnsSnapshotRefreshRequest,
    progress: &mut dyn QueryProgress,
) -> Result<SnsSnapshotRefreshReport, SnsHostError> {
    refresh_sns_neurons_cache_with_source_and_progress(request, &LiveSnsSource::default(), progress)
        .await
}

pub async fn refresh_sns_neurons_cache_with_source(
    request: &SnsSnapshotRefreshRequest,
    source: &dyn SnsNeuronsSource,
) -> Result<SnsSnapshotRefreshReport, SnsHostError> {
    let mut progress = IgnoreQueryProgress;
    refresh_sns_neurons_cache_with_source_and_progress(request, source, &mut progress).await
}

async fn refresh_sns_neurons_cache_with_source_and_progress(
    request: &SnsSnapshotRefreshRequest,
    source: &dyn SnsNeuronsSource,
    progress: &mut dyn QueryProgress,
) -> Result<SnsSnapshotRefreshReport, SnsHostError> {
    run_resolved_sns_snapshot_refresh::<SnsNeuronsCacheCollection, _>(
        request,
        source,
        DEFAULT_SNS_NEURONS_REFRESH_LOCK_STALE_SECONDS,
        async |context| {
            let complete =
                fetch_complete_sns_neurons(context.refresh_context(), source, progress).await?;
            publish_complete_sns_snapshot(
                context,
                crate::sns::report::neurons_cache::SNS_NEURONS_CACHE_SCHEMA_VERSION,
                complete.page_count,
                complete.rows.len(),
                complete.last_cursor,
                crate::sns::report::neurons_cache::model::SnsNeuronsCacheRows {
                    neurons: complete.rows,
                },
            )
        },
    )
    .await
}