114 lines
3.9 KiB
Rust
114 lines
3.9 KiB
Rust
use jsonrpc_core::Error;
|
|
use v1::helpers::errors::execution;
|
|
use v1::traits::Miner;
|
|
use v1::types::{BlockTemplate, BlockTemplateRequest};
|
|
use zebra_keys::Address;
|
|
use zebra_miner;
|
|
use zebra_sync;
|
|
|
|
pub struct MinerClient<T: MinerClientCoreApi> {
|
|
core: T,
|
|
}
|
|
|
|
pub trait MinerClientCoreApi: Send + Sync + 'static {
|
|
fn get_block_template(&self) -> Result<zebra_miner::BlockTemplate, String>;
|
|
}
|
|
|
|
pub struct MinerClientCore {
|
|
local_sync_node: zebra_sync::LocalNodeRef,
|
|
miner_address: Option<Address>,
|
|
}
|
|
|
|
impl MinerClientCore {
|
|
pub fn new(local_sync_node: zebra_sync::LocalNodeRef, miner_address: Option<Address>) -> Self {
|
|
MinerClientCore {
|
|
local_sync_node: local_sync_node,
|
|
miner_address: miner_address,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl MinerClientCoreApi for MinerClientCore {
|
|
fn get_block_template(&self) -> Result<zebra_miner::BlockTemplate, String> {
|
|
self.miner_address
|
|
.as_ref()
|
|
.ok_or_else(|| "miner address not set".into())
|
|
.and_then(|miner_address| self.local_sync_node.get_block_template(miner_address))
|
|
}
|
|
}
|
|
|
|
impl<T> MinerClient<T>
|
|
where
|
|
T: MinerClientCoreApi,
|
|
{
|
|
pub fn new(core: T) -> Self {
|
|
MinerClient { core: core }
|
|
}
|
|
}
|
|
|
|
impl<T> Miner for MinerClient<T>
|
|
where
|
|
T: MinerClientCoreApi,
|
|
{
|
|
fn get_block_template(&self, _request: BlockTemplateRequest) -> Result<BlockTemplate, Error> {
|
|
self.core
|
|
.get_block_template()
|
|
.map(Into::into)
|
|
.map_err(|err| execution(&err))
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
pub mod tests {
|
|
use super::*;
|
|
use jsonrpc_core::IoHandler;
|
|
use v1::traits::Miner;
|
|
use zebra_chain;
|
|
use zebra_miner;
|
|
use zebra_primitives::hash::H256;
|
|
|
|
#[derive(Default)]
|
|
struct SuccessMinerClientCore;
|
|
|
|
impl MinerClientCoreApi for SuccessMinerClientCore {
|
|
fn get_block_template(&self) -> Result<zebra_miner::BlockTemplate, String> {
|
|
let tx: zebra_chain::Transaction = "00000000013ba3edfd7a7b12b27ac72c3e67768f617fc81bc3888a51323a9fb8aa4b1e5e4a0000000000000000000101000000000000000000000000".into();
|
|
Ok(zebra_miner::BlockTemplate {
|
|
version: 777,
|
|
previous_header_hash: H256::from(1),
|
|
final_sapling_root_hash: H256::from(2),
|
|
time: 33,
|
|
bits: 44.into(),
|
|
height: 55,
|
|
transactions: vec![tx.into()],
|
|
coinbase_tx: Default::default(),
|
|
size_limit: 77,
|
|
sigop_limit: 88,
|
|
})
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn getblocktemplate_accepted() {
|
|
let client = MinerClient::new(SuccessMinerClientCore::default());
|
|
let mut handler = IoHandler::new();
|
|
handler.extend_with(client.to_delegate());
|
|
|
|
let sample = handler
|
|
.handle_request_sync(
|
|
&(r#"
|
|
{
|
|
"jsonrpc": "2.0",
|
|
"method": "getblocktemplate",
|
|
"params": [{}],
|
|
"id": 1
|
|
}"#),
|
|
)
|
|
.unwrap();
|
|
|
|
// direct hash is 0100000000000000000000000000000000000000000000000000000000000000
|
|
// but client expects reverse hash
|
|
assert_eq!(&sample, r#"{"jsonrpc":"2.0","result":{"bits":44,"coinbasetxn":{"data":"00000000000000000000","depends":null,"fee":null,"hash":null,"required":false,"sigops":null},"curtime":33,"finalsaplingroothash":"0000000000000000000000000000000000000000000000000000000000000000","height":55,"mintime":null,"mutable":null,"noncerange":null,"previousblockhash":"0000000000000000000000000000000000000000000000000000000000000001","sigoplimit":88,"sizelimit":77,"target":"0000000000000000000000000000000000000000000000000000000000000000","transactions":[{"data":"00000000013ba3edfd7a7b12b27ac72c3e67768f617fc81bc3888a51323a9fb8aa4b1e5e4a0000000000000000000101000000000000000000000000","depends":null,"fee":null,"hash":null,"required":false,"sigops":null}],"version":777},"id":1}"#);
|
|
}
|
|
}
|