-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #364 from teonite/copy_auction_state_from_types
Use local duplicate of `AuctionState` struct
- Loading branch information
Showing
5 changed files
with
247 additions
and
3 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,213 @@ | ||
| use std::collections::{btree_map::Entry, BTreeMap}; | ||
|
|
||
| use once_cell::sync::Lazy; | ||
| use schemars::JsonSchema; | ||
| use serde::{Deserialize, Serialize}; | ||
| use serde_map_to_array::KeyValueJsonSchema; | ||
| use serde_map_to_array::{BTreeMapToArray, KeyValueLabels}; | ||
|
|
||
| use casper_types::{ | ||
| system::auction::{ | ||
| Bid, BidKind, DelegatorBid, DelegatorKind, EraValidators, Staking, ValidatorBid, | ||
| }, | ||
| Digest, EraId, PublicKey, U512, | ||
| }; | ||
|
|
||
| use crate::rpcs::docs::DocExample; | ||
|
|
||
| static ERA_VALIDATORS: Lazy<EraValidators> = Lazy::new(|| { | ||
| use casper_types::SecretKey; | ||
|
|
||
| let secret_key_1 = SecretKey::ed25519_from_bytes([42; SecretKey::ED25519_LENGTH]).unwrap(); | ||
| let public_key_1 = PublicKey::from(&secret_key_1); | ||
|
|
||
| let mut validator_weights = BTreeMap::new(); | ||
| validator_weights.insert(public_key_1, U512::from(10)); | ||
|
|
||
| let mut era_validators = BTreeMap::new(); | ||
| era_validators.insert(EraId::from(10u64), validator_weights); | ||
|
|
||
| era_validators | ||
| }); | ||
|
|
||
| static AUCTION_INFO: Lazy<AuctionState> = Lazy::new(|| { | ||
| use casper_types::{system::auction::DelegationRate, AccessRights, SecretKey, URef}; | ||
| use num_traits::Zero; | ||
|
|
||
| let state_root_hash = Digest::from([11; Digest::LENGTH]); | ||
| let validator_secret_key = | ||
| SecretKey::ed25519_from_bytes([42; SecretKey::ED25519_LENGTH]).unwrap(); | ||
| let validator_public_key = PublicKey::from(&validator_secret_key); | ||
|
|
||
| let mut bids = vec![]; | ||
| let validator_bid = ValidatorBid::unlocked( | ||
| validator_public_key.clone(), | ||
| URef::new([250; 32], AccessRights::READ_ADD_WRITE), | ||
| U512::from(20), | ||
| DelegationRate::zero(), | ||
| 0, | ||
| u64::MAX, | ||
| 0, | ||
| ); | ||
| bids.push(BidKind::Validator(Box::new(validator_bid))); | ||
|
|
||
| let delegator_secret_key = | ||
| SecretKey::ed25519_from_bytes([43; SecretKey::ED25519_LENGTH]).unwrap(); | ||
| let delegator_public_key = PublicKey::from(&delegator_secret_key); | ||
| let delegator_bid = DelegatorBid::unlocked( | ||
| delegator_public_key.into(), | ||
| U512::from(10), | ||
| URef::new([251; 32], AccessRights::READ_ADD_WRITE), | ||
| validator_public_key, | ||
| ); | ||
| bids.push(BidKind::Delegator(Box::new(delegator_bid))); | ||
|
|
||
| let height: u64 = 10; | ||
| let era_validators = ERA_VALIDATORS.clone(); | ||
| AuctionState::new(state_root_hash, height, era_validators, bids) | ||
| }); | ||
|
|
||
| /// A validator's weight. | ||
| #[derive(Debug, Serialize, Deserialize, PartialEq, Eq, Clone, JsonSchema)] | ||
| #[serde(deny_unknown_fields)] | ||
| pub struct JsonValidatorWeights { | ||
| public_key: PublicKey, | ||
| weight: U512, | ||
| } | ||
|
|
||
| /// The validators for the given era. | ||
| #[derive(Debug, Serialize, Deserialize, PartialEq, Eq, Clone, JsonSchema)] | ||
| #[serde(deny_unknown_fields)] | ||
| pub struct JsonEraValidators { | ||
| era_id: EraId, | ||
| validator_weights: Vec<JsonValidatorWeights>, | ||
| } | ||
|
|
||
| /// Data structure summarizing auction contract data. | ||
| #[derive(Debug, Serialize, Deserialize, PartialEq, Eq, Clone, JsonSchema)] | ||
| #[serde(deny_unknown_fields)] | ||
| pub struct AuctionState { | ||
| /// Global state hash. | ||
| pub state_root_hash: Digest, | ||
| /// Block height. | ||
| pub block_height: u64, | ||
| /// Era validators. | ||
| pub era_validators: Vec<JsonEraValidators>, | ||
| /// All bids. | ||
| #[serde(with = "BTreeMapToArray::<PublicKey, Bid, BidLabels>")] | ||
| bids: BTreeMap<PublicKey, Bid>, | ||
| } | ||
|
|
||
| impl AuctionState { | ||
| /// Create new instance of `AuctionState` | ||
| pub fn new( | ||
| state_root_hash: Digest, | ||
| block_height: u64, | ||
| era_validators: EraValidators, | ||
| bids: Vec<BidKind>, | ||
| ) -> Self { | ||
| let mut json_era_validators: Vec<JsonEraValidators> = Vec::new(); | ||
| for (era_id, validator_weights) in era_validators.iter() { | ||
| let mut json_validator_weights: Vec<JsonValidatorWeights> = Vec::new(); | ||
| for (public_key, weight) in validator_weights.iter() { | ||
| json_validator_weights.push(JsonValidatorWeights { | ||
| public_key: public_key.clone(), | ||
| weight: *weight, | ||
| }); | ||
| } | ||
| json_era_validators.push(JsonEraValidators { | ||
| era_id: *era_id, | ||
| validator_weights: json_validator_weights, | ||
| }); | ||
| } | ||
|
|
||
| let staking = { | ||
| let mut staking: Staking = BTreeMap::new(); | ||
| for bid_kind in bids.iter().filter(|x| x.is_unified()) { | ||
| if let BidKind::Unified(bid) = bid_kind { | ||
| let public_key = bid.validator_public_key().clone(); | ||
| let validator_bid = ValidatorBid::unlocked( | ||
| bid.validator_public_key().clone(), | ||
| *bid.bonding_purse(), | ||
| *bid.staked_amount(), | ||
| *bid.delegation_rate(), | ||
| 0, | ||
| u64::MAX, | ||
| 0, | ||
| ); | ||
| let mut delegators: BTreeMap<DelegatorKind, DelegatorBid> = BTreeMap::new(); | ||
| for (delegator_public_key, delegator) in bid.delegators() { | ||
| delegators.insert( | ||
| DelegatorKind::PublicKey(delegator_public_key.clone()), | ||
| DelegatorBid::from(delegator.clone()), | ||
| ); | ||
| } | ||
| staking.insert(public_key, (validator_bid, delegators)); | ||
| } | ||
| } | ||
|
|
||
| for bid_kind in bids.iter().filter(|x| x.is_validator()) { | ||
| if let BidKind::Validator(validator_bid) = bid_kind { | ||
| let public_key = validator_bid.validator_public_key().clone(); | ||
| staking.insert(public_key, (*validator_bid.clone(), BTreeMap::new())); | ||
| } | ||
| } | ||
|
|
||
| for bid_kind in bids.iter().filter(|x| x.is_delegator()) { | ||
| if let BidKind::Delegator(delegator_bid) = bid_kind { | ||
| let validator_public_key = delegator_bid.validator_public_key().clone(); | ||
| if let Entry::Occupied(mut occupant) = | ||
| staking.entry(validator_public_key.clone()) | ||
| { | ||
| let (_, delegators) = occupant.get_mut(); | ||
| delegators.insert( | ||
| delegator_bid.delegator_kind().clone(), | ||
| *delegator_bid.clone(), | ||
| ); | ||
| } | ||
| } | ||
| } | ||
| staking | ||
| }; | ||
|
|
||
| let mut bids: BTreeMap<PublicKey, Bid> = BTreeMap::new(); | ||
| for (public_key, (validator_bid, delegators)) in staking { | ||
| let bid = Bid::from_non_unified(validator_bid, delegators); | ||
| bids.insert(public_key, bid); | ||
| } | ||
|
|
||
| AuctionState { | ||
| state_root_hash, | ||
| block_height, | ||
| era_validators: json_era_validators, | ||
| bids, | ||
| } | ||
| } | ||
|
|
||
| // This method is not intended to be used by third party crates. | ||
| #[doc(hidden)] | ||
| pub fn example() -> &'static Self { | ||
| &AUCTION_INFO | ||
| } | ||
| } | ||
|
|
||
| impl DocExample for AuctionState { | ||
| fn doc_example() -> &'static Self { | ||
| AuctionState::example() | ||
| } | ||
| } | ||
|
|
||
| struct BidLabels; | ||
|
|
||
| impl KeyValueLabels for BidLabels { | ||
| const KEY: &'static str = "public_key"; | ||
| const VALUE: &'static str = "bid"; | ||
| } | ||
|
|
||
| impl KeyValueJsonSchema for BidLabels { | ||
| const JSON_SCHEMA_KV_NAME: Option<&'static str> = Some("PublicKeyAndBid"); | ||
| const JSON_SCHEMA_KV_DESCRIPTION: Option<&'static str> = | ||
| Some("A bid associated with the given public key."); | ||
| const JSON_SCHEMA_KEY_DESCRIPTION: Option<&'static str> = Some("The public key of the bidder."); | ||
| const JSON_SCHEMA_VALUE_DESCRIPTION: Option<&'static str> = Some("The bid details."); | ||
| } |