Merge pull request #55 from martindevans/fix_query_world_mismatch

Partial Fix For Source Generated Queries
genaray · Dec 11, 2023 · f533a44 · f533a44
2 parents 01644ad + 8ded3f7
commit f533a44
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Arch.System.SourceGenerator/Query.cs b/Arch.System.SourceGenerator/Query.cs
@@ -260,15 +260,15 @@ partial class {{queryMethod.ClassName}}{
                         Exclusive = {{exclusiveTypeArray}}
                     };
 
-                    private {{staticModifier}} bool _{{queryMethod.MethodName}}_Initialized;
+                    private {{staticModifier}} World? _{{queryMethod.MethodName}}_Initialized;
                     private {{staticModifier}} Query _{{queryMethod.MethodName}}_Query;
 
                     [MethodImpl(MethodImplOptions.AggressiveInlining)]
                     public {{staticModifier}} void {{queryMethod.MethodName}}Query(World world {{data}}){
                      
-                        if(!_{{queryMethod.MethodName}}_Initialized){
+                        if(!ReferenceEquals(_{{queryMethod.MethodName}}_Initialized, world)) {
                             _{{queryMethod.MethodName}}_Query = world.Query(in {{queryMethod.MethodName}}_QueryDescription);
-                            _{{queryMethod.MethodName}}_Initialized = true;
+                            _{{queryMethod.MethodName}}_Initialized = world;
                         }
 
                         foreach(ref var chunk in _{{queryMethod.MethodName}}_Query.GetChunkIterator()){