pezkuwichain
b6d35f6faf
Updated 4763 files with dual copyright: - Parity Technologies (UK) Ltd. - Dijital Kurdistan Tech Institute
114 lines
3.6 KiB
Rust
114 lines
3.6 KiB
Rust
// Copyright (C) Parity Technologies (UK) Ltd. and Dijital Kurdistan Tech Institute
|
|
// This file is part of Pezcumulus.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
//! Migrates the storage from the previously deleted DMP pezpallet.
|
|
|
|
use crate::*;
|
|
use alloc::vec::Vec;
|
|
use pezcumulus_primitives_core::relay_chain::BlockNumber as RelayBlockNumber;
|
|
use pezframe_support::{pezpallet_prelude::*, storage_alias, traits::HandleMessage};
|
|
|
|
pub(crate) const LOG: &str = "runtime::dmp-queue-export-xcms";
|
|
|
|
/// The old `PageIndexData` struct.
|
|
#[derive(Copy, Clone, Eq, PartialEq, Default, Encode, Decode, RuntimeDebug, TypeInfo)]
|
|
pub struct PageIndexData {
|
|
/// The lowest used page index.
|
|
pub begin_used: PageCounter,
|
|
/// The lowest unused page index.
|
|
pub end_used: PageCounter,
|
|
/// The number of overweight messages ever recorded (and thus the lowest free index).
|
|
pub overweight_count: OverweightIndex,
|
|
}
|
|
|
|
/// The old `MigrationState` type.
|
|
pub type OverweightIndex = u64;
|
|
/// The old `MigrationState` type.
|
|
pub type PageCounter = u32;
|
|
|
|
/// The old `PageIndex` storage item.
|
|
#[storage_alias]
|
|
pub type PageIndex<T: Config> = StorageValue<Pezpallet<T>, PageIndexData, ValueQuery>;
|
|
|
|
/// The old `Pages` storage item.
|
|
#[storage_alias]
|
|
pub type Pages<T: Config> = StorageMap<
|
|
Pezpallet<T>,
|
|
Blake2_128Concat,
|
|
PageCounter,
|
|
Vec<(RelayBlockNumber, Vec<u8>)>,
|
|
ValueQuery,
|
|
>;
|
|
|
|
/// The old `Overweight` storage item.
|
|
#[storage_alias]
|
|
pub type Overweight<T: Config> = CountedStorageMap<
|
|
Pezpallet<T>,
|
|
Blake2_128Concat,
|
|
OverweightIndex,
|
|
(RelayBlockNumber, Vec<u8>),
|
|
OptionQuery,
|
|
>;
|
|
|
|
pub(crate) mod testing_only {
|
|
use super::*;
|
|
|
|
/// This alias is not used by the migration but only for testing.
|
|
///
|
|
/// Note that the alias type is wrong on purpose.
|
|
#[storage_alias]
|
|
pub type Configuration<T: Config> = StorageValue<Pezpallet<T>, u32>;
|
|
}
|
|
|
|
/// Migrates a single page to the `DmpSink`.
|
|
pub(crate) fn migrate_page<T: crate::Config>(p: PageCounter) -> Result<(), ()> {
|
|
let page = Pages::<T>::take(p);
|
|
log::debug!(target: LOG, "Migrating page #{p} with {} messages ...", page.len());
|
|
if page.is_empty() {
|
|
log::error!(target: LOG, "Page #{p}: EMPTY - storage corrupted?");
|
|
return Err(());
|
|
}
|
|
|
|
for (m, (block, msg)) in page.iter().enumerate() {
|
|
let Ok(bound) = BoundedVec::<u8, _>::try_from(msg.clone()) else {
|
|
log::error!(target: LOG, "[Page {p}] Message #{m}: TOO LONG - dropping");
|
|
continue;
|
|
};
|
|
|
|
T::DmpSink::handle_message(bound.as_bounded_slice());
|
|
log::debug!(target: LOG, "[Page {p}] Migrated message #{m} from block {block}");
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Migrates a single overweight message to the `DmpSink`.
|
|
pub(crate) fn migrate_overweight<T: crate::Config>(i: OverweightIndex) -> Result<(), ()> {
|
|
let Some((block, msg)) = Overweight::<T>::take(i) else {
|
|
log::error!(target: LOG, "[Overweight {i}] Message: EMPTY - storage corrupted?");
|
|
return Err(());
|
|
};
|
|
let Ok(bound) = BoundedVec::<u8, _>::try_from(msg) else {
|
|
log::error!(target: LOG, "[Overweight {i}] Message: TOO LONG - dropping");
|
|
return Err(());
|
|
};
|
|
|
|
T::DmpSink::handle_message(bound.as_bounded_slice());
|
|
log::debug!(target: LOG, "[Overweight {i}] Migrated message from block {block}");
|
|
|
|
Ok(())
|
|
}
|