Avoid race-condition where layers are loading and populating as map s…

…tarts up. Can cause `InvalidOperationException: Collection was modified; enumeration operation may not execute.` while map is loading. Also listen to collection changed events first, to ensure we're always up to date.
Esri · Jul 25, 2023 · 060c729 · 060c729
1 parent 5db7706
commit 060c729
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/src/Toolkit/Toolkit/LayerContentDataSource.cs b/src/Toolkit/Toolkit/LayerContentDataSource.cs
@@ -362,7 +362,18 @@ private IEnumerable<Action> TrackLayerContentsRecursive(IEnumerable<ILayerConten
         {
             if (layers != null)
             {
-                foreach (var layer in layers)
+                if (layers is INotifyCollectionChanged incc)
+                {
+                    var listener = new WeakEventListener<LayerContentDataSource<T>, INotifyCollectionChanged, object?, NotifyCollectionChangedEventArgs>(this, incc)
+                    {
+                        OnEventAction = static (instance, source, eventArgs) => instance.Layers_CollectionChanged(source, eventArgs),
+                        OnDetachAction = static (instance, source, weakEventListener) => source.CollectionChanged -= weakEventListener.OnEvent,
+                    };
+                    incc.CollectionChanged += listener.OnEvent;
+                    yield return listener.Detach;
+                }
+
+                foreach (var layer in layers.ToArray())
                 {
                     if (layer is INotifyPropertyChanged inpc)
                     {
@@ -375,22 +386,11 @@ private IEnumerable<Action> TrackLayerContentsRecursive(IEnumerable<ILayerConten
                         yield return listener.Detach;
                     }
 
-                    foreach (var sublayerAction in TrackLayerContentsRecursive(layer.SublayerContents))
+                    foreach (var sublayerAction in TrackLayerContentsRecursive(layer.SublayerContents).ToArray())
                     {
                         yield return sublayerAction;
                     }
                 }
-
-                if (layers is INotifyCollectionChanged incc)
-                {
-                    var listener = new WeakEventListener<LayerContentDataSource<T>, INotifyCollectionChanged, object?, NotifyCollectionChangedEventArgs>(this, incc)
-                    {
-                        OnEventAction = static (instance, source, eventArgs) => instance.Layers_CollectionChanged(source, eventArgs),
-                        OnDetachAction = static (instance, source, weakEventListener) => source.CollectionChanged -= weakEventListener.OnEvent,
-                    };
-                    incc.CollectionChanged += listener.OnEvent;
-                    yield return listener.Detach;
-                }
             }
         }