pezkuwichain/pezkuwi-subxt
1
0
Fork 0
mirror of https://github.com/pezkuwichain/pezkuwi-subxt.git synced 2026-05-30 07:01:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

session_ info: small fixes (#2106)

Browse Source 
* session_info: use proper timeout in test

* Revert "fix off-by-one error"

This reverts commit 35cb56305a19134acd8a8f881f3aabf999a09d74.

* session_info: use correct EarliestStoredSession when introduced on a live chain

* use saturating_sub

Co-authored-by: Bastian Köcher <bkchr@users.noreply.github.com>

* session_info: revert the timeout test

* session_info: rust is dumb

Co-authored-by: Bastian Köcher <bkchr@users.noreply.github.com>
This commit is contained in:
Andronik Ordian
2020-12-10 22:34:17 +01:00
committed by GitHub
parent be3e5b3b3b
commit 04e9489da6
1 changed files with 17 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
polkadot/runtime/parachains/src/session_info.rs
+17 -32
View File
@@ -25,7 +25,7 @@ use frame_support::{
weights::Weight, weights::Weight,
}; };
use crate::{configuration, paras, scheduler}; use crate::{configuration, paras, scheduler};
use sp_std::{cmp, vec::Vec}; use sp_std::vec::Vec;
pub trait Config: pub trait Config:
frame_system::Config frame_system::Config
@@ -100,17 +100,19 @@ impl<T: Config> Module<T> {
let new_session_index = notification.session_index; let new_session_index = notification.session_index;
let old_earliest_stored_session = EarliestStoredSession::get(); let old_earliest_stored_session = EarliestStoredSession::get();
let dispute_period = cmp::max(1, dispute_period); let new_earliest_stored_session = new_session_index.saturating_sub(dispute_period);
let new_earliest_stored_session = new_session_index.checked_sub(dispute_period - 1).unwrap_or(0); let new_earliest_stored_session = core::cmp::max(new_earliest_stored_session, old_earliest_stored_session);
let new_earliest_stored_session = cmp::max(new_earliest_stored_session, old_earliest_stored_session);
// update `EarliestStoredSession` based on `config.dispute_period`
EarliestStoredSession::set(new_earliest_stored_session);
// remove all entries from `Sessions` from the previous value up to the new value // remove all entries from `Sessions` from the previous value up to the new value
// avoid a potentially heavy loop when introduced on a live chain // avoid a potentially heavy loop when introduced on a live chain
if old_earliest_stored_session != 0 || Sessions::get(0).is_some() { if old_earliest_stored_session != 0 || Sessions::get(0).is_some() {
for idx in old_earliest_stored_session..new_earliest_stored_session { for idx in old_earliest_stored_session..new_earliest_stored_session {
Sessions::remove(&idx); Sessions::remove(&idx);
} }
// update `EarliestStoredSession` based on `config.dispute_period`
EarliestStoredSession::set(new_earliest_stored_session);
} else {
// just introduced on a live chain
EarliestStoredSession::set(new_session_index);
} }
// create a new entry in `Sessions` with information about the current session // create a new entry in `Sessions` with information about the current session
let new_session_info = SessionInfo { let new_session_info = SessionInfo {
@@ -249,24 +251,29 @@ mod tests {
#[test] #[test]
fn session_pruning_is_based_on_dispute_period() { fn session_pruning_is_based_on_dispute_period() {
new_test_ext(genesis_config()).execute_with(|| { new_test_ext(genesis_config()).execute_with(|| {
let default_info = primitives::v1::SessionInfo::default();
Sessions::insert(9, default_info);
run_to_block(100, session_changes); run_to_block(100, session_changes);
assert_eq!(EarliestStoredSession::get(), 9); // but the first session change is not based on dispute_period
assert_eq!(EarliestStoredSession::get(), 10);
// and we didn't prune the last changes
assert!(Sessions::get(9).is_some());
// changing dispute_period works // changing dispute_period works
let dispute_period = 5; let dispute_period = 5;
Configuration::set_dispute_period(Origin::root(), dispute_period).unwrap(); Configuration::set_dispute_period(Origin::root(), dispute_period).unwrap();
run_to_block(200, session_changes); run_to_block(200, session_changes);
assert_eq!(EarliestStoredSession::get(), 20 - dispute_period + 1); assert_eq!(EarliestStoredSession::get(), 20 - dispute_period);
// we don't have that many sessions stored // we don't have that many sessions stored
let new_dispute_period = 16; let new_dispute_period = 16;
Configuration::set_dispute_period(Origin::root(), new_dispute_period).unwrap(); Configuration::set_dispute_period(Origin::root(), new_dispute_period).unwrap();
run_to_block(300, session_changes); run_to_block(300, session_changes);
assert_eq!(EarliestStoredSession::get(), 20 - dispute_period + 1); assert_eq!(EarliestStoredSession::get(), 20 - dispute_period);
// now we do // now we do
run_to_block(400, session_changes); run_to_block(400, session_changes);
assert_eq!(EarliestStoredSession::get(), 40 - new_dispute_period + 1); assert_eq!(EarliestStoredSession::get(), 40 - new_dispute_period);
}) })
} }
@@ -284,26 +291,4 @@ mod tests {
assert_eq!(session.needed_approvals, 42); assert_eq!(session.needed_approvals, 42);
}) })
} }
#[test]
fn session_pruning_avoids_heavy_loop() {
new_test_ext(genesis_config()).execute_with(|| {
let start = 1_000_000_000;
System::on_initialize(start);
System::set_block_number(start);
if let Some(notification) = new_session_every_block(start) {
Configuration::initializer_on_new_session(&notification.validators, &notification.queued);
SessionInfo::initializer_on_new_session(&notification);
}
Configuration::initializer_initialize(start);
SessionInfo::initializer_initialize(start);
assert_eq!(EarliestStoredSession::get(), start - 1);
run_to_block(start + 1, new_session_every_block);
assert_eq!(EarliestStoredSession::get(), start);
})
}
} }