fix system context setting

NSSTC · Feb 8, 2024 · 4609155 · 4609155
1 parent 17f6749
commit 4609155
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/world/runtime/runtime-world.ts b/src/world/runtime/runtime-world.ts
@@ -125,6 +125,7 @@ export class RuntimeWorld implements IRuntimeWorld, IMutableWorld {
 
             for (scheduler of this.config.stateSchedulers.values()) {
                 for (system of scheduler.getSystems()) {
+                    system.setRuntimeContext(this);
                     this.systems.add(system);
                 }
             }
@@ -173,8 +174,7 @@ export class RuntimeWorld implements IRuntimeWorld, IMutableWorld {
         let i, stage, system, systems;
 
         schedulers
-            .map(scheduler => scheduler.pipeline.getGroups())
-            .flat()
+            .flatMap(scheduler => scheduler.pipeline.getGroups())
             .forEach(group => {
                 for (stage of group.stages) {
                     systems = stage.systems;
@@ -199,7 +199,6 @@ export class RuntimeWorld implements IRuntimeWorld, IMutableWorld {
             let query, system;
 
             for (system of this.config.defaultScheduler.getSystems()) {
-                system.setRuntimeContext(this);
                 for (query of getQueriesFromSystem(system)) {
                     this.queries.add(query);
                 }