libmotiva 0.1.0

Sanctioned entities matching utilities
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use std::{collections::HashSet, sync::Arc};

use ahash::RandomState;
use bon::Builder;
use tokio::sync::RwLock;

use crate::{
  catalog::Collections,
  error::MotivaError,
  index::{EntityHandle, IndexProvider},
  matching::MatchParams,
  model::{Entity, SearchEntity},
};

#[derive(Clone, Builder, Default)]
pub struct MockedElasticsearch {
  healthy: Option<bool>,
  #[builder(default)]
  entities: Vec<Entity>,
}

impl IndexProvider for MockedElasticsearch {
  async fn health(&self) -> Result<bool, MotivaError> {
    match self.healthy {
      None => Err(MotivaError::OtherError(anyhow::anyhow!("an error"))),
      Some(health) => Ok(health),
    }
  }

  async fn search(&self, _: &Arc<RwLock<Collections>>, _: &SearchEntity, _: &MatchParams) -> Result<Vec<Entity>, MotivaError> {
    Ok(self.entities.clone())
  }

  async fn get_entity(&self, _: &str) -> Result<EntityHandle, MotivaError> {
    unimplemented!();
  }

  async fn get_related_entities(&self, _: Option<&String>, _: &[String], _: &HashSet<String, RandomState>) -> Result<Vec<Entity>, MotivaError> {
    unimplemented!();
  }
}