spikard 0.3.2 → 0.3.3

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.
Files changed (180) hide show
  1. checksums.yaml +4 -4
  2. data/LICENSE +1 -1
  3. data/README.md +659 -659
  4. data/ext/spikard_rb/Cargo.toml +17 -17
  5. data/ext/spikard_rb/extconf.rb +10 -10
  6. data/ext/spikard_rb/src/lib.rs +6 -6
  7. data/lib/spikard/app.rb +386 -386
  8. data/lib/spikard/background.rb +27 -27
  9. data/lib/spikard/config.rb +396 -396
  10. data/lib/spikard/converters.rb +13 -13
  11. data/lib/spikard/handler_wrapper.rb +113 -113
  12. data/lib/spikard/provide.rb +214 -214
  13. data/lib/spikard/response.rb +173 -173
  14. data/lib/spikard/schema.rb +243 -243
  15. data/lib/spikard/sse.rb +111 -111
  16. data/lib/spikard/streaming_response.rb +44 -44
  17. data/lib/spikard/testing.rb +221 -221
  18. data/lib/spikard/upload_file.rb +131 -131
  19. data/lib/spikard/version.rb +5 -5
  20. data/lib/spikard/websocket.rb +59 -59
  21. data/lib/spikard.rb +43 -43
  22. data/sig/spikard.rbs +360 -360
  23. data/vendor/crates/spikard-core/Cargo.toml +40 -40
  24. data/vendor/crates/spikard-core/src/bindings/mod.rs +3 -3
  25. data/vendor/crates/spikard-core/src/bindings/response.rs +133 -133
  26. data/vendor/crates/spikard-core/src/debug.rs +63 -63
  27. data/vendor/crates/spikard-core/src/di/container.rs +726 -726
  28. data/vendor/crates/spikard-core/src/di/dependency.rs +273 -273
  29. data/vendor/crates/spikard-core/src/di/error.rs +118 -118
  30. data/vendor/crates/spikard-core/src/di/factory.rs +538 -538
  31. data/vendor/crates/spikard-core/src/di/graph.rs +545 -545
  32. data/vendor/crates/spikard-core/src/di/mod.rs +192 -192
  33. data/vendor/crates/spikard-core/src/di/resolved.rs +411 -411
  34. data/vendor/crates/spikard-core/src/di/value.rs +283 -283
  35. data/vendor/crates/spikard-core/src/errors.rs +39 -39
  36. data/vendor/crates/spikard-core/src/http.rs +153 -153
  37. data/vendor/crates/spikard-core/src/lib.rs +29 -29
  38. data/vendor/crates/spikard-core/src/lifecycle.rs +422 -422
  39. data/vendor/crates/spikard-core/src/parameters.rs +722 -722
  40. data/vendor/crates/spikard-core/src/problem.rs +310 -310
  41. data/vendor/crates/spikard-core/src/request_data.rs +189 -189
  42. data/vendor/crates/spikard-core/src/router.rs +249 -249
  43. data/vendor/crates/spikard-core/src/schema_registry.rs +183 -183
  44. data/vendor/crates/spikard-core/src/type_hints.rs +304 -304
  45. data/vendor/crates/spikard-core/src/validation.rs +699 -699
  46. data/vendor/crates/spikard-http/Cargo.toml +58 -58
  47. data/vendor/crates/spikard-http/src/auth.rs +247 -247
  48. data/vendor/crates/spikard-http/src/background.rs +249 -249
  49. data/vendor/crates/spikard-http/src/bindings/mod.rs +3 -3
  50. data/vendor/crates/spikard-http/src/bindings/response.rs +1 -1
  51. data/vendor/crates/spikard-http/src/body_metadata.rs +8 -8
  52. data/vendor/crates/spikard-http/src/cors.rs +490 -490
  53. data/vendor/crates/spikard-http/src/debug.rs +63 -63
  54. data/vendor/crates/spikard-http/src/di_handler.rs +423 -423
  55. data/vendor/crates/spikard-http/src/handler_response.rs +190 -190
  56. data/vendor/crates/spikard-http/src/handler_trait.rs +228 -228
  57. data/vendor/crates/spikard-http/src/handler_trait_tests.rs +284 -284
  58. data/vendor/crates/spikard-http/src/lib.rs +529 -529
  59. data/vendor/crates/spikard-http/src/lifecycle/adapter.rs +149 -149
  60. data/vendor/crates/spikard-http/src/lifecycle.rs +428 -428
  61. data/vendor/crates/spikard-http/src/middleware/mod.rs +285 -285
  62. data/vendor/crates/spikard-http/src/middleware/multipart.rs +86 -86
  63. data/vendor/crates/spikard-http/src/middleware/urlencoded.rs +147 -147
  64. data/vendor/crates/spikard-http/src/middleware/validation.rs +287 -287
  65. data/vendor/crates/spikard-http/src/openapi/mod.rs +309 -309
  66. data/vendor/crates/spikard-http/src/openapi/parameter_extraction.rs +190 -190
  67. data/vendor/crates/spikard-http/src/openapi/schema_conversion.rs +308 -308
  68. data/vendor/crates/spikard-http/src/openapi/spec_generation.rs +195 -195
  69. data/vendor/crates/spikard-http/src/parameters.rs +1 -1
  70. data/vendor/crates/spikard-http/src/problem.rs +1 -1
  71. data/vendor/crates/spikard-http/src/query_parser.rs +369 -369
  72. data/vendor/crates/spikard-http/src/response.rs +399 -399
  73. data/vendor/crates/spikard-http/src/router.rs +1 -1
  74. data/vendor/crates/spikard-http/src/schema_registry.rs +1 -1
  75. data/vendor/crates/spikard-http/src/server/handler.rs +87 -87
  76. data/vendor/crates/spikard-http/src/server/lifecycle_execution.rs +98 -98
  77. data/vendor/crates/spikard-http/src/server/mod.rs +805 -805
  78. data/vendor/crates/spikard-http/src/server/request_extraction.rs +119 -119
  79. data/vendor/crates/spikard-http/src/sse.rs +447 -447
  80. data/vendor/crates/spikard-http/src/testing/form.rs +14 -14
  81. data/vendor/crates/spikard-http/src/testing/multipart.rs +60 -60
  82. data/vendor/crates/spikard-http/src/testing/test_client.rs +285 -285
  83. data/vendor/crates/spikard-http/src/testing.rs +377 -377
  84. data/vendor/crates/spikard-http/src/type_hints.rs +1 -1
  85. data/vendor/crates/spikard-http/src/validation.rs +1 -1
  86. data/vendor/crates/spikard-http/src/websocket.rs +324 -324
  87. data/vendor/crates/spikard-rb/Cargo.toml +42 -42
  88. data/vendor/crates/spikard-rb/build.rs +8 -8
  89. data/vendor/crates/spikard-rb/src/background.rs +63 -63
  90. data/vendor/crates/spikard-rb/src/config.rs +294 -294
  91. data/vendor/crates/spikard-rb/src/conversion.rs +453 -453
  92. data/vendor/crates/spikard-rb/src/di.rs +409 -409
  93. data/vendor/crates/spikard-rb/src/handler.rs +625 -625
  94. data/vendor/crates/spikard-rb/src/lib.rs +2771 -2771
  95. data/vendor/crates/spikard-rb/src/lifecycle.rs +274 -274
  96. data/vendor/crates/spikard-rb/src/server.rs +283 -283
  97. data/vendor/crates/spikard-rb/src/sse.rs +231 -231
  98. data/vendor/crates/spikard-rb/src/test_client.rs +404 -404
  99. data/vendor/crates/spikard-rb/src/test_sse.rs +143 -143
  100. data/vendor/crates/spikard-rb/src/test_websocket.rs +221 -221
  101. data/vendor/crates/spikard-rb/src/websocket.rs +233 -233
  102. data/vendor/spikard-core/Cargo.toml +40 -40
  103. data/vendor/spikard-core/src/bindings/mod.rs +3 -3
  104. data/vendor/spikard-core/src/bindings/response.rs +133 -133
  105. data/vendor/spikard-core/src/debug.rs +63 -63
  106. data/vendor/spikard-core/src/di/container.rs +726 -726
  107. data/vendor/spikard-core/src/di/dependency.rs +273 -273
  108. data/vendor/spikard-core/src/di/error.rs +118 -118
  109. data/vendor/spikard-core/src/di/factory.rs +538 -538
  110. data/vendor/spikard-core/src/di/graph.rs +545 -545
  111. data/vendor/spikard-core/src/di/mod.rs +192 -192
  112. data/vendor/spikard-core/src/di/resolved.rs +411 -411
  113. data/vendor/spikard-core/src/di/value.rs +283 -283
  114. data/vendor/spikard-core/src/http.rs +153 -153
  115. data/vendor/spikard-core/src/lib.rs +28 -28
  116. data/vendor/spikard-core/src/lifecycle.rs +422 -422
  117. data/vendor/spikard-core/src/parameters.rs +719 -719
  118. data/vendor/spikard-core/src/problem.rs +310 -310
  119. data/vendor/spikard-core/src/request_data.rs +189 -189
  120. data/vendor/spikard-core/src/router.rs +249 -249
  121. data/vendor/spikard-core/src/schema_registry.rs +183 -183
  122. data/vendor/spikard-core/src/type_hints.rs +304 -304
  123. data/vendor/spikard-core/src/validation.rs +699 -699
  124. data/vendor/spikard-http/Cargo.toml +58 -58
  125. data/vendor/spikard-http/src/auth.rs +247 -247
  126. data/vendor/spikard-http/src/background.rs +249 -249
  127. data/vendor/spikard-http/src/bindings/mod.rs +3 -3
  128. data/vendor/spikard-http/src/bindings/response.rs +1 -1
  129. data/vendor/spikard-http/src/body_metadata.rs +8 -8
  130. data/vendor/spikard-http/src/cors.rs +490 -490
  131. data/vendor/spikard-http/src/debug.rs +63 -63
  132. data/vendor/spikard-http/src/di_handler.rs +423 -423
  133. data/vendor/spikard-http/src/handler_response.rs +190 -190
  134. data/vendor/spikard-http/src/handler_trait.rs +228 -228
  135. data/vendor/spikard-http/src/handler_trait_tests.rs +284 -284
  136. data/vendor/spikard-http/src/lib.rs +529 -529
  137. data/vendor/spikard-http/src/lifecycle/adapter.rs +149 -149
  138. data/vendor/spikard-http/src/lifecycle.rs +428 -428
  139. data/vendor/spikard-http/src/middleware/mod.rs +285 -285
  140. data/vendor/spikard-http/src/middleware/multipart.rs +86 -86
  141. data/vendor/spikard-http/src/middleware/urlencoded.rs +147 -147
  142. data/vendor/spikard-http/src/middleware/validation.rs +287 -287
  143. data/vendor/spikard-http/src/openapi/mod.rs +309 -309
  144. data/vendor/spikard-http/src/openapi/parameter_extraction.rs +190 -190
  145. data/vendor/spikard-http/src/openapi/schema_conversion.rs +308 -308
  146. data/vendor/spikard-http/src/openapi/spec_generation.rs +195 -195
  147. data/vendor/spikard-http/src/parameters.rs +1 -1
  148. data/vendor/spikard-http/src/problem.rs +1 -1
  149. data/vendor/spikard-http/src/query_parser.rs +369 -369
  150. data/vendor/spikard-http/src/response.rs +399 -399
  151. data/vendor/spikard-http/src/router.rs +1 -1
  152. data/vendor/spikard-http/src/schema_registry.rs +1 -1
  153. data/vendor/spikard-http/src/server/handler.rs +80 -80
  154. data/vendor/spikard-http/src/server/lifecycle_execution.rs +98 -98
  155. data/vendor/spikard-http/src/server/mod.rs +805 -805
  156. data/vendor/spikard-http/src/server/request_extraction.rs +119 -119
  157. data/vendor/spikard-http/src/sse.rs +447 -447
  158. data/vendor/spikard-http/src/testing/form.rs +14 -14
  159. data/vendor/spikard-http/src/testing/multipart.rs +60 -60
  160. data/vendor/spikard-http/src/testing/test_client.rs +285 -285
  161. data/vendor/spikard-http/src/testing.rs +377 -377
  162. data/vendor/spikard-http/src/type_hints.rs +1 -1
  163. data/vendor/spikard-http/src/validation.rs +1 -1
  164. data/vendor/spikard-http/src/websocket.rs +324 -324
  165. data/vendor/spikard-rb/Cargo.toml +42 -42
  166. data/vendor/spikard-rb/build.rs +8 -8
  167. data/vendor/spikard-rb/src/background.rs +63 -63
  168. data/vendor/spikard-rb/src/config.rs +294 -294
  169. data/vendor/spikard-rb/src/conversion.rs +392 -392
  170. data/vendor/spikard-rb/src/di.rs +409 -409
  171. data/vendor/spikard-rb/src/handler.rs +534 -534
  172. data/vendor/spikard-rb/src/lib.rs +2020 -2020
  173. data/vendor/spikard-rb/src/lifecycle.rs +267 -267
  174. data/vendor/spikard-rb/src/server.rs +283 -283
  175. data/vendor/spikard-rb/src/sse.rs +231 -231
  176. data/vendor/spikard-rb/src/test_client.rs +404 -404
  177. data/vendor/spikard-rb/src/test_sse.rs +143 -143
  178. data/vendor/spikard-rb/src/test_websocket.rs +221 -221
  179. data/vendor/spikard-rb/src/websocket.rs +233 -233
  180. 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
+ }