1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100

use std::{collections::HashSet, sync::Arc};

use super::core::TagCore;
use crate::{contract::Contract, events::EventMerge, node::core::NodeCore, object::{core::ObjectVersion, observe::{ObjectDeserializer, ObjectObserveState}}, observe_contract::ObserveContract, DefaultSerializerError, LocalNode, ObjectDescriptor, TagDescriptor};

/// Represents a wish, a __contract__ to observe all objects with specific tag
pub struct TagObserve<T: Send + Sync + 'static, E = DefaultSerializerError> {
	//TODO: say no to the Option
	contract: Option<Arc<TagObserveContract>>,
	deserializer: ObjectDeserializer<T, E>,
}

impl<T: Send + Sync + 'static, E> TagObserve<T, E> {
	pub(super) fn new(contract: Arc<TagObserveContract>, deserializer: ObjectDeserializer<T, E>) -> TagObserve<T, E> {
		TagObserve { contract: Some(contract), deserializer }
	}

	pub fn descriptor(&self) -> &TagDescriptor {
		&self.contract.as_ref().unwrap().tag_core.descriptor
	}

	pub fn object_descriptors(&self) -> HashSet<ObjectDescriptor> {
		self.contract.as_ref().unwrap().tag_core.active_objects.read().unwrap().keys().cloned().collect()
	}

	pub fn changes(&self) -> EventMerge<ObjectDescriptor> {
		return self.contract.as_ref().unwrap().tag_core.active_objects_changes();
	}

	pub fn object_state<'a>(&self, descriptor: impl Into<&'a ObjectDescriptor>) -> Result<Option<ObjectObserveState<T>>, E> {
		//TODO: release lock before calling state_get
		if let Some(object_core) = self.contract.as_ref().unwrap().tag_core.active_objects.read().unwrap().get(descriptor.into()) {
			let state = object_core.state_get();
			let cooked = if let Some(raw) = state.3.as_ref() { Some((self.deserializer)(state.2.as_ref().unwrap(), raw)?) } else { None };
			Ok(Some(ObjectObserveState { synchronized: state.0, data_version: state.1.clone(), data: cooked, data_type: state.2 }))
		} else {
			Ok(None)
		}
	}

	pub fn object_data<'a>(&self, descriptor: impl Into<&'a ObjectDescriptor>) -> Result<Option<T>, E> {
		Ok(self.object_state(descriptor)?.and_then(|state| state.data))
	}

	pub fn object_version<'a>(&self, descriptor: impl Into<&'a ObjectDescriptor>) -> Option<ObjectVersion> {
		self.contract.as_ref().unwrap().tag_core.active_objects.read().unwrap().get(descriptor.into()).and_then(|object_core| object_core.version_get())
	}
}

impl<T: Sync + Send, E> Drop for TagObserve<T, E> {
	fn drop(&mut self) {
		self.contract.take().unwrap().destroy();
	}
}



pub(super) struct TagObserveContract {
	pub(super) local_node: Option<LocalNode>,
	pub(super) tag_core: Arc<TagCore>,
}

impl TagObserveContract {
	pub(super) fn new(local_node: LocalNode, tag_core: Arc<TagCore>) -> Arc<TagObserveContract> {
		Arc::new(TagObserveContract { local_node: Some(local_node), tag_core })
	}

	pub(crate) fn destroy(mut self: Arc<Self>) {
		let dyn_self: Arc<dyn ObserveContract> = self.clone();
		self.tag_core.unlink_tag_observe_contract(&dyn_self);
		drop(dyn_self);
		let local_node = Arc::get_mut(&mut self).unwrap().local_node.take().unwrap();
		assert_eq!(Arc::strong_count(&self), 1);
		let tag_core = self.tag_core.clone();
		drop(self);
		let descriptor = tag_core.descriptor.clone();
		drop(tag_core);
		local_node.shared.tag_release(&descriptor);
	}
}


impl Contract for TagObserveContract {
	fn id(&self) -> usize {
		(self as *const Self) as usize
	}

	fn node_core(&self) -> &Arc<NodeCore> {
		self.local_node.as_ref().unwrap().shared.node_core.as_ref().unwrap()
	}
}

impl ObserveContract for TagObserveContract {}


impl std::fmt::Debug for TagObserveContract {
	fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
		formatter.write_str(&format!("TagObserveContract:{}", self.tag_core.descriptor))
	}
}