spikard 0.3.5 → 0.3.6
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/LICENSE +1 -1
- data/README.md +659 -659
- data/ext/spikard_rb/Cargo.toml +17 -17
- data/ext/spikard_rb/extconf.rb +10 -10
- data/ext/spikard_rb/src/lib.rs +6 -6
- data/lib/spikard/app.rb +386 -386
- data/lib/spikard/background.rb +27 -27
- data/lib/spikard/config.rb +396 -396
- data/lib/spikard/converters.rb +13 -13
- data/lib/spikard/handler_wrapper.rb +113 -113
- data/lib/spikard/provide.rb +214 -214
- data/lib/spikard/response.rb +173 -173
- data/lib/spikard/schema.rb +243 -243
- data/lib/spikard/sse.rb +111 -111
- data/lib/spikard/streaming_response.rb +44 -44
- data/lib/spikard/testing.rb +221 -221
- data/lib/spikard/upload_file.rb +131 -131
- data/lib/spikard/version.rb +5 -5
- data/lib/spikard/websocket.rb +59 -59
- data/lib/spikard.rb +43 -43
- data/sig/spikard.rbs +366 -360
- data/vendor/crates/spikard-core/Cargo.toml +40 -40
- data/vendor/crates/spikard-core/src/bindings/mod.rs +3 -3
- data/vendor/crates/spikard-core/src/bindings/response.rs +133 -133
- data/vendor/crates/spikard-core/src/debug.rs +63 -63
- data/vendor/crates/spikard-core/src/di/container.rs +726 -726
- data/vendor/crates/spikard-core/src/di/dependency.rs +273 -273
- data/vendor/crates/spikard-core/src/di/error.rs +118 -118
- data/vendor/crates/spikard-core/src/di/factory.rs +538 -538
- data/vendor/crates/spikard-core/src/di/graph.rs +545 -545
- data/vendor/crates/spikard-core/src/di/mod.rs +192 -192
- data/vendor/crates/spikard-core/src/di/resolved.rs +411 -411
- data/vendor/crates/spikard-core/src/di/value.rs +283 -283
- data/vendor/crates/spikard-core/src/errors.rs +39 -39
- data/vendor/crates/spikard-core/src/http.rs +153 -153
- data/vendor/crates/spikard-core/src/lib.rs +29 -29
- data/vendor/crates/spikard-core/src/lifecycle.rs +422 -422
- data/vendor/crates/spikard-core/src/parameters.rs +722 -722
- data/vendor/crates/spikard-core/src/problem.rs +310 -310
- data/vendor/crates/spikard-core/src/request_data.rs +189 -189
- data/vendor/crates/spikard-core/src/router.rs +249 -249
- data/vendor/crates/spikard-core/src/schema_registry.rs +183 -183
- data/vendor/crates/spikard-core/src/type_hints.rs +304 -304
- data/vendor/crates/spikard-core/src/validation.rs +699 -699
- data/vendor/crates/spikard-http/Cargo.toml +68 -68
- data/vendor/crates/spikard-http/src/auth.rs +247 -247
- data/vendor/crates/spikard-http/src/background.rs +249 -249
- data/vendor/crates/spikard-http/src/bindings/mod.rs +3 -3
- data/vendor/crates/spikard-http/src/bindings/response.rs +1 -1
- data/vendor/crates/spikard-http/src/body_metadata.rs +8 -8
- data/vendor/crates/spikard-http/src/cors.rs +490 -490
- data/vendor/crates/spikard-http/src/debug.rs +63 -63
- data/vendor/crates/spikard-http/src/di_handler.rs +423 -423
- data/vendor/crates/spikard-http/src/handler_response.rs +190 -190
- data/vendor/crates/spikard-http/src/handler_trait.rs +228 -228
- data/vendor/crates/spikard-http/src/handler_trait_tests.rs +284 -284
- data/vendor/crates/spikard-http/src/lib.rs +529 -529
- data/vendor/crates/spikard-http/src/lifecycle/adapter.rs +149 -149
- data/vendor/crates/spikard-http/src/lifecycle.rs +428 -428
- data/vendor/crates/spikard-http/src/middleware/mod.rs +285 -285
- data/vendor/crates/spikard-http/src/middleware/multipart.rs +86 -86
- data/vendor/crates/spikard-http/src/middleware/urlencoded.rs +147 -147
- data/vendor/crates/spikard-http/src/middleware/validation.rs +287 -287
- data/vendor/crates/spikard-http/src/openapi/mod.rs +309 -309
- data/vendor/crates/spikard-http/src/openapi/parameter_extraction.rs +190 -190
- data/vendor/crates/spikard-http/src/openapi/schema_conversion.rs +308 -308
- data/vendor/crates/spikard-http/src/openapi/spec_generation.rs +195 -195
- data/vendor/crates/spikard-http/src/parameters.rs +1 -1
- data/vendor/crates/spikard-http/src/problem.rs +1 -1
- data/vendor/crates/spikard-http/src/query_parser.rs +369 -369
- data/vendor/crates/spikard-http/src/response.rs +399 -399
- data/vendor/crates/spikard-http/src/router.rs +1 -1
- data/vendor/crates/spikard-http/src/schema_registry.rs +1 -1
- data/vendor/crates/spikard-http/src/server/handler.rs +87 -87
- data/vendor/crates/spikard-http/src/server/lifecycle_execution.rs +98 -98
- data/vendor/crates/spikard-http/src/server/mod.rs +805 -805
- data/vendor/crates/spikard-http/src/server/request_extraction.rs +119 -119
- data/vendor/crates/spikard-http/src/sse.rs +447 -447
- data/vendor/crates/spikard-http/src/testing/form.rs +14 -14
- data/vendor/crates/spikard-http/src/testing/multipart.rs +60 -60
- data/vendor/crates/spikard-http/src/testing/test_client.rs +285 -285
- data/vendor/crates/spikard-http/src/testing.rs +377 -377
- data/vendor/crates/spikard-http/src/type_hints.rs +1 -1
- data/vendor/crates/spikard-http/src/validation.rs +1 -1
- data/vendor/crates/spikard-http/src/websocket.rs +324 -324
- data/vendor/crates/spikard-rb/Cargo.toml +42 -42
- data/vendor/crates/spikard-rb/build.rs +8 -8
- data/vendor/crates/spikard-rb/src/background.rs +63 -63
- data/vendor/crates/spikard-rb/src/config.rs +294 -294
- data/vendor/crates/spikard-rb/src/conversion.rs +453 -453
- data/vendor/crates/spikard-rb/src/di.rs +409 -409
- data/vendor/crates/spikard-rb/src/handler.rs +625 -625
- data/vendor/crates/spikard-rb/src/lib.rs +2771 -2771
- data/vendor/crates/spikard-rb/src/lifecycle.rs +274 -274
- data/vendor/crates/spikard-rb/src/server.rs +283 -283
- data/vendor/crates/spikard-rb/src/sse.rs +231 -231
- data/vendor/crates/spikard-rb/src/test_client.rs +404 -404
- data/vendor/crates/spikard-rb/src/test_sse.rs +143 -143
- data/vendor/crates/spikard-rb/src/test_websocket.rs +221 -221
- data/vendor/crates/spikard-rb/src/websocket.rs +233 -233
- metadata +1 -1
|
@@ -1,411 +1,411 @@
|
|
|
1
|
-
//! Storage for resolved dependencies
|
|
2
|
-
//!
|
|
3
|
-
//! This module provides the `ResolvedDependencies` type which holds all dependencies
|
|
4
|
-
//! resolved for a particular request, with type-safe access and cleanup support.
|
|
5
|
-
|
|
6
|
-
use std::any::Any;
|
|
7
|
-
use std::collections::HashMap;
|
|
8
|
-
use std::future::Future;
|
|
9
|
-
use std::pin::Pin;
|
|
10
|
-
use std::sync::{Arc, Mutex};
|
|
11
|
-
|
|
12
|
-
type BoxFuture<'a, T> = Pin<Box<dyn Future<Output = T> + Send + 'a>>;
|
|
13
|
-
type CleanupTask = Box<dyn FnOnce() -> BoxFuture<'static, ()> + Send>;
|
|
14
|
-
|
|
15
|
-
/// Storage for resolved dependencies with type-safe access
|
|
16
|
-
///
|
|
17
|
-
/// This type stores all dependencies that have been resolved for a request,
|
|
18
|
-
/// allowing type-safe retrieval and supporting cleanup tasks for generator-pattern
|
|
19
|
-
/// dependencies.
|
|
20
|
-
///
|
|
21
|
-
/// Uses Arc<Mutex<>> internally for thread-safe shared access.
|
|
22
|
-
///
|
|
23
|
-
/// # Examples
|
|
24
|
-
///
|
|
25
|
-
/// ```ignore
|
|
26
|
-
/// use spikard_core::di::ResolvedDependencies;
|
|
27
|
-
/// use std::sync::Arc;
|
|
28
|
-
///
|
|
29
|
-
/// # tokio_test::block_on(async {
|
|
30
|
-
/// let mut resolved = ResolvedDependencies::new();
|
|
31
|
-
///
|
|
32
|
-
/// // Insert a dependency
|
|
33
|
-
/// let value = Arc::new(42i32);
|
|
34
|
-
/// resolved.insert("answer".to_string(), value.clone());
|
|
35
|
-
///
|
|
36
|
-
/// // Retrieve with type safety
|
|
37
|
-
/// let retrieved: Option<Arc<i32>> = resolved.get("answer");
|
|
38
|
-
/// assert_eq!(retrieved.map(|v| *v), Some(42));
|
|
39
|
-
///
|
|
40
|
-
/// // Type mismatch returns None
|
|
41
|
-
/// let wrong_type: Option<Arc<String>> = resolved.get("answer");
|
|
42
|
-
/// assert!(wrong_type.is_none());
|
|
43
|
-
///
|
|
44
|
-
/// // Cleanup
|
|
45
|
-
/// resolved.cleanup().await;
|
|
46
|
-
/// # });
|
|
47
|
-
/// ```
|
|
48
|
-
#[derive(Default, Clone)]
|
|
49
|
-
pub struct ResolvedDependencies {
|
|
50
|
-
/// Map of dependency keys to type-erased values
|
|
51
|
-
dependencies: Arc<Mutex<HashMap<String, Arc<dyn Any + Send + Sync>>>>,
|
|
52
|
-
/// Cleanup tasks to run when dependencies are dropped
|
|
53
|
-
cleanup_tasks: Arc<Mutex<Vec<CleanupTask>>>,
|
|
54
|
-
}
|
|
55
|
-
|
|
56
|
-
impl ResolvedDependencies {
|
|
57
|
-
/// Create a new empty resolved dependencies storage
|
|
58
|
-
///
|
|
59
|
-
/// # Examples
|
|
60
|
-
///
|
|
61
|
-
/// ```ignore
|
|
62
|
-
/// use spikard_core::di::ResolvedDependencies;
|
|
63
|
-
///
|
|
64
|
-
/// let resolved = ResolvedDependencies::new();
|
|
65
|
-
/// ```
|
|
66
|
-
#[must_use]
|
|
67
|
-
pub fn new() -> Self {
|
|
68
|
-
Self {
|
|
69
|
-
dependencies: Arc::new(Mutex::new(HashMap::new())),
|
|
70
|
-
cleanup_tasks: Arc::new(Mutex::new(Vec::new())),
|
|
71
|
-
}
|
|
72
|
-
}
|
|
73
|
-
|
|
74
|
-
/// Insert a dependency into the storage
|
|
75
|
-
///
|
|
76
|
-
/// # Arguments
|
|
77
|
-
///
|
|
78
|
-
/// * `key` - The unique key for this dependency
|
|
79
|
-
/// * `value` - The dependency value wrapped in Arc<dyn Any + Send + Sync>
|
|
80
|
-
///
|
|
81
|
-
/// # Examples
|
|
82
|
-
///
|
|
83
|
-
/// ```ignore
|
|
84
|
-
/// use spikard_core::di::ResolvedDependencies;
|
|
85
|
-
/// use std::sync::Arc;
|
|
86
|
-
///
|
|
87
|
-
/// let mut resolved = ResolvedDependencies::new();
|
|
88
|
-
/// let config = Arc::new("production".to_string());
|
|
89
|
-
/// resolved.insert("config".to_string(), config);
|
|
90
|
-
/// ```
|
|
91
|
-
pub fn insert(&mut self, key: String, value: Arc<dyn Any + Send + Sync>) {
|
|
92
|
-
self.dependencies.lock().unwrap().insert(key, value);
|
|
93
|
-
}
|
|
94
|
-
|
|
95
|
-
/// Get a dependency with type-safe downcasting
|
|
96
|
-
///
|
|
97
|
-
/// Returns `Some(Arc<T>)` if the dependency exists and is of type `T`,
|
|
98
|
-
/// or `None` if it doesn't exist or has a different type.
|
|
99
|
-
///
|
|
100
|
-
/// # Type Parameters
|
|
101
|
-
///
|
|
102
|
-
/// * `T` - The expected type of the dependency (must be Send + Sync)
|
|
103
|
-
///
|
|
104
|
-
/// # Arguments
|
|
105
|
-
///
|
|
106
|
-
/// * `key` - The key of the dependency to retrieve
|
|
107
|
-
///
|
|
108
|
-
/// # Examples
|
|
109
|
-
///
|
|
110
|
-
/// ```ignore
|
|
111
|
-
/// use spikard_core::di::ResolvedDependencies;
|
|
112
|
-
/// use std::sync::Arc;
|
|
113
|
-
///
|
|
114
|
-
/// let mut resolved = ResolvedDependencies::new();
|
|
115
|
-
/// resolved.insert("count".to_string(), Arc::new(100i32));
|
|
116
|
-
///
|
|
117
|
-
/// // Correct type
|
|
118
|
-
/// let count: Option<Arc<i32>> = resolved.get("count");
|
|
119
|
-
/// assert_eq!(count.map(|v| *v), Some(100));
|
|
120
|
-
///
|
|
121
|
-
/// // Wrong type
|
|
122
|
-
/// let wrong: Option<Arc<String>> = resolved.get("count");
|
|
123
|
-
/// assert!(wrong.is_none());
|
|
124
|
-
///
|
|
125
|
-
/// // Missing key
|
|
126
|
-
/// let missing: Option<Arc<i32>> = resolved.get("missing");
|
|
127
|
-
/// assert!(missing.is_none());
|
|
128
|
-
/// ```
|
|
129
|
-
pub fn get<T: Send + Sync + 'static>(&self, key: &str) -> Option<Arc<T>> {
|
|
130
|
-
self.dependencies
|
|
131
|
-
.lock()
|
|
132
|
-
.unwrap()
|
|
133
|
-
.get(key)
|
|
134
|
-
.and_then(|value| Arc::clone(value).downcast::<T>().ok())
|
|
135
|
-
}
|
|
136
|
-
|
|
137
|
-
/// Get a dependency as Arc<dyn Any> without type checking
|
|
138
|
-
///
|
|
139
|
-
/// This is useful when you need to pass dependencies around without
|
|
140
|
-
/// knowing their concrete type.
|
|
141
|
-
///
|
|
142
|
-
/// # Arguments
|
|
143
|
-
///
|
|
144
|
-
/// * `key` - The key of the dependency to retrieve
|
|
145
|
-
///
|
|
146
|
-
/// # Examples
|
|
147
|
-
///
|
|
148
|
-
/// ```ignore
|
|
149
|
-
/// use spikard_core::di::ResolvedDependencies;
|
|
150
|
-
/// use std::sync::Arc;
|
|
151
|
-
///
|
|
152
|
-
/// let mut resolved = ResolvedDependencies::new();
|
|
153
|
-
/// resolved.insert("data".to_string(), Arc::new(vec![1, 2, 3]));
|
|
154
|
-
///
|
|
155
|
-
/// let any_ref = resolved.get_arc("data");
|
|
156
|
-
/// assert!(any_ref.is_some());
|
|
157
|
-
/// ```
|
|
158
|
-
pub fn get_arc(&self, key: &str) -> Option<Arc<dyn Any + Send + Sync>> {
|
|
159
|
-
self.dependencies.lock().unwrap().get(key).cloned()
|
|
160
|
-
}
|
|
161
|
-
|
|
162
|
-
/// Check if a dependency exists
|
|
163
|
-
///
|
|
164
|
-
/// # Arguments
|
|
165
|
-
///
|
|
166
|
-
/// * `key` - The key to check
|
|
167
|
-
///
|
|
168
|
-
/// # Examples
|
|
169
|
-
///
|
|
170
|
-
/// ```ignore
|
|
171
|
-
/// use spikard_core::di::ResolvedDependencies;
|
|
172
|
-
/// use std::sync::Arc;
|
|
173
|
-
///
|
|
174
|
-
/// let mut resolved = ResolvedDependencies::new();
|
|
175
|
-
/// resolved.insert("exists".to_string(), Arc::new(true));
|
|
176
|
-
///
|
|
177
|
-
/// assert!(resolved.contains("exists"));
|
|
178
|
-
/// assert!(!resolved.contains("missing"));
|
|
179
|
-
/// ```
|
|
180
|
-
#[must_use]
|
|
181
|
-
pub fn contains(&self, key: &str) -> bool {
|
|
182
|
-
self.dependencies.lock().unwrap().contains_key(key)
|
|
183
|
-
}
|
|
184
|
-
|
|
185
|
-
/// Get all dependency keys
|
|
186
|
-
///
|
|
187
|
-
/// Returns a vector of all keys currently stored in this resolved dependencies.
|
|
188
|
-
/// Useful for iterating over all dependencies when you need to extract them.
|
|
189
|
-
///
|
|
190
|
-
/// # Examples
|
|
191
|
-
///
|
|
192
|
-
/// ```ignore
|
|
193
|
-
/// use spikard_core::di::ResolvedDependencies;
|
|
194
|
-
/// use std::sync::Arc;
|
|
195
|
-
///
|
|
196
|
-
/// let mut resolved = ResolvedDependencies::new();
|
|
197
|
-
/// resolved.insert("config".to_string(), Arc::new("prod".to_string()));
|
|
198
|
-
/// resolved.insert("db".to_string(), Arc::new(42i32));
|
|
199
|
-
///
|
|
200
|
-
/// let keys = resolved.keys();
|
|
201
|
-
/// assert_eq!(keys.len(), 2);
|
|
202
|
-
/// assert!(keys.contains(&"config".to_string()));
|
|
203
|
-
/// assert!(keys.contains(&"db".to_string()));
|
|
204
|
-
/// ```
|
|
205
|
-
#[must_use]
|
|
206
|
-
pub fn keys(&self) -> Vec<String> {
|
|
207
|
-
self.dependencies.lock().unwrap().keys().cloned().collect()
|
|
208
|
-
}
|
|
209
|
-
|
|
210
|
-
/// Add a cleanup task to be run when dependencies are cleaned up
|
|
211
|
-
///
|
|
212
|
-
/// Cleanup tasks are useful for generator-pattern dependencies that need
|
|
213
|
-
/// to perform cleanup (e.g., closing database connections, releasing locks).
|
|
214
|
-
///
|
|
215
|
-
/// # Arguments
|
|
216
|
-
///
|
|
217
|
-
/// * `task` - A function that returns a future performing cleanup
|
|
218
|
-
///
|
|
219
|
-
/// # Examples
|
|
220
|
-
///
|
|
221
|
-
/// ```ignore
|
|
222
|
-
/// use spikard_core::di::ResolvedDependencies;
|
|
223
|
-
///
|
|
224
|
-
/// # tokio_test::block_on(async {
|
|
225
|
-
/// let mut resolved = ResolvedDependencies::new();
|
|
226
|
-
///
|
|
227
|
-
/// resolved.add_cleanup_task(Box::new(|| {
|
|
228
|
-
/// Box::pin(async {
|
|
229
|
-
/// println!("Cleaning up resources");
|
|
230
|
-
/// })
|
|
231
|
-
/// }));
|
|
232
|
-
///
|
|
233
|
-
/// resolved.cleanup().await;
|
|
234
|
-
/// # });
|
|
235
|
-
/// ```
|
|
236
|
-
pub fn add_cleanup_task(&self, task: CleanupTask) {
|
|
237
|
-
self.cleanup_tasks.lock().unwrap().push(task);
|
|
238
|
-
}
|
|
239
|
-
|
|
240
|
-
/// Run all cleanup tasks in reverse order
|
|
241
|
-
///
|
|
242
|
-
/// Cleanup tasks are executed in LIFO order (last added, first executed)
|
|
243
|
-
/// to properly handle nested resource dependencies.
|
|
244
|
-
///
|
|
245
|
-
/// This consumes self to ensure cleanup is only run once.
|
|
246
|
-
///
|
|
247
|
-
/// # Examples
|
|
248
|
-
///
|
|
249
|
-
/// ```ignore
|
|
250
|
-
/// use spikard_core::di::ResolvedDependencies;
|
|
251
|
-
/// use std::sync::{Arc, Mutex};
|
|
252
|
-
///
|
|
253
|
-
/// # tokio_test::block_on(async {
|
|
254
|
-
/// let order = Arc::new(Mutex::new(Vec::new()));
|
|
255
|
-
///
|
|
256
|
-
/// let mut resolved = ResolvedDependencies::new();
|
|
257
|
-
///
|
|
258
|
-
/// let order1 = order.clone();
|
|
259
|
-
/// resolved.add_cleanup_task(Box::new(move || {
|
|
260
|
-
/// Box::pin(async move {
|
|
261
|
-
/// order1.lock().unwrap().push(1);
|
|
262
|
-
/// })
|
|
263
|
-
/// }));
|
|
264
|
-
///
|
|
265
|
-
/// let order2 = order.clone();
|
|
266
|
-
/// resolved.add_cleanup_task(Box::new(move || {
|
|
267
|
-
/// Box::pin(async move {
|
|
268
|
-
/// order2.lock().unwrap().push(2);
|
|
269
|
-
/// })
|
|
270
|
-
/// }));
|
|
271
|
-
///
|
|
272
|
-
/// resolved.cleanup().await;
|
|
273
|
-
///
|
|
274
|
-
/// // Tasks run in reverse order (LIFO)
|
|
275
|
-
/// assert_eq!(*order.lock().unwrap(), vec![2, 1]);
|
|
276
|
-
/// # });
|
|
277
|
-
/// ```
|
|
278
|
-
pub async fn cleanup(self) {
|
|
279
|
-
// Take ownership of cleanup tasks
|
|
280
|
-
let tasks = {
|
|
281
|
-
let mut cleanup_tasks = self.cleanup_tasks.lock().unwrap();
|
|
282
|
-
std::mem::take(&mut *cleanup_tasks)
|
|
283
|
-
};
|
|
284
|
-
|
|
285
|
-
// Run cleanup tasks in reverse order (LIFO)
|
|
286
|
-
for task in tasks.into_iter().rev() {
|
|
287
|
-
task().await;
|
|
288
|
-
}
|
|
289
|
-
}
|
|
290
|
-
}
|
|
291
|
-
|
|
292
|
-
impl std::fmt::Debug for ResolvedDependencies {
|
|
293
|
-
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
294
|
-
let deps = self.dependencies.lock().unwrap();
|
|
295
|
-
let tasks = self.cleanup_tasks.lock().unwrap();
|
|
296
|
-
f.debug_struct("ResolvedDependencies")
|
|
297
|
-
.field("dependencies", &deps.keys())
|
|
298
|
-
.field("cleanup_tasks_count", &tasks.len())
|
|
299
|
-
.finish()
|
|
300
|
-
}
|
|
301
|
-
}
|
|
302
|
-
|
|
303
|
-
#[cfg(test)]
|
|
304
|
-
mod tests {
|
|
305
|
-
use super::*;
|
|
306
|
-
|
|
307
|
-
#[test]
|
|
308
|
-
fn test_new() {
|
|
309
|
-
let resolved = ResolvedDependencies::new();
|
|
310
|
-
assert!(!resolved.contains("anything"));
|
|
311
|
-
}
|
|
312
|
-
|
|
313
|
-
#[test]
|
|
314
|
-
fn test_insert_and_get() {
|
|
315
|
-
let mut resolved = ResolvedDependencies::new();
|
|
316
|
-
let value = Arc::new(42i32);
|
|
317
|
-
resolved.insert("answer".to_string(), value.clone());
|
|
318
|
-
|
|
319
|
-
let retrieved: Option<Arc<i32>> = resolved.get("answer");
|
|
320
|
-
assert_eq!(retrieved.map(|v| *v), Some(42));
|
|
321
|
-
}
|
|
322
|
-
|
|
323
|
-
#[test]
|
|
324
|
-
fn test_get_type_mismatch() {
|
|
325
|
-
let mut resolved = ResolvedDependencies::new();
|
|
326
|
-
resolved.insert("number".to_string(), Arc::new(42i32));
|
|
327
|
-
|
|
328
|
-
// Wrong type returns None
|
|
329
|
-
let wrong: Option<Arc<String>> = resolved.get("number");
|
|
330
|
-
assert!(wrong.is_none());
|
|
331
|
-
}
|
|
332
|
-
|
|
333
|
-
#[test]
|
|
334
|
-
fn test_get_missing() {
|
|
335
|
-
let resolved = ResolvedDependencies::new();
|
|
336
|
-
let missing: Option<Arc<i32>> = resolved.get("missing");
|
|
337
|
-
assert!(missing.is_none());
|
|
338
|
-
}
|
|
339
|
-
|
|
340
|
-
#[test]
|
|
341
|
-
fn test_get_arc() {
|
|
342
|
-
let mut resolved = ResolvedDependencies::new();
|
|
343
|
-
resolved.insert("data".to_string(), Arc::new(vec![1, 2, 3]));
|
|
344
|
-
|
|
345
|
-
let any_ref = resolved.get_arc("data");
|
|
346
|
-
assert!(any_ref.is_some());
|
|
347
|
-
|
|
348
|
-
// Can downcast manually
|
|
349
|
-
let vec_ref = any_ref.unwrap().downcast::<Vec<i32>>().ok();
|
|
350
|
-
assert!(vec_ref.is_some());
|
|
351
|
-
}
|
|
352
|
-
|
|
353
|
-
#[test]
|
|
354
|
-
fn test_contains() {
|
|
355
|
-
let mut resolved = ResolvedDependencies::new();
|
|
356
|
-
resolved.insert("exists".to_string(), Arc::new(true));
|
|
357
|
-
|
|
358
|
-
assert!(resolved.contains("exists"));
|
|
359
|
-
assert!(!resolved.contains("missing"));
|
|
360
|
-
}
|
|
361
|
-
|
|
362
|
-
#[tokio::test]
|
|
363
|
-
async fn test_cleanup_order() {
|
|
364
|
-
let order = Arc::new(Mutex::new(Vec::new()));
|
|
365
|
-
|
|
366
|
-
let resolved = ResolvedDependencies::new();
|
|
367
|
-
|
|
368
|
-
let order1 = order.clone();
|
|
369
|
-
resolved.add_cleanup_task(Box::new(move || {
|
|
370
|
-
Box::pin(async move {
|
|
371
|
-
order1.lock().unwrap().push(1);
|
|
372
|
-
})
|
|
373
|
-
}));
|
|
374
|
-
|
|
375
|
-
let order2 = order.clone();
|
|
376
|
-
resolved.add_cleanup_task(Box::new(move || {
|
|
377
|
-
Box::pin(async move {
|
|
378
|
-
order2.lock().unwrap().push(2);
|
|
379
|
-
})
|
|
380
|
-
}));
|
|
381
|
-
|
|
382
|
-
let order3 = order.clone();
|
|
383
|
-
resolved.add_cleanup_task(Box::new(move || {
|
|
384
|
-
Box::pin(async move {
|
|
385
|
-
order3.lock().unwrap().push(3);
|
|
386
|
-
})
|
|
387
|
-
}));
|
|
388
|
-
|
|
389
|
-
resolved.cleanup().await;
|
|
390
|
-
|
|
391
|
-
// Tasks run in reverse order (LIFO)
|
|
392
|
-
assert_eq!(*order.lock().unwrap(), vec![3, 2, 1]);
|
|
393
|
-
}
|
|
394
|
-
|
|
395
|
-
#[tokio::test]
|
|
396
|
-
async fn test_cleanup_empty() {
|
|
397
|
-
let resolved = ResolvedDependencies::new();
|
|
398
|
-
resolved.cleanup().await; // Should not panic
|
|
399
|
-
}
|
|
400
|
-
|
|
401
|
-
#[test]
|
|
402
|
-
fn test_clone() {
|
|
403
|
-
let mut resolved1 = ResolvedDependencies::new();
|
|
404
|
-
resolved1.insert("key".to_string(), Arc::new(42i32));
|
|
405
|
-
|
|
406
|
-
// Clone shares the same underlying data
|
|
407
|
-
let resolved2 = resolved1.clone();
|
|
408
|
-
let value: Option<Arc<i32>> = resolved2.get("key");
|
|
409
|
-
assert_eq!(value.map(|v| *v), Some(42));
|
|
410
|
-
}
|
|
411
|
-
}
|
|
1
|
+
//! Storage for resolved dependencies
|
|
2
|
+
//!
|
|
3
|
+
//! This module provides the `ResolvedDependencies` type which holds all dependencies
|
|
4
|
+
//! resolved for a particular request, with type-safe access and cleanup support.
|
|
5
|
+
|
|
6
|
+
use std::any::Any;
|
|
7
|
+
use std::collections::HashMap;
|
|
8
|
+
use std::future::Future;
|
|
9
|
+
use std::pin::Pin;
|
|
10
|
+
use std::sync::{Arc, Mutex};
|
|
11
|
+
|
|
12
|
+
type BoxFuture<'a, T> = Pin<Box<dyn Future<Output = T> + Send + 'a>>;
|
|
13
|
+
type CleanupTask = Box<dyn FnOnce() -> BoxFuture<'static, ()> + Send>;
|
|
14
|
+
|
|
15
|
+
/// Storage for resolved dependencies with type-safe access
|
|
16
|
+
///
|
|
17
|
+
/// This type stores all dependencies that have been resolved for a request,
|
|
18
|
+
/// allowing type-safe retrieval and supporting cleanup tasks for generator-pattern
|
|
19
|
+
/// dependencies.
|
|
20
|
+
///
|
|
21
|
+
/// Uses Arc<Mutex<>> internally for thread-safe shared access.
|
|
22
|
+
///
|
|
23
|
+
/// # Examples
|
|
24
|
+
///
|
|
25
|
+
/// ```ignore
|
|
26
|
+
/// use spikard_core::di::ResolvedDependencies;
|
|
27
|
+
/// use std::sync::Arc;
|
|
28
|
+
///
|
|
29
|
+
/// # tokio_test::block_on(async {
|
|
30
|
+
/// let mut resolved = ResolvedDependencies::new();
|
|
31
|
+
///
|
|
32
|
+
/// // Insert a dependency
|
|
33
|
+
/// let value = Arc::new(42i32);
|
|
34
|
+
/// resolved.insert("answer".to_string(), value.clone());
|
|
35
|
+
///
|
|
36
|
+
/// // Retrieve with type safety
|
|
37
|
+
/// let retrieved: Option<Arc<i32>> = resolved.get("answer");
|
|
38
|
+
/// assert_eq!(retrieved.map(|v| *v), Some(42));
|
|
39
|
+
///
|
|
40
|
+
/// // Type mismatch returns None
|
|
41
|
+
/// let wrong_type: Option<Arc<String>> = resolved.get("answer");
|
|
42
|
+
/// assert!(wrong_type.is_none());
|
|
43
|
+
///
|
|
44
|
+
/// // Cleanup
|
|
45
|
+
/// resolved.cleanup().await;
|
|
46
|
+
/// # });
|
|
47
|
+
/// ```
|
|
48
|
+
#[derive(Default, Clone)]
|
|
49
|
+
pub struct ResolvedDependencies {
|
|
50
|
+
/// Map of dependency keys to type-erased values
|
|
51
|
+
dependencies: Arc<Mutex<HashMap<String, Arc<dyn Any + Send + Sync>>>>,
|
|
52
|
+
/// Cleanup tasks to run when dependencies are dropped
|
|
53
|
+
cleanup_tasks: Arc<Mutex<Vec<CleanupTask>>>,
|
|
54
|
+
}
|
|
55
|
+
|
|
56
|
+
impl ResolvedDependencies {
|
|
57
|
+
/// Create a new empty resolved dependencies storage
|
|
58
|
+
///
|
|
59
|
+
/// # Examples
|
|
60
|
+
///
|
|
61
|
+
/// ```ignore
|
|
62
|
+
/// use spikard_core::di::ResolvedDependencies;
|
|
63
|
+
///
|
|
64
|
+
/// let resolved = ResolvedDependencies::new();
|
|
65
|
+
/// ```
|
|
66
|
+
#[must_use]
|
|
67
|
+
pub fn new() -> Self {
|
|
68
|
+
Self {
|
|
69
|
+
dependencies: Arc::new(Mutex::new(HashMap::new())),
|
|
70
|
+
cleanup_tasks: Arc::new(Mutex::new(Vec::new())),
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
|
|
74
|
+
/// Insert a dependency into the storage
|
|
75
|
+
///
|
|
76
|
+
/// # Arguments
|
|
77
|
+
///
|
|
78
|
+
/// * `key` - The unique key for this dependency
|
|
79
|
+
/// * `value` - The dependency value wrapped in Arc<dyn Any + Send + Sync>
|
|
80
|
+
///
|
|
81
|
+
/// # Examples
|
|
82
|
+
///
|
|
83
|
+
/// ```ignore
|
|
84
|
+
/// use spikard_core::di::ResolvedDependencies;
|
|
85
|
+
/// use std::sync::Arc;
|
|
86
|
+
///
|
|
87
|
+
/// let mut resolved = ResolvedDependencies::new();
|
|
88
|
+
/// let config = Arc::new("production".to_string());
|
|
89
|
+
/// resolved.insert("config".to_string(), config);
|
|
90
|
+
/// ```
|
|
91
|
+
pub fn insert(&mut self, key: String, value: Arc<dyn Any + Send + Sync>) {
|
|
92
|
+
self.dependencies.lock().unwrap().insert(key, value);
|
|
93
|
+
}
|
|
94
|
+
|
|
95
|
+
/// Get a dependency with type-safe downcasting
|
|
96
|
+
///
|
|
97
|
+
/// Returns `Some(Arc<T>)` if the dependency exists and is of type `T`,
|
|
98
|
+
/// or `None` if it doesn't exist or has a different type.
|
|
99
|
+
///
|
|
100
|
+
/// # Type Parameters
|
|
101
|
+
///
|
|
102
|
+
/// * `T` - The expected type of the dependency (must be Send + Sync)
|
|
103
|
+
///
|
|
104
|
+
/// # Arguments
|
|
105
|
+
///
|
|
106
|
+
/// * `key` - The key of the dependency to retrieve
|
|
107
|
+
///
|
|
108
|
+
/// # Examples
|
|
109
|
+
///
|
|
110
|
+
/// ```ignore
|
|
111
|
+
/// use spikard_core::di::ResolvedDependencies;
|
|
112
|
+
/// use std::sync::Arc;
|
|
113
|
+
///
|
|
114
|
+
/// let mut resolved = ResolvedDependencies::new();
|
|
115
|
+
/// resolved.insert("count".to_string(), Arc::new(100i32));
|
|
116
|
+
///
|
|
117
|
+
/// // Correct type
|
|
118
|
+
/// let count: Option<Arc<i32>> = resolved.get("count");
|
|
119
|
+
/// assert_eq!(count.map(|v| *v), Some(100));
|
|
120
|
+
///
|
|
121
|
+
/// // Wrong type
|
|
122
|
+
/// let wrong: Option<Arc<String>> = resolved.get("count");
|
|
123
|
+
/// assert!(wrong.is_none());
|
|
124
|
+
///
|
|
125
|
+
/// // Missing key
|
|
126
|
+
/// let missing: Option<Arc<i32>> = resolved.get("missing");
|
|
127
|
+
/// assert!(missing.is_none());
|
|
128
|
+
/// ```
|
|
129
|
+
pub fn get<T: Send + Sync + 'static>(&self, key: &str) -> Option<Arc<T>> {
|
|
130
|
+
self.dependencies
|
|
131
|
+
.lock()
|
|
132
|
+
.unwrap()
|
|
133
|
+
.get(key)
|
|
134
|
+
.and_then(|value| Arc::clone(value).downcast::<T>().ok())
|
|
135
|
+
}
|
|
136
|
+
|
|
137
|
+
/// Get a dependency as Arc<dyn Any> without type checking
|
|
138
|
+
///
|
|
139
|
+
/// This is useful when you need to pass dependencies around without
|
|
140
|
+
/// knowing their concrete type.
|
|
141
|
+
///
|
|
142
|
+
/// # Arguments
|
|
143
|
+
///
|
|
144
|
+
/// * `key` - The key of the dependency to retrieve
|
|
145
|
+
///
|
|
146
|
+
/// # Examples
|
|
147
|
+
///
|
|
148
|
+
/// ```ignore
|
|
149
|
+
/// use spikard_core::di::ResolvedDependencies;
|
|
150
|
+
/// use std::sync::Arc;
|
|
151
|
+
///
|
|
152
|
+
/// let mut resolved = ResolvedDependencies::new();
|
|
153
|
+
/// resolved.insert("data".to_string(), Arc::new(vec![1, 2, 3]));
|
|
154
|
+
///
|
|
155
|
+
/// let any_ref = resolved.get_arc("data");
|
|
156
|
+
/// assert!(any_ref.is_some());
|
|
157
|
+
/// ```
|
|
158
|
+
pub fn get_arc(&self, key: &str) -> Option<Arc<dyn Any + Send + Sync>> {
|
|
159
|
+
self.dependencies.lock().unwrap().get(key).cloned()
|
|
160
|
+
}
|
|
161
|
+
|
|
162
|
+
/// Check if a dependency exists
|
|
163
|
+
///
|
|
164
|
+
/// # Arguments
|
|
165
|
+
///
|
|
166
|
+
/// * `key` - The key to check
|
|
167
|
+
///
|
|
168
|
+
/// # Examples
|
|
169
|
+
///
|
|
170
|
+
/// ```ignore
|
|
171
|
+
/// use spikard_core::di::ResolvedDependencies;
|
|
172
|
+
/// use std::sync::Arc;
|
|
173
|
+
///
|
|
174
|
+
/// let mut resolved = ResolvedDependencies::new();
|
|
175
|
+
/// resolved.insert("exists".to_string(), Arc::new(true));
|
|
176
|
+
///
|
|
177
|
+
/// assert!(resolved.contains("exists"));
|
|
178
|
+
/// assert!(!resolved.contains("missing"));
|
|
179
|
+
/// ```
|
|
180
|
+
#[must_use]
|
|
181
|
+
pub fn contains(&self, key: &str) -> bool {
|
|
182
|
+
self.dependencies.lock().unwrap().contains_key(key)
|
|
183
|
+
}
|
|
184
|
+
|
|
185
|
+
/// Get all dependency keys
|
|
186
|
+
///
|
|
187
|
+
/// Returns a vector of all keys currently stored in this resolved dependencies.
|
|
188
|
+
/// Useful for iterating over all dependencies when you need to extract them.
|
|
189
|
+
///
|
|
190
|
+
/// # Examples
|
|
191
|
+
///
|
|
192
|
+
/// ```ignore
|
|
193
|
+
/// use spikard_core::di::ResolvedDependencies;
|
|
194
|
+
/// use std::sync::Arc;
|
|
195
|
+
///
|
|
196
|
+
/// let mut resolved = ResolvedDependencies::new();
|
|
197
|
+
/// resolved.insert("config".to_string(), Arc::new("prod".to_string()));
|
|
198
|
+
/// resolved.insert("db".to_string(), Arc::new(42i32));
|
|
199
|
+
///
|
|
200
|
+
/// let keys = resolved.keys();
|
|
201
|
+
/// assert_eq!(keys.len(), 2);
|
|
202
|
+
/// assert!(keys.contains(&"config".to_string()));
|
|
203
|
+
/// assert!(keys.contains(&"db".to_string()));
|
|
204
|
+
/// ```
|
|
205
|
+
#[must_use]
|
|
206
|
+
pub fn keys(&self) -> Vec<String> {
|
|
207
|
+
self.dependencies.lock().unwrap().keys().cloned().collect()
|
|
208
|
+
}
|
|
209
|
+
|
|
210
|
+
/// Add a cleanup task to be run when dependencies are cleaned up
|
|
211
|
+
///
|
|
212
|
+
/// Cleanup tasks are useful for generator-pattern dependencies that need
|
|
213
|
+
/// to perform cleanup (e.g., closing database connections, releasing locks).
|
|
214
|
+
///
|
|
215
|
+
/// # Arguments
|
|
216
|
+
///
|
|
217
|
+
/// * `task` - A function that returns a future performing cleanup
|
|
218
|
+
///
|
|
219
|
+
/// # Examples
|
|
220
|
+
///
|
|
221
|
+
/// ```ignore
|
|
222
|
+
/// use spikard_core::di::ResolvedDependencies;
|
|
223
|
+
///
|
|
224
|
+
/// # tokio_test::block_on(async {
|
|
225
|
+
/// let mut resolved = ResolvedDependencies::new();
|
|
226
|
+
///
|
|
227
|
+
/// resolved.add_cleanup_task(Box::new(|| {
|
|
228
|
+
/// Box::pin(async {
|
|
229
|
+
/// println!("Cleaning up resources");
|
|
230
|
+
/// })
|
|
231
|
+
/// }));
|
|
232
|
+
///
|
|
233
|
+
/// resolved.cleanup().await;
|
|
234
|
+
/// # });
|
|
235
|
+
/// ```
|
|
236
|
+
pub fn add_cleanup_task(&self, task: CleanupTask) {
|
|
237
|
+
self.cleanup_tasks.lock().unwrap().push(task);
|
|
238
|
+
}
|
|
239
|
+
|
|
240
|
+
/// Run all cleanup tasks in reverse order
|
|
241
|
+
///
|
|
242
|
+
/// Cleanup tasks are executed in LIFO order (last added, first executed)
|
|
243
|
+
/// to properly handle nested resource dependencies.
|
|
244
|
+
///
|
|
245
|
+
/// This consumes self to ensure cleanup is only run once.
|
|
246
|
+
///
|
|
247
|
+
/// # Examples
|
|
248
|
+
///
|
|
249
|
+
/// ```ignore
|
|
250
|
+
/// use spikard_core::di::ResolvedDependencies;
|
|
251
|
+
/// use std::sync::{Arc, Mutex};
|
|
252
|
+
///
|
|
253
|
+
/// # tokio_test::block_on(async {
|
|
254
|
+
/// let order = Arc::new(Mutex::new(Vec::new()));
|
|
255
|
+
///
|
|
256
|
+
/// let mut resolved = ResolvedDependencies::new();
|
|
257
|
+
///
|
|
258
|
+
/// let order1 = order.clone();
|
|
259
|
+
/// resolved.add_cleanup_task(Box::new(move || {
|
|
260
|
+
/// Box::pin(async move {
|
|
261
|
+
/// order1.lock().unwrap().push(1);
|
|
262
|
+
/// })
|
|
263
|
+
/// }));
|
|
264
|
+
///
|
|
265
|
+
/// let order2 = order.clone();
|
|
266
|
+
/// resolved.add_cleanup_task(Box::new(move || {
|
|
267
|
+
/// Box::pin(async move {
|
|
268
|
+
/// order2.lock().unwrap().push(2);
|
|
269
|
+
/// })
|
|
270
|
+
/// }));
|
|
271
|
+
///
|
|
272
|
+
/// resolved.cleanup().await;
|
|
273
|
+
///
|
|
274
|
+
/// // Tasks run in reverse order (LIFO)
|
|
275
|
+
/// assert_eq!(*order.lock().unwrap(), vec![2, 1]);
|
|
276
|
+
/// # });
|
|
277
|
+
/// ```
|
|
278
|
+
pub async fn cleanup(self) {
|
|
279
|
+
// Take ownership of cleanup tasks
|
|
280
|
+
let tasks = {
|
|
281
|
+
let mut cleanup_tasks = self.cleanup_tasks.lock().unwrap();
|
|
282
|
+
std::mem::take(&mut *cleanup_tasks)
|
|
283
|
+
};
|
|
284
|
+
|
|
285
|
+
// Run cleanup tasks in reverse order (LIFO)
|
|
286
|
+
for task in tasks.into_iter().rev() {
|
|
287
|
+
task().await;
|
|
288
|
+
}
|
|
289
|
+
}
|
|
290
|
+
}
|
|
291
|
+
|
|
292
|
+
impl std::fmt::Debug for ResolvedDependencies {
|
|
293
|
+
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
294
|
+
let deps = self.dependencies.lock().unwrap();
|
|
295
|
+
let tasks = self.cleanup_tasks.lock().unwrap();
|
|
296
|
+
f.debug_struct("ResolvedDependencies")
|
|
297
|
+
.field("dependencies", &deps.keys())
|
|
298
|
+
.field("cleanup_tasks_count", &tasks.len())
|
|
299
|
+
.finish()
|
|
300
|
+
}
|
|
301
|
+
}
|
|
302
|
+
|
|
303
|
+
#[cfg(test)]
|
|
304
|
+
mod tests {
|
|
305
|
+
use super::*;
|
|
306
|
+
|
|
307
|
+
#[test]
|
|
308
|
+
fn test_new() {
|
|
309
|
+
let resolved = ResolvedDependencies::new();
|
|
310
|
+
assert!(!resolved.contains("anything"));
|
|
311
|
+
}
|
|
312
|
+
|
|
313
|
+
#[test]
|
|
314
|
+
fn test_insert_and_get() {
|
|
315
|
+
let mut resolved = ResolvedDependencies::new();
|
|
316
|
+
let value = Arc::new(42i32);
|
|
317
|
+
resolved.insert("answer".to_string(), value.clone());
|
|
318
|
+
|
|
319
|
+
let retrieved: Option<Arc<i32>> = resolved.get("answer");
|
|
320
|
+
assert_eq!(retrieved.map(|v| *v), Some(42));
|
|
321
|
+
}
|
|
322
|
+
|
|
323
|
+
#[test]
|
|
324
|
+
fn test_get_type_mismatch() {
|
|
325
|
+
let mut resolved = ResolvedDependencies::new();
|
|
326
|
+
resolved.insert("number".to_string(), Arc::new(42i32));
|
|
327
|
+
|
|
328
|
+
// Wrong type returns None
|
|
329
|
+
let wrong: Option<Arc<String>> = resolved.get("number");
|
|
330
|
+
assert!(wrong.is_none());
|
|
331
|
+
}
|
|
332
|
+
|
|
333
|
+
#[test]
|
|
334
|
+
fn test_get_missing() {
|
|
335
|
+
let resolved = ResolvedDependencies::new();
|
|
336
|
+
let missing: Option<Arc<i32>> = resolved.get("missing");
|
|
337
|
+
assert!(missing.is_none());
|
|
338
|
+
}
|
|
339
|
+
|
|
340
|
+
#[test]
|
|
341
|
+
fn test_get_arc() {
|
|
342
|
+
let mut resolved = ResolvedDependencies::new();
|
|
343
|
+
resolved.insert("data".to_string(), Arc::new(vec![1, 2, 3]));
|
|
344
|
+
|
|
345
|
+
let any_ref = resolved.get_arc("data");
|
|
346
|
+
assert!(any_ref.is_some());
|
|
347
|
+
|
|
348
|
+
// Can downcast manually
|
|
349
|
+
let vec_ref = any_ref.unwrap().downcast::<Vec<i32>>().ok();
|
|
350
|
+
assert!(vec_ref.is_some());
|
|
351
|
+
}
|
|
352
|
+
|
|
353
|
+
#[test]
|
|
354
|
+
fn test_contains() {
|
|
355
|
+
let mut resolved = ResolvedDependencies::new();
|
|
356
|
+
resolved.insert("exists".to_string(), Arc::new(true));
|
|
357
|
+
|
|
358
|
+
assert!(resolved.contains("exists"));
|
|
359
|
+
assert!(!resolved.contains("missing"));
|
|
360
|
+
}
|
|
361
|
+
|
|
362
|
+
#[tokio::test]
|
|
363
|
+
async fn test_cleanup_order() {
|
|
364
|
+
let order = Arc::new(Mutex::new(Vec::new()));
|
|
365
|
+
|
|
366
|
+
let resolved = ResolvedDependencies::new();
|
|
367
|
+
|
|
368
|
+
let order1 = order.clone();
|
|
369
|
+
resolved.add_cleanup_task(Box::new(move || {
|
|
370
|
+
Box::pin(async move {
|
|
371
|
+
order1.lock().unwrap().push(1);
|
|
372
|
+
})
|
|
373
|
+
}));
|
|
374
|
+
|
|
375
|
+
let order2 = order.clone();
|
|
376
|
+
resolved.add_cleanup_task(Box::new(move || {
|
|
377
|
+
Box::pin(async move {
|
|
378
|
+
order2.lock().unwrap().push(2);
|
|
379
|
+
})
|
|
380
|
+
}));
|
|
381
|
+
|
|
382
|
+
let order3 = order.clone();
|
|
383
|
+
resolved.add_cleanup_task(Box::new(move || {
|
|
384
|
+
Box::pin(async move {
|
|
385
|
+
order3.lock().unwrap().push(3);
|
|
386
|
+
})
|
|
387
|
+
}));
|
|
388
|
+
|
|
389
|
+
resolved.cleanup().await;
|
|
390
|
+
|
|
391
|
+
// Tasks run in reverse order (LIFO)
|
|
392
|
+
assert_eq!(*order.lock().unwrap(), vec![3, 2, 1]);
|
|
393
|
+
}
|
|
394
|
+
|
|
395
|
+
#[tokio::test]
|
|
396
|
+
async fn test_cleanup_empty() {
|
|
397
|
+
let resolved = ResolvedDependencies::new();
|
|
398
|
+
resolved.cleanup().await; // Should not panic
|
|
399
|
+
}
|
|
400
|
+
|
|
401
|
+
#[test]
|
|
402
|
+
fn test_clone() {
|
|
403
|
+
let mut resolved1 = ResolvedDependencies::new();
|
|
404
|
+
resolved1.insert("key".to_string(), Arc::new(42i32));
|
|
405
|
+
|
|
406
|
+
// Clone shares the same underlying data
|
|
407
|
+
let resolved2 = resolved1.clone();
|
|
408
|
+
let value: Option<Arc<i32>> = resolved2.get("key");
|
|
409
|
+
assert_eq!(value.map(|v| *v), Some(42));
|
|
410
|
+
}
|
|
411
|
+
}
|