diff --git a/gradually-update/src/lib.rs b/gradually-update/src/lib.rs
index 497e25e85f3d35981a01bbf9ce79feea88ff0b5b..6290afde7ddcbd9d517bfb46885dfa10a2bb65de 100644
--- a/gradually-update/src/lib.rs
+++ b/gradually-update/src/lib.rs
@@ -3,7 +3,10 @@
#![allow(clippy::string_lit_as_bytes)]
use codec::{Decode, Encode};
-use frame_support::{decl_error, decl_event, decl_module, decl_storage, ensure, storage, traits::Get};
+use frame_support::{
+ decl_error, decl_event, decl_module, decl_storage, ensure, storage,
+ traits::{EnsureOrigin, Get},
+};
use frame_system::{self as system, ensure_root};
use sp_runtime::{traits::SaturatedConversion, RuntimeDebug};
@@ -25,6 +28,7 @@ pub struct GraduallyUpdate {
pub trait Trait: frame_system::Trait {
type Event: From<Event<Self>> + Into<<Self as frame_system::Trait>::Event>;
type UpdateFrequency: Get<Self::BlockNumber>;
+ type DispatchOrigin: EnsureOrigin<Self::Origin>;
}
decl_storage! {
@@ -66,10 +70,10 @@ decl_module! {
const UpdateFrequency: T::BlockNumber = T::UpdateFrequency::get();
- /// Add gradually_update to adjust numeric parameter. This is a root call.
+ /// Add gradually_update to adjust numeric parameter.
#[weight = 0]
pub fn gradually_update(origin, update: GraduallyUpdate) {
- ensure_root(origin)?;
+ T::DispatchOrigin::try_origin(origin).map(|_| ()).or_else(ensure_root)?;
// Support max value is u128, ensure per_block and target_value <= 16 bytes.
ensure!(update.per_block.len() == update.target_value.len() && update.per_block.len() <= 16, Error::<T>::InvalidPerBlockOrTargetValue);
@@ -88,10 +92,10 @@ decl_module! {
Self::deposit_event(RawEvent::GraduallyUpdate(update.key, update.per_block, update.target_value));
}
- /// Cancel gradually_update to adjust numeric parameter. This is a root call.
+ /// Cancel gradually_update to adjust numeric parameter.
#[weight = 0]
pub fn cancel_gradually_update(origin, key: StorageKey) {
- ensure_root(origin)?;
+ T::DispatchOrigin::try_origin(origin).map(|_| ()).or_else(ensure_root)?;
let gradually_updates: Vec<GraduallyUpdate> = GraduallyUpdates::get()
.into_iter()
@@ -117,41 +121,47 @@ impl<T: Trait> Module<T> {
return;
}
- let mut gradually_updates = GraduallyUpdates::get();
+ let gradually_updates = GraduallyUpdates::get();
+ let initial_count = gradually_updates.len();
- #[allow(clippy::redundant_clone)] // FIXME: This looks like a bug in clippy
- for (i, update) in gradually_updates.clone().iter().enumerate() {
- let current_value = storage::unhashed::get::<StorageValue>(&update.key).unwrap_or_default();
- let current_value_u128 = u128::from_le_bytes(Self::convert_vec_to_u8(¤t_value));
+ let gradually_updates = gradually_updates
+ .into_iter()
+ .filter(|update| {
+ let mut keep = true;
+ let current_value = storage::unhashed::get::<StorageValue>(&update.key).unwrap_or_default();
+ let current_value_u128 = u128::from_le_bytes(Self::convert_vec_to_u8(¤t_value));
- let frequency_u128: u128 = T::UpdateFrequency::get().saturated_into();
+ let frequency_u128: u128 = T::UpdateFrequency::get().saturated_into();
- let step = u128::from_le_bytes(Self::convert_vec_to_u8(&update.per_block));
- let step_u128 = step.checked_mul(frequency_u128).unwrap();
+ let step = u128::from_le_bytes(Self::convert_vec_to_u8(&update.per_block));
+ let step_u128 = step.checked_mul(frequency_u128).unwrap();
- let target_u128 = u128::from_le_bytes(Self::convert_vec_to_u8(&update.target_value));
+ let target_u128 = u128::from_le_bytes(Self::convert_vec_to_u8(&update.target_value));
- let new_value_u128 = if current_value_u128 > target_u128 {
- (current_value_u128.checked_sub(step_u128).unwrap()).max(target_u128)
- } else {
- (current_value_u128.checked_add(step_u128).unwrap()).min(target_u128)
- };
+ let new_value_u128 = if current_value_u128 > target_u128 {
+ (current_value_u128.checked_sub(step_u128).unwrap()).max(target_u128)
+ } else {
+ (current_value_u128.checked_add(step_u128).unwrap()).min(target_u128)
+ };
- // current_value equal target_value, remove gradually_update
- if new_value_u128 == target_u128 {
- gradually_updates.remove(i);
- }
+ // current_value equal target_value, remove gradually_update
+ if new_value_u128 == target_u128 {
+ keep = false;
+ }
- let mut value = new_value_u128.encode();
- value.truncate(update.target_value.len());
+ let mut value = new_value_u128.encode();
+ value.truncate(update.target_value.len());
- storage::unhashed::put(&update.key, &value);
+ storage::unhashed::put(&update.key, &value);
- Self::deposit_event(RawEvent::GraduallyUpdateBlockNumber(now, update.key.clone(), value));
- }
+ Self::deposit_event(RawEvent::GraduallyUpdateBlockNumber(now, update.key.clone(), value));
+
+ keep
+ })
+ .collect::<Vec<_>>();
// gradually_update has finished. Remove it from GraduallyUpdates.
- if gradually_updates.len() < GraduallyUpdates::decode_len().unwrap_or_default() {
+ if gradually_updates.len() < initial_count {
GraduallyUpdates::put(gradually_updates);
}
diff --git a/gradually-update/src/mock.rs b/gradually-update/src/mock.rs
index 40ab82f5a756412cecebb4de742934efba8c40ef..2c7abd5e458ba7e8f7c7d717e509e6cb1044fcdc 100644
--- a/gradually-update/src/mock.rs
+++ b/gradually-update/src/mock.rs
@@ -71,6 +71,7 @@ parameter_types! {
impl Trait for Runtime {
type Event = TestEvent;
type UpdateFrequency = UpdateFrequency;
+ type DispatchOrigin = system::EnsureRoot<AccountId>;
}
pub type GraduallyUpdateModule = Module<Runtime>;
diff --git a/gradually-update/src/tests.rs b/gradually-update/src/tests.rs
index 0e518be1902f0cb49957151f6c2df37b189c70c4..9d8b0e1267ee49d3e41107b34d579f0ac75ff837 100644
--- a/gradually-update/src/tests.rs
+++ b/gradually-update/src/tests.rs
@@ -330,3 +330,54 @@ fn fixedu128_should_work() {
);
});
}
+
+#[test]
+fn finish_multiple_on_finalize_should_work() {
+ ExtBuilder::default().build().execute_with(|| {
+ System::set_block_number(1);
+
+ let update = GraduallyUpdate {
+ key: vec![10],
+ target_value: vec![30],
+ per_block: vec![1],
+ };
+ let update2 = GraduallyUpdate {
+ key: vec![20],
+ target_value: vec![60],
+ per_block: vec![2],
+ };
+ let update3 = GraduallyUpdate {
+ key: vec![30],
+ target_value: vec![100],
+ per_block: vec![3],
+ };
+ assert_ok!(GraduallyUpdateModule::gradually_update(Origin::ROOT, update.clone()));
+ assert_ok!(GraduallyUpdateModule::gradually_update(Origin::ROOT, update2.clone()));
+ assert_ok!(GraduallyUpdateModule::gradually_update(Origin::ROOT, update3.clone()));
+
+ GraduallyUpdateModule::on_finalize(10);
+ assert_eq!(storage_get(&update.key), vec![10]);
+ assert_eq!(storage_get(&update2.key), vec![20]);
+ assert_eq!(storage_get(&update3.key), vec![30]);
+
+ GraduallyUpdateModule::on_finalize(15);
+ assert_eq!(storage_get(&update.key), vec![10]);
+ assert_eq!(storage_get(&update2.key), vec![20]);
+ assert_eq!(storage_get(&update3.key), vec![30]);
+
+ GraduallyUpdateModule::on_finalize(20);
+ assert_eq!(storage_get(&update.key), vec![20]);
+ assert_eq!(storage_get(&update2.key), vec![40]);
+ assert_eq!(storage_get(&update3.key), vec![60]);
+
+ GraduallyUpdateModule::on_finalize(40);
+ assert_eq!(storage_get(&update.key), vec![30]);
+ assert_eq!(storage_get(&update2.key), vec![60]);
+ assert_eq!(storage_get(&update3.key), vec![90]);
+
+ GraduallyUpdateModule::on_finalize(50);
+ assert_eq!(storage_get(&update.key), vec![30]);
+ assert_eq!(storage_get(&update2.key), vec![60]);
+ assert_eq!(storage_get(&update3.key), vec![100]);
+ });
+}