spikard 0.5.0 → 0.6.1

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 (135) hide show
  1. checksums.yaml +4 -4
  2. data/LICENSE +1 -1
  3. data/README.md +674 -674
  4. data/ext/spikard_rb/Cargo.toml +17 -17
  5. data/ext/spikard_rb/extconf.rb +13 -10
  6. data/ext/spikard_rb/src/lib.rs +6 -6
  7. data/lib/spikard/app.rb +405 -405
  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 +256 -256
  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 +366 -366
  23. data/vendor/crates/spikard-bindings-shared/Cargo.toml +63 -63
  24. data/vendor/crates/spikard-bindings-shared/examples/config_extraction.rs +132 -132
  25. data/vendor/crates/spikard-bindings-shared/src/config_extractor.rs +752 -752
  26. data/vendor/crates/spikard-bindings-shared/src/conversion_traits.rs +194 -194
  27. data/vendor/crates/spikard-bindings-shared/src/di_traits.rs +246 -246
  28. data/vendor/crates/spikard-bindings-shared/src/error_response.rs +401 -401
  29. data/vendor/crates/spikard-bindings-shared/src/handler_base.rs +238 -238
  30. data/vendor/crates/spikard-bindings-shared/src/lib.rs +24 -24
  31. data/vendor/crates/spikard-bindings-shared/src/lifecycle_base.rs +292 -292
  32. data/vendor/crates/spikard-bindings-shared/src/lifecycle_executor.rs +616 -616
  33. data/vendor/crates/spikard-bindings-shared/src/response_builder.rs +305 -305
  34. data/vendor/crates/spikard-bindings-shared/src/test_client_base.rs +248 -248
  35. data/vendor/crates/spikard-bindings-shared/src/validation_helpers.rs +351 -351
  36. data/vendor/crates/spikard-bindings-shared/tests/comprehensive_coverage.rs +454 -454
  37. data/vendor/crates/spikard-bindings-shared/tests/error_response_edge_cases.rs +383 -383
  38. data/vendor/crates/spikard-bindings-shared/tests/handler_base_integration.rs +280 -280
  39. data/vendor/crates/spikard-core/Cargo.toml +40 -40
  40. data/vendor/crates/spikard-core/src/bindings/mod.rs +3 -3
  41. data/vendor/crates/spikard-core/src/bindings/response.rs +133 -133
  42. data/vendor/crates/spikard-core/src/debug.rs +127 -127
  43. data/vendor/crates/spikard-core/src/di/container.rs +702 -702
  44. data/vendor/crates/spikard-core/src/di/dependency.rs +273 -273
  45. data/vendor/crates/spikard-core/src/di/error.rs +118 -118
  46. data/vendor/crates/spikard-core/src/di/factory.rs +534 -534
  47. data/vendor/crates/spikard-core/src/di/graph.rs +506 -506
  48. data/vendor/crates/spikard-core/src/di/mod.rs +192 -192
  49. data/vendor/crates/spikard-core/src/di/resolved.rs +405 -405
  50. data/vendor/crates/spikard-core/src/di/value.rs +281 -281
  51. data/vendor/crates/spikard-core/src/errors.rs +69 -69
  52. data/vendor/crates/spikard-core/src/http.rs +415 -415
  53. data/vendor/crates/spikard-core/src/lib.rs +29 -29
  54. data/vendor/crates/spikard-core/src/lifecycle.rs +1186 -1186
  55. data/vendor/crates/spikard-core/src/metadata.rs +389 -389
  56. data/vendor/crates/spikard-core/src/parameters.rs +2525 -2525
  57. data/vendor/crates/spikard-core/src/problem.rs +344 -344
  58. data/vendor/crates/spikard-core/src/request_data.rs +1154 -1154
  59. data/vendor/crates/spikard-core/src/router.rs +510 -510
  60. data/vendor/crates/spikard-core/src/schema_registry.rs +183 -183
  61. data/vendor/crates/spikard-core/src/type_hints.rs +304 -304
  62. data/vendor/crates/spikard-core/src/validation/error_mapper.rs +696 -688
  63. data/vendor/crates/spikard-core/src/validation/mod.rs +457 -457
  64. data/vendor/crates/spikard-http/Cargo.toml +62 -64
  65. data/vendor/crates/spikard-http/examples/sse-notifications.rs +148 -148
  66. data/vendor/crates/spikard-http/examples/websocket-chat.rs +92 -92
  67. data/vendor/crates/spikard-http/src/auth.rs +296 -296
  68. data/vendor/crates/spikard-http/src/background.rs +1860 -1860
  69. data/vendor/crates/spikard-http/src/bindings/mod.rs +3 -3
  70. data/vendor/crates/spikard-http/src/bindings/response.rs +1 -1
  71. data/vendor/crates/spikard-http/src/body_metadata.rs +8 -8
  72. data/vendor/crates/spikard-http/src/cors.rs +1005 -1005
  73. data/vendor/crates/spikard-http/src/debug.rs +128 -128
  74. data/vendor/crates/spikard-http/src/di_handler.rs +1668 -1668
  75. data/vendor/crates/spikard-http/src/handler_response.rs +901 -901
  76. data/vendor/crates/spikard-http/src/handler_trait.rs +838 -830
  77. data/vendor/crates/spikard-http/src/handler_trait_tests.rs +290 -290
  78. data/vendor/crates/spikard-http/src/lib.rs +534 -534
  79. data/vendor/crates/spikard-http/src/lifecycle/adapter.rs +230 -230
  80. data/vendor/crates/spikard-http/src/lifecycle.rs +1193 -1193
  81. data/vendor/crates/spikard-http/src/middleware/mod.rs +560 -540
  82. data/vendor/crates/spikard-http/src/middleware/multipart.rs +912 -912
  83. data/vendor/crates/spikard-http/src/middleware/urlencoded.rs +513 -513
  84. data/vendor/crates/spikard-http/src/middleware/validation.rs +768 -735
  85. data/vendor/crates/spikard-http/src/openapi/mod.rs +309 -309
  86. data/vendor/crates/spikard-http/src/openapi/parameter_extraction.rs +535 -535
  87. data/vendor/crates/spikard-http/src/openapi/schema_conversion.rs +1363 -1363
  88. data/vendor/crates/spikard-http/src/openapi/spec_generation.rs +665 -665
  89. data/vendor/crates/spikard-http/src/query_parser.rs +793 -793
  90. data/vendor/crates/spikard-http/src/response.rs +720 -720
  91. data/vendor/crates/spikard-http/src/server/handler.rs +1650 -1650
  92. data/vendor/crates/spikard-http/src/server/lifecycle_execution.rs +234 -234
  93. data/vendor/crates/spikard-http/src/server/mod.rs +1593 -1502
  94. data/vendor/crates/spikard-http/src/server/request_extraction.rs +789 -770
  95. data/vendor/crates/spikard-http/src/server/routing_factory.rs +629 -599
  96. data/vendor/crates/spikard-http/src/sse.rs +1409 -1409
  97. data/vendor/crates/spikard-http/src/testing/form.rs +52 -52
  98. data/vendor/crates/spikard-http/src/testing/multipart.rs +64 -60
  99. data/vendor/crates/spikard-http/src/testing/test_client.rs +311 -283
  100. data/vendor/crates/spikard-http/src/testing.rs +406 -377
  101. data/vendor/crates/spikard-http/src/websocket.rs +1404 -1375
  102. data/vendor/crates/spikard-http/tests/background_behavior.rs +832 -832
  103. data/vendor/crates/spikard-http/tests/common/handlers.rs +309 -309
  104. data/vendor/crates/spikard-http/tests/common/mod.rs +26 -26
  105. data/vendor/crates/spikard-http/tests/di_integration.rs +192 -192
  106. data/vendor/crates/spikard-http/tests/doc_snippets.rs +5 -5
  107. data/vendor/crates/spikard-http/tests/lifecycle_execution.rs +1093 -1093
  108. data/vendor/crates/spikard-http/tests/multipart_behavior.rs +656 -656
  109. data/vendor/crates/spikard-http/tests/server_config_builder.rs +314 -314
  110. data/vendor/crates/spikard-http/tests/sse_behavior.rs +620 -620
  111. data/vendor/crates/spikard-http/tests/websocket_behavior.rs +663 -663
  112. data/vendor/crates/spikard-rb/Cargo.toml +48 -48
  113. data/vendor/crates/spikard-rb/build.rs +199 -199
  114. data/vendor/crates/spikard-rb/src/background.rs +63 -63
  115. data/vendor/crates/spikard-rb/src/config/mod.rs +5 -5
  116. data/vendor/crates/spikard-rb/src/config/server_config.rs +285 -285
  117. data/vendor/crates/spikard-rb/src/conversion.rs +554 -554
  118. data/vendor/crates/spikard-rb/src/di/builder.rs +100 -100
  119. data/vendor/crates/spikard-rb/src/di/mod.rs +375 -375
  120. data/vendor/crates/spikard-rb/src/handler.rs +618 -618
  121. data/vendor/crates/spikard-rb/src/integration/mod.rs +3 -3
  122. data/vendor/crates/spikard-rb/src/lib.rs +1806 -1810
  123. data/vendor/crates/spikard-rb/src/lifecycle.rs +275 -275
  124. data/vendor/crates/spikard-rb/src/metadata/mod.rs +5 -5
  125. data/vendor/crates/spikard-rb/src/metadata/route_extraction.rs +442 -447
  126. data/vendor/crates/spikard-rb/src/runtime/mod.rs +5 -5
  127. data/vendor/crates/spikard-rb/src/runtime/server_runner.rs +324 -324
  128. data/vendor/crates/spikard-rb/src/server.rs +305 -308
  129. data/vendor/crates/spikard-rb/src/sse.rs +231 -231
  130. data/vendor/crates/spikard-rb/src/testing/client.rs +538 -551
  131. data/vendor/crates/spikard-rb/src/testing/mod.rs +7 -7
  132. data/vendor/crates/spikard-rb/src/testing/sse.rs +143 -143
  133. data/vendor/crates/spikard-rb/src/testing/websocket.rs +608 -635
  134. data/vendor/crates/spikard-rb/src/websocket.rs +377 -374
  135. metadata +15 -1
