Tratifying Miner

This commit is contained in:
Tomasz Drwięga 2016-03-09 13:28:37 +01:00
parent 5db84c3233
commit 9d664336b5
6 changed files with 65 additions and 41 deletions

View File

@ -31,7 +31,8 @@ mod transaction_queue;
use std::ops::*; use std::ops::*;
use std::sync::*; use std::sync::*;
pub use miner::Miner; pub use miner::{Miner, MinerService};
pub struct EthMiner { pub struct EthMiner {
miner: Miner, miner: Miner,
@ -45,6 +46,7 @@ impl EthMiner {
}) })
} }
} }
impl Deref for EthMiner { impl Deref for EthMiner {
type Target = Miner; type Target = Miner;

View File

@ -17,16 +17,45 @@
use util::*; use util::*;
use std::sync::atomic::AtomicBool; use std::sync::atomic::AtomicBool;
use rayon::prelude::*; use rayon::prelude::*;
use ethcore::views::{HeaderView, BlockView}; use ethcore::views::{BlockView};
use ethcore::header::{BlockNumber, Header as BlockHeader}; use ethcore::client::{BlockChainClient, BlockId};
use ethcore::client::{BlockChainClient, BlockStatus, BlockId, BlockChainInfo};
use ethcore::block::*; use ethcore::block::*;
use ethcore::error::*; use ethcore::error::*;
use ethcore::transaction::SignedTransaction; use ethcore::transaction::SignedTransaction;
use transaction_queue::{TransactionQueue, TransactionQueueStatus}; use transaction_queue::{TransactionQueue};
pub trait MinerService {
fn status(&self) -> MinerStatus;
fn import_transactions<T>(&self, transactions: Vec<SignedTransaction>, fetch_nonce: T)
where T: Fn(&Address) -> U256;
/// called when blocks are imported to chain, updates transactions queue
fn chain_new_blocks(&self, chain: &BlockChainClient, good: &[H256], bad: &[H256], _retracted: &[H256]);
/// Set the author that we will seal blocks as.
fn set_author(&self, author: Address);
/// Set the extra_data that we will seal blocks with.
fn set_extra_data(&self, extra_data: Bytes);
/// New chain head event. Restart mining operation.
fn prepare_sealing(&self, chain: &BlockChainClient);
/// Grab the `ClosedBlock` that we want to be sealed. Comes as a mutex that you have to lock.
fn sealing_block(&self, chain: &BlockChainClient) -> &Mutex<Option<ClosedBlock>>;
/// Submit `seal` as a valid solution for the header of `pow_hash`.
/// Will check the seal, but not actually insert the block into the chain.
fn submit_seal(&self, chain: &BlockChainClient, pow_hash: H256, seal: Vec<Bytes>) -> Result<(), Error>;
}
pub struct MinerStatus {
pub transaction_queue_pending: usize,
pub transaction_queue_future: usize,
}
pub struct Miner { pub struct Miner {
/// Transactions Queue
transaction_queue: Mutex<TransactionQueue>, transaction_queue: Mutex<TransactionQueue>,
// for sealing... // for sealing...
@ -36,12 +65,8 @@ pub struct Miner {
extra_data: RwLock<Bytes>, extra_data: RwLock<Bytes>,
} }
pub struct MinerStatus {
pub transaction_queue_pending: usize,
pub transaction_queue_future: usize,
}
impl Miner { impl Miner {
/// Creates new instance of miner
pub fn new() -> Miner { pub fn new() -> Miner {
Miner { Miner {
transaction_queue: Mutex::new(TransactionQueue::new()), transaction_queue: Mutex::new(TransactionQueue::new()),
@ -52,7 +77,20 @@ impl Miner {
} }
} }
pub fn status(&self) -> MinerStatus { /// Get the author that we will seal blocks as.
fn author(&self) -> Address {
*self.author.read().unwrap()
}
/// Get the extra_data that we will seal blocks wuth.
fn extra_data(&self) -> Bytes {
self.extra_data.read().unwrap().clone()
}
}
impl MinerService for Miner {
fn status(&self) -> MinerStatus {
let status = self.transaction_queue.lock().unwrap().status(); let status = self.transaction_queue.lock().unwrap().status();
MinerStatus { MinerStatus {
transaction_queue_pending: status.pending, transaction_queue_pending: status.pending,
@ -60,34 +98,22 @@ impl Miner {
} }
} }
pub fn import_transactions<T>(&self, transactions: Vec<SignedTransaction>, fetch_nonce: T) fn import_transactions<T>(&self, transactions: Vec<SignedTransaction>, fetch_nonce: T)
where T: Fn(&Address) -> U256 { where T: Fn(&Address) -> U256 {
let mut transaction_queue = self.transaction_queue.lock().unwrap(); let mut transaction_queue = self.transaction_queue.lock().unwrap();
transaction_queue.add_all(transactions, fetch_nonce); transaction_queue.add_all(transactions, fetch_nonce);
} }
/// Get the author that we will seal blocks as. fn set_author(&self, author: Address) {
pub fn author(&self) -> Address {
*self.author.read().unwrap()
}
/// Set the author that we will seal blocks as.
pub fn set_author(&self, author: Address) {
*self.author.write().unwrap() = author; *self.author.write().unwrap() = author;
} }
/// Get the extra_data that we will seal blocks wuth.
pub fn extra_data(&self) -> Bytes {
self.extra_data.read().unwrap().clone()
}
/// Set the extra_data that we will seal blocks with. fn set_extra_data(&self, extra_data: Bytes) {
pub fn set_extra_data(&self, extra_data: Bytes) {
*self.extra_data.write().unwrap() = extra_data; *self.extra_data.write().unwrap() = extra_data;
} }
/// New chain head event. Restart mining operation. fn prepare_sealing(&self, chain: &BlockChainClient) {
pub fn prepare_sealing(&self, chain: &BlockChainClient) {
let no_of_transactions = 128; let no_of_transactions = 128;
let transactions = self.transaction_queue.lock().unwrap().top_transactions(no_of_transactions); let transactions = self.transaction_queue.lock().unwrap().top_transactions(no_of_transactions);
@ -99,8 +125,7 @@ impl Miner {
*self.sealing_block.lock().unwrap() = b; *self.sealing_block.lock().unwrap() = b;
} }
/// Grab the `ClosedBlock` that we want to be sealed. Comes as a mutex that you have to lock. fn sealing_block(&self, chain: &BlockChainClient) -> &Mutex<Option<ClosedBlock>> {
pub fn sealing_block(&self, chain: &BlockChainClient) -> &Mutex<Option<ClosedBlock>> {
if self.sealing_block.lock().unwrap().is_none() { if self.sealing_block.lock().unwrap().is_none() {
self.sealing_enabled.store(true, atomic::Ordering::Relaxed); self.sealing_enabled.store(true, atomic::Ordering::Relaxed);
// TODO: Above should be on a timer that resets after two blocks have arrived without being asked for. // TODO: Above should be on a timer that resets after two blocks have arrived without being asked for.
@ -109,9 +134,7 @@ impl Miner {
&self.sealing_block &self.sealing_block
} }
/// Submit `seal` as a valid solution for the header of `pow_hash`. fn submit_seal(&self, chain: &BlockChainClient, pow_hash: H256, seal: Vec<Bytes>) -> Result<(), Error> {
/// Will check the seal, but not actually insert the block into the chain.
pub fn submit_seal(&self, chain: &BlockChainClient, pow_hash: H256, seal: Vec<Bytes>) -> Result<(), Error> {
let mut maybe_b = self.sealing_block.lock().unwrap(); let mut maybe_b = self.sealing_block.lock().unwrap();
match *maybe_b { match *maybe_b {
Some(ref b) if b.hash() == pow_hash => {} Some(ref b) if b.hash() == pow_hash => {}
@ -132,8 +155,7 @@ impl Miner {
} }
} }
/// called when block is imported to chain, updates transactions queue and propagates the blocks fn chain_new_blocks(&self, chain: &BlockChainClient, good: &[H256], bad: &[H256], _retracted: &[H256]) {
pub fn chain_new_blocks(&self, chain: &BlockChainClient, good: &[H256], bad: &[H256], _retracted: &[H256]) {
fn fetch_transactions(chain: &BlockChainClient, hash: &H256) -> Vec<SignedTransaction> { fn fetch_transactions(chain: &BlockChainClient, hash: &H256) -> Vec<SignedTransaction> {
let block = chain let block = chain
.block(BlockId::Hash(hash.clone())) .block(BlockId::Hash(hash.clone()))

View File

@ -50,7 +50,7 @@ use ethcore::client::*;
use ethcore::service::{ClientService, NetSyncMessage}; use ethcore::service::{ClientService, NetSyncMessage};
use ethcore::ethereum; use ethcore::ethereum;
use ethsync::{EthSync, SyncConfig}; use ethsync::{EthSync, SyncConfig};
use ethminer::{EthMiner}; use ethminer::{EthMiner, MinerService};
use docopt::Docopt; use docopt::Docopt;
use daemonize::Daemonize; use daemonize::Daemonize;
use number_prefix::{binary_prefix, Standalone, Prefixed}; use number_prefix::{binary_prefix, Standalone, Prefixed};

View File

@ -19,7 +19,7 @@ use std::collections::HashMap;
use std::sync::{Arc, Weak, Mutex, RwLock}; use std::sync::{Arc, Weak, Mutex, RwLock};
use std::ops::Deref; use std::ops::Deref;
use ethsync::{EthSync, SyncState}; use ethsync::{EthSync, SyncState};
use ethminer::{EthMiner}; use ethminer::{EthMiner, MinerService};
use jsonrpc_core::*; use jsonrpc_core::*;
use util::numbers::*; use util::numbers::*;
use util::sha3::*; use util::sha3::*;

View File

@ -38,7 +38,7 @@ use range_collection::{RangeCollection, ToUsize, FromUsize};
use ethcore::error::*; use ethcore::error::*;
use ethcore::transaction::SignedTransaction; use ethcore::transaction::SignedTransaction;
use ethcore::block::Block; use ethcore::block::Block;
use ethminer::EthMiner; use ethminer::{EthMiner, MinerService};
use io::SyncIo; use io::SyncIo;
use time; use time;
use super::SyncConfig; use super::SyncConfig;
@ -1285,7 +1285,7 @@ mod tests {
use super::{PeerInfo, PeerAsking}; use super::{PeerInfo, PeerAsking};
use ethcore::header::*; use ethcore::header::*;
use ethcore::client::*; use ethcore::client::*;
use ethminer::EthMiner; use ethminer::{EthMiner, MinerService};
fn get_dummy_block(order: u32, parent_hash: H256) -> Bytes { fn get_dummy_block(order: u32, parent_hash: H256) -> Bytes {
let mut header = Header::new(); let mut header = Header::new();

View File

@ -311,11 +311,11 @@ impl BlockChainClient for TestBlockChainClient {
} }
} }
fn prepare_sealing(&self, author: Address, extra_data: Bytes, transactions: Vec<SignedTransaction>) -> Option<ClosedBlock> { fn prepare_sealing(&self, _author: Address, _extra_data: Bytes, _transactions: Vec<SignedTransaction>) -> Option<ClosedBlock> {
unimplemented!() unimplemented!()
} }
fn try_seal(&self, block: ClosedBlock, seal: Vec<Bytes>) -> Result<SealedBlock, ClosedBlock> { fn try_seal(&self, _block: ClosedBlock, _seal: Vec<Bytes>) -> Result<SealedBlock, ClosedBlock> {
unimplemented!() unimplemented!()
} }
} }