canic-core 0.85.1

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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

use crate::{
    cdk::structures::{DefaultMemoryImpl, cell::Cell, memory::VirtualMemory},
    role_contract::allocation::memory::env::SUBNET_STATE_ID,
    storage::prelude::*,
};
use std::cell::RefCell;

//
// SUBNET_STATE
//

eager_static! {
    static SUBNET_STATE: RefCell<Cell<SubnetStateRecord, VirtualMemory<DefaultMemoryImpl>>> =
        RefCell::new(Cell::init(
            crate::ic_memory_key!(authority = CANIC_CORE_MEMORY_AUTHORITY, key = "canic.core.subnet_state.v1", ty = SubnetState, id = SUBNET_STATE_ID),
            SubnetStateRecord::default(),
        ));
}

///
/// SubnetAuthStateRecord
///

#[derive(Clone, Debug, Default, Deserialize, Eq, PartialEq, Serialize)]
pub struct SubnetAuthStateRecord {}

///
/// SubnetStateRecord
///

#[derive(Clone, Debug, Default, Deserialize, Eq, PartialEq, Serialize)]
pub struct SubnetStateRecord {
    pub auth: SubnetAuthStateRecord,
}

impl_storable_bounded!(SubnetStateRecord, 1024, true);

impl SubnetStateRecord {
    pub const STATE_CONTRACT_NAME: &'static str = "SubnetStateRecord";
}

///
/// SubnetStateData
///
/// Canonical subnet-state import/export snapshot.
///

#[derive(Clone, Debug, Default, Eq, PartialEq)]
pub struct SubnetStateData {
    pub record: SubnetStateRecord,
}

impl SubnetStateData {
    pub const STATE_CONTRACT_NAME: &'static str = "SubnetStateData";
}

///
/// SubnetState
///

pub struct SubnetState;

impl SubnetState {
    pub(crate) fn import(data: SubnetStateData) {
        SUBNET_STATE.with_borrow_mut(|cell| cell.set(data.record));
    }

    #[must_use]
    pub(crate) fn export() -> SubnetStateData {
        SubnetStateData {
            record: SUBNET_STATE.with_borrow(|cell| cell.get().clone()),
        }
    }
}

// -----------------------------------------------------------------------------
// Tests
// -----------------------------------------------------------------------------

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn subnet_state_round_trip() {
        let record = SubnetStateRecord {
            auth: SubnetAuthStateRecord {},
        };

        SubnetState::import(SubnetStateData {
            record: record.clone(),
        });
        assert_eq!(SubnetState::export(), SubnetStateData { record });
    }
}