2017-08-12 07:57:07 -07:00
|
|
|
use std::sync::Arc;
|
2017-08-12 11:03:48 -07:00
|
|
|
use futures::{Future, Stream, Poll, Async};
|
2017-08-12 07:57:07 -07:00
|
|
|
use futures::future::{JoinAll, join_all};
|
|
|
|
use web3::Transport;
|
|
|
|
use web3::helpers::CallResult;
|
2017-08-13 04:09:28 -07:00
|
|
|
use web3::types::{TransactionRequest, H256, Log, Address};
|
2017-08-12 07:57:07 -07:00
|
|
|
use api::{LogStream, self};
|
|
|
|
use error::{Error, ErrorKind};
|
2017-08-12 11:03:48 -07:00
|
|
|
use database::Database;
|
2017-08-12 07:57:07 -07:00
|
|
|
use app::App;
|
2017-08-10 08:55:46 -07:00
|
|
|
|
2017-08-13 04:09:28 -07:00
|
|
|
/// State of deposits relay.
|
|
|
|
enum DepositRelayState<T: Transport> {
|
|
|
|
/// Deposit relay is waiting for logs.
|
2017-08-12 07:57:07 -07:00
|
|
|
Wait,
|
2017-08-13 04:09:28 -07:00
|
|
|
/// Relaying deposits in progress.
|
2017-08-12 07:57:07 -07:00
|
|
|
RelayDeposits {
|
|
|
|
future: JoinAll<Vec<CallResult<H256, T::Out>>>,
|
|
|
|
block: u64,
|
|
|
|
},
|
2017-08-13 04:09:28 -07:00
|
|
|
/// All deposits from given block has been relayed.
|
2017-08-12 07:57:07 -07:00
|
|
|
Yield(Option<u64>),
|
|
|
|
}
|
|
|
|
|
2017-08-13 04:09:28 -07:00
|
|
|
fn deposit_relay<T: Transport + Clone>(app: Arc<App<T>>, init: &Database) -> DepositRelay<T> {
|
|
|
|
let logs_init = api::LogStreamInit {
|
|
|
|
after: init.checked_deposit_relay,
|
|
|
|
poll_interval: app.config.mainnet.poll_interval,
|
|
|
|
confirmations: app.config.mainnet.required_confirmations,
|
|
|
|
filter: app.mainnet_bridge().deposits_filter(init.mainnet_contract_address.clone()),
|
|
|
|
};
|
|
|
|
DepositRelay {
|
|
|
|
logs: api::log_stream(app.connections.mainnet.clone(), logs_init),
|
|
|
|
testnet_contract: init.testnet_contract_address.clone(),
|
|
|
|
state: DepositRelayState::Wait,
|
|
|
|
app,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-08-12 07:57:07 -07:00
|
|
|
pub struct DepositRelay<T: Transport> {
|
|
|
|
app: Arc<App<T>>,
|
|
|
|
logs: LogStream<T>,
|
|
|
|
state: DepositRelayState<T>,
|
2017-08-13 04:09:28 -07:00
|
|
|
testnet_contract: Address,
|
2017-08-12 07:57:07 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
impl<T: Transport> Stream for DepositRelay<T> {
|
|
|
|
type Item = u64;
|
|
|
|
type Error = Error;
|
|
|
|
|
|
|
|
fn poll(&mut self) -> Poll<Option<Self::Item>, Self::Error> {
|
|
|
|
loop {
|
|
|
|
let next_state = match self.state {
|
2017-08-12 11:03:48 -07:00
|
|
|
DepositRelayState::Wait => {
|
|
|
|
let item = try_stream!(self.logs.poll());
|
|
|
|
let deposits = item.logs
|
|
|
|
.into_iter()
|
|
|
|
.map(|log| self.app.mainnet_bridge().deposit_from_log(log))
|
|
|
|
.collect::<Result<Vec<_>, _>>()?
|
|
|
|
.into_iter()
|
|
|
|
.map(|deposit| self.app.testnet_bridge().deposit_payload(deposit))
|
|
|
|
.map(|payload| TransactionRequest {
|
|
|
|
from: self.app.config.testnet.account.clone(),
|
2017-08-13 04:09:28 -07:00
|
|
|
to: Some(self.testnet_contract.clone()),
|
2017-08-12 11:03:48 -07:00
|
|
|
gas: Some(self.app.config.testnet.txs.deposit.gas.into()),
|
|
|
|
gas_price: Some(self.app.config.testnet.txs.deposit.gas_price.into()),
|
|
|
|
value: Some(self.app.config.testnet.txs.deposit.value.into()),
|
|
|
|
data: Some(payload),
|
|
|
|
nonce: None,
|
|
|
|
condition: None,
|
|
|
|
})
|
|
|
|
.map(|request| api::send_transaction(&self.app.connections.testnet, request))
|
|
|
|
.collect::<Vec<_>>();
|
2017-08-12 07:57:07 -07:00
|
|
|
|
2017-08-12 11:03:48 -07:00
|
|
|
DepositRelayState::RelayDeposits {
|
|
|
|
future: join_all(deposits),
|
|
|
|
block: item.to
|
|
|
|
}
|
2017-08-12 07:57:07 -07:00
|
|
|
},
|
|
|
|
DepositRelayState::RelayDeposits { ref mut future, block } => {
|
|
|
|
let _ = try_ready!(future.poll().map_err(ErrorKind::Web3));
|
|
|
|
DepositRelayState::Yield(Some(block))
|
|
|
|
},
|
|
|
|
DepositRelayState::Yield(ref mut block) => match block.take() {
|
|
|
|
None => DepositRelayState::Wait,
|
|
|
|
some => return Ok(some.into()),
|
|
|
|
}
|
|
|
|
};
|
|
|
|
self.state = next_state;
|
|
|
|
}
|
|
|
|
}
|
2017-08-10 08:55:46 -07:00
|
|
|
}
|