pezkuwichain/pezkuwi-subxt
1
0
Fork 0
mirror of https://github.com/pezkuwichain/pezkuwi-subxt.git synced 2026-05-31 15:41:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use a struct!

Browse Source
This commit is contained in:
Gav
2018-03-07 19:36:13 +01:00
parent 3ad3288508
commit 8c0c504997
1 changed files with 35 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
substrate/demo/runtime/src/runtime/staking.rs
+34 -21
View File
@@ -141,7 +141,20 @@ pub fn total_stake() -> Balance {
pub mod public { pub mod public {
use super::*; use super::*;
type State = BTreeMap<AccountId, (Option<Balance>, Option<Vec<u8>>, BTreeMap<Vec<u8>, Option<Vec<u8>>>)>; #[derive(Default)]
struct ChangeEntry {
balance: Option<Balance>,
code: Option<Vec<u8>>,
storage: BTreeMap<Vec<u8>, Option<Vec<u8>>>,
}
impl ChangeEntry {
pub fn balance_changed(b: Balance) -> Self {
ChangeEntry { balance: Some(b), code: None, storage: Default::default() }
}
}
type State = BTreeMap<AccountId, ChangeEntry>;
trait Externalities { trait Externalities {
fn get_storage(&self, account: &AccountId, location: &[u8]) -> Option<Vec<u8>>; fn get_storage(&self, account: &AccountId, location: &[u8]) -> Option<Vec<u8>>;
@@ -191,15 +204,15 @@ pub mod public {
} }
fn commit_state(s: State) { fn commit_state(s: State) {
for (address, (maybe_balance, maybe_code, storage)) in s.into_iter() { for (address, changed) in s.into_iter() {
if let Some(balance) = maybe_balance { if let Some(balance) = changed.balance {
storage::put(&address.to_keyed_vec(BALANCE_OF), &balance); storage::put(&address.to_keyed_vec(BALANCE_OF), &balance);
} }
if let Some(code) = maybe_code { if let Some(code) = changed.code {
storage::put(&address.to_keyed_vec(CODE_OF), &code); storage::put(&address.to_keyed_vec(CODE_OF), &code);
} }
let storage_key = address.to_keyed_vec(STORAGE_OF); let storage_key = address.to_keyed_vec(STORAGE_OF);
for (k, v) in storage.into_iter() { for (k, v) in changed.storage.into_iter() {
let mut key = storage_key.clone(); let mut key = storage_key.clone();
key.extend(k); key.extend(k);
if let Some(value) = v { if let Some(value) = v {
@@ -212,20 +225,20 @@ pub mod public {
} }
fn merge_state(commit_state: State, local: &mut State) { fn merge_state(commit_state: State, local: &mut State) {
for (address, (maybe_balance, maybe_code, storage)) in commit_state.into_iter() { for (address, changed) in commit_state.into_iter() {
match local.entry(address) { match local.entry(address) {
Entry::Occupied(e) => { Entry::Occupied(e) => {
let mut value = e.into_mut(); let mut value = e.into_mut();
if maybe_balance.is_some() { if changed.balance.is_some() {
value.0 = maybe_balance; value.balance = changed.balance;
} }
if maybe_code.is_some() { if changed.code.is_some() {
value.1 = maybe_code; value.code = changed.code;
} }
value.2.extend(storage.into_iter()); value.storage.extend(changed.storage.into_iter());
} }
Entry::Vacant(e) => { Entry::Vacant(e) => {
e.insert((maybe_balance, maybe_code, storage)); e.insert(changed);
} }
} }
} }
@@ -262,8 +275,8 @@ pub mod public {
// two inserts are safe // two inserts are safe
assert!(&dest != transactor); assert!(&dest != transactor);
local.insert(dest, (Some(value), Some(code.to_vec()), Default::default())); local.insert(dest, ChangeEntry { balance: Some(value), code: Some(code.to_vec()), storage: Default::default() });
local.insert(transactor.clone(), (Some(from_balance - value), None, Default::default())); local.insert(transactor.clone(), ChangeEntry::balance_changed(from_balance - value));
Some(local) Some(local)
} }
@@ -298,8 +311,8 @@ pub mod public {
if transactor != dest { if transactor != dest {
let mut local = local.borrow_mut(); let mut local = local.borrow_mut();
local.insert(transactor.clone(), (Some(from_balance - value), None, Default::default())); local.insert(transactor.clone(), ChangeEntry::balance_changed(from_balance - value));
local.insert(dest.clone(), (Some(to_balance + value), None, Default::default())); local.insert(dest.clone(), ChangeEntry::balance_changed(to_balance + value));
} }
let should_commit = { let should_commit = {
@@ -307,16 +320,16 @@ pub mod public {
let ext = || Ext { let ext = || Ext {
do_get_storage: |account: &AccountId, location: &[u8]| do_get_storage: |account: &AccountId, location: &[u8]|
local.borrow().get(account) local.borrow().get(account)
.and_then(|a| a.2.get(location)) .and_then(|a| a.storage.get(location))
.cloned() .cloned()
.unwrap_or_else(|| ext.get_storage(account, location)), .unwrap_or_else(|| ext.get_storage(account, location)),
do_get_code: |account: &AccountId| do_get_code: |account: &AccountId|
local.borrow().get(account) local.borrow().get(account)
.and_then(|a| a.1.clone()) .and_then(|a| a.code.clone())
.unwrap_or_else(|| ext.get_code(account)), .unwrap_or_else(|| ext.get_code(account)),
do_get_balance: |account: &AccountId| do_get_balance: |account: &AccountId|
local.borrow().get(account) local.borrow().get(account)
.and_then(|a| a.0) .and_then(|a| a.balance)
.unwrap_or_else(|| ext.get_balance(account)), .unwrap_or_else(|| ext.get_balance(account)),
}; };
let mut transfer = |inner_dest: &AccountId, value: Balance| { let mut transfer = |inner_dest: &AccountId, value: Balance| {
@@ -332,8 +345,8 @@ pub mod public {
let mut put_storage = |location: Vec<u8>, value: Option<Vec<u8>>| { let mut put_storage = |location: Vec<u8>, value: Option<Vec<u8>>| {
local.borrow_mut() local.borrow_mut()
.entry(dest.clone()) .entry(dest.clone())
.or_insert((None, None, Default::default())) .or_insert(Default::default())
.2.insert(location, value); .storage.insert(location, value);
}; };
// TODO: logging (logs are just appended into a notable storage-based vector and cleared every // TODO: logging (logs are just appended into a notable storage-based vector and cleared every