109 lines
3.1 KiB
Rust
109 lines
3.1 KiB
Rust
//! [`tower::Service`] for zebra-scan.
|
|
|
|
use std::{future::Future, pin::Pin, task::Poll};
|
|
|
|
use futures::future::FutureExt;
|
|
use tower::Service;
|
|
|
|
use zebra_chain::parameters::Network;
|
|
use zebra_state::ChainTipChange;
|
|
|
|
use crate::{init::ScanTask, scan, storage::Storage, Config, Request, Response};
|
|
|
|
/// Zebra-scan [`tower::Service`]
|
|
#[derive(Debug)]
|
|
pub struct ScanService {
|
|
/// On-disk storage
|
|
db: Storage,
|
|
|
|
/// Handle to scan task that's responsible for writing results
|
|
scan_task: ScanTask,
|
|
}
|
|
|
|
impl ScanService {
|
|
/// Create a new [`ScanService`].
|
|
pub fn new(
|
|
config: &Config,
|
|
network: Network,
|
|
state: scan::State,
|
|
chain_tip_change: ChainTipChange,
|
|
) -> Self {
|
|
Self {
|
|
db: Storage::new(config, network, false),
|
|
scan_task: ScanTask::spawn(config, network, state, chain_tip_change),
|
|
}
|
|
}
|
|
|
|
/// Create a new [`ScanService`] with a mock `ScanTask`
|
|
pub fn new_with_mock_scanner(config: &Config, network: Network) -> Self {
|
|
Self {
|
|
db: Storage::new(config, network, false),
|
|
scan_task: ScanTask::mock(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Service<Request> for ScanService {
|
|
type Response = Response;
|
|
type Error = Box<dyn std::error::Error + Send + Sync + 'static>;
|
|
type Future =
|
|
Pin<Box<dyn Future<Output = Result<Self::Response, Self::Error>> + Send + 'static>>;
|
|
|
|
fn poll_ready(&mut self, _cx: &mut std::task::Context<'_>) -> Poll<Result<(), Self::Error>> {
|
|
// TODO: If scan task returns an error, add error to the panic message
|
|
assert!(
|
|
!self.scan_task.handle.is_finished(),
|
|
"scan task finished unexpectedly"
|
|
);
|
|
|
|
self.db.check_for_panics();
|
|
|
|
Poll::Ready(Ok(()))
|
|
}
|
|
|
|
fn call(&mut self, req: Request) -> Self::Future {
|
|
match req {
|
|
Request::Info => {
|
|
let db = self.db.clone();
|
|
|
|
return async move {
|
|
Ok(Response::Info {
|
|
min_sapling_birthday_height: db.min_sapling_birthday_height(),
|
|
})
|
|
}
|
|
.boxed();
|
|
}
|
|
|
|
Request::CheckKeyHashes(_key_hashes) => {
|
|
// TODO: check that these entries exist in db
|
|
}
|
|
|
|
Request::RegisterKeys(_viewing_key_with_hashes) => {
|
|
// TODO:
|
|
// - add these keys as entries in db
|
|
// - send new keys to scan task
|
|
}
|
|
|
|
Request::DeleteKeys(_key_hashes) => {
|
|
// TODO:
|
|
// - delete these keys and their results from db
|
|
// - send deleted keys to scan task
|
|
}
|
|
|
|
Request::Results(_key_hashes) => {
|
|
// TODO: read results from db
|
|
}
|
|
|
|
Request::SubscribeResults(_key_hashes) => {
|
|
// TODO: send key_hashes and mpsc::Sender to scanner task, return mpsc::Receiver to caller
|
|
}
|
|
|
|
Request::ClearResults(_key_hashes) => {
|
|
// TODO: clear results for these keys from db
|
|
}
|
|
}
|
|
|
|
async move { Ok(Response::Results(vec![])) }.boxed()
|
|
}
|
|
}
|