Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Format code using 'cargo fmt' #1

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Format code using 'cargo fmt' #1

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
613 changes: 342 additions & 271 deletions examples/counter/src/main.rs
View file
Open in desktop

Large diffs are not rendered by default.

283 changes: 143 additions & 140 deletions examples/counter/src/runtime.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,178 +1,181 @@
use primitive_types::H256;
use blockchain::traits::{
Block as BlockT, BlockExecutor,
BuilderExecutor, StorageExternalities,
};
use codec::{Encode, Decode};
use blockchain::traits::{Block as BlockT, BlockExecutor, BuilderExecutor, StorageExternalities};
use codec::{Decode, Encode};
use codec_derive::{Decode, Encode};
use primitive_types::H256;
use sha3::{Digest, Sha3_256};

#[derive(Clone, Debug, Encode, Decode)]
pub struct Block {
hash: H256,
parent_hash: Option<H256>,
extrinsics: Vec<Extrinsic>,
hash: H256,
parent_hash: Option<H256>,
extrinsics: Vec<Extrinsic>,
}

impl Block {
pub fn calculate_hash(&self) -> H256 {
let data = (self.parent_hash.clone(), self.extrinsics.clone()).encode();
H256::from_slice(Sha3_256::digest(&data).as_slice())
}

pub fn verify_hash(&self) -> bool {
self.hash == self.calculate_hash()
}

pub fn fix_hash(&mut self) {
self.hash = self.calculate_hash();
}

pub fn genesis() -> Self {
let mut block = Block {
hash: H256::default(),
parent_hash: None,
extrinsics: Vec::new(),
};
block.fix_hash();
block
}
pub fn calculate_hash(&self) -> H256 {
let data = (self.parent_hash.clone(), self.extrinsics.clone()).encode();
H256::from_slice(Sha3_256::digest(&data).as_slice())
}

pub fn verify_hash(&self) -> bool {
self.hash == self.calculate_hash()
}

pub fn fix_hash(&mut self) {
self.hash = self.calculate_hash();
}

pub fn genesis() -> Self {
let mut block = Block {
hash: H256::default(),
parent_hash: None,
extrinsics: Vec::new(),
};
block.fix_hash();
block
}
}

impl BlockT for Block {
type Identifier = H256;
type Identifier = H256;

fn parent_id(&self) -> Option<H256> {
self.parent_hash
}
fn parent_id(&self) -> Option<H256> {
self.parent_hash
}

fn id(&self) -> H256 {
self.hash
}
fn id(&self) -> H256 {
self.hash
}
}

#[derive(Clone, Debug, Encode, Decode)]
pub enum Extrinsic {
Add(u128),
Add(u128),
}

#[derive(Debug)]
pub enum Error {
Backend(Box<std::error::Error>),
HashMismatch,
StateCorruption,
Backend(Box<std::error::Error>),
HashMismatch,
StateCorruption,
}

impl std::fmt::Display for Error {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
match self {
Error::HashMismatch => "Hash mismatch".fmt(f)?,
Error::StateCorruption => "State is corrupted".fmt(f)?,
Error::Backend(_) => "Backend error".fmt(f)?,
}

Ok(())
}
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
match self {
Error::HashMismatch => "Hash mismatch".fmt(f)?,
Error::StateCorruption => "State is corrupted".fmt(f)?,
Error::Backend(_) => "Backend error".fmt(f)?,
}

Ok(())
}
}

impl std::error::Error for Error { }
impl std::error::Error for Error {}

#[derive(Clone)]
pub struct Executor;

