MeiliSearch/meilisearch-lib/src/index_controller/indexes/error.rs

use std::fmt;

use meilisearch_error::{Code, ErrorCode};

use crate::{error::MilliError, index::error::IndexError};

pub type Result<T> = std::result::Result<T, IndexActorError>;

#[derive(thiserror::Error, Debug)]
pub enum IndexActorError {
    #[error("{0}")]
    IndexError(#[from] IndexError),
    #[error("Index already exists")]
    IndexAlreadyExists,
    #[error("Index not found")]
    UnexistingIndex,
    #[error("A primary key is already present. It's impossible to update it")]
    ExistingPrimaryKey,
    #[error("Internal Error: {0}")]
    Internal(Box<dyn std::error::Error + Send + Sync + 'static>),
    #[error("{0}")]
    Milli(#[from] milli::Error),
}

impl<T> From<tokio::sync::mpsc::error::SendError<T>> for IndexActorError
where T: Send + Sync + 'static + fmt::Debug
{
    fn from(other: tokio::sync::mpsc::error::SendError<T>) -> Self {
        Self::Internal(Box::new(other))
    }
}

impl From<tokio::sync::oneshot::error::RecvError> for IndexActorError {
    fn from(other: tokio::sync::oneshot::error::RecvError) -> Self {
        Self::Internal(Box::new(other))
    }
}

macro_rules! internal_error {
    ($($other:path), *) => {
        $(
            impl From<$other> for IndexActorError {
                fn from(other: $other) -> Self {
                    Self::Internal(Box::new(other))
                }
            }
        )*
    }
}

internal_error!(heed::Error, tokio::task::JoinError, std::io::Error);

impl ErrorCode for IndexActorError {
    fn error_code(&self) -> Code {
        match self {
            IndexActorError::IndexError(e) => e.error_code(),
            IndexActorError::IndexAlreadyExists => Code::IndexAlreadyExists,
            IndexActorError::UnexistingIndex => Code::IndexNotFound,
            IndexActorError::ExistingPrimaryKey => Code::PrimaryKeyAlreadyPresent,
            IndexActorError::Internal(_) => Code::Internal,
            IndexActorError::Milli(e) => MilliError(e).error_code(),
        }
    }
}