51 lines
1.5 KiB
Rust
51 lines
1.5 KiB
Rust
use super::{DumpActor, DumpActorHandle, DumpInfo, DumpMsg, DumpResult};
|
|
use actix_web::web::Bytes;
|
|
use std::path::Path;
|
|
use tokio::sync::{mpsc, oneshot};
|
|
|
|
#[derive(Clone)]
|
|
pub struct DumpActorHandleImpl {
|
|
sender: mpsc::Sender<DumpMsg>,
|
|
}
|
|
|
|
#[async_trait::async_trait]
|
|
impl DumpActorHandle for DumpActorHandleImpl {
|
|
async fn create_dump(&self) -> DumpResult<DumpInfo> {
|
|
let (ret, receiver) = oneshot::channel();
|
|
let msg = DumpMsg::CreateDump { ret };
|
|
let _ = self.sender.send(msg).await;
|
|
receiver.await.expect("IndexActor has been killed")
|
|
}
|
|
|
|
async fn dump_info(&self, uid: String) -> DumpResult<DumpInfo> {
|
|
let (ret, receiver) = oneshot::channel();
|
|
let msg = DumpMsg::DumpInfo { ret, uid };
|
|
let _ = self.sender.send(msg).await;
|
|
receiver.await.expect("IndexActor has been killed")
|
|
}
|
|
}
|
|
|
|
impl DumpActorHandleImpl {
|
|
pub fn new(
|
|
path: impl AsRef<Path>,
|
|
uuid_resolver: crate::index_controller::uuid_resolver::UuidResolverHandleImpl,
|
|
update: crate::index_controller::update_actor::UpdateActorHandleImpl<Bytes>,
|
|
index_db_size: u64,
|
|
update_db_size: u64,
|
|
) -> anyhow::Result<Self> {
|
|
let (sender, receiver) = mpsc::channel(10);
|
|
let actor = DumpActor::new(
|
|
receiver,
|
|
uuid_resolver,
|
|
update,
|
|
path,
|
|
index_db_size,
|
|
update_db_size,
|
|
);
|
|
|
|
tokio::task::spawn(actor.run());
|
|
|
|
Ok(Self { sender })
|
|
}
|
|
}
|