impl Executor {
fn read_counter(&self, state: &mut <Self as BlockExecutor>::Externalities) -> Result<u128, Error> {
Ok(
match state.read_storage(b"counter").map_err(|e| Error::Backend(e))? {
Some(counter) => {
u128::decode(&mut counter.as_slice()).ok_or(Error::StateCorruption)?
},
None => 0,
}
)
}

fn write_counter(&self, counter: u128, state: &mut <Self as BlockExecutor>::Externalities) {
state.write_storage(b"counter".to_vec(), counter.encode());
}
fn read_counter(
&self,
state: &mut <Self as BlockExecutor>::Externalities,
) -> Result<u128, Error> {
Ok(
match state
.read_storage(b"counter")
.map_err(|e| Error::Backend(e))?
{
Some(counter) => {
u128::decode(&mut counter.as_slice()).ok_or(Error::StateCorruption)?
}
None => 0,
},
)
}

fn write_counter(&self, counter: u128, state: &mut <Self as BlockExecutor>::Externalities) {
state.write_storage(b"counter".to_vec(), counter.encode());
}
}

impl BlockExecutor for Executor {
type Error = Error;
type Block = Block;
type Externalities = dyn StorageExternalities + 'static;

fn execute_block(
&self,
block: &Self::Block,
state: &mut Self::Externalities,
) -> Result<(), Error> {
if !block.verify_hash() {
return Err(Error::HashMismatch);
}

let mut counter = self.read_counter(state)?;

for extrinsic in &block.extrinsics {
match extrinsic {
Extrinsic::Add(add) => counter += add,
}
}

self.write_counter(counter, state);

Ok(())
}
type Error = Error;
type Block = Block;
type Externalities = dyn StorageExternalities + 'static;

fn execute_block(
&self,
block: &Self::Block,
state: &mut Self::Externalities,
) -> Result<(), Error> {
if !block.verify_hash() {
return Err(Error::HashMismatch);
}

let mut counter = self.read_counter(state)?;

for extrinsic in &block.extrinsics {
match extrinsic {
Extrinsic::Add(add) => counter += add,
}
}

self.write_counter(counter, state);

Ok(())
}
}

impl BuilderExecutor for Executor {
type Error = Error;
type Block = Block;
type BuildBlock = Block;
type Externalities = dyn StorageExternalities + 'static;
type Extrinsic = Extrinsic;
type Inherent = ();

fn initialize_block(
&self,
block: &Self::Block,
_state: &mut Self::Externalities,
_inherent: (),
) -> Result<Self::BuildBlock, Self::Error> {
let mut block = block.clone();
block.parent_hash = Some(block.hash);
block.fix_hash();

Ok(block)
}

fn apply_extrinsic(
&self,
block: &mut Self::Block,
extrinsic: Self::Extrinsic,
state: &mut Self::Externalities,
) -> Result<(), Self::Error> {
let mut counter = self.read_counter(state)?;

match extrinsic {
Extrinsic::Add(add) => {
counter += add;
},
}

self.write_counter(counter, state);
block.fix_hash();

Ok(())
}

fn finalize_block(
&self,
block: &mut Self::Block,
_state: &mut Self::Externalities,
) -> Result<(), Self::Error> {
block.fix_hash();

Ok(())
}
type Error = Error;
type Block = Block;
type BuildBlock = Block;
type Externalities = dyn StorageExternalities + 'static;
type Extrinsic = Extrinsic;
type Inherent = ();

fn initialize_block(
&self,
block: &Self::Block,
_state: &mut Self::Externalities,
_inherent: (),
) -> Result<Self::BuildBlock, Self::Error> {
let mut block = block.clone();
block.parent_hash = Some(block.hash);
block.fix_hash();

Ok(block)
}

fn apply_extrinsic(
&self,
block: &mut Self::Block,
extrinsic: Self::Extrinsic,
state: &mut Self::Externalities,
) -> Result<(), Self::Error> {
let mut counter = self.read_counter(state)?;

match extrinsic {
Extrinsic::Add(add) => {
counter += add;
}
}

self.write_counter(counter, state);
block.fix_hash();

Ok(())
}

fn finalize_block(
&self,
block: &mut Self::Block,
_state: &mut Self::Externalities,
) -> Result<(), Self::Error> {
block.fix_hash();

Ok(())
}
}
Loading