MeiliSearch/meilisearch-http/src/routes/dump.rs

48 lines
1.7 KiB
Rust
Raw Normal View History

2022-09-27 16:33:37 +02:00
use actix_web::web::Data;
2021-10-14 11:32:55 +02:00
use actix_web::{web, HttpRequest, HttpResponse};
2022-09-27 16:33:37 +02:00
use index_scheduler::IndexScheduler;
use log::debug;
2022-10-13 15:02:59 +02:00
use meilisearch_auth::AuthController;
use meilisearch_types::error::ResponseError;
2022-10-12 03:21:25 +02:00
use meilisearch_types::tasks::KindWithContent;
2021-10-14 11:32:55 +02:00
use serde_json::json;
2022-10-13 15:02:59 +02:00
use time::macros::format_description;
use time::OffsetDateTime;
2020-12-12 13:32:06 +01:00
2021-10-14 11:32:55 +02:00
use crate::analytics::Analytics;
2021-06-24 15:06:58 +02:00
use crate::extractors::authentication::{policies::*, GuardedData};
2022-03-04 20:12:44 +01:00
use crate::extractors::sequential_extractor::SeqHandler;
2022-10-13 15:02:59 +02:00
use crate::routes::SummarizedTaskView;
2020-12-12 13:32:06 +01:00
2021-07-05 14:29:20 +02:00
pub fn configure(cfg: &mut web::ServiceConfig) {
2022-05-19 20:19:34 +02:00
cfg.service(web::resource("").route(web::post().to(SeqHandler(create_dump))));
2020-12-12 13:32:06 +01:00
}
2021-09-28 22:22:59 +02:00
pub async fn create_dump(
2022-09-27 16:33:37 +02:00
index_scheduler: GuardedData<ActionPolicy<{ actions::DUMPS_CREATE }>, Data<IndexScheduler>>,
2022-10-13 15:02:59 +02:00
auth_controller: GuardedData<ActionPolicy<{ actions::DUMPS_CREATE }>, AuthController>,
2021-10-14 11:32:55 +02:00
req: HttpRequest,
2021-10-29 16:10:58 +02:00
analytics: web::Data<dyn Analytics>,
2021-09-28 22:22:59 +02:00
) -> Result<HttpResponse, ResponseError> {
2021-10-14 11:32:55 +02:00
analytics.publish("Dump Created".to_string(), json!({}), Some(&req));
2022-10-13 15:02:59 +02:00
let dump_uid = OffsetDateTime::now_utc()
.format(format_description!(
"[year repr:full][month repr:numerical][day padding:zero]-[hour padding:zero][minute padding:zero][second padding:zero][subsecond digits:3]"
))
.unwrap();
let task = KindWithContent::DumpExport {
2022-10-13 15:02:59 +02:00
keys: auth_controller.list_keys()?,
instance_uid: analytics.instance_uid().cloned(),
dump_uid,
};
2022-10-13 15:02:59 +02:00
let task: SummarizedTaskView =
tokio::task::spawn_blocking(move || index_scheduler.register(task))
.await??
.into();
2021-05-05 14:11:56 +02:00
2022-10-13 15:02:59 +02:00
debug!("returns: {:?}", task);
Ok(HttpResponse::Accepted().json(task))
2020-12-12 13:32:06 +01:00
}