mirror of
https://github.com/meilisearch/MeiliSearch
synced 2024-11-10 06:58:56 +01:00
101 lines
3.5 KiB
Rust
101 lines
3.5 KiB
Rust
use actix_web::web::{self, Data};
|
|
use actix_web::{HttpRequest, HttpResponse};
|
|
use deserr::actix_web::AwebJson;
|
|
use deserr::Deserr;
|
|
use index_scheduler::IndexScheduler;
|
|
use meilisearch_types::deserr::DeserrJsonError;
|
|
use meilisearch_types::error::ResponseError;
|
|
use meilisearch_types::keys::actions;
|
|
use serde_json::json;
|
|
use tracing::debug;
|
|
|
|
use crate::analytics::Analytics;
|
|
use crate::extractors::authentication::policies::ActionPolicy;
|
|
use crate::extractors::authentication::GuardedData;
|
|
use crate::extractors::sequential_extractor::SeqHandler;
|
|
|
|
pub fn configure(cfg: &mut web::ServiceConfig) {
|
|
cfg.service(
|
|
web::resource("")
|
|
.route(web::get().to(SeqHandler(get_features)))
|
|
.route(web::patch().to(SeqHandler(patch_features))),
|
|
);
|
|
}
|
|
|
|
async fn get_features(
|
|
index_scheduler: GuardedData<
|
|
ActionPolicy<{ actions::EXPERIMENTAL_FEATURES_GET }>,
|
|
Data<IndexScheduler>,
|
|
>,
|
|
req: HttpRequest,
|
|
analytics: Data<dyn Analytics>,
|
|
) -> HttpResponse {
|
|
let features = index_scheduler.features();
|
|
|
|
analytics.publish("Experimental features Seen".to_string(), json!(null), Some(&req));
|
|
let features = features.runtime_features();
|
|
debug!(returns = ?features, "Get features");
|
|
HttpResponse::Ok().json(features)
|
|
}
|
|
|
|
#[derive(Debug, Deserr)]
|
|
#[deserr(error = DeserrJsonError, rename_all = camelCase, deny_unknown_fields)]
|
|
pub struct RuntimeTogglableFeatures {
|
|
#[deserr(default)]
|
|
pub vector_store: Option<bool>,
|
|
#[deserr(default)]
|
|
pub metrics: Option<bool>,
|
|
#[deserr(default)]
|
|
pub logs_route: Option<bool>,
|
|
#[deserr(default)]
|
|
pub export_puffin_reports: Option<bool>,
|
|
}
|
|
|
|
async fn patch_features(
|
|
index_scheduler: GuardedData<
|
|
ActionPolicy<{ actions::EXPERIMENTAL_FEATURES_UPDATE }>,
|
|
Data<IndexScheduler>,
|
|
>,
|
|
new_features: AwebJson<RuntimeTogglableFeatures, DeserrJsonError>,
|
|
req: HttpRequest,
|
|
analytics: Data<dyn Analytics>,
|
|
) -> Result<HttpResponse, ResponseError> {
|
|
let features = index_scheduler.features();
|
|
debug!(parameters = ?new_features, "Patch features");
|
|
|
|
let old_features = features.runtime_features();
|
|
let new_features = meilisearch_types::features::RuntimeTogglableFeatures {
|
|
vector_store: new_features.0.vector_store.unwrap_or(old_features.vector_store),
|
|
metrics: new_features.0.metrics.unwrap_or(old_features.metrics),
|
|
logs_route: new_features.0.logs_route.unwrap_or(old_features.logs_route),
|
|
export_puffin_reports: new_features
|
|
.0
|
|
.export_puffin_reports
|
|
.unwrap_or(old_features.export_puffin_reports),
|
|
};
|
|
|
|
// explicitly destructure for analytics rather than using the `Serialize` implementation, because
|
|
// the it renames to camelCase, which we don't want for analytics.
|
|
// **Do not** ignore fields with `..` or `_` here, because we want to add them in the future.
|
|
let meilisearch_types::features::RuntimeTogglableFeatures {
|
|
vector_store,
|
|
metrics,
|
|
logs_route,
|
|
export_puffin_reports,
|
|
} = new_features;
|
|
|
|
analytics.publish(
|
|
"Experimental features Updated".to_string(),
|
|
json!({
|
|
"vector_store": vector_store,
|
|
"metrics": metrics,
|
|
"logs_route": logs_route,
|
|
"export_puffin_reports": export_puffin_reports,
|
|
}),
|
|
Some(&req),
|
|
);
|
|
index_scheduler.put_runtime_features(new_features)?;
|
|
debug!(returns = ?new_features, "Patch features");
|
|
Ok(HttpResponse::Ok().json(new_features))
|
|
}
|