MeiliSearch/meilisearch-http/src/main.rs

use std::env::VarError::NotPresent;
use std::{env, thread};

use async_std::task;
use log::info;
use main_error::MainError;
use structopt::StructOpt;
use tide::middleware::RequestLogger;

use meilisearch_http::data::Data;
use meilisearch_http::option::Opt;
use meilisearch_http::routes;
use meilisearch_http::routes::index::index_update_callback;

use cors::Cors;

mod analytics;
mod cors;

#[cfg(target_os = "linux")]
#[global_allocator]
static ALLOC: jemallocator::Jemalloc = jemallocator::Jemalloc;

pub fn main() -> Result<(), MainError> {
    env_logger::init();

    let opt = Opt::from_args();
    let data = Data::new(opt.clone());

    if env::var("MEILI_NO_ANALYTICS") == Err(NotPresent) {
        thread::spawn(|| analytics::analytics_sender());
    }

    let data_cloned = data.clone();
    data.db.set_update_callback(Box::new(move |name, status| {
        index_update_callback(name, &data_cloned, status);
    }));

    let mut app = tide::with_state(data);

    app.middleware(Cors::new());
    app.middleware(RequestLogger::new());
    // app.middleware(tide_compression::Compression::new());
    // app.middleware(tide_compression::Decompression::new());

    routes::load_routes(&mut app);

    info!("Server HTTP enabled");

    task::block_on(async {
        app.listen(opt.http_addr).await.unwrap();
    });
    Ok(())
}