2021-05-10 20:25:09 +02:00
|
|
|
use std::collections::BTreeMap;
|
2021-09-30 11:29:27 +02:00
|
|
|
use std::fmt;
|
2021-09-14 18:39:02 +02:00
|
|
|
use std::path::{Path, PathBuf};
|
2021-03-06 12:57:56 +01:00
|
|
|
use std::sync::Arc;
|
|
|
|
use std::time::Duration;
|
2021-03-04 12:03:06 +01:00
|
|
|
|
2021-09-14 18:39:02 +02:00
|
|
|
use actix_web::error::PayloadError;
|
|
|
|
use bytes::Bytes;
|
2021-05-10 20:24:14 +02:00
|
|
|
use chrono::{DateTime, Utc};
|
2021-09-14 18:39:02 +02:00
|
|
|
use futures::Stream;
|
2021-03-24 11:29:11 +01:00
|
|
|
use log::info;
|
2021-09-14 18:39:02 +02:00
|
|
|
use milli::update::IndexDocumentsMethod;
|
2021-03-15 18:11:10 +01:00
|
|
|
use serde::{Deserialize, Serialize};
|
2021-09-24 11:53:11 +02:00
|
|
|
use tokio::task::spawn_blocking;
|
2021-03-06 12:57:56 +01:00
|
|
|
use tokio::time::sleep;
|
2021-03-18 09:09:26 +01:00
|
|
|
use uuid::Uuid;
|
2021-01-16 15:09:48 +01:00
|
|
|
|
2021-05-10 20:25:09 +02:00
|
|
|
use dump_actor::DumpActorHandle;
|
2021-05-26 20:42:09 +02:00
|
|
|
pub use dump_actor::{DumpInfo, DumpStatus};
|
2021-09-14 18:39:02 +02:00
|
|
|
use snapshot::load_snapshot;
|
2021-04-01 16:44:42 +02:00
|
|
|
|
2021-09-28 22:22:59 +02:00
|
|
|
use crate::index::error::Result as IndexResult;
|
|
|
|
use crate::index::{
|
|
|
|
Checked, Document, IndexMeta, IndexStats, SearchQuery, SearchResult, Settings, Unchecked,
|
|
|
|
};
|
2021-09-24 11:53:11 +02:00
|
|
|
use crate::index_controller::index_resolver::create_index_resolver;
|
2021-09-27 16:48:03 +02:00
|
|
|
use crate::index_controller::snapshot::SnapshotService;
|
2021-09-22 15:07:04 +02:00
|
|
|
use crate::options::IndexerOpts;
|
2021-06-23 14:48:33 +02:00
|
|
|
use error::Result;
|
2021-04-01 16:44:42 +02:00
|
|
|
|
2021-05-27 14:30:20 +02:00
|
|
|
use self::dump_actor::load_dump;
|
2021-09-24 11:53:11 +02:00
|
|
|
use self::index_resolver::error::IndexResolverError;
|
2021-10-06 13:01:02 +02:00
|
|
|
use self::index_resolver::index_store::{IndexStore, MapIndexStore};
|
|
|
|
use self::index_resolver::uuid_store::{HeedUuidStore, UuidStore};
|
|
|
|
use self::index_resolver::IndexResolver;
|
2021-09-22 11:52:29 +02:00
|
|
|
use self::updates::status::UpdateStatus;
|
2021-09-22 15:07:04 +02:00
|
|
|
use self::updates::UpdateMsg;
|
2021-05-27 14:30:20 +02:00
|
|
|
|
2021-05-26 20:42:09 +02:00
|
|
|
mod dump_actor;
|
2021-06-15 17:39:07 +02:00
|
|
|
pub mod error;
|
2021-09-28 22:22:59 +02:00
|
|
|
mod index_resolver;
|
2021-04-01 16:44:42 +02:00
|
|
|
mod snapshot;
|
2021-09-22 15:07:04 +02:00
|
|
|
pub mod update_file_store;
|
2021-09-22 11:52:29 +02:00
|
|
|
pub mod updates;
|
2021-09-14 18:39:02 +02:00
|
|
|
|
2021-10-06 13:01:02 +02:00
|
|
|
/// Concrete implementation of the IndexController, exposed by meilisearch-lib
|
|
|
|
pub type MeiliSearch =
|
|
|
|
IndexController<HeedUuidStore, MapIndexStore, dump_actor::DumpActorHandleImpl>;
|
|
|
|
|
2021-09-22 15:07:04 +02:00
|
|
|
pub type Payload = Box<
|
|
|
|
dyn Stream<Item = std::result::Result<Bytes, PayloadError>> + Send + Sync + 'static + Unpin,
|
|
|
|
>;
|
2021-02-01 19:51:47 +01:00
|
|
|
|
2021-02-03 17:44:20 +01:00
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone)]
|
|
|
|
#[serde(rename_all = "camelCase")]
|
|
|
|
pub struct IndexMetadata {
|
2021-04-14 18:55:04 +02:00
|
|
|
#[serde(skip)]
|
|
|
|
pub uuid: Uuid,
|
2021-04-01 16:44:42 +02:00
|
|
|
pub uid: String,
|
2021-03-15 18:35:16 +01:00
|
|
|
name: String,
|
2021-03-06 20:12:20 +01:00
|
|
|
#[serde(flatten)]
|
2021-09-24 11:53:11 +02:00
|
|
|
pub meta: IndexMeta,
|
2021-02-03 17:44:20 +01:00
|
|
|
}
|
|
|
|
|
2021-02-09 16:08:13 +01:00
|
|
|
#[derive(Clone, Debug)]
|
2021-02-09 11:41:26 +01:00
|
|
|
pub struct IndexSettings {
|
2021-03-11 22:47:29 +01:00
|
|
|
pub uid: Option<String>,
|
2021-02-09 11:41:26 +01:00
|
|
|
pub primary_key: Option<String>,
|
|
|
|
}
|
2021-03-04 12:03:06 +01:00
|
|
|
|
2021-09-22 15:07:04 +02:00
|
|
|
#[derive(Debug)]
|
2021-09-14 18:39:02 +02:00
|
|
|
pub enum DocumentAdditionFormat {
|
|
|
|
Json,
|
2021-09-22 16:01:21 +02:00
|
|
|
Csv,
|
2021-09-29 10:17:52 +02:00
|
|
|
Ndjson,
|
2021-09-14 18:39:02 +02:00
|
|
|
}
|
|
|
|
|
2021-09-30 11:29:27 +02:00
|
|
|
impl fmt::Display for DocumentAdditionFormat {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
|
|
match self {
|
|
|
|
DocumentAdditionFormat::Json => write!(f, "json"),
|
|
|
|
DocumentAdditionFormat::Ndjson => write!(f, "ndjson"),
|
|
|
|
DocumentAdditionFormat::Csv => write!(f, "csv"),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-06-23 12:18:34 +02:00
|
|
|
#[derive(Serialize, Debug)]
|
2021-04-15 19:54:25 +02:00
|
|
|
#[serde(rename_all = "camelCase")]
|
2021-04-14 18:55:04 +02:00
|
|
|
pub struct Stats {
|
|
|
|
pub database_size: u64,
|
|
|
|
pub last_update: Option<DateTime<Utc>>,
|
|
|
|
pub indexes: BTreeMap<String, IndexStats>,
|
|
|
|
}
|
|
|
|
|
2021-09-28 22:22:59 +02:00
|
|
|
#[allow(clippy::large_enum_variant)]
|
2021-09-22 15:07:04 +02:00
|
|
|
#[derive(derivative::Derivative)]
|
|
|
|
#[derivative(Debug)]
|
2021-09-14 18:39:02 +02:00
|
|
|
pub enum Update {
|
2021-09-24 15:21:07 +02:00
|
|
|
DeleteDocuments(Vec<String>),
|
|
|
|
ClearDocuments,
|
2021-09-24 14:55:57 +02:00
|
|
|
Settings(Settings<Unchecked>),
|
2021-09-14 18:39:02 +02:00
|
|
|
DocumentAddition {
|
2021-09-28 22:22:59 +02:00
|
|
|
#[derivative(Debug = "ignore")]
|
2021-09-14 18:39:02 +02:00
|
|
|
payload: Payload,
|
|
|
|
primary_key: Option<String>,
|
|
|
|
method: IndexDocumentsMethod,
|
|
|
|
format: DocumentAdditionFormat,
|
2021-09-22 15:07:04 +02:00
|
|
|
},
|
2021-09-14 18:39:02 +02:00
|
|
|
}
|
|
|
|
|
2021-09-21 13:23:22 +02:00
|
|
|
#[derive(Default, Debug)]
|
|
|
|
pub struct IndexControllerBuilder {
|
|
|
|
max_index_size: Option<usize>,
|
|
|
|
max_update_store_size: Option<usize>,
|
|
|
|
snapshot_dir: Option<PathBuf>,
|
|
|
|
import_snapshot: Option<PathBuf>,
|
2021-09-27 16:48:03 +02:00
|
|
|
snapshot_interval: Option<Duration>,
|
2021-09-21 13:23:22 +02:00
|
|
|
ignore_snapshot_if_db_exists: bool,
|
|
|
|
ignore_missing_snapshot: bool,
|
2021-09-27 16:48:03 +02:00
|
|
|
schedule_snapshot: bool,
|
2021-09-21 13:23:22 +02:00
|
|
|
dump_src: Option<PathBuf>,
|
|
|
|
dump_dst: Option<PathBuf>,
|
|
|
|
}
|
2021-03-17 12:01:56 +01:00
|
|
|
|
2021-09-21 13:23:22 +02:00
|
|
|
impl IndexControllerBuilder {
|
2021-09-22 15:07:04 +02:00
|
|
|
pub fn build(
|
|
|
|
self,
|
|
|
|
db_path: impl AsRef<Path>,
|
|
|
|
indexer_options: IndexerOpts,
|
2021-10-06 13:01:02 +02:00
|
|
|
) -> anyhow::Result<MeiliSearch> {
|
2021-09-22 15:07:04 +02:00
|
|
|
let index_size = self
|
|
|
|
.max_index_size
|
|
|
|
.ok_or_else(|| anyhow::anyhow!("Missing index size"))?;
|
|
|
|
let update_store_size = self
|
|
|
|
.max_index_size
|
|
|
|
.ok_or_else(|| anyhow::anyhow!("Missing update database size"))?;
|
2021-09-21 13:23:22 +02:00
|
|
|
|
|
|
|
if let Some(ref path) = self.import_snapshot {
|
2021-03-24 11:29:11 +01:00
|
|
|
info!("Loading from snapshot {:?}", path);
|
2021-03-23 16:37:46 +01:00
|
|
|
load_snapshot(
|
2021-09-21 13:23:22 +02:00
|
|
|
db_path.as_ref(),
|
2021-03-23 16:37:46 +01:00
|
|
|
path,
|
2021-09-21 13:23:22 +02:00
|
|
|
self.ignore_snapshot_if_db_exists,
|
|
|
|
self.ignore_missing_snapshot,
|
2021-03-23 16:37:46 +01:00
|
|
|
)?;
|
2021-09-21 13:23:22 +02:00
|
|
|
} else if let Some(ref src_path) = self.dump_src {
|
2021-05-27 14:30:20 +02:00
|
|
|
load_dump(
|
2021-09-21 13:23:22 +02:00
|
|
|
db_path.as_ref(),
|
2021-05-27 14:30:20 +02:00
|
|
|
src_path,
|
2021-09-21 13:23:22 +02:00
|
|
|
index_size,
|
|
|
|
update_store_size,
|
|
|
|
&indexer_options,
|
2021-05-27 14:30:20 +02:00
|
|
|
)?;
|
2021-03-22 19:19:37 +01:00
|
|
|
}
|
|
|
|
|
2021-09-21 13:23:22 +02:00
|
|
|
std::fs::create_dir_all(db_path.as_ref())?;
|
2021-03-23 17:23:57 +01:00
|
|
|
|
2021-09-28 22:22:59 +02:00
|
|
|
let index_resolver = Arc::new(create_index_resolver(
|
|
|
|
&db_path,
|
|
|
|
index_size,
|
|
|
|
&indexer_options,
|
|
|
|
)?);
|
2021-09-22 11:52:29 +02:00
|
|
|
|
|
|
|
#[allow(unreachable_code)]
|
2021-09-28 22:22:59 +02:00
|
|
|
let update_sender =
|
|
|
|
updates::create_update_handler(index_resolver.clone(), &db_path, update_store_size)?;
|
2021-09-21 13:23:22 +02:00
|
|
|
|
2021-09-28 22:22:59 +02:00
|
|
|
let dump_path = self
|
|
|
|
.dump_dst
|
|
|
|
.ok_or_else(|| anyhow::anyhow!("Missing dump directory path"))?;
|
2021-05-26 20:42:09 +02:00
|
|
|
let dump_handle = dump_actor::DumpActorHandleImpl::new(
|
2021-09-24 11:53:11 +02:00
|
|
|
dump_path,
|
|
|
|
index_resolver.clone(),
|
2021-09-27 16:48:03 +02:00
|
|
|
update_sender.clone(),
|
2021-09-21 13:23:22 +02:00
|
|
|
index_size,
|
|
|
|
update_store_size,
|
2021-05-26 20:42:09 +02:00
|
|
|
)?;
|
2021-03-17 12:01:56 +01:00
|
|
|
|
2021-10-06 13:01:02 +02:00
|
|
|
let dump_handle = Arc::new(dump_handle);
|
|
|
|
|
2021-09-27 16:48:03 +02:00
|
|
|
if self.schedule_snapshot {
|
|
|
|
let snapshot_service = SnapshotService::new(
|
|
|
|
index_resolver.clone(),
|
|
|
|
update_sender.clone(),
|
2021-09-28 22:22:59 +02:00
|
|
|
self.snapshot_interval
|
|
|
|
.ok_or_else(|| anyhow::anyhow!("Snapshot interval not provided."))?,
|
|
|
|
self.snapshot_dir
|
|
|
|
.ok_or_else(|| anyhow::anyhow!("Snapshot path not provided."))?,
|
2021-09-27 16:48:03 +02:00
|
|
|
db_path
|
2021-09-28 22:22:59 +02:00
|
|
|
.as_ref()
|
|
|
|
.file_name()
|
|
|
|
.map(|n| n.to_owned().into_string().expect("invalid path"))
|
|
|
|
.unwrap_or_else(|| String::from("data.ms")),
|
2021-09-27 16:48:03 +02:00
|
|
|
);
|
|
|
|
|
|
|
|
tokio::task::spawn(snapshot_service.run());
|
|
|
|
}
|
2021-03-17 12:01:56 +01:00
|
|
|
|
2021-09-21 13:23:22 +02:00
|
|
|
Ok(IndexController {
|
2021-09-24 11:53:11 +02:00
|
|
|
index_resolver,
|
2021-09-27 16:48:03 +02:00
|
|
|
update_sender,
|
2021-05-10 20:25:09 +02:00
|
|
|
dump_handle,
|
2021-03-15 18:11:10 +01:00
|
|
|
})
|
2021-03-04 12:03:06 +01:00
|
|
|
}
|
|
|
|
|
2021-09-21 13:23:22 +02:00
|
|
|
/// Set the index controller builder's max update store size.
|
|
|
|
pub fn set_max_update_store_size(&mut self, max_update_store_size: usize) -> &mut Self {
|
|
|
|
self.max_update_store_size.replace(max_update_store_size);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn set_max_index_size(&mut self, size: usize) -> &mut Self {
|
|
|
|
self.max_index_size.replace(size);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Set the index controller builder's snapshot path.
|
|
|
|
pub fn set_snapshot_dir(&mut self, snapshot_dir: PathBuf) -> &mut Self {
|
|
|
|
self.snapshot_dir.replace(snapshot_dir);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Set the index controller builder's ignore snapshot if db exists.
|
2021-09-22 15:07:04 +02:00
|
|
|
pub fn set_ignore_snapshot_if_db_exists(
|
|
|
|
&mut self,
|
|
|
|
ignore_snapshot_if_db_exists: bool,
|
|
|
|
) -> &mut Self {
|
2021-09-21 13:23:22 +02:00
|
|
|
self.ignore_snapshot_if_db_exists = ignore_snapshot_if_db_exists;
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Set the index controller builder's ignore missing snapshot.
|
|
|
|
pub fn set_ignore_missing_snapshot(&mut self, ignore_missing_snapshot: bool) -> &mut Self {
|
|
|
|
self.ignore_missing_snapshot = ignore_missing_snapshot;
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Set the index controller builder's dump src.
|
|
|
|
pub fn set_dump_src(&mut self, dump_src: PathBuf) -> &mut Self {
|
|
|
|
self.dump_src.replace(dump_src);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Set the index controller builder's dump dst.
|
|
|
|
pub fn set_dump_dst(&mut self, dump_dst: PathBuf) -> &mut Self {
|
|
|
|
self.dump_dst.replace(dump_dst);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Set the index controller builder's import snapshot.
|
|
|
|
pub fn set_import_snapshot(&mut self, import_snapshot: PathBuf) -> &mut Self {
|
|
|
|
self.import_snapshot.replace(import_snapshot);
|
|
|
|
self
|
|
|
|
}
|
2021-09-27 16:48:03 +02:00
|
|
|
|
|
|
|
/// Set the index controller builder's snapshot interval sec.
|
|
|
|
pub fn set_snapshot_interval(&mut self, snapshot_interval: Duration) -> &mut Self {
|
|
|
|
self.snapshot_interval = Some(snapshot_interval);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Set the index controller builder's schedule snapshot.
|
|
|
|
pub fn set_schedule_snapshot(&mut self) -> &mut Self {
|
|
|
|
self.schedule_snapshot = true;
|
|
|
|
self
|
|
|
|
}
|
2021-09-21 13:23:22 +02:00
|
|
|
}
|
|
|
|
|
2021-10-06 14:34:14 +02:00
|
|
|
// We are using derivative here to derive Clone, because U, I and D do not necessarily implement
|
|
|
|
// Clone themselves.
|
2021-10-06 13:01:02 +02:00
|
|
|
#[derive(derivative::Derivative)]
|
|
|
|
#[derivative(Clone(bound = ""))]
|
|
|
|
pub struct IndexController<U, I, D> {
|
|
|
|
index_resolver: Arc<IndexResolver<U, I>>,
|
|
|
|
update_sender: updates::UpdateSender,
|
|
|
|
dump_handle: Arc<D>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<U, I, D> IndexController<U, I, D>
|
|
|
|
where
|
|
|
|
U: UuidStore + Sync + Send + 'static,
|
|
|
|
I: IndexStore + Sync + Send + 'static,
|
|
|
|
D: DumpActorHandle + Send + Sync,
|
|
|
|
{
|
2021-09-21 13:23:22 +02:00
|
|
|
pub fn builder() -> IndexControllerBuilder {
|
|
|
|
IndexControllerBuilder::default()
|
|
|
|
}
|
|
|
|
|
2021-09-28 22:22:59 +02:00
|
|
|
pub async fn register_update(
|
|
|
|
&self,
|
|
|
|
uid: String,
|
|
|
|
update: Update,
|
|
|
|
create_index: bool,
|
|
|
|
) -> Result<UpdateStatus> {
|
2021-09-24 14:55:57 +02:00
|
|
|
match self.index_resolver.get_uuid(uid).await {
|
2021-09-14 18:39:02 +02:00
|
|
|
Ok(uuid) => {
|
2021-09-27 16:48:03 +02:00
|
|
|
let update_result = UpdateMsg::update(&self.update_sender, uuid, update).await?;
|
2021-09-14 18:39:02 +02:00
|
|
|
Ok(update_result)
|
2021-09-22 15:07:04 +02:00
|
|
|
}
|
2021-09-24 11:53:11 +02:00
|
|
|
Err(IndexResolverError::UnexistingIndex(name)) => {
|
2021-09-28 18:10:09 +02:00
|
|
|
if create_index {
|
|
|
|
let index = self.index_resolver.create_index(name, None).await?;
|
2021-09-28 22:22:59 +02:00
|
|
|
let update_result =
|
2021-10-06 13:01:02 +02:00
|
|
|
UpdateMsg::update(&self.update_sender, index.uuid(), update).await?;
|
2021-09-28 18:10:09 +02:00
|
|
|
Ok(update_result)
|
|
|
|
} else {
|
2021-09-28 22:22:59 +02:00
|
|
|
Err(IndexResolverError::UnexistingIndex(name).into())
|
2021-09-28 18:10:09 +02:00
|
|
|
}
|
2021-03-04 12:03:06 +01:00
|
|
|
}
|
2021-03-18 09:09:26 +01:00
|
|
|
Err(e) => Err(e.into()),
|
|
|
|
}
|
2021-03-04 12:03:06 +01:00
|
|
|
}
|
|
|
|
|
2021-06-14 21:26:35 +02:00
|
|
|
pub async fn update_status(&self, uid: String, id: u64) -> Result<UpdateStatus> {
|
2021-09-24 11:53:11 +02:00
|
|
|
let uuid = self.index_resolver.get_uuid(uid).await?;
|
2021-09-27 16:48:03 +02:00
|
|
|
let result = UpdateMsg::get_update(&self.update_sender, uuid, id).await?;
|
2021-03-06 10:51:52 +01:00
|
|
|
Ok(result)
|
2021-03-04 12:03:06 +01:00
|
|
|
}
|
|
|
|
|
2021-06-14 21:26:35 +02:00
|
|
|
pub async fn all_update_status(&self, uid: String) -> Result<Vec<UpdateStatus>> {
|
2021-09-24 11:53:11 +02:00
|
|
|
let uuid = self.index_resolver.get_uuid(uid).await?;
|
2021-09-27 16:48:03 +02:00
|
|
|
let result = UpdateMsg::list_updates(&self.update_sender, uuid).await?;
|
2021-03-05 18:34:04 +01:00
|
|
|
Ok(result)
|
2021-03-04 12:03:06 +01:00
|
|
|
}
|
|
|
|
|
2021-06-14 21:26:35 +02:00
|
|
|
pub async fn list_indexes(&self) -> Result<Vec<IndexMetadata>> {
|
2021-09-24 11:53:11 +02:00
|
|
|
let indexes = self.index_resolver.list().await?;
|
2021-03-06 20:12:20 +01:00
|
|
|
let mut ret = Vec::new();
|
2021-09-24 11:53:11 +02:00
|
|
|
for (uid, index) in indexes {
|
|
|
|
let meta = index.meta()?;
|
2021-03-22 10:17:38 +01:00
|
|
|
let meta = IndexMetadata {
|
2021-10-04 12:15:21 +02:00
|
|
|
uuid: index.uuid(),
|
2021-03-22 10:17:38 +01:00
|
|
|
name: uid.clone(),
|
|
|
|
uid,
|
|
|
|
meta,
|
|
|
|
};
|
2021-03-15 16:52:05 +01:00
|
|
|
ret.push(meta);
|
2021-03-06 20:12:20 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Ok(ret)
|
2021-03-04 12:03:06 +01:00
|
|
|
}
|
|
|
|
|
2021-06-14 21:26:35 +02:00
|
|
|
pub async fn settings(&self, uid: String) -> Result<Settings<Checked>> {
|
2021-09-24 11:53:11 +02:00
|
|
|
let index = self.index_resolver.get_index(uid).await?;
|
|
|
|
let settings = spawn_blocking(move || index.settings()).await??;
|
2021-03-04 12:38:55 +01:00
|
|
|
Ok(settings)
|
|
|
|
}
|
|
|
|
|
2021-03-04 14:20:19 +01:00
|
|
|
pub async fn documents(
|
|
|
|
&self,
|
2021-03-11 22:47:29 +01:00
|
|
|
uid: String,
|
2021-03-04 14:20:19 +01:00
|
|
|
offset: usize,
|
|
|
|
limit: usize,
|
|
|
|
attributes_to_retrieve: Option<Vec<String>>,
|
2021-06-14 21:26:35 +02:00
|
|
|
) -> Result<Vec<Document>> {
|
2021-09-24 11:53:11 +02:00
|
|
|
let index = self.index_resolver.get_index(uid).await?;
|
2021-09-28 22:22:59 +02:00
|
|
|
let documents =
|
|
|
|
spawn_blocking(move || index.retrieve_documents(offset, limit, attributes_to_retrieve))
|
|
|
|
.await??;
|
2021-03-04 14:20:19 +01:00
|
|
|
Ok(documents)
|
|
|
|
}
|
|
|
|
|
2021-03-04 15:09:00 +01:00
|
|
|
pub async fn document(
|
|
|
|
&self,
|
2021-03-11 22:47:29 +01:00
|
|
|
uid: String,
|
2021-03-04 15:09:00 +01:00
|
|
|
doc_id: String,
|
|
|
|
attributes_to_retrieve: Option<Vec<String>>,
|
2021-06-14 21:26:35 +02:00
|
|
|
) -> Result<Document> {
|
2021-09-24 11:53:11 +02:00
|
|
|
let index = self.index_resolver.get_index(uid).await?;
|
2021-09-28 22:22:59 +02:00
|
|
|
let document =
|
|
|
|
spawn_blocking(move || index.retrieve_document(doc_id, attributes_to_retrieve))
|
|
|
|
.await??;
|
2021-03-04 15:09:00 +01:00
|
|
|
Ok(document)
|
|
|
|
}
|
|
|
|
|
2021-03-15 18:11:10 +01:00
|
|
|
pub async fn update_index(
|
|
|
|
&self,
|
|
|
|
uid: String,
|
2021-06-21 13:57:32 +02:00
|
|
|
mut index_settings: IndexSettings,
|
2021-06-14 21:26:35 +02:00
|
|
|
) -> Result<IndexMetadata> {
|
2021-09-24 11:53:11 +02:00
|
|
|
index_settings.uid.take();
|
|
|
|
|
|
|
|
let index = self.index_resolver.get_index(uid.clone()).await?;
|
2021-10-04 12:15:21 +02:00
|
|
|
let uuid = index.uuid();
|
2021-09-28 22:22:59 +02:00
|
|
|
let meta =
|
|
|
|
spawn_blocking(move || index.update_primary_key(index_settings.primary_key)).await??;
|
2021-03-22 10:17:38 +01:00
|
|
|
let meta = IndexMetadata {
|
2021-04-14 18:55:04 +02:00
|
|
|
uuid,
|
2021-03-22 10:17:38 +01:00
|
|
|
name: uid.clone(),
|
|
|
|
uid,
|
|
|
|
meta,
|
|
|
|
};
|
2021-03-12 14:48:43 +01:00
|
|
|
Ok(meta)
|
2021-03-04 12:03:06 +01:00
|
|
|
}
|
|
|
|
|
2021-06-14 21:26:35 +02:00
|
|
|
pub async fn search(&self, uid: String, query: SearchQuery) -> Result<SearchResult> {
|
2021-09-28 18:10:09 +02:00
|
|
|
let index = self.index_resolver.get_index(uid.clone()).await?;
|
|
|
|
let result = spawn_blocking(move || index.perform_search(query)).await??;
|
2021-03-04 12:03:06 +01:00
|
|
|
Ok(result)
|
|
|
|
}
|
2021-03-06 20:17:58 +01:00
|
|
|
|
2021-06-14 21:26:35 +02:00
|
|
|
pub async fn get_index(&self, uid: String) -> Result<IndexMetadata> {
|
2021-09-24 11:53:11 +02:00
|
|
|
let index = self.index_resolver.get_index(uid.clone()).await?;
|
2021-10-04 12:15:21 +02:00
|
|
|
let uuid = index.uuid();
|
2021-09-24 11:53:11 +02:00
|
|
|
let meta = spawn_blocking(move || index.meta()).await??;
|
2021-03-22 10:17:38 +01:00
|
|
|
let meta = IndexMetadata {
|
2021-04-14 18:55:04 +02:00
|
|
|
uuid,
|
2021-03-22 10:17:38 +01:00
|
|
|
name: uid.clone(),
|
|
|
|
uid,
|
|
|
|
meta,
|
|
|
|
};
|
2021-03-15 16:52:05 +01:00
|
|
|
Ok(meta)
|
2021-03-06 20:17:58 +01:00
|
|
|
}
|
2021-04-01 16:44:42 +02:00
|
|
|
|
2021-06-14 21:26:35 +02:00
|
|
|
pub async fn get_index_stats(&self, uid: String) -> Result<IndexStats> {
|
2021-09-27 16:48:03 +02:00
|
|
|
let update_infos = UpdateMsg::get_info(&self.update_sender).await?;
|
2021-09-24 11:53:11 +02:00
|
|
|
let index = self.index_resolver.get_index(uid).await?;
|
2021-10-04 12:15:21 +02:00
|
|
|
let uuid = index.uuid();
|
2021-09-24 11:53:11 +02:00
|
|
|
let mut stats = spawn_blocking(move || index.stats()).await??;
|
|
|
|
// Check if the currently indexing update is from our index.
|
2021-04-22 10:14:29 +02:00
|
|
|
stats.is_indexing = Some(Some(uuid) == update_infos.processing);
|
2021-04-14 18:55:04 +02:00
|
|
|
Ok(stats)
|
2021-04-09 14:41:24 +02:00
|
|
|
}
|
|
|
|
|
2021-06-14 21:26:35 +02:00
|
|
|
pub async fn get_all_stats(&self) -> Result<Stats> {
|
2021-09-27 16:48:03 +02:00
|
|
|
let update_infos = UpdateMsg::get_info(&self.update_sender).await?;
|
2021-09-28 12:05:22 +02:00
|
|
|
let mut database_size = self.index_resolver.get_uuids_size().await? + update_infos.size;
|
2021-04-14 18:55:04 +02:00
|
|
|
let mut last_update: Option<DateTime<_>> = None;
|
|
|
|
let mut indexes = BTreeMap::new();
|
|
|
|
|
2021-09-24 11:53:11 +02:00
|
|
|
for (index_uid, index) in self.index_resolver.list().await? {
|
2021-10-04 12:15:21 +02:00
|
|
|
let uuid = index.uuid();
|
2021-09-24 11:53:11 +02:00
|
|
|
let (mut stats, meta) = spawn_blocking::<_, IndexResult<_>>(move || {
|
|
|
|
let stats = index.stats()?;
|
|
|
|
let meta = index.meta()?;
|
|
|
|
Ok((stats, meta))
|
2021-09-28 22:22:59 +02:00
|
|
|
})
|
|
|
|
.await??;
|
2021-09-24 11:53:11 +02:00
|
|
|
|
|
|
|
database_size += stats.size;
|
2021-04-14 18:55:04 +02:00
|
|
|
|
2021-09-24 11:53:11 +02:00
|
|
|
last_update = last_update.map_or(Some(meta.updated_at), |last| {
|
|
|
|
Some(last.max(meta.updated_at))
|
2021-04-14 18:55:04 +02:00
|
|
|
});
|
|
|
|
|
2021-09-24 11:53:11 +02:00
|
|
|
// Check if the currently indexing update is from our index.
|
|
|
|
stats.is_indexing = Some(Some(uuid) == update_infos.processing);
|
2021-04-14 18:55:04 +02:00
|
|
|
|
2021-09-24 11:53:11 +02:00
|
|
|
indexes.insert(index_uid, stats);
|
2021-04-14 18:55:04 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
Ok(Stats {
|
|
|
|
database_size,
|
|
|
|
last_update,
|
|
|
|
indexes,
|
|
|
|
})
|
2021-04-09 14:41:24 +02:00
|
|
|
}
|
2021-05-10 20:25:09 +02:00
|
|
|
|
2021-06-14 21:26:35 +02:00
|
|
|
pub async fn create_dump(&self) -> Result<DumpInfo> {
|
2021-05-10 20:25:09 +02:00
|
|
|
Ok(self.dump_handle.create_dump().await?)
|
|
|
|
}
|
|
|
|
|
2021-06-14 21:26:35 +02:00
|
|
|
pub async fn dump_info(&self, uid: String) -> Result<DumpInfo> {
|
2021-05-10 20:25:09 +02:00
|
|
|
Ok(self.dump_handle.dump_info(uid).await?)
|
|
|
|
}
|
2021-09-28 18:10:09 +02:00
|
|
|
|
2021-09-28 22:22:59 +02:00
|
|
|
pub async fn create_index(
|
|
|
|
&self,
|
|
|
|
uid: String,
|
|
|
|
primary_key: Option<String>,
|
|
|
|
) -> Result<IndexMetadata> {
|
|
|
|
let index = self
|
|
|
|
.index_resolver
|
|
|
|
.create_index(uid.clone(), primary_key)
|
|
|
|
.await?;
|
2021-09-28 18:10:09 +02:00
|
|
|
let meta = spawn_blocking(move || -> IndexResult<_> {
|
|
|
|
let meta = index.meta()?;
|
|
|
|
let meta = IndexMetadata {
|
2021-10-04 12:15:21 +02:00
|
|
|
uuid: index.uuid(),
|
2021-09-28 18:10:09 +02:00
|
|
|
uid: uid.clone(),
|
|
|
|
name: uid,
|
|
|
|
meta,
|
|
|
|
};
|
|
|
|
Ok(meta)
|
2021-09-28 22:22:59 +02:00
|
|
|
})
|
|
|
|
.await??;
|
2021-09-28 18:10:09 +02:00
|
|
|
|
|
|
|
Ok(meta)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub async fn delete_index(&self, uid: String) -> Result<()> {
|
|
|
|
let uuid = self.index_resolver.delete_index(uid).await?;
|
|
|
|
|
|
|
|
let update_sender = self.update_sender.clone();
|
|
|
|
tokio::spawn(async move {
|
|
|
|
let _ = UpdateMsg::delete(&update_sender, uuid).await;
|
|
|
|
});
|
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
2021-03-04 12:03:06 +01:00
|
|
|
}
|
2021-03-06 12:57:56 +01:00
|
|
|
|
|
|
|
pub async fn get_arc_ownership_blocking<T>(mut item: Arc<T>) -> T {
|
|
|
|
loop {
|
|
|
|
match Arc::try_unwrap(item) {
|
|
|
|
Ok(item) => return item,
|
|
|
|
Err(item_arc) => {
|
|
|
|
item = item_arc;
|
|
|
|
sleep(Duration::from_millis(100)).await;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2021-10-06 13:01:02 +02:00
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod test {
|
|
|
|
use futures::future::ok;
|
|
|
|
use mockall::predicate::eq;
|
|
|
|
use tokio::sync::mpsc;
|
|
|
|
|
|
|
|
use crate::index::error::Result as IndexResult;
|
|
|
|
use crate::index::test::Mocker;
|
|
|
|
use crate::index::Index;
|
|
|
|
use crate::index_controller::dump_actor::MockDumpActorHandle;
|
|
|
|
use crate::index_controller::index_resolver::index_store::MockIndexStore;
|
|
|
|
use crate::index_controller::index_resolver::uuid_store::MockUuidStore;
|
|
|
|
|
|
|
|
use super::updates::UpdateSender;
|
|
|
|
use super::*;
|
|
|
|
|
|
|
|
impl<D: DumpActorHandle> IndexController<MockUuidStore, MockIndexStore, D> {
|
|
|
|
pub fn mock(
|
|
|
|
index_resolver: IndexResolver<MockUuidStore, MockIndexStore>,
|
|
|
|
update_sender: UpdateSender,
|
|
|
|
dump_handle: D,
|
|
|
|
) -> Self {
|
|
|
|
IndexController {
|
|
|
|
index_resolver: Arc::new(index_resolver),
|
|
|
|
update_sender,
|
|
|
|
dump_handle: Arc::new(dump_handle),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[actix_rt::test]
|
|
|
|
async fn test_search_simple() {
|
|
|
|
let index_uid = "test";
|
|
|
|
let index_uuid = Uuid::new_v4();
|
|
|
|
let query = SearchQuery {
|
|
|
|
q: Some(String::from("hello world")),
|
|
|
|
offset: Some(10),
|
|
|
|
limit: 0,
|
|
|
|
attributes_to_retrieve: Some(vec!["string".to_owned()].into_iter().collect()),
|
|
|
|
attributes_to_crop: None,
|
|
|
|
crop_length: 18,
|
|
|
|
attributes_to_highlight: None,
|
|
|
|
matches: true,
|
|
|
|
filter: None,
|
|
|
|
sort: None,
|
|
|
|
facets_distribution: None,
|
|
|
|
};
|
|
|
|
|
|
|
|
let result = SearchResult {
|
|
|
|
hits: vec![],
|
|
|
|
nb_hits: 29,
|
|
|
|
exhaustive_nb_hits: true,
|
|
|
|
query: "hello world".to_string(),
|
|
|
|
limit: 24,
|
|
|
|
offset: 0,
|
|
|
|
processing_time_ms: 50,
|
|
|
|
facets_distribution: None,
|
|
|
|
exhaustive_facets_count: Some(true),
|
|
|
|
};
|
|
|
|
|
|
|
|
let mut uuid_store = MockUuidStore::new();
|
|
|
|
uuid_store
|
|
|
|
.expect_get_uuid()
|
|
|
|
.with(eq(index_uid.to_owned()))
|
|
|
|
.returning(move |s| Box::pin(ok((s, Some(index_uuid)))));
|
|
|
|
|
|
|
|
let mut index_store = MockIndexStore::new();
|
|
|
|
let result_clone = result.clone();
|
|
|
|
let query_clone = query.clone();
|
|
|
|
index_store
|
|
|
|
.expect_get()
|
|
|
|
.with(eq(index_uuid))
|
|
|
|
.returning(move |_uuid| {
|
|
|
|
let result = result_clone.clone();
|
|
|
|
let query = query_clone.clone();
|
|
|
|
let mocker = Mocker::default();
|
|
|
|
mocker
|
|
|
|
.when::<SearchQuery, IndexResult<SearchResult>>("perform_search")
|
|
|
|
.once()
|
|
|
|
.then(move |q| {
|
|
|
|
assert_eq!(&q, &query);
|
|
|
|
Ok(result.clone())
|
|
|
|
});
|
|
|
|
let index = Index::faux(mocker);
|
|
|
|
Box::pin(ok(Some(index)))
|
|
|
|
});
|
|
|
|
|
|
|
|
let index_resolver = IndexResolver::new(uuid_store, index_store);
|
|
|
|
let (update_sender, _) = mpsc::channel(1);
|
|
|
|
let dump_actor = MockDumpActorHandle::new();
|
|
|
|
let index_controller = IndexController::mock(index_resolver, update_sender, dump_actor);
|
|
|
|
|
|
|
|
let r = index_controller
|
|
|
|
.search(index_uid.to_owned(), query.clone())
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
assert_eq!(r, result);
|
|
|
|
}
|
|
|
|
}
|