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