canic-core 0.21.6

Canic — a canister orchestration and management toolkit for the Internet Computer
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

use crate::{
    dto::placement::scaling::{ScalingRegistryEntry, ScalingRegistryResponse},
    ops::{placement::scaling::mapper::WorkerEntryRecordMapper, prelude::*},
    storage::stable::scaling::{ScalingRegistry, ScalingRegistryRecord, WorkerEntryRecord},
    view::placement::scaling::ScalingWorkerPlanEntry,
};

///
/// ScalingRegistryOps
/// Stable storage wrapper for the scaling worker registry.
///

pub struct ScalingRegistryOps;

impl ScalingRegistryOps {
    #[expect(dead_code)]
    pub fn upsert(pid: Principal, entry: WorkerEntryRecord) {
        ScalingRegistry::upsert(pid, entry);
    }

    pub fn upsert_from_plan(pid: Principal, plan: ScalingWorkerPlanEntry, created_at_secs: u64) {
        let entry = WorkerEntryRecordMapper::validated_to_record(plan, created_at_secs);
        ScalingRegistry::upsert(pid, entry);
    }

    /// Lookup all workers in a given pool
    #[must_use]
    pub fn find_by_pool(pool: &str) -> Vec<(Principal, WorkerEntryRecord)> {
        ScalingRegistry::export()
            .entries
            .into_iter()
            .filter(|(_, entry)| entry.pool.as_ref() == pool)
            .collect()
    }

    #[must_use]
    #[expect(clippy::cast_possible_truncation)]
    pub fn count_by_pool(pool: &str) -> u32 {
        Self::find_by_pool(pool).len() as u32
    }

    #[must_use]
    #[expect(dead_code)]
    pub fn export() -> ScalingRegistryRecord {
        ScalingRegistry::export()
    }

    #[must_use]
    pub fn entries_response() -> ScalingRegistryResponse {
        let entries = ScalingRegistry::export()
            .entries
            .into_iter()
            .map(|(pid, entry)| ScalingRegistryEntry {
                pid,
                entry: WorkerEntryRecordMapper::record_to_view(&entry),
            })
            .collect();

        ScalingRegistryResponse(entries)
    }
}