1072 lines
32 KiB
Rust
1072 lines
32 KiB
Rust
use async_trait::async_trait;
|
|
use colored::Colorize;
|
|
use core::fmt;
|
|
use flume::{bounded, unbounded, Receiver, Sender};
|
|
use interner::shared::{StringPool, VecStringPool};
|
|
use mize_macros::mize_part;
|
|
use std::any::Any;
|
|
use std::collections::HashMap;
|
|
use std::fs::create_dir;
|
|
use std::fs::File;
|
|
use std::ops::{Deref, DerefMut};
|
|
use std::path::{Path, PathBuf};
|
|
use std::sync::{Arc, Mutex};
|
|
use std::thread::panicking;
|
|
use std::{thread, vec};
|
|
use tracing::{debug, error, info, trace, warn, Instrument};
|
|
use uuid::Uuid;
|
|
|
|
use crate::config::ConfigOptNameAndMize;
|
|
use crate::config::{gather_config, ConfigOpt};
|
|
use crate::error::{IntoMizeResult, MizeError, MizeResult, MizeResultTrait};
|
|
use crate::id::{IntoMizeId, MizeId, Namespace};
|
|
use crate::instance::store::Store;
|
|
use crate::instance::subscription::Subscription;
|
|
use crate::instance::updater::Operation;
|
|
use crate::instance::updater::{updater_thread, updater_thread_async};
|
|
use crate::item::{Item, ItemData};
|
|
use crate::memstore::MemStore;
|
|
use crate::mize_err;
|
|
use crate::proto::MizeMessage;
|
|
|
|
use self::connection::{ConnListener, Connection};
|
|
use self::updater::handle_operation;
|
|
|
|
#[cfg(feature = "async")]
|
|
use tokio::runtime::Handle;
|
|
#[cfg(feature = "async")]
|
|
use tokio::runtime::Runtime;
|
|
|
|
use core::future::Future;
|
|
use std::thread::JoinHandle;
|
|
|
|
pub mod connection;
|
|
pub mod msg_thread;
|
|
pub mod store;
|
|
pub mod subscription;
|
|
pub mod updater;
|
|
|
|
#[cfg(test)]
|
|
mod tests;
|
|
|
|
static MSG_CHANNEL_SIZE: usize = 200;
|
|
|
|
static BUILD_TIME_CONFIG: &str = include_str!(std::env!("MIZE_BUILD_CONFIG"));
|
|
|
|
/// The Instance type is the heart of the mize system
|
|
#[derive(Clone)]
|
|
pub struct Mize {
|
|
// a bit a lot of Mutexes isn't it???
|
|
pub(crate) store: Arc<Mutex<Box<dyn Store>>>,
|
|
connections: Arc<Mutex<Vec<Connection>>>,
|
|
next_con_id: Arc<Mutex<u64>>,
|
|
subs: Arc<Mutex<HashMap<MizeId, Vec<Subscription>>>>,
|
|
pub(crate) id_pool: Arc<Mutex<VecStringPool>>,
|
|
pub(crate) namespace_pool: Arc<Mutex<StringPool>>,
|
|
|
|
pub(crate) listeners: Arc<
|
|
Mutex<HashMap<String, Arc<dyn Fn(MizeMessage, &mut Mize) -> MizeResult<()> + Send + Sync>>>,
|
|
>,
|
|
|
|
pub parts: Arc<Mutex<HashMap<&'static str, Option<Box<dyn MizePart + Sync + Send>>>>>,
|
|
//pub(crate) platform: Arc<Mutex<Box<dyn MizePlatform>>>, // should we do that instead of all the #cfg(faturs=targt-smth) in lib.rs
|
|
part_names: Arc<Mutex<Vec<&'static str>>>, // exists, so that we know the ordering in which parts were loaded
|
|
|
|
pub(crate) config_opts: Arc<Mutex<HashMap<String, ConfigOpt>>>,
|
|
|
|
// the namespace the instance operates in
|
|
pub(crate) namespace: Arc<Mutex<Namespace>>,
|
|
|
|
// the namespace of the instance itself
|
|
// TODO: set to a random uuid
|
|
pub(crate) self_namespace: Arc<Mutex<Namespace>>,
|
|
pub(crate) op_tx: Sender<Operation>,
|
|
threads: Arc<Mutex<Vec<(u32, String, Option<JoinHandle<MizeResult<()>>>)>>>,
|
|
next_thread_id: Arc<Mutex<u32>>,
|
|
give_msg_wait: Arc<Mutex<HashMap<MizeId, Vec<Sender<ItemData>>>>>,
|
|
create_msg_wait: Arc<Mutex<Option<Sender<MizeId>>>>,
|
|
|
|
#[cfg(feature = "async")]
|
|
pub(crate) runtime: Arc<Mutex<Runtime>>,
|
|
}
|
|
|
|
// the public Mize API
|
|
|
|
pub struct InstanceRef {
|
|
inner: Arc<Mutex<Mize>>,
|
|
}
|
|
|
|
pub struct InstanceAsync {
|
|
inner: Arc<Mutex<Mize>>,
|
|
}
|
|
|
|
pub struct DynMizePartGuard {
|
|
pub mize: Mize,
|
|
pub part: Option<Box<dyn MizePart + Send + Sync>>,
|
|
}
|
|
|
|
pub struct MizePartGuard<T: MizePart + Send + Sync + 'static> {
|
|
pub mize: Mize,
|
|
pub part: Option<T>,
|
|
}
|
|
|
|
impl<T: MizePart + Send + Sync> Deref for MizePartGuard<T> {
|
|
type Target = T;
|
|
|
|
fn deref(&self) -> &Self::Target {
|
|
&*self.part.as_ref().unwrap()
|
|
}
|
|
}
|
|
|
|
impl<T: MizePart + Send + Sync> DerefMut for MizePartGuard<T> {
|
|
fn deref_mut(&mut self) -> &mut Self::Target {
|
|
&mut *self.part.as_mut().unwrap()
|
|
}
|
|
}
|
|
|
|
impl<T: MizePart + Send + Sync> Drop for MizePartGuard<T> {
|
|
fn drop(&mut self) {
|
|
if self.part.is_none() {
|
|
return;
|
|
}
|
|
let part = self.part.take().unwrap();
|
|
self.mize.give_back_part(Box::new(part));
|
|
}
|
|
}
|
|
|
|
impl Deref for DynMizePartGuard {
|
|
type Target = dyn MizePart + Send + Sync;
|
|
|
|
fn deref(&self) -> &Self::Target {
|
|
&*self.part.as_deref().unwrap()
|
|
}
|
|
}
|
|
|
|
impl DerefMut for DynMizePartGuard {
|
|
fn deref_mut(&mut self) -> &mut Self::Target {
|
|
&mut *self.part.as_deref_mut().unwrap()
|
|
}
|
|
}
|
|
|
|
impl Drop for DynMizePartGuard {
|
|
fn drop(&mut self) {
|
|
if self.part.is_none() {
|
|
return;
|
|
}
|
|
self.mize.give_back_part(self.part.take().unwrap());
|
|
}
|
|
}
|
|
|
|
pub trait MizePartCreateGenerated: Sized {
|
|
fn create_generated(mize: Mize) -> Self;
|
|
}
|
|
pub trait MizePartGenerated {
|
|
fn name_generated(&self) -> &'static str;
|
|
fn get_mize_generated(&mut self) -> &mut Mize;
|
|
fn as_any_generated(&self) -> &dyn Any;
|
|
fn as_any_mut_generated(&mut self) -> &mut dyn Any;
|
|
fn into_any_generated(self: Box<Self>) -> Box<dyn Any>;
|
|
}
|
|
|
|
pub trait MizePartCreate: MizePartCreateGenerated + MizePart + 'static {
|
|
fn create(mize: &mut Mize) -> MizeResult<Box<dyn MizePart>> {
|
|
Ok(Box::new(Self::create_generated(mize.clone())))
|
|
}
|
|
}
|
|
|
|
pub trait MizePartCreateThisSucks: MizePartCreate + MizePart + 'static {
|
|
fn create_this_sucks(mize: &mut Mize) -> MizeResult<Box<dyn MizePart>> {
|
|
Self::create(mize)
|
|
}
|
|
}
|
|
|
|
pub struct EmptyPart {
|
|
pub(crate) mize: Mize,
|
|
}
|
|
impl MizePartGenerated for EmptyPart {
|
|
fn name_generated(&self) -> &'static str {
|
|
"empty"
|
|
}
|
|
fn get_mize_generated(&mut self) -> &mut Mize {
|
|
&mut self.mize
|
|
}
|
|
fn as_any_generated(&self) -> &dyn Any {
|
|
self
|
|
}
|
|
fn as_any_mut_generated(&mut self) -> &mut dyn Any {
|
|
self
|
|
}
|
|
fn into_any_generated(self: Box<Self>) -> Box<dyn Any> {
|
|
self
|
|
}
|
|
}
|
|
|
|
impl MizePart for EmptyPart {}
|
|
|
|
#[async_trait]
|
|
pub trait MizePart: MizePartGenerated {
|
|
fn name(&self) -> &'static str {
|
|
self.name_generated()
|
|
}
|
|
fn get_mize(&mut self) -> &mut Mize {
|
|
self.get_mize_generated()
|
|
}
|
|
fn as_any(&self) -> &dyn Any {
|
|
self.as_any_generated()
|
|
}
|
|
fn as_any_mut(&mut self) -> &mut dyn Any {
|
|
self.as_any_mut_generated()
|
|
}
|
|
fn into_any(self: Box<Self>) -> Box<dyn Any> {
|
|
self.into_any_generated()
|
|
}
|
|
fn init(&mut self, mize: &mut Mize) -> MizeResult<()> {
|
|
Ok(())
|
|
}
|
|
fn run(&mut self, mize: &mut Mize) -> MizeResult<()> {
|
|
Ok(())
|
|
}
|
|
fn deps(&self) -> &'static [&'static str] {
|
|
&[]
|
|
}
|
|
fn opts(&self, mize: &mut Mize) {
|
|
()
|
|
}
|
|
async fn async_run(&mut self, mize: &mut Mize) -> MizeResult<()> {
|
|
Ok(())
|
|
}
|
|
async fn async_init(&mut self, mize: &mut Mize) -> MizeResult<()> {
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl Mize {
|
|
pub fn empty() -> MizeResult<Mize> {
|
|
let id_pool = Arc::new(Mutex::new(VecStringPool::default()));
|
|
let namespace_pool_raw = StringPool::default();
|
|
let connections = Arc::new(Mutex::new(Vec::new()));
|
|
let subs = Arc::new(Mutex::new(HashMap::new()));
|
|
let (op_tx, op_rx) = unbounded();
|
|
let give_msg_wait = Arc::new(Mutex::new(HashMap::new()));
|
|
let create_msg_wait = Arc::new(Mutex::new(None));
|
|
let namespace = Arc::new(Mutex::new(Namespace(
|
|
namespace_pool_raw.get("mize.default.namespace"),
|
|
)));
|
|
let self_namespace = Arc::new(Mutex::new(Namespace(
|
|
namespace_pool_raw.get("mize.default.namespace"),
|
|
)));
|
|
|
|
let mut instance = Mize {
|
|
store: Arc::new(Mutex::new(Box::new(MemStore::new()))),
|
|
parts: Arc::new(Mutex::new(HashMap::new())),
|
|
part_names: Arc::new(Mutex::new(Vec::new())),
|
|
config_opts: Arc::new(Mutex::new(HashMap::new())),
|
|
listeners: Arc::new(Mutex::new(HashMap::new())),
|
|
connections,
|
|
subs,
|
|
id_pool,
|
|
namespace,
|
|
self_namespace,
|
|
op_tx,
|
|
namespace_pool: Arc::new(Mutex::new(namespace_pool_raw)),
|
|
threads: Arc::new(Mutex::new(Vec::new())),
|
|
next_thread_id: Arc::new(Mutex::new(0)),
|
|
next_con_id: Arc::new(Mutex::new(1)),
|
|
give_msg_wait,
|
|
create_msg_wait,
|
|
|
|
#[cfg(feature = "async")]
|
|
#[cfg(not(feature = "target-js"))]
|
|
runtime: Arc::new(Mutex::new(
|
|
Runtime::new().mres_msg("Could not create async runtime")?,
|
|
)),
|
|
};
|
|
|
|
#[cfg(feature = "target-os")]
|
|
{
|
|
let instance_clone = instance.clone();
|
|
let op_rx_clone = op_rx.clone();
|
|
let closure = move || updater_thread(op_rx_clone, &instance_clone);
|
|
instance.spawn_background("updater_thread", closure)?;
|
|
|
|
let instance_clone_two = instance.clone();
|
|
let closure_two = move || updater_thread(op_rx, &instance_clone_two);
|
|
instance.spawn_background("updater_thread", closure_two)?;
|
|
}
|
|
|
|
// set up async update "threads" when using wasm
|
|
#[cfg(feature = "target-wasm ")]
|
|
{
|
|
let instance_clone = instance.clone();
|
|
let op_rx_clone = op_rx.clone();
|
|
wasm_bindgen_futures::spawn_local(updater_thread_async(op_rx_clone, instance_clone));
|
|
|
|
let instance_clone_two = instance.clone();
|
|
wasm_bindgen_futures::spawn_local(updater_thread_async(op_rx, instance_clone_two));
|
|
}
|
|
|
|
// will move the msg stuff into it's own thread
|
|
// like this it can deadlock.... if a msg waits on an operation to complete
|
|
//let msg_instance_clone = instance.clone();
|
|
//let msg_closure = move || msg_thread(op_rx, instance_clone);
|
|
//instance.spawn("msg_thread", closure)?;
|
|
|
|
#[cfg(feature = "target-wasm ")]
|
|
console_log!("before loading build time config");
|
|
|
|
// load the config from build time
|
|
let config = ItemData::from_toml(BUILD_TIME_CONFIG)?;
|
|
instance.set_blocking("0", config);
|
|
|
|
#[cfg(feature = "target-wasm ")]
|
|
console_log!("after loading build time config");
|
|
|
|
return Ok(instance);
|
|
}
|
|
|
|
pub fn new() -> MizeResult<Mize> {
|
|
trace!("[ {} ] Instance::new()", "CALL".yellow());
|
|
|
|
let mut instance = Mize::empty()?;
|
|
|
|
instance.init()?;
|
|
|
|
debug!(
|
|
"instance inited with config: {}",
|
|
instance.get("0/config")?.as_data_full()?
|
|
);
|
|
|
|
return Ok(instance);
|
|
}
|
|
|
|
pub fn get_part(&mut self, name: &str) -> MizeResult<DynMizePartGuard> {
|
|
match self.parts.lock().unwrap().get_mut(name) {
|
|
Some(part) => Ok(DynMizePartGuard {
|
|
part: Some(part.take().unwrap()),
|
|
mize: self.clone(),
|
|
}),
|
|
None => Err(mize_err!("Part '{name}' not found or currently taken")),
|
|
}
|
|
}
|
|
pub fn has_part(&mut self, name: &str) -> bool {
|
|
self.parts.lock().unwrap().contains_key(name)
|
|
}
|
|
pub fn add_part(&mut self, part: Box<dyn MizePart + Send + Sync>) -> MizeResult<()> {
|
|
self.part_names.lock().unwrap().push(part.name());
|
|
self.parts.lock().unwrap().insert(part.name(), Some(part));
|
|
Ok(())
|
|
}
|
|
pub fn register_part(&mut self, part: Box<dyn MizePart + Send + Sync>) -> MizeResult<()> {
|
|
self.part_names.lock().unwrap().push(part.name());
|
|
self.parts.lock().unwrap().insert(part.name(), Some(part));
|
|
Ok(())
|
|
}
|
|
pub fn add_listener(
|
|
&mut self,
|
|
name: &str,
|
|
handler: Box<dyn Fn(MizeMessage, &mut Mize) -> MizeResult<()> + Send + Sync>,
|
|
) -> MizeResult<()> {
|
|
println!("adding listener: {}", name);
|
|
let name = name.to_owned();
|
|
let mut listeners = self.listeners.lock().unwrap();
|
|
listeners.insert(name, Arc::new(handler));
|
|
|
|
Ok(())
|
|
}
|
|
fn part_names(&mut self) -> Vec<&'static str> {
|
|
self.part_names.lock().unwrap().clone()
|
|
}
|
|
|
|
pub(crate) fn give_back_part(&mut self, part: Box<dyn MizePart + Send + Sync>) {
|
|
self.parts.lock().unwrap().insert(part.name(), Some(part));
|
|
}
|
|
|
|
pub fn init(&mut self) -> MizeResult<()> {
|
|
// gather config options from all parts
|
|
gather_config(self)?;
|
|
|
|
// platform specific init code
|
|
crate::platform::any::instance_init(self)?;
|
|
|
|
// end of platform specific init code
|
|
|
|
// load the modules, as specified in the parts config
|
|
match self
|
|
.get_config("parts")
|
|
.unwrap_or(ItemData::new())
|
|
.value_string()
|
|
{
|
|
Ok(parts_to_load) => {
|
|
for part_name in parts_to_load.split(" ") {
|
|
if part_name.trim().is_empty() {
|
|
continue;
|
|
}
|
|
self.load_part(part_name)?;
|
|
}
|
|
}
|
|
Err(err) => {
|
|
// no parts to load
|
|
}
|
|
}
|
|
|
|
debug!("MIZE INIT DONE");
|
|
Ok(())
|
|
}
|
|
|
|
pub fn with_config(config: ItemData) -> MizeResult<Mize> {
|
|
trace!("[ {} ] Instance::with_config()", "CALL".yellow());
|
|
trace!("config: {}", config);
|
|
println!("here ....");
|
|
|
|
let mut mize = Mize::empty()?;
|
|
|
|
let mut opts = mize.config_opts.lock().unwrap();
|
|
let mut config = config.clone();
|
|
for mut path in config.get_paths_recursive()? {
|
|
if path.starts_with("/") {
|
|
path.remove(0);
|
|
}
|
|
let path_split = path
|
|
.split("/")
|
|
.map(|s| s.to_owned())
|
|
.collect::<Vec<String>>();
|
|
let path_string = path_split.join(".");
|
|
if let Ok(val) = config.get_path(path_split) {
|
|
let opt = ConfigOpt {
|
|
name: path.clone(),
|
|
val: Some(val),
|
|
thunk: None,
|
|
};
|
|
opts.insert(path_string, opt);
|
|
}
|
|
}
|
|
drop(opts);
|
|
|
|
mize.init()?;
|
|
|
|
Ok(mize)
|
|
}
|
|
|
|
pub fn migrate_to_store(&self, new_store: Box<dyn Store>) -> MizeResult<()> {
|
|
info!("MIGRATING");
|
|
let mut old_store = self.store.lock()?;
|
|
|
|
let id = self.id_from_string("0".to_owned())?;
|
|
let inst_data = old_store.get_value_data_full(id.clone())?;
|
|
new_store.set(id, inst_data.to_owned())?;
|
|
|
|
for id in old_store.id_iter()? {
|
|
let data = old_store.get_value_data_full(self.id_from_string(id?)?)?;
|
|
|
|
let id_of_new_store = new_store.new_id()?;
|
|
new_store.set(self.id_from_string(id_of_new_store)?, data.to_owned())?;
|
|
}
|
|
|
|
*old_store = new_store;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn new_item(&self) -> MizeResult<Item> {
|
|
if self.get_namespace()? != self.get_self_namespace()? {
|
|
// need to send create msg and wait for it
|
|
let mut connection = self.get_connection_by_ns(self.get_namespace()?)?;
|
|
|
|
let msg = MizeMessage::new_create(connection.id);
|
|
|
|
connection.send(msg)?;
|
|
|
|
let (tx, rx) = bounded::<MizeId>(1);
|
|
|
|
let mut msg_wait_inner = self.create_msg_wait.lock()?;
|
|
*msg_wait_inner = Some(tx);
|
|
drop(msg_wait_inner);
|
|
|
|
let id = rx.recv()?;
|
|
println!("new_item namespace: {:?}", id.namespace());
|
|
|
|
return Ok(Item::new(id, self));
|
|
}
|
|
|
|
let store_inner = self.store.lock()?;
|
|
let id = self.id_from_string(store_inner.new_id()?)?;
|
|
return Ok(Item::new(id, self));
|
|
}
|
|
|
|
pub fn get<I: IntoMizeId>(&self, id: I) -> MizeResult<Item> {
|
|
let id = id.to_mize_id(self)?;
|
|
return Ok(Item::new(id, &self));
|
|
}
|
|
|
|
pub fn set<I: IntoMizeId, V: Into<ItemData>>(&self, id: I, value: V) -> MizeResult<()> {
|
|
let id = id.to_mize_id(self)?;
|
|
self.op_tx.send(Operation::Set(id, value.into(), None));
|
|
Ok(())
|
|
}
|
|
|
|
pub fn set_blocking<I: IntoMizeId, V: Into<ItemData>>(
|
|
&self,
|
|
id: I,
|
|
value: V,
|
|
) -> MizeResult<()> {
|
|
handle_operation(
|
|
&mut Operation::Set(id.to_mize_id(self)?, value.into(), None),
|
|
self,
|
|
)?;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn sub<I: IntoMizeId>(&self, id: I, sub: Subscription) -> MizeResult<()> {
|
|
let mut subs_inner = self.subs.lock()?;
|
|
let id = id.to_mize_id(self)?;
|
|
match subs_inner.get_mut(&id) {
|
|
Some(vec) => {
|
|
vec.push(sub);
|
|
}
|
|
None => {
|
|
subs_inner.insert(id.clone(), vec![sub]);
|
|
}
|
|
}
|
|
|
|
// if we are not the owner of this item, send a sub msg to them
|
|
if !self.we_are_namespace()? {
|
|
let con = self.get_connection_by_ns(id.namespace())?;
|
|
let msg = MizeMessage::new_sub(id, con.id);
|
|
con.send(msg)?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn new_id<T: IntoMizeId>(&self, value: T) -> MizeResult<MizeId> {
|
|
value.to_mize_id(self)
|
|
}
|
|
|
|
pub fn id_from_string(&self, string: String) -> MizeResult<MizeId> {
|
|
let vec_string: Vec<String> = string
|
|
.split("/")
|
|
.map(|v| v.to_owned())
|
|
.filter(|el| el.as_str() != "")
|
|
.collect();
|
|
self.id_from_vec_string(vec_string)
|
|
}
|
|
|
|
pub fn id_from_vec_string(&self, mut vec_string: Vec<String>) -> MizeResult<MizeId> {
|
|
let id_pool_inner = self.id_pool.lock()?;
|
|
let namespace_inner = self.namespace.lock()?;
|
|
let first_el = vec_string
|
|
.first_mut()
|
|
.ok_or(mize_err!("MizeId was empty"))?;
|
|
|
|
let id = if first_el.contains(":") {
|
|
// first el is a namespace + store_part
|
|
let new_first_el = first_el.clone();
|
|
let vec: Vec<&str> = new_first_el.split(":").collect();
|
|
let ns_part = vec
|
|
.iter()
|
|
.nth(0)
|
|
.ok_or(mize_err!("should really not happen"))?;
|
|
let store_part = vec.iter().nth(1).ok_or(mize_err!(
|
|
"mizeid was like 'namespace:/hi', why are you doing that"
|
|
))?;
|
|
*first_el = store_part.to_owned().to_owned();
|
|
|
|
MizeId {
|
|
path: id_pool_inner.get(vec_string),
|
|
namespace: self.namespace_from_string(ns_part.to_owned().to_owned())?,
|
|
}
|
|
} else {
|
|
MizeId {
|
|
path: id_pool_inner.get(vec_string),
|
|
namespace: namespace_inner.clone(),
|
|
}
|
|
};
|
|
trace!("new MizeId made: {:?}", id);
|
|
|
|
Ok(id)
|
|
}
|
|
|
|
pub fn load_part(&mut self, name: &str) -> MizeResult<()> {
|
|
if let Ok(path) = self
|
|
.get_config(&format!("part_dir.{}", name))?
|
|
.value_string()
|
|
{
|
|
crate::platform::any::load_part(self, name, Some(path))
|
|
} else {
|
|
crate::platform::any::load_part(self, name, None)
|
|
}
|
|
}
|
|
|
|
pub fn fetch_part(&mut self, name: &str) -> MizeResult<String> {
|
|
// platform specific way to fetch a module
|
|
crate::platform::any::fetch_part(self, name)
|
|
}
|
|
|
|
pub fn load_part_at(&mut self, name: &str, path: String) -> MizeResult<()> {
|
|
// platform specific init code
|
|
crate::platform::any::load_part(self, name, Some(path))
|
|
}
|
|
|
|
pub fn namespace_from_string(&self, ns_str: String) -> MizeResult<Namespace> {
|
|
let namespace_pool_inner = self.namespace_pool.lock()?;
|
|
let namespace = Namespace(namespace_pool_inner.get(ns_str));
|
|
Ok(namespace)
|
|
}
|
|
|
|
pub fn set_namespace(&self, ns: Namespace) -> MizeResult<()> {
|
|
let mut namespace_inner = self.namespace.lock()?;
|
|
*namespace_inner = ns;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn get_namespace(&self) -> MizeResult<Namespace> {
|
|
let mut namespace_inner = self.namespace.lock()?;
|
|
return Ok(namespace_inner.clone());
|
|
}
|
|
|
|
pub fn get_self_namespace(&self) -> MizeResult<Namespace> {
|
|
let mut self_namespace_inner = self.self_namespace.lock()?;
|
|
return Ok(self_namespace_inner.clone());
|
|
}
|
|
|
|
pub fn we_are_namespace(&self) -> MizeResult<bool> {
|
|
Ok(self.get_namespace()? == self.get_self_namespace()?)
|
|
}
|
|
|
|
pub fn add_conn_listener<T: ConnListener + 'static>(&mut self, listener: T) -> MizeResult<()> {
|
|
let mut instance_clone = self.clone();
|
|
self.spawn_background("some_listener", move || listener.listen(instance_clone));
|
|
Ok(())
|
|
}
|
|
|
|
pub fn new_connection(&self, tx: Sender<MizeMessage>) -> MizeResult<u64> {
|
|
let mut conn_inner = self.connections.lock()?;
|
|
let mut next_con_id = self.next_con_id.lock()?;
|
|
let old_next_con_id = *next_con_id;
|
|
|
|
let connection = Connection {
|
|
id: next_con_id.to_owned(),
|
|
tx,
|
|
ns: None,
|
|
};
|
|
conn_inner.push(connection);
|
|
*next_con_id += 1;
|
|
Ok(old_next_con_id)
|
|
}
|
|
|
|
pub fn new_connection_join_namespace(&self, tx: Sender<MizeMessage>) -> MizeResult<u64> {
|
|
let conn_id = self.new_connection(tx)?;
|
|
|
|
let ns_of_peer_str = self
|
|
.get(format!(
|
|
"inst/con_by_id/{}/peer/self/config/namespace",
|
|
conn_id
|
|
))?
|
|
.value_string()?;
|
|
let ns_of_peer = self.namespace_from_string(ns_of_peer_str)?;
|
|
|
|
self.connection_set_namespace(conn_id, ns_of_peer.clone());
|
|
self.set_namespace(ns_of_peer);
|
|
|
|
Ok(conn_id)
|
|
}
|
|
|
|
pub fn connection_set_namespace(&self, conn_id: u64, namespace: Namespace) -> MizeResult<()> {
|
|
let mut connection = self.get_connection(conn_id)?;
|
|
connection.ns = Some(namespace);
|
|
self.set_connection(conn_id, connection);
|
|
Ok(())
|
|
}
|
|
|
|
pub fn got_msg(&self, msg: MizeMessage) -> MizeResult<()> {
|
|
Ok(self.op_tx.send(Operation::Msg(msg))?)
|
|
}
|
|
|
|
pub fn report_err(&self, err: MizeError) {
|
|
#[cfg(feature = "target-js")] unsafe {
|
|
use wasm_bindgen::prelude::wasm_bindgen;
|
|
#[wasm_bindgen]
|
|
extern "C" {
|
|
#[wasm_bindgen(js_namespace = console)]
|
|
fn log(s: &str);
|
|
}
|
|
|
|
macro_rules! console_log {
|
|
// Note that this is using the `log` function imported above during
|
|
// `bare_bones`
|
|
($($t:tt)*) => ({ log(&format_args!($($t)*).to_string())})
|
|
}
|
|
console_log!("reported_err: {:?}", err);
|
|
}
|
|
err.log();
|
|
}
|
|
|
|
fn set_connection(&self, conn_id: u64, new_connection: Connection) -> MizeResult<()> {
|
|
let mut conn_inner = self.connections.lock()?;
|
|
|
|
for connection in conn_inner.iter_mut() {
|
|
if connection.id == conn_id {
|
|
*connection = new_connection;
|
|
return Ok(());
|
|
}
|
|
}
|
|
|
|
return Err(mize_err!(
|
|
"Connection with id {} not known to instance",
|
|
conn_id
|
|
));
|
|
Ok(())
|
|
}
|
|
|
|
pub fn get_connection(&self, conn_id: u64) -> MizeResult<Connection> {
|
|
let mut conn_inner = self.connections.lock()?;
|
|
|
|
for connection in conn_inner.iter() {
|
|
if connection.id == conn_id {
|
|
return Ok(connection.clone());
|
|
}
|
|
}
|
|
|
|
return Err(mize_err!(
|
|
"Connection with id {} not known to instance",
|
|
conn_id
|
|
));
|
|
}
|
|
|
|
pub fn get_connection_by_ns(&self, ns: Namespace) -> MizeResult<Connection> {
|
|
let mut conn_inner = self.connections.lock()?;
|
|
|
|
for connection in conn_inner.iter() {
|
|
if connection.ns == Some(ns.clone()) {
|
|
return Ok(connection.clone());
|
|
}
|
|
}
|
|
|
|
return Err(mize_err!(
|
|
"Connection with namespace {} not known to instance",
|
|
ns.as_string()
|
|
));
|
|
}
|
|
|
|
pub fn spawn_and_wait(
|
|
&mut self,
|
|
name: &str,
|
|
func: impl FnOnce() -> MizeResult<()> + Send + 'static,
|
|
) -> MizeResult<()> {
|
|
let mize_clone = self.clone();
|
|
let mut threads_inner = self.threads.lock()?;
|
|
let mut next_thread_id = self.next_thread_id.lock()?;
|
|
|
|
let my_thread_id_no_mutex_guard = *next_thread_id;
|
|
let thread_mutex = self.threads.clone();
|
|
let name_to_move = name.to_owned();
|
|
let to_spawn = move || -> MizeResult<()> {
|
|
debug!("spawning thread: {}", name_to_move);
|
|
let my_thread_id = my_thread_id_no_mutex_guard;
|
|
|
|
if let Err(err) = func() {
|
|
mize_clone.report_err(err);
|
|
}
|
|
|
|
//let mut threads_inner = thread_mutex.lock()?;
|
|
/*
|
|
*threads_inner = threads_inner
|
|
.clone()
|
|
.into_iter()
|
|
.filter(|el| match el {
|
|
(my_thread_id, _) => false,
|
|
(_, _) => true,
|
|
})
|
|
.collect();
|
|
*/
|
|
debug!("thread '{}' stopped", name_to_move);
|
|
Ok(())
|
|
};
|
|
|
|
*next_thread_id += 1;
|
|
|
|
#[cfg(feature = "target-os")]
|
|
let handle = thread::spawn(move || to_spawn());
|
|
|
|
#[cfg(feature = "target-os")]
|
|
threads_inner.push((*next_thread_id, name.to_owned(), Some(handle)));
|
|
|
|
#[cfg(feature = "target-wasm ")]
|
|
{
|
|
//console_log!("in instance::spawn with wasm target")
|
|
}
|
|
//NOT WELL SUPPORTED
|
|
//crate::platform::wasm::wasm_spawn(to_spawn)?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn spawn_background(
|
|
&mut self,
|
|
name: &str,
|
|
func: impl FnOnce() -> MizeResult<()> + Send + 'static,
|
|
) -> MizeResult<()> {
|
|
let mut threads_inner = self.threads.lock()?;
|
|
let mut next_thread_id = self.next_thread_id.lock()?;
|
|
|
|
let my_thread_id_no_mutex_guard = *next_thread_id;
|
|
let thread_mutex = self.threads.clone();
|
|
let name_to_move = name.to_owned();
|
|
let to_spawn = move || -> () {
|
|
debug!("spawning thread: {}", name_to_move);
|
|
let my_thread_id = my_thread_id_no_mutex_guard;
|
|
|
|
let res = func();
|
|
if let Err(e) = res {
|
|
e.log();
|
|
}
|
|
|
|
//let mut threads_inner = thread_mutex.lock()?;
|
|
/*
|
|
*threads_inner = threads_inner
|
|
.clone()
|
|
.into_iter()
|
|
.filter(|el| match el {
|
|
(my_thread_id, _) => false,
|
|
(_, _) => true,
|
|
})
|
|
.collect();
|
|
*/
|
|
debug!("thread '{}' stopped", name_to_move);
|
|
};
|
|
|
|
*next_thread_id += 1;
|
|
|
|
#[cfg(feature = "target-os")]
|
|
thread::spawn(move || to_spawn());
|
|
|
|
threads_inner.push((*next_thread_id, name.to_owned(), None));
|
|
|
|
#[cfg(feature = "target-wasm ")]
|
|
{
|
|
//console_log!("in instance::spawn with wasm target")
|
|
}
|
|
//NOT WELL SUPPORTED
|
|
//crate::platform::wasm::wasm_spawn(to_spawn)?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn give_msg_wait(&self, id: MizeId) -> MizeResult<ItemData> {
|
|
let mut give_msg_wait_inner = self.give_msg_wait.lock()?;
|
|
|
|
let (tx, rx) = bounded::<ItemData>(1);
|
|
|
|
let vec = match give_msg_wait_inner.get_mut(&id) {
|
|
Some(vec) => vec,
|
|
None => {
|
|
give_msg_wait_inner.insert(id.clone(), Vec::new());
|
|
give_msg_wait_inner.get_mut(&id).unwrap()
|
|
}
|
|
};
|
|
|
|
vec.push(tx);
|
|
|
|
// so that another thread can also give_msg_wait(), while we wait in the recv() of rx
|
|
drop(vec);
|
|
drop(give_msg_wait_inner);
|
|
|
|
let data = rx.recv()?;
|
|
|
|
return Ok(data);
|
|
}
|
|
|
|
#[cfg(feature = "async")]
|
|
pub fn spawn_async<F: Future<Output = impl Send + 'static> + Send + 'static>(
|
|
&mut self,
|
|
name: &str,
|
|
func: F,
|
|
) -> MizeResult<()> {
|
|
let mut threads_inner = self.threads.lock()?;
|
|
let mut next_thread_id = self.next_thread_id.lock()?;
|
|
|
|
threads_inner.push((*next_thread_id, name.to_owned(), None));
|
|
let runtime_inner = self.runtime.lock().unwrap();
|
|
runtime_inner.spawn(func);
|
|
|
|
*next_thread_id += 1;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(feature = "async")]
|
|
pub fn async_get_handle(&self) -> Handle {
|
|
let runtime_inner = self.runtime.lock().unwrap();
|
|
let handle = runtime_inner.handle().to_owned();
|
|
handle
|
|
}
|
|
|
|
#[cfg(feature = "async")]
|
|
pub fn spawn_async_blocking<
|
|
F: Future<Output = impl Send + Sync + 'static> + Send + Sync + 'static,
|
|
>(
|
|
&mut self,
|
|
name: &str,
|
|
func: F,
|
|
) -> F::Output {
|
|
let mut threads_inner = self
|
|
.threads
|
|
.lock()
|
|
.expect("mutex lock failed in spawn_async_blocking");
|
|
let mut next_thread_id = self
|
|
.next_thread_id
|
|
.lock()
|
|
.expect("mutex lock failed in spawn_async_blocking");
|
|
|
|
threads_inner.push((*next_thread_id, name.to_owned(), None));
|
|
let runtime_inner = self.runtime.lock().unwrap();
|
|
let handle = runtime_inner.handle().to_owned();
|
|
*next_thread_id += 1;
|
|
|
|
drop(runtime_inner);
|
|
drop(threads_inner);
|
|
drop(next_thread_id);
|
|
|
|
let result = handle.block_on(func);
|
|
|
|
return result;
|
|
}
|
|
|
|
pub fn wait(&self) {
|
|
info!("Instance main thread waiting");
|
|
loop {
|
|
thread::sleep_ms(10000000)
|
|
}
|
|
}
|
|
|
|
pub fn report_error(err: MizeError) {
|
|
err.log();
|
|
}
|
|
|
|
pub fn get_config(&mut self, name: &str) -> MizeResult<ItemData> {
|
|
let mut config_opts = self.config_opts.lock().unwrap();
|
|
|
|
let opt = config_opts
|
|
.get_mut(name)
|
|
.ok_or_else(|| mize_err!("Config opt for {name} not in config_opts"))?;
|
|
|
|
// get the cached value
|
|
if let Some(val) = opt.val.clone() {
|
|
//println!("get_config: {name} has cached value: {val}");
|
|
return Ok(val);
|
|
}
|
|
|
|
// Evaluate the thunk
|
|
let thunk = opt
|
|
.thunk
|
|
.take()
|
|
.ok_or_else(|| mize_err!("ConfigOpt {name} does not have a thunk"))?;
|
|
let result = thunk();
|
|
|
|
// Cache the result
|
|
opt.val = Some(result.clone());
|
|
|
|
return Ok(result);
|
|
}
|
|
|
|
pub fn new_opt(&mut self, name: &str) -> ConfigOptNameAndMize {
|
|
let mut config_opts = self.config_opts.lock().unwrap();
|
|
match config_opts.get_mut(name) {
|
|
Some(opt) => {}
|
|
None => {
|
|
let opt = ConfigOpt {
|
|
name: name.to_owned(),
|
|
val: None,
|
|
thunk: None,
|
|
};
|
|
config_opts.insert(name.to_string(), opt);
|
|
}
|
|
}
|
|
ConfigOptNameAndMize {
|
|
name: name.to_string(),
|
|
mize: self.clone(),
|
|
}
|
|
}
|
|
|
|
pub fn run(&mut self) -> MizeResult<()> {
|
|
let part_names = self.part_names();
|
|
for part_name in part_names {
|
|
let mut part = self.get_part(part_name)?;
|
|
//println!("running part {part_name}");
|
|
part.run(&mut self.clone())?;
|
|
}
|
|
let mut threads_inner = self.threads.lock().unwrap();
|
|
for (id, name, handle) in threads_inner.drain(..) {
|
|
if let Some(handle) = handle {
|
|
handle.join().unwrap()?;
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
pub fn get_part_native<T: MizePart + Send + Sync + 'static>(
|
|
&mut self,
|
|
name: &str,
|
|
) -> MizeResult<MizePartGuard<T>> {
|
|
let mut dyn_guard = self.get_part(name)?;
|
|
let part = dyn_guard.part.take().unwrap();
|
|
drop(dyn_guard);
|
|
let concrete_part = part
|
|
.into_any()
|
|
.downcast::<T>()
|
|
.map_err(|_| mize_err!("failed to downcast"))?;
|
|
Ok(MizePartGuard {
|
|
mize: self.clone(),
|
|
part: Some(*concrete_part),
|
|
})
|
|
}
|
|
pub fn add_name_only_part(&mut self, name: &'static str) {
|
|
let part = NameOnlyPart {
|
|
mize: self.clone(),
|
|
name,
|
|
};
|
|
self.add_part(Box::new(part));
|
|
}
|
|
pub fn id_path(&self, id: &MizeId) -> MizeResult<String> {
|
|
Ok(id.path().join("/"))
|
|
}
|
|
}
|
|
|
|
#[derive(Default)]
|
|
struct NameOnlyPart {
|
|
mize: Mize,
|
|
name: &'static str,
|
|
}
|
|
|
|
impl MizePart for NameOnlyPart {}
|
|
impl MizePartGenerated for NameOnlyPart {
|
|
fn as_any_generated(&self) -> &dyn Any {
|
|
self.as_any()
|
|
}
|
|
fn as_any_mut_generated(&mut self) -> &mut dyn Any {
|
|
self.as_any_mut()
|
|
}
|
|
fn get_mize_generated(&mut self) -> &mut Mize {
|
|
&mut self.mize
|
|
}
|
|
fn into_any_generated(self: Box<Self>) -> Box<dyn Any> {
|
|
self.into_any()
|
|
}
|
|
fn name_generated(&self) -> &'static str {
|
|
self.name
|
|
}
|
|
}
|
|
|
|
impl fmt::Debug for Mize {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
write!(f, "Mize Instance with subs: {:?}", self.subs,)
|
|
}
|
|
}
|
|
|
|
impl Default for Mize {
|
|
fn default() -> Self {
|
|
Self::empty().unwrap()
|
|
}
|
|
}
|