Simplifies UnityInjector

gustavopsantos · Dec 19, 2023 · 086dab8 · 086dab8
1 parent 8c3b9da
commit 086dab8
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/Assets/Reflex/Injectors/UnityInjector.cs b/Assets/Reflex/Injectors/UnityInjector.cs
@@ -37,8 +37,7 @@ void InjectScene(Scene scene, LoadSceneMode mode = default)
             void DisposeScene(Scene scene)
             {
                 ReflexLogger.Log($"Scene {scene.name} ({scene.GetHashCode()}) unloaded", LogLevel.Development);
-                var sceneContainer = containersByScene[scene];
-                containersByScene.Remove(scene);
+                containersByScene.Remove(scene, out var sceneContainer);
                 sceneContainer.Dispose();
             }
 
@@ -86,9 +85,8 @@ private static Container CreateSceneContainer(Scene scene, Container projectCont
             {
                 builder.SetName($"{scene.name} ({scene.GetHashCode()})");
 
-                if (ScenePreInstaller.TryGetValue(scene, out var preInstaller))
+                if (ScenePreInstaller.Remove(scene, out var preInstaller))
                 {
-                    ScenePreInstaller.Remove(scene);                    
                     preInstaller.Invoke(builder);
                 }