Method upgrade

core/upgrader/impl/src/upgrade.rs:62–83

(&self, ps: ChangeParams)

#[async_trait]
impl Upgrade for Upgrader {
    async fn upgrade(&self, ps: ChangeParams) -> Result<(), UpgradeError> {
        let target_canister = get_target_canister();

        match ps {
            ChangeParams::Upgrade(ps) => install_chunked_code(
                target_canister,
                ps.install_mode,
                ps.module,
                ps.module_extra_chunks,
                ps.arg,
            )
            .await
            .map_err(|e| anyhow!(e).into()),
            ChangeParams::Restore(ps) => load_canister_snapshot(LoadCanisterSnapshotArgs {
                canister_id: target_canister,
                snapshot_id: ps.snapshot_id,
                sender_canister_version: Some(canister_version()),
            })
            .await
            .map_err(|(_, e)| anyhow!(e).into()),
        }
    }
}

pub struct WithSnapshot<T>(pub T);