100 lines
2.4 KiB
Rust
100 lines
2.4 KiB
Rust
use std::ops::Deref;
|
|
use std::sync::Arc;
|
|
|
|
use meilisearch_core::{Database, DatabaseOptions};
|
|
use sha2::Digest;
|
|
use sysinfo::Pid;
|
|
|
|
use crate::index_update_callback;
|
|
use crate::option::Opt;
|
|
|
|
#[derive(Clone)]
|
|
pub struct Data {
|
|
inner: Arc<DataInner>,
|
|
}
|
|
|
|
impl Deref for Data {
|
|
type Target = DataInner;
|
|
|
|
fn deref(&self) -> &Self::Target {
|
|
&self.inner
|
|
}
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct DataInner {
|
|
pub db: Arc<Database>,
|
|
pub db_path: String,
|
|
pub api_keys: ApiKeys,
|
|
pub server_pid: Pid,
|
|
pub http_payload_size_limit: usize,
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct ApiKeys {
|
|
pub public: Option<String>,
|
|
pub private: Option<String>,
|
|
pub master: Option<String>,
|
|
}
|
|
|
|
impl ApiKeys {
|
|
pub fn generate_missing_api_keys(&mut self) {
|
|
if let Some(master_key) = &self.master {
|
|
if self.private.is_none() {
|
|
let key = format!("{}-private", master_key);
|
|
let sha = sha2::Sha256::digest(key.as_bytes());
|
|
self.private = Some(format!("{:x}", sha));
|
|
}
|
|
if self.public.is_none() {
|
|
let key = format!("{}-public", master_key);
|
|
let sha = sha2::Sha256::digest(key.as_bytes());
|
|
self.public = Some(format!("{:x}", sha));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Data {
|
|
pub fn new(opt: Opt) -> Data {
|
|
let db_path = opt.db_path.clone();
|
|
let server_pid = sysinfo::get_current_pid().unwrap();
|
|
|
|
let db_opt = DatabaseOptions {
|
|
main_map_size: opt.main_map_size,
|
|
update_map_size: opt.update_map_size,
|
|
};
|
|
|
|
let http_payload_size_limit = opt.http_payload_size_limit;
|
|
|
|
let db = Arc::new(Database::open_or_create(opt.db_path, db_opt).unwrap());
|
|
|
|
let mut api_keys = ApiKeys {
|
|
master: opt.master_key,
|
|
private: None,
|
|
public: None,
|
|
};
|
|
|
|
api_keys.generate_missing_api_keys();
|
|
|
|
let inner_data = DataInner {
|
|
db: db.clone(),
|
|
db_path,
|
|
api_keys,
|
|
server_pid,
|
|
http_payload_size_limit,
|
|
};
|
|
|
|
let data = Data {
|
|
inner: Arc::new(inner_data),
|
|
};
|
|
|
|
let callback_context = data.clone();
|
|
db.set_update_callback(Box::new(move |index_uid, status| {
|
|
index_update_callback(&index_uid, &callback_context, status);
|
|
}));
|
|
|
|
data
|
|
}
|
|
}
|
|
|