#44 Disposable Instances Handling

readme/accumulators.md

@@ -5,7 +5,7 @@
 Accumulators allow you to accumulate instances of certain types and lifetimes.
 
 ```c#
-interface IAccumulating { }
+interface IAccumulating;
 
 class MyAccumulator: List<IAccumulating>;
 
@@ -48,13 +48,10 @@ DI.Setup(nameof(Composition))
     .Root<(IService service, MyAccumulator accumulator)>("Root");
 
 var composition = new Composition();
-var root = composition.Root;
-var service = root.service;
-var accumulator = root.accumulator;
-accumulator.Count.ShouldBe(3);
+var (service, accumulator) = composition.Root;
+accumulator.Count.ShouldBe(2);
 accumulator[0].ShouldBeOfType<AbcDependency>();
-accumulator[1].ShouldBeOfType<AbcDependency>();
-accumulator[2].ShouldBeOfType<Service>();
+accumulator[1].ShouldBeOfType<Service>();
 
 ```
 
@@ -130,7 +127,6 @@ partial class Composition
     {
       var accumulatorM03D22di41 = new Pure.DI.UsageTests.Advanced.AccumulatorScenario.MyAccumulator();
       Pure.DI.UsageTests.Advanced.AccumulatorScenario.AbcDependency perBlockM03D22di4_AbcDependency = new Pure.DI.UsageTests.Advanced.AccumulatorScenario.AbcDependency();
-      accumulatorM03D22di41.Add(perBlockM03D22di4_AbcDependency);
       if (ReferenceEquals(_rootM03D22di._singletonM03D22di38_XyzDependency, null))
       {
           lock (_lockM03D22di)

readme/tracking-disposable-instances-in-delegates.md

@@ -134,7 +134,6 @@ partial class Composition
               var value_M03D22di3 = transientM03D22di3_Dependency;
               perBlockM03D22di1_Owned = new Owned<Pure.DI.UsageTests.Basics.TrackingDisposableInDelegatesScenario.IDependency>(value_M03D22di3, owned_M03D22di2);
           }
-          accumulatorM03D22di42.Add(perBlockM03D22di1_Owned);
           var factory_M03D22di1 = perBlockM03D22di1_Owned;
           return factory_M03D22di1;
       });

readme/tracking-disposable-instances-per-a-composition-root.md

@@ -121,7 +121,6 @@ partial class Composition
           var value_M03D22di2 = new Pure.DI.UsageTests.Basics.TrackingDisposableScenario.Service(transientM03D22di3_Dependency);
           perBlockM03D22di0_Owned = new Owned<Pure.DI.UsageTests.Basics.TrackingDisposableScenario.IService>(value_M03D22di2, owned_M03D22di1);
       }
-      accumulatorM03D22di39.Add(perBlockM03D22di0_Owned);
       return perBlockM03D22di0_Owned;
     }
   }

src/Pure.DI.Core/Core/Code/Accumulator.cs

@@ -5,4 +5,5 @@ internal record Accumulator(
     string Name,
     bool IsDeclared,
     ITypeSymbol Type,
+    Lifetime Lifetime,
     ITypeSymbol AccumulatorType);

src/Pure.DI.Core/Core/Code/BlockCodeBuilder.cs

@@ -36,19 +36,18 @@ variable.Node.Lifetime is Lifetime.Singleton or Lifetime.Scoped
             {
                 parent = $"{Names.ParentFieldName}.";
             }
-
-            var accumulators = new List<Accumulator>();
+
             var uniqueAccumulators = ctx.Accumulators
                 .Where(accumulator => !accumulator.IsDeclared)
-                .GroupBy(i => i.AccumulatorType, SymbolEqualityComparer.Default)
+                .GroupBy(i => i.Name)
                 .Select(i => i.First());
 
             foreach (var accumulator in uniqueAccumulators)
             {
                 code.AppendLine($"var {accumulator.Name} = new {accumulator.AccumulatorType}();");
-                accumulators.Add(accumulator with { IsDeclared = true });
             }
 
