MeiliSearch/meilisearch-http/src/lib.rs

175 lines
5.8 KiB
Rust
Raw Normal View History

#![allow(rustdoc::private_intra_doc_links)]
2021-06-15 17:55:27 +02:00
#[macro_use]
2020-12-12 13:32:06 +01:00
pub mod error;
pub mod analytics;
mod task;
2021-06-24 14:22:12 +02:00
#[macro_use]
2021-06-23 14:56:02 +02:00
pub mod extractors;
2020-12-12 13:32:06 +01:00
pub mod helpers;
2021-03-15 18:11:10 +01:00
pub mod option;
pub mod routes;
2022-01-19 11:21:19 +01:00
use std::sync::{atomic::AtomicBool, Arc};
2021-09-28 18:10:09 +02:00
use std::time::Duration;
use crate::error::MeilisearchHttpError;
use actix_web::error::JsonPayloadError;
use analytics::Analytics;
use error::PayloadError;
use http::header::CONTENT_TYPE;
2021-03-15 18:11:10 +01:00
pub use option::Opt;
2021-03-10 11:56:51 +01:00
use actix_web::{web, HttpRequest};
2021-06-23 14:56:02 +02:00
2021-06-24 16:25:52 +02:00
use extractors::payload::PayloadConfig;
use meilisearch_auth::AuthController;
use meilisearch_lib::MeiliSearch;
2021-09-20 15:31:03 +02:00
2022-01-19 11:21:19 +01:00
pub static AUTOBATCHING_ENABLED: AtomicBool = AtomicBool::new(false);
2021-09-28 18:10:09 +02:00
pub fn setup_meilisearch(opt: &Opt) -> anyhow::Result<MeiliSearch> {
let mut meilisearch = MeiliSearch::builder();
2022-01-19 11:21:19 +01:00
// enable autobatching?
let _ = AUTOBATCHING_ENABLED.store(
opt.scheduler_options.enable_autobatching,
std::sync::atomic::Ordering::Relaxed,
);
2021-09-28 18:10:09 +02:00
meilisearch
.set_max_index_size(opt.max_index_size.get_bytes() as usize)
.set_max_task_store_size(opt.max_task_db_size.get_bytes() as usize)
// snapshot
2021-09-28 18:10:09 +02:00
.set_ignore_missing_snapshot(opt.ignore_missing_snapshot)
.set_ignore_snapshot_if_db_exists(opt.ignore_snapshot_if_db_exists)
.set_snapshot_interval(Duration::from_secs(opt.snapshot_interval_sec))
.set_snapshot_dir(opt.snapshot_dir.clone())
// dump
.set_ignore_missing_dump(opt.ignore_missing_dump)
.set_ignore_dump_if_db_exists(opt.ignore_dump_if_db_exists)
.set_dump_dst(opt.dumps_dir.clone());
2021-09-28 18:10:09 +02:00
if let Some(ref path) = opt.import_snapshot {
meilisearch.set_import_snapshot(path.clone());
}
if let Some(ref path) = opt.import_dump {
meilisearch.set_dump_src(path.clone());
}
if opt.schedule_snapshot {
meilisearch.set_schedule_snapshot();
}
2022-01-19 11:21:19 +01:00
meilisearch.build(
opt.db_path.clone(),
opt.indexer_options.clone(),
opt.scheduler_options.clone(),
)
2021-09-28 18:10:09 +02:00
}
pub fn configure_data(
config: &mut web::ServiceConfig,
data: MeiliSearch,
auth: AuthController,
opt: &Opt,
analytics: Arc<dyn Analytics>,
) {
2021-09-20 15:31:03 +02:00
let http_payload_size_limit = opt.http_payload_size_limit.get_bytes() as usize;
2021-06-23 13:21:48 +02:00
config
2021-06-24 14:22:12 +02:00
.app_data(data)
.app_data(auth)
.app_data(web::Data::from(analytics))
2021-06-23 13:21:48 +02:00
.app_data(
web::JsonConfig::default()
.content_type(|mime| mime == mime::APPLICATION_JSON)
.error_handler(|err, req: &HttpRequest| match err {
JsonPayloadError::ContentType => match req.headers().get(CONTENT_TYPE) {
Some(content_type) => MeilisearchHttpError::InvalidContentType(
content_type.to_str().unwrap_or("unknown").to_string(),
vec![mime::APPLICATION_JSON.to_string()],
)
.into(),
None => MeilisearchHttpError::MissingContentType(vec![
mime::APPLICATION_JSON.to_string(),
])
.into(),
},
err => PayloadError::from(err).into(),
}),
2021-06-23 13:21:48 +02:00
)
2021-06-23 13:58:22 +02:00
.app_data(PayloadConfig::new(http_payload_size_limit))
2021-06-23 13:21:48 +02:00
.app_data(
web::QueryConfig::default().error_handler(|err, _req| PayloadError::from(err).into()),
2021-06-23 13:21:48 +02:00
);
}
2021-03-10 11:56:51 +01:00
2021-06-23 13:21:48 +02:00
#[cfg(feature = "mini-dashboard")]
pub fn dashboard(config: &mut web::ServiceConfig, enable_frontend: bool) {
2021-06-23 13:55:16 +02:00
use actix_web::HttpResponse;
use static_files::Resource;
2021-04-21 13:49:21 +02:00
2021-06-23 14:48:33 +02:00
mod generated {
2021-06-23 13:21:48 +02:00
include!(concat!(env!("OUT_DIR"), "/generated.rs"));
}
2021-04-21 13:49:21 +02:00
2021-06-23 13:21:48 +02:00
if enable_frontend {
2021-06-23 14:48:33 +02:00
let generated = generated::generate();
2021-06-24 16:25:52 +02:00
// Generate routes for mini-dashboard assets
for (path, resource) in generated.into_iter() {
let Resource {
mime_type, data, ..
} = resource;
// Redirect index.html to /
if path == "index.html" {
config.service(web::resource("/").route(
web::get().to(move || HttpResponse::Ok().content_type(mime_type).body(data)),
));
} else {
config.service(web::resource(path).route(
2021-06-24 16:25:52 +02:00
web::get().to(move || HttpResponse::Ok().content_type(mime_type).body(data)),
));
2021-06-23 13:21:48 +02:00
}
2021-06-24 16:25:52 +02:00
}
2021-06-23 13:21:48 +02:00
} else {
2021-06-24 19:02:28 +02:00
config.service(web::resource("/").route(web::get().to(routes::running)));
2021-06-23 13:21:48 +02:00
}
}
#[cfg(not(feature = "mini-dashboard"))]
pub fn dashboard(config: &mut web::ServiceConfig, _enable_frontend: bool) {
2021-06-24 19:02:28 +02:00
config.service(web::resource("/").route(web::get().to(routes::running)));
2021-06-23 13:21:48 +02:00
}
2021-04-21 13:49:21 +02:00
2021-06-23 13:21:48 +02:00
#[macro_export]
macro_rules! create_app {
($data:expr, $auth:expr, $enable_frontend:expr, $opt:expr, $analytics:expr) => {{
2021-06-23 13:21:48 +02:00
use actix_cors::Cors;
use actix_web::middleware::TrailingSlash;
use actix_web::App;
use actix_web::{middleware, web};
use meilisearch_error::ResponseError;
use meilisearch_http::error::MeilisearchHttpError;
2021-07-05 14:29:20 +02:00
use meilisearch_http::routes;
use meilisearch_http::{configure_data, dashboard};
2021-04-21 13:49:21 +02:00
2021-06-23 13:21:48 +02:00
App::new()
.configure(|s| configure_data(s, $data.clone(), $auth.clone(), &$opt, $analytics))
2021-07-05 14:29:20 +02:00
.configure(routes::configure)
2021-06-23 13:21:48 +02:00
.configure(|s| dashboard(s, $enable_frontend))
.wrap(
2021-06-22 21:48:51 +02:00
Cors::default()
2021-06-23 14:48:33 +02:00
.send_wildcard()
2022-01-05 15:26:17 +01:00
.allow_any_header()
2021-06-23 14:48:33 +02:00
.allow_any_origin()
.allow_any_method()
.max_age(86_400), // 24h
2021-06-22 21:48:51 +02:00
)
2021-06-23 13:21:48 +02:00
.wrap(middleware::Logger::default())
.wrap(middleware::Compress::default())
2021-06-23 14:48:33 +02:00
.wrap(middleware::NormalizePath::new(
middleware::TrailingSlash::Trim,
))
2021-06-23 13:21:48 +02:00
}};
2021-03-10 11:56:51 +01:00
}