use std::collections::HashSet;
use v_individual_model::onto::individual::Individual;
use v_individual_model::onto::onto_impl::Onto;
use v_individual_model::onto::onto_index::OntoIndex;
use crate::storage::async_storage::{get_individual_from_db, AStorage};
use v_storage::VStorage;
use crate::v_api::common_type::ResultCode;
use chrono::Utc;
use v_queue::consumer::Consumer;
use v_queue::record::Mode;
pub const DATA_BASE_PATH: &str = "./data";
pub async fn c_load_onto(storage: &AStorage, onto: &mut Onto) -> bool {
let onto_index = OntoIndex::load();
info!("load {} onto elements", onto_index.len());
for id in onto_index.data.keys() {
if let Ok((mut indv, res)) = get_individual_from_db(id, "", storage, None).await {
if res == ResultCode::Ok {
onto.update(&mut indv);
}
}
}
info!("add to hierarchy {} elements", onto.relations.len());
let keys: Vec<String> = onto.relations.keys().cloned().collect();
for el in keys.iter() {
let mut buf: HashSet<String> = HashSet::new();
onto.get_subs(el, &mut buf);
if !buf.is_empty() {
onto.update_subs(el, &mut buf);
}
}
info!("end update subs");
true
}
pub fn load_onto(storage: &mut VStorage, onto: &mut Onto) -> bool {
let onto_index = OntoIndex::load();
info!("load {} onto elements", onto_index.len());
for id in onto_index.data.keys() {
let mut indv: Individual = Individual::default();
if storage.get_individual(id, &mut indv).is_ok() {
onto.update(&mut indv);
}
}
info!("add to hierarchy {} elements", onto.relations.len());
let keys: Vec<String> = onto.relations.keys().cloned().collect();
for el in keys.iter() {
let mut buf: HashSet<String> = HashSet::new();
onto.get_subs(el, &mut buf);
if !buf.is_empty() {
onto.update_subs(el, &mut buf);
}
}
info!("end update subs");
true
}
pub use v_module_queue::sys_sig_listener;
const MAIN_QUEUE_NAME: &str = "individuals-flow";
pub fn get_queue_status(id: &str) -> Individual {
let mut out_indv = Individual::default();
if let Some(consumer_name) = id.strip_prefix("srv:queue-state-") {
let base_path: &str = &(DATA_BASE_PATH.to_owned() + "/queue");
if let Ok(mut c) = Consumer::new_with_mode(base_path, consumer_name, MAIN_QUEUE_NAME, Mode::Read) {
c.open(false);
c.get_info();
if c.queue.get_info_of_part(c.id, false).is_ok() {
out_indv.set_id(id);
out_indv.add_uri("rdf:type", "v-s:AppInfo");
out_indv.add_datetime("v-s:created", Utc::now().timestamp());
out_indv.add_uri("srv:queue", &("srv:".to_owned() + consumer_name));
out_indv.add_integer("srv:total_count", c.queue.count_pushed as i64);
out_indv.add_integer("srv:current_count", c.count_popped as i64);
}
} else {
error!("fail open consumer {}", consumer_name);
}
}
out_indv
}
#[macro_export]
macro_rules! init_module_log {
($module_name:expr) => {{
use git_version::git_version;
use version::version;
init_log($module_name);
info!("{} {} {}", $module_name, version!(), git_version!());
}};
}