openethereum/crates/concensus/miner/src/work_notify.rs
2021-01-14 15:30:57 +01:00

106 lines
3.4 KiB
Rust

// Copyright 2015-2020 Parity Technologies (UK) Ltd.
// This file is part of OpenEthereum.
// OpenEthereum is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// OpenEthereum is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// You should have received a copy of the GNU General Public License
// along with OpenEthereum. If not, see <http://www.gnu.org/licenses/>.
//! Sends HTTP notifications to a list of URLs every time new work is available.
extern crate ethash;
extern crate fetch;
extern crate hyper;
extern crate parity_runtime;
extern crate url;
use self::{
ethash::SeedHashCompute,
fetch::{Client as FetchClient, Fetch, Method, Request},
hyper::header::{self, HeaderValue},
parity_runtime::Executor,
url::Url,
};
use ethereum_types::{H256, U256};
use parking_lot::Mutex;
use futures::Future;
/// Trait for notifying about new mining work
pub trait NotifyWork: Send + Sync {
/// Fired when new mining job available
fn notify(&self, pow_hash: H256, difficulty: U256, number: u64);
}
/// POSTs info about new work to given urls.
pub struct WorkPoster {
urls: Vec<Url>,
client: FetchClient,
executor: Executor,
seed_compute: Mutex<SeedHashCompute>,
}
impl WorkPoster {
/// Create new `WorkPoster`.
pub fn new(urls: &[String], fetch: FetchClient, executor: Executor) -> Self {
let urls = urls
.into_iter()
.filter_map(|u| match Url::parse(u) {
Ok(url) => Some(url),
Err(e) => {
warn!("Error parsing URL {} : {}", u, e);
None
}
})
.collect();
WorkPoster {
client: fetch,
executor: executor,
urls: urls,
seed_compute: Mutex::new(SeedHashCompute::default()),
}
}
}
impl NotifyWork for WorkPoster {
fn notify(&self, pow_hash: H256, difficulty: U256, number: u64) {
// TODO: move this to engine
let target = ethash::difficulty_to_boundary(&difficulty);
let seed_hash = &self.seed_compute.lock().hash_block_number(number);
let seed_hash = H256::from_slice(&seed_hash[..]);
let body = format!(
r#"{{ "result": ["0x{:x}","0x{:x}","0x{:x}","0x{:x}"] }}"#,
pow_hash, seed_hash, target, number
);
for u in &self.urls {
let u = u.clone();
self.executor.spawn(
self.client
.fetch(
Request::new(u.clone(), Method::POST)
.with_header(
header::CONTENT_TYPE,
HeaderValue::from_static("application/json"),
)
.with_body(body.clone()),
Default::default(),
)
.map_err(move |e| {
warn!("Error sending HTTP notification to {} : {}, retrying", u, e);
})
.map(|_| ()),
);
}
}
}