Clean up getting entry to window map for winit

Aceeri · Aceeri · commit 81e68f675104 · 2022-08-09T23:08:04.000-07:00
diff --git a/crates/bevy_winit/src/winit_windows.rs b/crates/bevy_winit/src/winit_windows.rs
@@ -142,15 +142,10 @@ impl WinitWindows {
             }
         }
 
-        // TODO: Might be more elegant ways to get return the reference of the winit-window
-        let id = winit_window.id();
-        self.windows.insert(winit_window.id(), winit_window);
-        let created_window = self
-            .windows
-            .get(&id)
-            .expect("Winit should alway have the window it just created");
-
-        created_window
+        self.windows
+            .entry(winit_window.id())
+            .insert(winit_window)
+            .into_mut()
     }
 
     /// Get the winit window that is associated with our entity.

Original file line number	Diff line number	Diff line change
`@@ -142,15 +142,10 @@ impl WinitWindows {`
`142`	`142`	`}`
`143`	`143`	`}`
`144`	`144`
`145`		`- // TODO: Might be more elegant ways to get return the reference of the winit-window`
`146`		`- let id = winit_window.id();`
`147`		`- self.windows.insert(winit_window.id(), winit_window);`
`148`		`- let created_window = self`
`149`		`- .windows`
`150`		`- .get(&id)`
`151`		`- .expect("Winit should alway have the window it just created");`
`152`		`-`
`153`		`- created_window`
	`145`	`+ self.windows`
	`146`	`+ .entry(winit_window.id())`
	`147`	`+ .insert(winit_window)`
	`148`	`+ .into_mut()`
`154`	`149`	`}`
`155`	`150`
`156`	`151`	`/// Get the winit window that is associated with our entity.`