Add async version to Query

Author: emelrad12

Arch.System.SourceGenerator/Query.cs

@@ -281,7 +281,9 @@ public static StringBuilder AppendQueryMethod(this StringBuilder sb, ref QueryMe
         var getFirstElements = new StringBuilder().GetFirstElements(queryMethod.Components);
         var getComponents = new StringBuilder().GetComponents(queryMethod.Components);
         var insertParams = new StringBuilder().InsertParams(queryMethod.Parameters);
-        
+        var jobParameters = new StringBuilder().JobParameters(queryMethod.Parameters);
+        var jobParametersAssigment = new StringBuilder().JobParametersAssigment(queryMethod.Parameters, queryMethod.IsStatic);
+
         var allTypeArray = new StringBuilder().GetTypeArray(queryMethod.AllFilteredTypes);
         var anyTypeArray = new StringBuilder().GetTypeArray(queryMethod.AnyFilteredTypes);
         var noneTypeArray = new StringBuilder().GetTypeArray(queryMethod.NoneFilteredTypes);
@@ -296,6 +298,7 @@ public static StringBuilder AppendQueryMethod(this StringBuilder sb, ref QueryMe
             using Arch.Core;
             using Arch.Core.Extensions;
             using Arch.Core.Utils;
+            using Schedulers;
             using ArrayExtensions = CommunityToolkit.HighPerformance.ArrayExtensions;
             using Component = Arch.Core.Component;
             {{(!queryMethod.IsGlobalNamespace ? $"namespace {queryMethod.Namespace} {{" : "")}}
@@ -310,27 +313,44 @@ partial class {{queryMethod.ClassName}}{
 
                     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(!ReferenceEquals(_{{queryMethod.MethodName}}_Initialized, world)) {
-                            _{{queryMethod.MethodName}}_Query = world.Query(in {{queryMethod.MethodName}}_QueryDescription);
-                            _{{queryMethod.MethodName}}_Initialized = world;
-                        }
-
-                        foreach(ref var chunk in _{{queryMethod.MethodName}}_Query){
-                            
+                    private struct {{queryMethod.MethodName}}QueryJobChunk : IJob 
+                    {
+                        {{jobParameters}}
+                        public {{queryMethod.ClassName}} self;
+                        public void Execute() {
+                            foreach(ref var chunk in self._{{queryMethod.MethodName}}_Query){
                             {{(queryMethod.IsEntityQuery ? "ref var entityFirstElement = ref chunk.Entity(0);" : "")}}
                             {{getFirstElements}}
-
+                        
                             foreach(var entityIndex in chunk)
                             {
                                 {{(queryMethod.IsEntityQuery ? $"ref readonly var {queryMethod.EntityParameter.Name.ToLower()} = ref Unsafe.Add(ref entityFirstElement, entityIndex);" : "")}}
                                 {{getComponents}}
-                                {{queryMethod.MethodName}}({{insertParams}});
+                                {{(queryMethod.IsStatic ? "" : "self.")}}{{queryMethod.MethodName}}({{insertParams}});
                             }
                         }
+                        }
+                    }
+                    
+                    public {{staticModifier}} void {{queryMethod.MethodName}}Query(World world {{data}}){
+                     
+                        if(!ReferenceEquals(_{{queryMethod.MethodName}}_Initialized, world)) {
+                            _{{queryMethod.MethodName}}_Query = world.Query(in {{queryMethod.MethodName}}_QueryDescription);
+                            _{{queryMethod.MethodName}}_Initialized = world;
+                        }
+                        var job = new {{queryMethod.MethodName}}QueryJobChunk() { {{jobParametersAssigment}} };
+                        job.Execute();
+                    }
+                    
+                    public {{staticModifier}} JobHandle {{queryMethod.MethodName}}QueryAsync(World world {{data}}){
+                     
+                        if(!ReferenceEquals(_{{queryMethod.MethodName}}_Initialized, world)) {
+                            _{{queryMethod.MethodName}}_Query = world.Query(in {{queryMethod.MethodName}}_QueryDescription);
+                            _{{queryMethod.MethodName}}_Initialized = world;
+                        }
+                        var job = new {{queryMethod.MethodName}}QueryJobChunk() { {{jobParametersAssigment}} };
+                        var handle = World.SharedJobScheduler.Schedule(job);
+                        return handle;
                     }
                 }
             {{(!queryMethod.IsGlobalNamespace ? "}" : "")}}