mirror of
https://github.com/meilisearch/MeiliSearch
synced 2024-06-02 10:58:02 +02:00
104 lines
2.8 KiB
Rust
104 lines
2.8 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use actix_web::get;
|
|
use actix_web::web;
|
|
use actix_web::HttpResponse;
|
|
use chrono::{DateTime, Utc};
|
|
use milli::FieldsDistribution;
|
|
use serde::Serialize;
|
|
|
|
use crate::error::ResponseError;
|
|
use crate::helpers::Authentication;
|
|
use crate::index_controller::IndexStats;
|
|
use crate::routes::IndexParam;
|
|
use crate::Data;
|
|
use crate::data::Stats;
|
|
|
|
pub fn services(cfg: &mut web::ServiceConfig) {
|
|
cfg.service(get_index_stats)
|
|
.service(get_stats)
|
|
.service(get_version);
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
struct IndexStatsResponse {
|
|
number_of_documents: u64,
|
|
is_indexing: bool,
|
|
fields_distribution: FieldsDistribution,
|
|
}
|
|
|
|
impl From<IndexStats> for IndexStatsResponse {
|
|
fn from(stats: IndexStats) -> Self {
|
|
Self {
|
|
number_of_documents: stats.number_of_documents,
|
|
is_indexing: stats.is_indexing,
|
|
fields_distribution: stats.fields_distribution,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[get("/indexes/{index_uid}/stats", wrap = "Authentication::Private")]
|
|
async fn get_index_stats(
|
|
data: web::Data<Data>,
|
|
path: web::Path<IndexParam>,
|
|
) -> Result<HttpResponse, ResponseError> {
|
|
let response: IndexStatsResponse = data.get_index_stats(path.index_uid.clone()).await?.into();
|
|
|
|
Ok(HttpResponse::Ok().json(response))
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
struct StatsResponse {
|
|
database_size: u64,
|
|
last_update: Option<DateTime<Utc>>,
|
|
indexes: HashMap<String, IndexStatsResponse>,
|
|
}
|
|
|
|
impl From<Stats> for StatsResponse {
|
|
fn from(stats: Stats) -> Self {
|
|
Self {
|
|
database_size: stats.database_size,
|
|
last_update: stats.last_update,
|
|
indexes: stats.indexes
|
|
.into_iter()
|
|
.map(|(uid, index_stats)| (uid, index_stats.into()))
|
|
.collect(),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[get("/stats", wrap = "Authentication::Private")]
|
|
async fn get_stats(data: web::Data<Data>) -> Result<HttpResponse, ResponseError> {
|
|
let response: StatsResponse = data.get_stats().await?.into();
|
|
|
|
Ok(HttpResponse::Ok().json(response))
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
struct VersionResponse {
|
|
commit_sha: String,
|
|
build_date: String,
|
|
pkg_version: String,
|
|
}
|
|
|
|
#[get("/version", wrap = "Authentication::Private")]
|
|
async fn get_version() -> HttpResponse {
|
|
let commit_sha = match option_env!("COMMIT_SHA") {
|
|
Some("") | None => env!("VERGEN_SHA"),
|
|
Some(commit_sha) => commit_sha,
|
|
};
|
|
let commit_date = match option_env!("COMMIT_DATE") {
|
|
Some("") | None => env!("VERGEN_COMMIT_DATE"),
|
|
Some(commit_date) => commit_date,
|
|
};
|
|
|
|
HttpResponse::Ok().json(VersionResponse {
|
|
commit_sha: commit_sha.to_string(),
|
|
build_date: commit_date.to_string(),
|
|
pkg_version: env!("CARGO_PKG_VERSION").to_string(),
|
|
})
|
|
}
|