Htrace 2.0.0

simple tracing system
Documentation 
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160

use crate::components::context::Context;
use crate::thread_manager::{ThreadManager, MAIN_THREAD};
use dashmap::DashMap;
use std::sync::OnceLock;

pub(crate) struct ContextManager
{
	data: DashMap<String, Vec<Context>>,
}

static SINGLETON: OnceLock<ContextManager> = OnceLock::new();

impl ContextManager
{
	/// get singleton
	pub fn singleton() -> &'static Self
	{
		return SINGLETON.get_or_init(|| Self::new());
	}

	pub fn add(&self, context: Context) -> u64
	{
		let threadName = ThreadManager::local_getName().unwrap_or(MAIN_THREAD.to_string());
		return match self.data.get_mut(&threadName)
		{
			None =>
			{
				self.data.insert(threadName, vec![context.clone()]);
				0
			}
			Some(mut inner) =>
			{
				let size = inner.len();
				inner.push(context.clone());
				size as u64
			}
		};
	}

	pub fn remove(&self, contextId: u64)
	{
		let threadName = ThreadManager::local_getName().unwrap_or(MAIN_THREAD.to_string());
		if let Some(mut inner) = self.data.get_mut(&threadName)
		{
			if(inner.iter().len()==1 && threadName==MAIN_THREAD)
			{
				return;
			}
			// remove last element until the context is removed (normally it's always the last, but this is resilient of something wrong happened)
			while (contextId < inner.iter().len() as u64)
			{
				inner.pop();
			}
		}
	}

	/// reset the global context
	pub(crate) fn global_set(&self, context: Context)
	{
		let globalName = MAIN_THREAD.to_string();
		self.data.insert(globalName,vec![context]);
	}

	/// get the global context
	pub(crate) fn global_get(&self) -> Option<Context>
	{
		let globalName = MAIN_THREAD.to_string();
		match self.data.get(&globalName) {
			None => None,
			Some(contextArray) => contextArray.get(0).cloned()
		}
	}

	/// resolve a trace context
	/// it starts from the global context, go into the current thread, and go down do the last context (the closest one)
	/// each context add its own information/modules/extras
	pub fn resolve(&self) -> Context
	{
		let Some(threadName) = ThreadManager::local_getName()
		else
		{
			return self.resolve_internal(MAIN_THREAD.to_string());
		};

		return self.resolve_internal(threadName);
	}

	////////////// PRIVATE ///////////

	fn new() -> Self
	{
		return ContextManager {
			data: DashMap::new(),
		};
	}

	/// resolve the context
	fn resolve_internal(&self, threadName: String) -> Context
	{
		// we always start from the global context
		let mut resolvedContext = self.resolve_main();

		// if this threadname have no existing context, we're using the global one
		let Some(contextArray) = self.data.get(&threadName)
		else
		{
			return resolvedContext;
		};

		// going throw each context
		contextArray.iter().for_each(|oneContext| {
			oneContext
				.modules_get()
				.iter()
				.for_each(|(key, context)| {
					match context
					{
						None => resolvedContext.module_discard(key),
						Some(inner) => resolvedContext.module_add_arc(key, inner.clone()),
					}
				});
			resolvedContext.level_setMin(oneContext.level_getMin().cloned());
			if let Some(name) = oneContext.threadName_get()
			{
				resolvedContext.threadName_set(name);
			}
			if let Some(name) = oneContext.name_get()
			{
				resolvedContext.name_set(name);
			}
			resolvedContext.extra_merge(oneContext.extra_getAll());
		});

		return resolvedContext;
	}

	/// resolve the main context
	fn resolve_main(&self) -> Context
	{
		let defaultContext = || {
			let mut context = Context::default();
			context.threadName_set(MAIN_THREAD);
			return context;
		};

		let Some(contextArray) = self.data.get(&MAIN_THREAD.to_string())
		else
		{
			return defaultContext();
		};

		let Some(context) = contextArray.get(0)
		else
		{
			return defaultContext();
		};

		return context.clone();
	}
}