fix: Enumerator can be invalid when an item is deleted (#565)

homuler · May 13, 2022 · d8e061a · d8e061a
1 parent dcee215
commit d8e061a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Packages/com.github.homuler.mediapipe/Runtime/Scripts/Unity/GlobalInstanceTable.cs b/Packages/com.github.homuler.mediapipe/Runtime/Scripts/Unity/GlobalInstanceTable.cs
@@ -81,7 +81,7 @@ private void ClearUnusedKeys()
     {
       lock (((ICollection)_table).SyncRoot)
       {
-        var deadKeys = _table.Where(x => !x.Value.TryGetTarget(out var target)).Select(x => x.Key);
+        var deadKeys = _table.Where(x => !x.Value.TryGetTarget(out var target)).Select(x => x.Key).ToArray();
 
         foreach (var key in deadKeys)
         {

diff --git a/Packages/com.github.homuler.mediapipe/Runtime/Scripts/Unity/OutputStream.cs b/Packages/com.github.homuler.mediapipe/Runtime/Scripts/Unity/OutputStream.cs
@@ -470,7 +470,7 @@ protected static void CompressCallbackStatus()
     {
       lock (((ICollection)_CallbackStatus).SyncRoot)
       {
-        var deadKeys = _CallbackStatus.Where(x => !_InstanceTable.ContainsKey(x.Key)).Select(x => x.Key);
+        var deadKeys = _CallbackStatus.Where(x => !_InstanceTable.ContainsKey(x.Key)).Select(x => x.Key).ToArray();
 
         foreach (var key in deadKeys)
         {