+            var accumulators = ctx.Accumulators.Select(accumulator => accumulator with { IsDeclared = true }).ToList();
             if (accumulators.Count > 0)
             {
                 ctx = ctx with { Accumulators = accumulators.ToImmutableArray() };

src/Pure.DI.Core/Core/Code/BuildTools.cs

@@ -54,6 +54,9 @@ public IEnumerable<Line> OnCreated(BuildContext ctx, Variable variable)
         var lines = ctx.Accumulators
             .Where(i => FilterAccumulator(i, variable.Node.Lifetime))
             .Where(i => baseTypes.Contains(i.Type))
+            .GroupBy(i => i.Name)
+            .Select(i => i.First())
+            .OrderBy(i => i.Name)
             .Select(i => new Line(0, $"{i.Name}.Add({variable.VariableName});"))
             .ToList();
 
@@ -76,9 +79,20 @@ public IEnumerable<Line> OnCreated(BuildContext ctx, Variable variable)
         return lines;
     }
 
-    private static bool FilterAccumulator(Accumulator accumulator, Lifetime lifetime) => 
-        accumulator.IsRoot
-        || lifetime is not (Lifetime.Singleton or Lifetime.Scoped or Lifetime.PerResolve);
+    private static bool FilterAccumulator(Accumulator accumulator, Lifetime lifetime)
+    {
+        if (accumulator.Lifetime != lifetime)
+        {
+            return false;
+        }
+
+        if (accumulator.IsRoot)
+        {
+            return true;
+        }
+
+        return lifetime is not (Lifetime.Singleton or Lifetime.Scoped or Lifetime.PerResolve);
+    }
 
     private static object? GetTag(BuildContext ctx, Variable variable)
     {

src/Pure.DI.Core/Core/Code/VariablesBuilder.cs

@@ -36,7 +36,7 @@ public Block Build(
 
                     case Variable variable:
                     {
-                        var isAccumulator = IsAccumulator(variable, out var construct);
+                        var isAccumulator = IsAccumulator(variable, out var construct, out var mdAccumulators);
                         IReadOnlyCollection<Dependency> dependencies = Array.Empty<Dependency>();
                         if (!isAccumulator)
                         {
@@ -73,7 +73,18 @@ public Block Build(
                         accumulators ??= rootNode.Accumulators;
                         if (isAccumulator)
                         {
-                            accumulators.Add(new Accumulator(isRoot, GetAccumulatorName(variable), false, construct.ElementType, construct.Type)); 
+                            var name = GetAccumulatorName(variable);
+                            foreach (var mdAccumulator in mdAccumulators)
+                            {
+                                accumulators.Add(
+                                    new Accumulator(
+                                        isRoot,
+                                        name,
+                                        false,
+                                        construct.ElementType,
+                                        mdAccumulator.Lifetime,
+                                        construct.Type));
+                            }
                         }
 
                         foreach (var (isDepResolved, depNode, depInjection, _) in dependencies)
@@ -139,15 +150,17 @@ public Block Build(
         return rootBlock;
     }
 
-    private static bool IsAccumulator(Variable variable, out MdConstruct mdConstruct)
+    private static bool IsAccumulator(Variable variable, out MdConstruct mdConstruct, out IReadOnlyCollection<MdAccumulator> accumulators)
     {
         if(variable.Node.Construct?.Source is { Kind: MdConstructKind.Accumulator } construct)
         {
             mdConstruct = construct;
+            accumulators = construct.State as IReadOnlyCollection<MdAccumulator> ?? ImmutableArray<MdAccumulator>.Empty; 
             return true;
         }
 
         mdConstruct = default;
+        accumulators = ImmutableArray<MdAccumulator>.Empty;
         return false;
     }
 

src/Pure.DI.Core/Core/DependencyGraphBuilder.cs

@@ -2,6 +2,7 @@
 // ReSharper disable InvertIf
 // ReSharper disable ClassNeverInstantiated.Global
 // ReSharper disable ForeachCanBePartlyConvertedToQueryUsingAnotherGetEnumerator
+// ReSharper disable IdentifierTypo
 namespace Pure.DI.Core;
 
 internal sealed class DependencyGraphBuilder(
@@ -52,10 +53,16 @@ public IEnumerable<DependencyNode> TryBuild(
             }
         }
 
-        var accumulators = new Dictionary<AccumulatorKey, MdAccumulator>();
+        var accumulators = new Dictionary<ITypeSymbol, List<MdAccumulator>>(SymbolEqualityComparer.Default);
         foreach (var accumulator in setup.Accumulators)
         {
-            accumulators[new AccumulatorKey(accumulator.AccumulatorType, accumulator.Lifetime)] = accumulator;
+            if (!accumulators.TryGetValue(accumulator.AccumulatorType, out var accs))
+            {
+                accs = [];
+                accumulators.Add(accumulator.AccumulatorType, accs);
+            }
+
+            accs.Add(accumulator);
         }
 
         var processed = new HashSet<ProcessingNode>();
@@ -76,13 +83,13 @@ public IEnumerable<DependencyNode> TryBuild(
                     }
                 }
 
-                if (accumulators.TryGetValue(new AccumulatorKey(injection.Type, node.Node.Lifetime), out var accumulator))
+                if (accumulators.TryGetValue(injection.Type, out var accs))
                 {
                     var accumulatorBinding = CreateAccumulatorBinding(
                         setup,
                         targetNode,
                         ref maxId,
-                        accumulator,
+                        accs,
                         hasExplicitDefaultValue,
                         explicitDefaultValue);
 
@@ -143,16 +150,6 @@ public IEnumerable<DependencyNode> TryBuild(
                                 case MdConstructKind.None:
                                     break;
 
-                                case MdConstructKind.Accumulator:
-                                    var accumulatorBinding = CreateAccumulatorBinding(
-                                        setup,
-                                        targetNode,
-                                        ref maxId,
-                                        accumulator,
-                                        hasExplicitDefaultValue,
-                                        explicitDefaultValue);
-                                    return CreateNodes(setup, accumulatorBinding);
-
                                 default:
                                     var lifetime = constructKind == MdConstructKind.Enumerable ? Lifetime.PerBlock : Lifetime.Transient;
                                     var constructBinding = CreateConstructBinding(
@@ -439,10 +436,12 @@ private MdBinding CreateGenericBinding(
     private static MdBinding CreateAccumulatorBinding(MdSetup setup,
         DependencyNode targetNode,
         ref int maxId,
-        MdAccumulator accumulator,
+        IReadOnlyCollection<MdAccumulator> accumulators,
         bool hasExplicitDefaultValue,
-        object? explicitDefaultValue) =>
-        new(
+        object? explicitDefaultValue)
+    {
+        var accumulator = accumulators.First();
+        return new MdBinding(
             ++maxId,
             targetNode.Binding.Source,
             setup,
@@ -459,9 +458,11 @@ private static MdBinding CreateAccumulatorBinding(MdSetup setup,
                 accumulator.AccumulatorType,
                 accumulator.Type,
                 MdConstructKind.Accumulator,
-                ImmutableArray<MdContract>.Empty, 
+                ImmutableArray<MdContract>.Empty,
                 hasExplicitDefaultValue,
-                explicitDefaultValue));
+                explicitDefaultValue,
+                accumulators));
+    }
 
     private MdBinding CreateAutoBinding(
         MdSetup setup,

src/Pure.DI.Core/Core/Models/MdConstruct.cs

@@ -9,7 +9,8 @@ internal readonly record struct MdConstruct(
     MdConstructKind Kind,
     ImmutableArray<MdContract> Dependencies,
     bool HasExplicitDefaultValue,
-    object? ExplicitDefaultValue)
+    object? ExplicitDefaultValue,
+    object? State = default)
 {
     public override string ToString() => $"To<{Kind}<{Type}>>()";
 }