Resolution reporting to Behavior

Author: bradfol

Sources/Swinject/Behavior.swift

@@ -4,6 +4,11 @@
 
 /// Protocol for adding functionality to the container
 public protocol Behavior {
+    /// Whether the container should report resolutions to this Behavior.
+    /// If `true`, `container(_:didResolve:name:)` will be called after each successful resolution.
+    /// Defaults to `false`.
+    var shouldReportResolution: Bool { get }
+
     /// This will be invoked on each behavior added to the `container` for each `entry` added to the container using
     /// one of the `register()` or type forwarding methods
     ///
@@ -21,4 +26,31 @@ public protocol Behavior {
         toService entry: ServiceEntry<Service>,
         withName name: String?
     )
+
+    /// This will be invoked after a successful resolution if `shouldReportResolution` is `true`.
+    ///
+    /// - Parameters:
+    ///     - container: container from which the service was resolved
+    ///     - serviceType: The type that was resolved
+    ///     - name: The registration name used for resolution (if any)
+    ///     - duration: The time taken to resolve the service
+    func container<Service>(
+        _ container: SwinjectContainer,
+        didResolve serviceType: Service.Type,
+        name: String?,
+        duration: Duration
+    )
+}
+
+// MARK: - Default Implementations
+
+public extension Behavior {
+    var shouldReportResolution: Bool { false }
+
+    func container<Service>(
+        _ container: SwinjectContainer,
+        didResolve serviceType: Service.Type,
+        name: String?,
+        duration: Duration
+    ) {}
 }

Sources/Swinject/SwinjectContainer.swift

@@ -227,7 +227,8 @@ extension SwinjectContainer: _Resolver {
     ) -> Service? {
         // No need to use weak self since the resolution will be executed before
         // this function exits.
-        sync {
+        let start = ContinuousClock.now
+        return sync {
             var resolvedInstance: Service?
             let key = ServiceKey(serviceType: Service.self, argumentsType: Arguments.self, name: name, option: option)
 
@@ -249,6 +250,11 @@ extension SwinjectContainer: _Resolver {
                     key: key,
                     availableRegistrations: getRegistrations()
                 )
+            } else {
+                let duration = ContinuousClock.now - start
+                behaviors
+                    .filter { $0.shouldReportResolution }
+                    .forEach { $0.container(self, didResolve: Service.self, name: name, duration: duration) }
             }
 
             return resolvedInstance