57 lines
1.5 KiB
Rust
57 lines
1.5 KiB
Rust
use async_trait::async_trait;
|
|
|
|
pub use handlers::empty_handler::EmptyBatchHandler;
|
|
pub use handlers::snapshot_handler::SnapshotHandler;
|
|
pub use scheduler::Scheduler;
|
|
pub use task_store::TaskFilter;
|
|
|
|
#[cfg(test)]
|
|
pub use task_store::test::MockTaskStore as TaskStore;
|
|
#[cfg(not(test))]
|
|
pub use task_store::TaskStore;
|
|
|
|
use batch::Batch;
|
|
use error::Result;
|
|
|
|
pub mod batch;
|
|
pub mod error;
|
|
mod handlers;
|
|
mod scheduler;
|
|
pub mod task;
|
|
mod task_store;
|
|
pub mod update_loop;
|
|
|
|
#[cfg_attr(test, mockall::automock(type Error=test::DebugError;))]
|
|
#[async_trait]
|
|
pub trait BatchHandler: Sync + Send + 'static {
|
|
/// return whether this handler can accept this batch
|
|
fn accept(&self, batch: &Batch) -> bool;
|
|
|
|
/// Processes the `Task` batch returning the batch with the `Task` updated.
|
|
///
|
|
/// It is ok for this function to panic if a batch is handed that hasn't been verified by
|
|
/// `accept` beforehand.
|
|
async fn process_batch(&self, batch: Batch) -> Batch;
|
|
|
|
/// `finish` is called when the result of `process` has been committed to the task store. This
|
|
/// method can be used to perform cleanup after the update has been completed for example.
|
|
async fn finish(&self, batch: &Batch);
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use serde::{Deserialize, Serialize};
|
|
use std::fmt::Display;
|
|
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
pub struct DebugError;
|
|
|
|
impl Display for DebugError {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
f.write_str("an error")
|
|
}
|
|
}
|
|
|
|
impl std::error::Error for DebugError {}
|
|
}
|