fix: Cached queries empty after world reset (#172)

🐛 core: restore cached queries after world reset

Co-authored-by: Kris Baumgartner <kjbaumgartner@gmail.com>

Author: jerzakm

packages/core/src/world/world.ts

@@ -180,6 +180,12 @@ export class World {
 		// Create new world entity.
 		ctx.worldEntity = createEntity(this, IsExcluded);
 
+		// Restore cached queries.
+		for (const [hash, parameters] of universe.cachedQueries) {
+			const query = createQuery(this, parameters);
+			ctx.queriesHashMap.set(hash, query);
+		}
+
 		for (const sub of ctx.resetSubscriptions) {
 			sub(this);
 		}

packages/core/tests/query.test.ts

@@ -424,4 +424,24 @@ describe('Query', () => {
 			expect(entities[i].id()).toBe(i + 1);
 		}
 	});
+
+	it('cached query should return values after reset', () => {
+		const movementQuery = cacheQuery(Foo);
+
+		const spawnEntities = () => {
+			for (let i = 0; i < 100; i++) {
+				world.spawn(Foo);
+			}
+		};
+
+		spawnEntities();
+		const resultsBefore = world.query(movementQuery);
+
+		world.reset();
+		spawnEntities();
+
+		const resultsAfter = world.query(movementQuery);
+
+		expect(resultsAfter.length).toBe(resultsBefore.length);
+	});
 });