ic-canister-kit 1.1.0

//! 升级代码需要持久化

// ================== 功能 ==================

/// 版本升级
pub trait StableHeap {
    /// 升级前，将内存中需要持久化的数据转换为字节数组
    fn heap_to_bytes(&self) -> Vec<u8>;

    /// 升级后，将字节数组还原为内存中需要持久化的数据
    fn heap_from_bytes(&mut self, bytes: &[u8]);
}

// ================== 工具方法 ==================

#[inline]
fn type_key<T: ?Sized + 'static>() -> String {
    use std::any::TypeId;
    format!("{:?}", TypeId::of::<T>())
}

/// 序列化
pub fn to_bytes<T: 'static + serde::Serialize>(value: &T) -> Result<Vec<u8>, String> {
    let mut bytes = vec![];
    ciborium::ser::into_writer(value, &mut bytes).map_err(|e| {
        format!(
            "{}: {}",
            match e {
                ciborium::ser::Error::Io(_) => "write bytes failed.",
                ciborium::ser::Error::Value(_) => "serialize failed.",
            },
            type_key::<T>()
        )
    })?;
    Ok(bytes)
}

/// 序列化
pub fn from_bytes<T: 'static + serde::de::DeserializeOwned>(bytes: &[u8]) -> Result<T, String> {
    ciborium::de::from_reader(bytes).map_err(|_| format!("deserialize failed: {}", type_key::<T>()))
}