@@ -1,702 +1,702 @@
1
- //! Dependency injection container
2
- //!
3
- //! This module provides the main `DependencyContainer` which manages dependency
4
- //! registration, resolution, and caching.
5
-
6
- use super::dependency::Dependency;
7
- use super::error::DependencyError;
8
- use super::graph::DependencyGraph;
9
- use super::resolved::ResolvedDependencies;
10
- use crate::request_data::RequestData;
11
- use http::Request;
12
- use indexmap::IndexMap;
13
- use std::any::Any;
14
- use std::collections::HashMap;
15
- use std::sync::Arc;
16
- use tokio::sync::RwLock;
17
-
18
- /// Main dependency injection container
19
- ///
20
- /// The container manages:
21
- /// - Registration of dependencies with cycle detection
22
- /// - Batched parallel resolution using topological sorting
23
- /// - Singleton caching (global across all requests)
24
- /// - Request-scoped caching (within a single request)
25
- ///
26
- /// # Thread Safety
27
- ///
28
- /// The container is thread-safe and can be shared across multiple threads
29
- /// using `Arc<DependencyContainer>`.
30
- ///
31
- /// # Examples
32
- ///
33
- /// ```ignore
34
- /// use spikard_core::di::{DependencyContainer, ValueDependency};
35
- /// use std::sync::Arc;
36
- ///
37
- /// # tokio_test::block_on(async {
38
- /// let mut container = DependencyContainer::new();
39
- ///
40
- /// // Register a simple value dependency
41
- /// let config = ValueDependency::new("port", 8080u16);
42
- /// container.register("port".to_string(), Arc::new(config)).unwrap();
43
- ///
44
- /// // Resolve dependencies for a handler
45
- /// use http::Request;
46
- /// use crate::request_data::RequestData;
47
- /// use std::collections::HashMap;
48
- ///
49
- /// let request = Request::builder().body(()).unwrap();
50
- /// let request_data = RequestData {
51
- /// path_params: Arc::new(HashMap::new()),
52
- /// query_params: serde_json::Value::Null,
53
- /// validated_params: None,
54
- /// raw_query_params: Arc::new(HashMap::new()),
55
- /// body: serde_json::Value::Null,
56
- /// raw_body: None,
57
- /// headers: Arc::new(HashMap::new()),
58
- /// cookies: Arc::new(HashMap::new()),
59
- /// method: "GET".to_string(),
60
- /// path: "/".to_string(),
61
- /// };
62
- ///
63
- /// let resolved = container
64
- /// .resolve_for_handler(&["port".to_string()], &request, &request_data)
65
- /// .await
66
- /// .unwrap();
67
- ///
68
- /// let port: Option<Arc<u16>> = resolved.get("port");
69
- /// assert_eq!(port.map(|p| *p), Some(8080));
70
- /// # });
71
- /// ```
72
- pub struct DependencyContainer {
73
- /// Registered dependencies by key (preserves insertion order)
74
- dependencies: IndexMap<String, Arc<dyn Dependency>>,
75
- /// Dependency graph for topological sorting and cycle detection
76
- dependency_graph: DependencyGraph,
77
- /// Global singleton cache
78
- singleton_cache: Arc<RwLock<HashMap<String, Arc<dyn Any + Send + Sync>>>>,
79
- }
80
-
81
- impl DependencyContainer {
82
- /// Create a new empty dependency container
83
- ///
84
- /// # Examples
85
- ///
86
- /// ```ignore
87
- /// use spikard_core::di::DependencyContainer;
88
- ///
89
- /// let container = DependencyContainer::new();
90
- /// ```
91
- #[must_use]
92
- pub fn new() -> Self {
93
- Self {
94
- dependencies: IndexMap::new(),
95
- dependency_graph: DependencyGraph::new(),
96
- singleton_cache: Arc::new(RwLock::new(HashMap::new())),
97
- }
98
- }
99
-
100
- /// Register a dependency in the container
101
- ///
102
- /// This will validate that:
103
- /// - The key is not already registered
104
- /// - Adding this dependency won't create a circular dependency
105
- ///
106
- /// # Arguments
107
- ///
108
- /// * `key` - The unique key for this dependency
109
- /// * `dep` - The dependency implementation
110
- ///
111
- /// # Returns
112
- ///
113
- /// Returns `&mut Self` for method chaining.
114
- ///
115
- /// # Errors
116
- ///
117
- /// - `DependencyError::DuplicateKey` if a dependency with this key exists
118
- /// - `DependencyError::CircularDependency` if this would create a cycle
119
- ///
120
- /// # Examples
121
- ///
122
- /// ```ignore
123
- /// use spikard_core::di::{DependencyContainer, ValueDependency};
124
- /// use std::sync::Arc;
125
- ///
126
- /// let mut container = DependencyContainer::new();
127
- ///
128
- /// let config = ValueDependency::new("config", "production".to_string());
129
- /// container.register("config".to_string(), Arc::new(config)).unwrap();
130
- /// ```
131
- pub fn register(&mut self, key: String, dep: Arc<dyn Dependency>) -> Result<&mut Self, DependencyError> {
132
- self.dependency_graph.add_dependency(&key, dep.depends_on())?;
133
-
134
- self.dependencies.insert(key, dep);
135
-
136
- Ok(self)
137
- }
138
-
139
- /// Resolve dependencies for a handler
140
- ///
141
- /// This method:
142
- /// 1. Calculates the optimal batched resolution order using topological sorting
143
- /// 2. Resolves dependencies in batches (dependencies in the same batch run in parallel)
144
- /// 3. Caches singleton dependencies globally
145
- /// 4. Caches per-request dependencies within the returned `ResolvedDependencies`
146
- ///
147
- /// # Arguments
148
- ///
149
- /// * `deps` - The dependency keys needed by the handler
150
- /// * `req` - The HTTP request being handled
151
- /// * `data` - Extracted request data
152
- ///
153
- /// # Returns
154
- ///
155
- /// A `ResolvedDependencies` instance containing all resolved dependencies.
156
- ///
157
- /// # Errors
158
- ///
159
- /// - `DependencyError::NotFound` if a required dependency is not registered
160
- /// - `DependencyError::CircularDependency` if there's a cycle in dependencies
161
- /// - `DependencyError::ResolutionFailed` if a dependency fails to resolve
162
- ///
163
- /// # Examples
164
- ///
165
- /// ```ignore
166
- /// use spikard_core::di::{DependencyContainer, ValueDependency, FactoryDependency};
167
- /// use std::sync::Arc;
168
- ///
169
- /// # tokio_test::block_on(async {
170
- /// let mut container = DependencyContainer::new();
171
- ///
172
- /// // Register dependencies
173
- /// let config = ValueDependency::new("config", "production".to_string());
174
- /// container.register("config".to_string(), Arc::new(config)).unwrap();
175
- ///
176
- /// let db = FactoryDependency::builder("database")
177
- /// .depends_on(vec!["config".to_string()])
178
- /// .factory(|_req, _data, resolved| {
179
- /// Box::pin(async move {
180
- /// let config: Arc<String> = resolved.get("config").unwrap();
181
- /// let db = format!("DB connected to {}", *config);
182
- /// Ok(Arc::new(db) as Arc<dyn std::any::Any + Send + Sync>)
183
- /// })
184
- /// })
185
- /// .build();
186
- /// container.register("database".to_string(), Arc::new(db)).unwrap();
187
- ///
188
- /// // Resolve for handler
189
- /// use http::Request;
190
- /// use crate::request_data::RequestData;
191
- /// use std::collections::HashMap;
192
- ///
193
- /// let request = Request::builder().body(()).unwrap();
194
- /// let request_data = RequestData {
195
- /// path_params: Arc::new(HashMap::new()),
196
- /// query_params: serde_json::Value::Null,
197
- /// validated_params: None,
198
- /// raw_query_params: Arc::new(HashMap::new()),
199
- /// body: serde_json::Value::Null,
200
- /// raw_body: None,
201
- /// headers: Arc::new(HashMap::new()),
202
- /// cookies: Arc::new(HashMap::new()),
203
- /// method: "GET".to_string(),
204
- /// path: "/".to_string(),
205
- /// };
206
- ///
207
- /// let resolved = container
208
- /// .resolve_for_handler(&["database".to_string()], &request, &request_data)
209
- /// .await
210
- /// .unwrap();
211
- ///
212
- /// let db: Option<Arc<String>> = resolved.get("database");
213
- /// assert!(db.is_some());
214
- /// # });
215
- /// ```
216
- pub async fn resolve_for_handler(
217
- &self,
218
- deps: &[String],
219
- req: &Request<()>,
220
- data: &RequestData,
221
- ) -> Result<ResolvedDependencies, DependencyError> {
222
- for key in deps {
223
- if !self.dependencies.contains_key(key) {
224
- return Err(DependencyError::NotFound { key: key.clone() });
225
- }
226
- }
227
-
228
- let batches = self.dependency_graph.calculate_batches(deps)?;
229
-
230
- let mut resolved = ResolvedDependencies::new();
231
- let mut request_cache: HashMap<String, Arc<dyn Any + Send + Sync>> = HashMap::new();
232
-
233
- for batch in batches {
234
- // NOTE: We resolve sequentially within each batch to ensure cleanup tasks
235
- let mut sorted_keys: Vec<_> = batch.iter().collect();
236
-
237
- sorted_keys.sort_by_key(|key| self.dependencies.get_index_of(*key).unwrap_or(usize::MAX));
238
-
239
- for key in sorted_keys {
240
- let dep = self
241
- .dependencies
242
- .get(key)
243
- .ok_or_else(|| DependencyError::NotFound { key: key.clone() })?;
244
-
245
- if dep.singleton() {
246
- let cache = self.singleton_cache.read().await;
247
- if let Some(cached) = cache.get(key) {
248
- resolved.insert(key.clone(), Arc::clone(cached));
249
- continue;
250
- }
251
- }
252
-
253
- if dep.cacheable()
254
- && let Some(cached) = request_cache.get(key)
255
- {
256
- resolved.insert(key.clone(), Arc::clone(cached));
257
- continue;
258
- }
259
-
260
- let result = dep.resolve(req, data, &resolved).await?;
261
-
262
- if dep.singleton() {
263
- let mut cache = self.singleton_cache.write().await;
264
- cache.insert(key.clone(), Arc::clone(&result));
265
- } else if dep.cacheable() {
266
- request_cache.insert(key.clone(), Arc::clone(&result));
267
- }
268
-
269
- resolved.insert(key.clone(), result);
270
- }
271
- }
272
-
273
- Ok(resolved)
274
- }
275
-
276
- /// Get the number of registered dependencies
277
- ///
278
- /// # Examples
279
- ///
280
- /// ```ignore
281
- /// use spikard_core::di::{DependencyContainer, ValueDependency};
282
- /// use std::sync::Arc;
283
- ///
284
- /// let mut container = DependencyContainer::new();
285
- /// assert_eq!(container.len(), 0);
286
- ///
287
- /// let dep = ValueDependency::new("test", 42);
288
- /// container.register("test".to_string(), Arc::new(dep)).unwrap();
289
- /// assert_eq!(container.len(), 1);
290
- /// ```
291
- #[must_use]
292
- pub fn len(&self) -> usize {
293
- self.dependencies.len()
294
- }
295
-
296
- /// Check if the container is empty
297
- ///
298
- /// # Examples
299
- ///
300
- /// ```ignore
301
- /// use spikard_core::di::DependencyContainer;
302
- ///
303
- /// let container = DependencyContainer::new();
304
- /// assert!(container.is_empty());
305
- /// ```
306
- #[must_use]
307
- pub fn is_empty(&self) -> bool {
308
- self.dependencies.is_empty()
309
- }
310
-
311
- /// Check if a dependency is registered
312
- ///
313
- /// # Examples
314
- ///
315
- /// ```ignore
316
- /// use spikard_core::di::{DependencyContainer, ValueDependency};
317
- /// use std::sync::Arc;
318
- ///
319
- /// let mut container = DependencyContainer::new();
320
- /// assert!(!container.contains("config"));
321
- ///
322
- /// let dep = ValueDependency::new("config", "value");
323
- /// container.register("config".to_string(), Arc::new(dep)).unwrap();
324
- /// assert!(container.contains("config"));
325
- /// ```
326
- #[must_use]
327
- pub fn contains(&self, key: &str) -> bool {
328
- self.dependencies.contains_key(key)
329
- }
330
-
331
- /// Get the keys of all registered dependencies
332
- #[must_use]
333
- pub fn keys(&self) -> Vec<String> {
334
- self.dependencies.keys().cloned().collect()
335
- }
336
-
337
- /// Clear the singleton cache
338
- ///
339
- /// This is useful for testing or when you need to force re-resolution
340
- /// of singleton dependencies.
341
- ///
342
- /// # Examples
343
- ///
344
- /// ```ignore
345
- /// use spikard_core::di::DependencyContainer;
346
- ///
347
- /// # tokio_test::block_on(async {
348
- /// let container = DependencyContainer::new();
349
- /// container.clear_singleton_cache().await;
350
- /// # });
351
- /// ```
352
- pub async fn clear_singleton_cache(&self) {
353
- let mut cache = self.singleton_cache.write().await;
354
- cache.clear();
355
- }
356
- }
357
-
358
- impl Default for DependencyContainer {
359
- fn default() -> Self {
360
- Self::new()
361
- }
362
- }
363
-
364
- impl std::fmt::Debug for DependencyContainer {
365
- fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
366
- f.debug_struct("DependencyContainer")
367
- .field("dependencies", &self.dependencies.keys())
368
- .finish()
369
- }
370
- }
371
-
372
- #[cfg(test)]
373
- mod tests {
374
- use super::*;
375
- use crate::di::{FactoryDependency, ValueDependency};
376
- use std::collections::HashMap;
377
- use std::sync::atomic::{AtomicU32, Ordering};
378
-
379
- fn make_request() -> Request<()> {
380
- Request::builder().body(()).unwrap()
381
- }
382
-
383
- fn make_request_data() -> RequestData {
384
- RequestData {
385
- path_params: Arc::new(HashMap::new()),
386
- query_params: serde_json::Value::Null,
387
- validated_params: None,
388
- raw_query_params: Arc::new(HashMap::new()),
389
- body: serde_json::Value::Null,
390
- raw_body: None,
391
- headers: Arc::new(HashMap::new()),
392
- cookies: Arc::new(HashMap::new()),
393
- method: "GET".to_string(),
394
- path: "/".to_string(),
395
- #[cfg(feature = "di")]
396
- dependencies: None,
397
- }
398
- }
399
-
400
- #[test]
401
- fn test_new() {
402
- let container = DependencyContainer::new();
403
- assert!(container.is_empty());
404
- assert_eq!(container.len(), 0);
405
- }
406
-
407
- #[test]
408
- fn test_register_simple() {
409
- let mut container = DependencyContainer::new();
410
- let dep = ValueDependency::new("test", 42i32);
411
-
412
- assert!(container.register("test".to_string(), Arc::new(dep)).is_ok());
413
- assert_eq!(container.len(), 1);
414
- assert!(container.contains("test"));
415
- }
416
-
417
- #[test]
418
- fn test_register_duplicate() {
419
- let mut container = DependencyContainer::new();
420
- let dep1 = ValueDependency::new("test", 42i32);
421
- let dep2 = ValueDependency::new("test", 100i32);
422
-
423
- container.register("test".to_string(), Arc::new(dep1)).unwrap();
424
- let result = container.register("test".to_string(), Arc::new(dep2));
425
-
426
- assert!(matches!(result, Err(DependencyError::DuplicateKey { .. })));
427
- }
428
-
429
- #[tokio::test]
430
- async fn test_register_circular() {
431
- let mut container = DependencyContainer::new();
432
-
433
- let dep_a = FactoryDependency::builder("a")
434
- .depends_on(vec!["b".to_string()])
435
- .factory(|_req, _data, _resolved| Box::pin(async { Ok(Arc::new(1i32) as Arc<dyn Any + Send + Sync>) }))
436
- .build();
437
-
438
- let dep_b = FactoryDependency::builder("b")
439
- .depends_on(vec!["a".to_string()])
440
- .factory(|_req, _data, _resolved| Box::pin(async { Ok(Arc::new(2i32) as Arc<dyn Any + Send + Sync>) }))
441
- .build();
442
-
443
- container.register("a".to_string(), Arc::new(dep_a)).unwrap();
444
- container.register("b".to_string(), Arc::new(dep_b)).unwrap();
445
-
446
- let request = make_request();
447
- let request_data = make_request_data();
448
- let result = container
449
- .resolve_for_handler(&["a".to_string()], &request, &request_data)
450
- .await;
451
-
452
- assert!(matches!(result, Err(DependencyError::CircularDependency { .. })));
453
- }
454
-
455
- #[tokio::test]
456
- async fn test_resolve_value() {
457
- let mut container = DependencyContainer::new();
458
- let dep = ValueDependency::new("answer", 42i32);
459
- container.register("answer".to_string(), Arc::new(dep)).unwrap();
460
-
461
- let request = make_request();
462
- let request_data = make_request_data();
463
-
464
- let resolved = container
465
- .resolve_for_handler(&["answer".to_string()], &request, &request_data)
466
- .await
467
- .unwrap();
468
-
469
- let value: Option<Arc<i32>> = resolved.get("answer");
470
- assert_eq!(value.map(|v| *v), Some(42));
471
- }
472
-
473
- #[tokio::test]
474
- async fn test_resolve_factory() {
475
- let mut container = DependencyContainer::new();
476
-
477
- let factory = FactoryDependency::builder("computed")
478
- .factory(|_req, _data, _resolved| Box::pin(async { Ok(Arc::new(100i32) as Arc<dyn Any + Send + Sync>) }))
479
- .build();
480
-
481
- container.register("computed".to_string(), Arc::new(factory)).unwrap();
482
-
483
- let request = make_request();
484
- let request_data = make_request_data();
485
-
486
- let resolved = container
487
- .resolve_for_handler(&["computed".to_string()], &request, &request_data)
488
- .await
489
- .unwrap();
490
-
491
- let value: Option<Arc<i32>> = resolved.get("computed");
492
- assert_eq!(value.map(|v| *v), Some(100));
493
- }
494
-
495
- #[tokio::test]
496
- async fn test_resolve_nested() {
497
- let mut container = DependencyContainer::new();
498
-
499
- let config = ValueDependency::new("config", "production".to_string());
500
- container.register("config".to_string(), Arc::new(config)).unwrap();
501
-
502
- let database = FactoryDependency::builder("database")
503
- .depends_on(vec!["config".to_string()])
504
- .factory(|_req, _data, resolved| {
505
- let resolved = resolved.clone();
506
- Box::pin(async move {
507
- let config: Arc<String> = resolved.get("config").unwrap();
508
- let db = format!("DB:{}", *config);
509
- Ok(Arc::new(db) as Arc<dyn Any + Send + Sync>)
510
- })
511
- })
512
- .build();
513
- container.register("database".to_string(), Arc::new(database)).unwrap();
514
-
515
- let request = make_request();
516
- let request_data = make_request_data();
517
-
518
- let resolved = container
519
- .resolve_for_handler(&["database".to_string()], &request, &request_data)
520
- .await
521
- .unwrap();
522
-
523
- let db: Option<Arc<String>> = resolved.get("database");
524
- assert_eq!(db.as_ref().map(|v| v.as_str()), Some("DB:production"));
525
- }
526
-
527
- #[tokio::test]
528
- async fn test_resolve_batched() {
529
- let mut container = DependencyContainer::new();
530
-
531
- let counter = Arc::new(AtomicU32::new(0));
532
-
533
- let counter1 = Arc::clone(&counter);
534
- let config = FactoryDependency::builder("config")
535
- .factory(move |_req, _data, _resolved| {
536
- let c = Arc::clone(&counter1);
537
- Box::pin(async move {
538
- let order = c.fetch_add(1, Ordering::SeqCst);
539
- Ok(Arc::new(order) as Arc<dyn Any + Send + Sync>)
540
- })
541
- })
542
- .build();
543
- container.register("config".to_string(), Arc::new(config)).unwrap();
544
-
545
- let counter2 = Arc::clone(&counter);
546
- let database = FactoryDependency::builder("database")
547
- .depends_on(vec!["config".to_string()])
548
- .factory(move |_req, _data, _resolved| {
549
- let c = Arc::clone(&counter2);
550
- Box::pin(async move {
551
- let order = c.fetch_add(1, Ordering::SeqCst);
552
- Ok(Arc::new(order) as Arc<dyn Any + Send + Sync>)
553
- })
554
- })
555
- .build();
556
- container.register("database".to_string(), Arc::new(database)).unwrap();
557
-
558
- let counter3 = Arc::clone(&counter);
559
- let cache = FactoryDependency::builder("cache")
560
- .depends_on(vec!["config".to_string()])
561
- .factory(move |_req, _data, _resolved| {
562
- let c = Arc::clone(&counter3);
563
- Box::pin(async move {
564
- let order = c.fetch_add(1, Ordering::SeqCst);
565
- Ok(Arc::new(order) as Arc<dyn Any + Send + Sync>)
566
- })
567
- })
568
- .build();
569
- container.register("cache".to_string(), Arc::new(cache)).unwrap();
570
-
571
- let request = make_request();
572
- let request_data = make_request_data();
573
-
574
- let resolved = container
575
- .resolve_for_handler(&["database".to_string(), "cache".to_string()], &request, &request_data)
576
- .await
577
- .unwrap();
578
-
579
- let config_order: Arc<u32> = resolved.get("config").unwrap();
580
- assert_eq!(*config_order, 0);
581
-
582
- let db_order: Arc<u32> = resolved.get("database").unwrap();
583
- let cache_order: Arc<u32> = resolved.get("cache").unwrap();
584
- assert!(*db_order >= 1);
585
- assert!(*cache_order >= 1);
586
- }
587
-
588
- #[tokio::test]
589
- async fn test_singleton_cache() {
590
- let mut container = DependencyContainer::new();
591
-
592
- let counter = Arc::new(AtomicU32::new(0));
593
- let counter_clone = Arc::clone(&counter);
594
-
595
- let singleton = FactoryDependency::builder("singleton")
596
- .singleton(true)
597
- .factory(move |_req, _data, _resolved| {
598
- let c = Arc::clone(&counter_clone);
599
- Box::pin(async move {
600
- let value = c.fetch_add(1, Ordering::SeqCst);
601
- Ok(Arc::new(value) as Arc<dyn Any + Send + Sync>)
602
- })
603
- })
604
- .build();
605
-
606
- container
607
- .register("singleton".to_string(), Arc::new(singleton))
608
- .unwrap();
609
-
610
- let request = make_request();
611
- let request_data = make_request_data();
612
-
613
- for _ in 0..3 {
614
- let resolved = container
615
- .resolve_for_handler(&["singleton".to_string()], &request, &request_data)
616
- .await
617
- .unwrap();
618
-
619
- let value: Arc<u32> = resolved.get("singleton").unwrap();
620
- assert_eq!(*value, 0);
621
- }
622
-
623
- assert_eq!(counter.load(Ordering::SeqCst), 1);
624
- }
625
-
626
- #[tokio::test]
627
- async fn test_clear_singleton_cache() {
628
- let mut container = DependencyContainer::new();
629
-
630
- let counter = Arc::new(AtomicU32::new(0));
631
- let counter_clone = Arc::clone(&counter);
632
-
633
- let singleton = FactoryDependency::builder("singleton")
634
- .singleton(true)
635
- .factory(move |_req, _data, _resolved| {
636
- let c = Arc::clone(&counter_clone);
637
- Box::pin(async move {
638
- let value = c.fetch_add(1, Ordering::SeqCst);
639
- Ok(Arc::new(value) as Arc<dyn Any + Send + Sync>)
640
- })
641
- })
642
- .build();
643
-
644
- container
645
- .register("singleton".to_string(), Arc::new(singleton))
646
- .unwrap();
647
-
648
- let request = make_request();
649
- let request_data = make_request_data();
650
-
651
- let resolved1 = container
652
- .resolve_for_handler(&["singleton".to_string()], &request, &request_data)
653
- .await
654
- .unwrap();
655
- let value1: Arc<u32> = resolved1.get("singleton").unwrap();
656
- assert_eq!(*value1, 0);
657
-
658
- container.clear_singleton_cache().await;
659
-
660
- let resolved2 = container
661
- .resolve_for_handler(&["singleton".to_string()], &request, &request_data)
662
- .await
663
- .unwrap();
664
- let value2: Arc<u32> = resolved2.get("singleton").unwrap();
665
- assert_eq!(*value2, 1);
666
- }
667
-
668
- #[tokio::test]
669
- async fn test_resolve_not_found() {
670
- let container = DependencyContainer::new();
671
- let request = make_request();
672
- let request_data = make_request_data();
673
-
674
- let result = container
675
- .resolve_for_handler(&["missing".to_string()], &request, &request_data)
676
- .await;
677
-
678
- assert!(matches!(result, Err(DependencyError::NotFound { .. })));
679
- }
680
-
681
- #[test]
682
- fn test_contains() {
683
- let mut container = DependencyContainer::new();
684
- assert!(!container.contains("test"));
685
-
686
- let dep = ValueDependency::new("test", 42i32);
687
- container.register("test".to_string(), Arc::new(dep)).unwrap();
688
-
689
- assert!(container.contains("test"));
690
- assert!(!container.contains("other"));
691
- }
692
-
693
- #[test]
694
- fn test_debug() {
695
- let mut container = DependencyContainer::new();
696
- let dep = ValueDependency::new("test", 42i32);
697
- container.register("test".to_string(), Arc::new(dep)).unwrap();
698
-
699
- let debug_str = format!("{:?}", container);
700
- assert!(debug_str.contains("DependencyContainer"));
701
- }
702
- }
1
+ //! Dependency injection container
2
+ //!
3
+ //! This module provides the main `DependencyContainer` which manages dependency
4
+ //! registration, resolution, and caching.
5
+
6
+ use super::dependency::Dependency;
7
+ use super::error::DependencyError;
8
+ use super::graph::DependencyGraph;
9
+ use super::resolved::ResolvedDependencies;
10
+ use crate::request_data::RequestData;
11
+ use http::Request;
12
+ use indexmap::IndexMap;
13
+ use std::any::Any;
14
+ use std::collections::HashMap;
15
+ use std::sync::Arc;
16
+ use tokio::sync::RwLock;
17
+
18
+ /// Main dependency injection container
19
+ ///
20
+ /// The container manages:
21
+ /// - Registration of dependencies with cycle detection
22
+ /// - Batched parallel resolution using topological sorting
23
+ /// - Singleton caching (global across all requests)
24
+ /// - Request-scoped caching (within a single request)
25
+ ///
26
+ /// # Thread Safety
27
+ ///
28
+ /// The container is thread-safe and can be shared across multiple threads
29
+ /// using `Arc<DependencyContainer>`.
30
+ ///
31
+ /// # Examples
32
+ ///
33
+ /// ```ignore
34
+ /// use spikard_core::di::{DependencyContainer, ValueDependency};
35
+ /// use std::sync::Arc;
36
+ ///
37
+ /// # tokio_test::block_on(async {
38
+ /// let mut container = DependencyContainer::new();
39
+ ///
40
+ /// // Register a simple value dependency
41
+ /// let config = ValueDependency::new("port", 8080u16);
42
+ /// container.register("port".to_string(), Arc::new(config)).unwrap();
43
+ ///
44
+ /// // Resolve dependencies for a handler
45
+ /// use http::Request;
46
+ /// use crate::request_data::RequestData;
47
+ /// use std::collections::HashMap;
48
+ ///
49
+ /// let request = Request::builder().body(()).unwrap();
50
+ /// let request_data = RequestData {
51
+ /// path_params: Arc::new(HashMap::new()),
52
+ /// query_params: serde_json::Value::Null,
53
+ /// validated_params: None,
54
+ /// raw_query_params: Arc::new(HashMap::new()),
55
+ /// body: serde_json::Value::Null,
56
+ /// raw_body: None,
57
+ /// headers: Arc::new(HashMap::new()),
58
+ /// cookies: Arc::new(HashMap::new()),
59
+ /// method: "GET".to_string(),
60
+ /// path: "/".to_string(),
61
+ /// };
62
+ ///
63
+ /// let resolved = container
64
+ /// .resolve_for_handler(&["port".to_string()], &request, &request_data)
65
+ /// .await
66
+ /// .unwrap();
67
+ ///
68
+ /// let port: Option<Arc<u16>> = resolved.get("port");
69
+ /// assert_eq!(port.map(|p| *p), Some(8080));
70
+ /// # });
71
+ /// ```
72
+ pub struct DependencyContainer {
73
+ /// Registered dependencies by key (preserves insertion order)
74
+ dependencies: IndexMap<String, Arc<dyn Dependency>>,
75
+ /// Dependency graph for topological sorting and cycle detection
76
+ dependency_graph: DependencyGraph,
77
+ /// Global singleton cache
78
+ singleton_cache: Arc<RwLock<HashMap<String, Arc<dyn Any + Send + Sync>>>>,
79
+ }
80
+
81
+ impl DependencyContainer {
82
+ /// Create a new empty dependency container
83
+ ///
84
+ /// # Examples
85
+ ///
86
+ /// ```ignore
87
+ /// use spikard_core::di::DependencyContainer;
88
+ ///
89
+ /// let container = DependencyContainer::new();
90
+ /// ```
91
+ #[must_use]
92
+ pub fn new() -> Self {
93
+ Self {
94
+ dependencies: IndexMap::new(),
95
+ dependency_graph: DependencyGraph::new(),
96
+ singleton_cache: Arc::new(RwLock::new(HashMap::new())),
97
+ }
98
+ }
99
+
100
+ /// Register a dependency in the container
101
+ ///
102
+ /// This will validate that:
103
+ /// - The key is not already registered
104
+ /// - Adding this dependency won't create a circular dependency
105
+ ///
106
+ /// # Arguments
107
+ ///
108
+ /// * `key` - The unique key for this dependency
109
+ /// * `dep` - The dependency implementation
110
+ ///
111
+ /// # Returns
112
+ ///
113
+ /// Returns `&mut Self` for method chaining.
114
+ ///
115
+ /// # Errors
116
+ ///
117
+ /// - `DependencyError::DuplicateKey` if a dependency with this key exists
118
+ /// - `DependencyError::CircularDependency` if this would create a cycle
119
+ ///
120
+ /// # Examples
121
+ ///
122
+ /// ```ignore
123
+ /// use spikard_core::di::{DependencyContainer, ValueDependency};
124
+ /// use std::sync::Arc;
125
+ ///
126
+ /// let mut container = DependencyContainer::new();
127
+ ///
128
+ /// let config = ValueDependency::new("config", "production".to_string());
129
+ /// container.register("config".to_string(), Arc::new(config)).unwrap();
130
+ /// ```
131
+ pub fn register(&mut self, key: String, dep: Arc<dyn Dependency>) -> Result<&mut Self, DependencyError> {
132
+ self.dependency_graph.add_dependency(&key, dep.depends_on())?;
133
+
134
+ self.dependencies.insert(key, dep);
135
+
136
+ Ok(self)
137
+ }
138
+
139
+ /// Resolve dependencies for a handler
140
+ ///
141
+ /// This method:
142
+ /// 1. Calculates the optimal batched resolution order using topological sorting
143
+ /// 2. Resolves dependencies in batches (dependencies in the same batch run in parallel)
144
+ /// 3. Caches singleton dependencies globally
145
+ /// 4. Caches per-request dependencies within the returned `ResolvedDependencies`
146
+ ///
147
+ /// # Arguments
148
+ ///
149
+ /// * `deps` - The dependency keys needed by the handler
150
+ /// * `req` - The HTTP request being handled
151
+ /// * `data` - Extracted request data
152
+ ///
153
+ /// # Returns
154
+ ///
155
+ /// A `ResolvedDependencies` instance containing all resolved dependencies.
156
+ ///
157
+ /// # Errors
158
+ ///
159
+ /// - `DependencyError::NotFound` if a required dependency is not registered
160
+ /// - `DependencyError::CircularDependency` if there's a cycle in dependencies
161
+ /// - `DependencyError::ResolutionFailed` if a dependency fails to resolve
162
+ ///
163
+ /// # Examples
164
+ ///
165
+ /// ```ignore
166
+ /// use spikard_core::di::{DependencyContainer, ValueDependency, FactoryDependency};
167
+ /// use std::sync::Arc;
168
+ ///
169
+ /// # tokio_test::block_on(async {
170
+ /// let mut container = DependencyContainer::new();
171
+ ///
172
+ /// // Register dependencies
173
+ /// let config = ValueDependency::new("config", "production".to_string());
174
+ /// container.register("config".to_string(), Arc::new(config)).unwrap();
175
+ ///
176
+ /// let db = FactoryDependency::builder("database")
177
+ /// .depends_on(vec!["config".to_string()])
178
+ /// .factory(|_req, _data, resolved| {
179
+ /// Box::pin(async move {
180
+ /// let config: Arc<String> = resolved.get("config").unwrap();
181
+ /// let db = format!("DB connected to {}", *config);
182
+ /// Ok(Arc::new(db) as Arc<dyn std::any::Any + Send + Sync>)
183
+ /// })
184
+ /// })
185
+ /// .build();
186
+ /// container.register("database".to_string(), Arc::new(db)).unwrap();
187
+ ///
188
+ /// // Resolve for handler
189
+ /// use http::Request;
190
+ /// use crate::request_data::RequestData;
191
+ /// use std::collections::HashMap;
192
+ ///
193
+ /// let request = Request::builder().body(()).unwrap();
194
+ /// let request_data = RequestData {
195
+ /// path_params: Arc::new(HashMap::new()),
196
+ /// query_params: serde_json::Value::Null,
197
+ /// validated_params: None,
198
+ /// raw_query_params: Arc::new(HashMap::new()),
199
+ /// body: serde_json::Value::Null,
200
+ /// raw_body: None,
201
+ /// headers: Arc::new(HashMap::new()),
202
+ /// cookies: Arc::new(HashMap::new()),
203
+ /// method: "GET".to_string(),
204
+ /// path: "/".to_string(),
205
+ /// };
206
+ ///
207
+ /// let resolved = container
208
+ /// .resolve_for_handler(&["database".to_string()], &request, &request_data)
209
+ /// .await
210
+ /// .unwrap();
211
+ ///
212
+ /// let db: Option<Arc<String>> = resolved.get("database");
213
+ /// assert!(db.is_some());
214
+ /// # });
215
+ /// ```
216
+ pub async fn resolve_for_handler(
217
+ &self,
218
+ deps: &[String],
219
+ req: &Request<()>,
220
+ data: &RequestData,
221
+ ) -> Result<ResolvedDependencies, DependencyError> {
222
+ for key in deps {
223
+ if !self.dependencies.contains_key(key) {
224
+ return Err(DependencyError::NotFound { key: key.clone() });
225
+ }
226
+ }
227
+
228
+ let batches = self.dependency_graph.calculate_batches(deps)?;
229
+
230
+ let mut resolved = ResolvedDependencies::new();
231
+ let mut request_cache: HashMap<String, Arc<dyn Any + Send + Sync>> = HashMap::new();
232
+
233
+ for batch in batches {
234
+ // NOTE: We resolve sequentially within each batch to ensure cleanup tasks
235
+ let mut sorted_keys: Vec<_> = batch.iter().collect();
236
+
237
+ sorted_keys.sort_by_key(|key| self.dependencies.get_index_of(*key).unwrap_or(usize::MAX));
238
+
239
+ for key in sorted_keys {
240
+ let dep = self
241
+ .dependencies
242
+ .get(key)
243
+ .ok_or_else(|| DependencyError::NotFound { key: key.clone() })?;
244
+
245
+ if dep.singleton() {
246
+ let cache = self.singleton_cache.read().await;
247
+ if let Some(cached) = cache.get(key) {
248
+ resolved.insert(key.clone(), Arc::clone(cached));
249
+ continue;
250
+ }
251
+ }
252
+
253
+ if dep.cacheable()
254
+ && let Some(cached) = request_cache.get(key)
255
+ {
256
+ resolved.insert(key.clone(), Arc::clone(cached));
257
+ continue;
258
+ }
259
+
260
+ let result = dep.resolve(req, data, &resolved).await?;
261
+
262
+ if dep.singleton() {
263
+ let mut cache = self.singleton_cache.write().await;
264
+ cache.insert(key.clone(), Arc::clone(&result));
265
+ } else if dep.cacheable() {
266
+ request_cache.insert(key.clone(), Arc::clone(&result));
267
+ }
268
+
269
+ resolved.insert(key.clone(), result);
270
+ }
271
+ }
272
+
273
+ Ok(resolved)
274
+ }
275
+
276
+ /// Get the number of registered dependencies
277
+ ///
278
+ /// # Examples
279
+ ///
280
+ /// ```ignore
281
+ /// use spikard_core::di::{DependencyContainer, ValueDependency};
282
+ /// use std::sync::Arc;
283
+ ///
284
+ /// let mut container = DependencyContainer::new();
285
+ /// assert_eq!(container.len(), 0);
286
+ ///
287
+ /// let dep = ValueDependency::new("test", 42);
288
+ /// container.register("test".to_string(), Arc::new(dep)).unwrap();
289
+ /// assert_eq!(container.len(), 1);
290
+ /// ```
291
+ #[must_use]
292
+ pub fn len(&self) -> usize {
293
+ self.dependencies.len()
294
+ }
295
+
296
+ /// Check if the container is empty
297
+ ///
298
+ /// # Examples
299
+ ///
300
+ /// ```ignore
301
+ /// use spikard_core::di::DependencyContainer;
302
+ ///
303
+ /// let container = DependencyContainer::new();
304
+ /// assert!(container.is_empty());
305
+ /// ```
306
+ #[must_use]
307
+ pub fn is_empty(&self) -> bool {
308
+ self.dependencies.is_empty()
309
+ }
310
+
311
+ /// Check if a dependency is registered
312
+ ///
313
+ /// # Examples
314
+ ///
315
+ /// ```ignore
316
+ /// use spikard_core::di::{DependencyContainer, ValueDependency};
317
+ /// use std::sync::Arc;
318
+ ///
319
+ /// let mut container = DependencyContainer::new();
320
+ /// assert!(!container.contains("config"));
321
+ ///
322
+ /// let dep = ValueDependency::new("config", "value");
323
+ /// container.register("config".to_string(), Arc::new(dep)).unwrap();
324
+ /// assert!(container.contains("config"));
325
+ /// ```
326
+ #[must_use]
327
+ pub fn contains(&self, key: &str) -> bool {
328
+ self.dependencies.contains_key(key)
329
+ }
330
+
331
+ /// Get the keys of all registered dependencies
332
+ #[must_use]
333
+ pub fn keys(&self) -> Vec<String> {
334
+ self.dependencies.keys().cloned().collect()
335
+ }
336
+
337
+ /// Clear the singleton cache
338
+ ///
339
+ /// This is useful for testing or when you need to force re-resolution
340
+ /// of singleton dependencies.
341
+ ///
342
+ /// # Examples
343
+ ///
344
+ /// ```ignore
345
+ /// use spikard_core::di::DependencyContainer;
346
+ ///
347
+ /// # tokio_test::block_on(async {
348
+ /// let container = DependencyContainer::new();
349
+ /// container.clear_singleton_cache().await;
350
+ /// # });
351
+ /// ```
352
+ pub async fn clear_singleton_cache(&self) {
353
+ let mut cache = self.singleton_cache.write().await;
354
+ cache.clear();
355
+ }
356
+ }
357
+
358
+ impl Default for DependencyContainer {
359
+ fn default() -> Self {
360
+ Self::new()
361
+ }
362
+ }
363
+
364
+ impl std::fmt::Debug for DependencyContainer {
365
+ fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
366
+ f.debug_struct("DependencyContainer")
367
+ .field("dependencies", &self.dependencies.keys())
368
+ .finish()
369
+ }
370
+ }
371
+
372
+ #[cfg(test)]
373
+ mod tests {
374
+ use super::*;
375
+ use crate::di::{FactoryDependency, ValueDependency};
376
+ use std::collections::HashMap;
377
+ use std::sync::atomic::{AtomicU32, Ordering};
378
+
379
+ fn make_request() -> Request<()> {
380
+ Request::builder().body(()).unwrap()
381
+ }
382
+
383
+ fn make_request_data() -> RequestData {
384
+ RequestData {
385
+ path_params: Arc::new(HashMap::new()),
386
+ query_params: serde_json::Value::Null,
387
+ validated_params: None,
388
+ raw_query_params: Arc::new(HashMap::new()),
389
+ body: serde_json::Value::Null,
390
+ raw_body: None,
391
+ headers: Arc::new(HashMap::new()),
392
+ cookies: Arc::new(HashMap::new()),
393
+ method: "GET".to_string(),
394
+ path: "/".to_string(),
395
+ #[cfg(feature = "di")]
396
+ dependencies: None,
397
+ }
398
+ }
399
+
400
+ #[test]
401
+ fn test_new() {
402
+ let container = DependencyContainer::new();
403
+ assert!(container.is_empty());
404
+ assert_eq!(container.len(), 0);
405
+ }
406
+
407
+ #[test]
408
+ fn test_register_simple() {
409
+ let mut container = DependencyContainer::new();
410
+ let dep = ValueDependency::new("test", 42i32);
411
+
412
+ assert!(container.register("test".to_string(), Arc::new(dep)).is_ok());
413
+ assert_eq!(container.len(), 1);
414
+ assert!(container.contains("test"));
415
+ }
416
+
417
+ #[test]
418
+ fn test_register_duplicate() {
419
+ let mut container = DependencyContainer::new();
420
+ let dep1 = ValueDependency::new("test", 42i32);
421
+ let dep2 = ValueDependency::new("test", 100i32);
422
+
423
+ container.register("test".to_string(), Arc::new(dep1)).unwrap();
424
+ let result = container.register("test".to_string(), Arc::new(dep2));
425
+
426
+ assert!(matches!(result, Err(DependencyError::DuplicateKey { .. })));
427
+ }
428
+
429
+ #[tokio::test]
430
+ async fn test_register_circular() {
431
+ let mut container = DependencyContainer::new();
432
+
433
+ let dep_a = FactoryDependency::builder("a")
434
+ .depends_on(vec!["b".to_string()])
435
+ .factory(|_req, _data, _resolved| Box::pin(async { Ok(Arc::new(1i32) as Arc<dyn Any + Send + Sync>) }))
436
+ .build();
437
+
438
+ let dep_b = FactoryDependency::builder("b")
439
+ .depends_on(vec!["a".to_string()])
440
+ .factory(|_req, _data, _resolved| Box::pin(async { Ok(Arc::new(2i32) as Arc<dyn Any + Send + Sync>) }))
441
+ .build();
442
+
443
+ container.register("a".to_string(), Arc::new(dep_a)).unwrap();
444
+ container.register("b".to_string(), Arc::new(dep_b)).unwrap();
445
+
446
+ let request = make_request();
447
+ let request_data = make_request_data();
448
+ let result = container
449
+ .resolve_for_handler(&["a".to_string()], &request, &request_data)
450
+ .await;
451
+
452
+ assert!(matches!(result, Err(DependencyError::CircularDependency { .. })));
453
+ }
454
+
455
+ #[tokio::test]
456
+ async fn test_resolve_value() {
457
+ let mut container = DependencyContainer::new();
458
+ let dep = ValueDependency::new("answer", 42i32);
459
+ container.register("answer".to_string(), Arc::new(dep)).unwrap();
460
+
461
+ let request = make_request();
462
+ let request_data = make_request_data();
463
+
464
+ let resolved = container
465
+ .resolve_for_handler(&["answer".to_string()], &request, &request_data)
466
+ .await
467
+ .unwrap();
468
+
469
+ let value: Option<Arc<i32>> = resolved.get("answer");
470
+ assert_eq!(value.map(|v| *v), Some(42));
471
+ }
472
+
473
+ #[tokio::test]
474
+ async fn test_resolve_factory() {
475
+ let mut container = DependencyContainer::new();
476
+
477
+ let factory = FactoryDependency::builder("computed")
478
+ .factory(|_req, _data, _resolved| Box::pin(async { Ok(Arc::new(100i32) as Arc<dyn Any + Send + Sync>) }))
479
+ .build();
480
+
481
+ container.register("computed".to_string(), Arc::new(factory)).unwrap();
482
+
483
+ let request = make_request();
484
+ let request_data = make_request_data();
485
+
486
+ let resolved = container
487
+ .resolve_for_handler(&["computed".to_string()], &request, &request_data)
488
+ .await
489
+ .unwrap();
490
+
491
+ let value: Option<Arc<i32>> = resolved.get("computed");
492
+ assert_eq!(value.map(|v| *v), Some(100));
493
+ }
494
+
495
+ #[tokio::test]
496
+ async fn test_resolve_nested() {
497
+ let mut container = DependencyContainer::new();
498
+
499
+ let config = ValueDependency::new("config", "production".to_string());
500
+ container.register("config".to_string(), Arc::new(config)).unwrap();
501
+
502
+ let database = FactoryDependency::builder("database")
503
+ .depends_on(vec!["config".to_string()])
504
+ .factory(|_req, _data, resolved| {
505
+ let resolved = resolved.clone();
506
+ Box::pin(async move {
507
+ let config: Arc<String> = resolved.get("config").unwrap();
508
+ let db = format!("DB:{}", *config);
509
+ Ok(Arc::new(db) as Arc<dyn Any + Send + Sync>)
510
+ })
511
+ })
512
+ .build();
513
+ container.register("database".to_string(), Arc::new(database)).unwrap();
514
+
515
+ let request = make_request();
516
+ let request_data = make_request_data();
517
+
518
+ let resolved = container
519
+ .resolve_for_handler(&["database".to_string()], &request, &request_data)
520
+ .await
521
+ .unwrap();
522
+
523
+ let db: Option<Arc<String>> = resolved.get("database");
524
+ assert_eq!(db.as_ref().map(|v| v.as_str()), Some("DB:production"));
525
+ }
526
+
527
+ #[tokio::test]
528
+ async fn test_resolve_batched() {
529
+ let mut container = DependencyContainer::new();
530
+
531
+ let counter = Arc::new(AtomicU32::new(0));
532
+
533
+ let counter1 = Arc::clone(&counter);
534
+ let config = FactoryDependency::builder("config")
535
+ .factory(move |_req, _data, _resolved| {
536
+ let c = Arc::clone(&counter1);
537
+ Box::pin(async move {
538
+ let order = c.fetch_add(1, Ordering::SeqCst);
539
+ Ok(Arc::new(order) as Arc<dyn Any + Send + Sync>)
540
+ })
541
+ })
542
+ .build();
543
+ container.register("config".to_string(), Arc::new(config)).unwrap();
544
+
545
+ let counter2 = Arc::clone(&counter);
546
+ let database = FactoryDependency::builder("database")
547
+ .depends_on(vec!["config".to_string()])
548
+ .factory(move |_req, _data, _resolved| {
549
+ let c = Arc::clone(&counter2);
550
+ Box::pin(async move {
551
+ let order = c.fetch_add(1, Ordering::SeqCst);
552
+ Ok(Arc::new(order) as Arc<dyn Any + Send + Sync>)
553
+ })
554
+ })
555
+ .build();
556
+ container.register("database".to_string(), Arc::new(database)).unwrap();
557
+
558
+ let counter3 = Arc::clone(&counter);
559
+ let cache = FactoryDependency::builder("cache")
560
+ .depends_on(vec!["config".to_string()])
561
+ .factory(move |_req, _data, _resolved| {
562
+ let c = Arc::clone(&counter3);
563
+ Box::pin(async move {
564
+ let order = c.fetch_add(1, Ordering::SeqCst);
565
+ Ok(Arc::new(order) as Arc<dyn Any + Send + Sync>)
566
+ })
567
+ })
568
+ .build();
569
+ container.register("cache".to_string(), Arc::new(cache)).unwrap();
570
+
571
+ let request = make_request();
572
+ let request_data = make_request_data();
573
+
574
+ let resolved = container
575
+ .resolve_for_handler(&["database".to_string(), "cache".to_string()], &request, &request_data)
576
+ .await
577
+ .unwrap();
578
+
579
+ let config_order: Arc<u32> = resolved.get("config").unwrap();
580
+ assert_eq!(*config_order, 0);
581
+
582
+ let db_order: Arc<u32> = resolved.get("database").unwrap();
583
+ let cache_order: Arc<u32> = resolved.get("cache").unwrap();
584
+ assert!(*db_order >= 1);
585
+ assert!(*cache_order >= 1);
586
+ }
587
+
588
+ #[tokio::test]
589
+ async fn test_singleton_cache() {
590
+ let mut container = DependencyContainer::new();
591
+
592
+ let counter = Arc::new(AtomicU32::new(0));
593
+ let counter_clone = Arc::clone(&counter);
594
+
595
+ let singleton = FactoryDependency::builder("singleton")
596
+ .singleton(true)
597
+ .factory(move |_req, _data, _resolved| {
598
+ let c = Arc::clone(&counter_clone);
599
+ Box::pin(async move {
600
+ let value = c.fetch_add(1, Ordering::SeqCst);
601
+ Ok(Arc::new(value) as Arc<dyn Any + Send + Sync>)
602
+ })
603
+ })
604
+ .build();
605
+
606
+ container
607
+ .register("singleton".to_string(), Arc::new(singleton))
608
+ .unwrap();
609
+
610
+ let request = make_request();
611
+ let request_data = make_request_data();
612
+
613
+ for _ in 0..3 {
614
+ let resolved = container
615
+ .resolve_for_handler(&["singleton".to_string()], &request, &request_data)
616
+ .await
617
+ .unwrap();
618
+
619
+ let value: Arc<u32> = resolved.get("singleton").unwrap();
620
+ assert_eq!(*value, 0);
621
+ }
622
+
623
+ assert_eq!(counter.load(Ordering::SeqCst), 1);
624
+ }
625
+
626
+ #[tokio::test]
627
+ async fn test_clear_singleton_cache() {
628
+ let mut container = DependencyContainer::new();
629
+
630
+ let counter = Arc::new(AtomicU32::new(0));
631
+ let counter_clone = Arc::clone(&counter);
632
+
633
+ let singleton = FactoryDependency::builder("singleton")
634
+ .singleton(true)
635
+ .factory(move |_req, _data, _resolved| {
636
+ let c = Arc::clone(&counter_clone);
637
+ Box::pin(async move {
638
+ let value = c.fetch_add(1, Ordering::SeqCst);
639
+ Ok(Arc::new(value) as Arc<dyn Any + Send + Sync>)
640
+ })
641
+ })
642
+ .build();
643
+
644
+ container
645
+ .register("singleton".to_string(), Arc::new(singleton))
646
+ .unwrap();
647
+
648
+ let request = make_request();
649
+ let request_data = make_request_data();
650
+
651
+ let resolved1 = container
652
+ .resolve_for_handler(&["singleton".to_string()], &request, &request_data)
653
+ .await
654
+ .unwrap();
655
+ let value1: Arc<u32> = resolved1.get("singleton").unwrap();
656
+ assert_eq!(*value1, 0);
657
+
658
+ container.clear_singleton_cache().await;
659
+
660
+ let resolved2 = container
661
+ .resolve_for_handler(&["singleton".to_string()], &request, &request_data)
662
+ .await
663
+ .unwrap();
664
+ let value2: Arc<u32> = resolved2.get("singleton").unwrap();
665
+ assert_eq!(*value2, 1);
666
+ }
667
+
668
+ #[tokio::test]
669
+ async fn test_resolve_not_found() {
670
+ let container = DependencyContainer::new();
671
+ let request = make_request();
672
+ let request_data = make_request_data();
673
+
674
+ let result = container
675
+ .resolve_for_handler(&["missing".to_string()], &request, &request_data)
676
+ .await;
677
+
678
+ assert!(matches!(result, Err(DependencyError::NotFound { .. })));
679
+ }
680
+
681
+ #[test]
682
+ fn test_contains() {
683
+ let mut container = DependencyContainer::new();
684
+ assert!(!container.contains("test"));
685
+
686
+ let dep = ValueDependency::new("test", 42i32);
687
+ container.register("test".to_string(), Arc::new(dep)).unwrap();
688
+
689
+ assert!(container.contains("test"));
690
+ assert!(!container.contains("other"));
691
+ }
692
+
693
+ #[test]
694
+ fn test_debug() {
695
+ let mut container = DependencyContainer::new();
696
+ let dep = ValueDependency::new("test", 42i32);
697
+ container.register("test".to_string(), Arc::new(dep)).unwrap();
698
+
699
+ let debug_str = format!("{:?}", container);
700
+ assert!(debug_str.contains("DependencyContainer"));
701
+ }
702
+ }