- 修复 WithTempQuery + ToChunkAsyncEnumerable 报错问题；

Author: 2881099

Examples/base_entity/Program.cs

@@ -621,14 +621,17 @@ static void Main(string[] args)
             BaseEntity.Initialization(fsql, () => _asyncUow.Value);
             #endregion
 
+            fsql.Select<User1>().WithTempQuery(a => new { a.Nickname, a.Username }).ToChunk(10, e =>
+            {
+                foreach (var item in e.Object)
+                    Console.WriteLine(item.Nickname);
+            });
             Task.Run(async () =>
             {
-                await foreach (var xxs1 in fsql.Select<User1>().ToChunkAsyncEnumerable(10))
+                await foreach (var xxs1 in fsql.Select<User1>().WithTempQuery(a => new { a.Nickname, a.Username }).ToChunkAsyncEnumerable(10))
                 {
                     foreach (var item in xxs1)
-                    {
                         Console.WriteLine(item.Nickname);
-                    }
                 }
             }).Wait();
 

FreeSql.DbContext/FreeSql.DbContext.xml

@@ -234,6 +234,23 @@ internal void ToListChunkPrivate(int chunkSize, Action<FetchCallbackArgs<List<T1
         }
         public void ToChunk(int size, Action<FetchCallbackArgs<List<T1>>> done)
         {
+            if (_diymemexpWithTempQuery != null && _diymemexpWithTempQuery is WithTempQueryParser withTempQueryParser)
+            {
+                if (withTempQueryParser._outsideTable[0] != _tables[0])
+                {
+                    var tps = _tables.Select(a =>
+                    {
+                        var tp = Expression.Parameter(a.Table.Type, a.Alias);
+                        a.Parameter = tp;
+                        return tp;
+                    }).ToArray();
+                    this.InternalToChunk<T1>(tps[0], size, done);
+                    return;
+                }
+                var af = withTempQueryParser._insideSelectList[0].InsideAf;
+                this.ToListMrChunkPrivate(size, done, this.ToSql(af.field), af);
+                return;
+            }
             if (_selectExpression != null) throw new ArgumentException(CoreErrorStrings.Before_Chunk_Cannot_Use_Select);
             this.ToListChunkPrivate(size, done, this.GetAllFieldExpressionTreeLevel2(), null);
         }
@@ -1769,6 +1786,24 @@ internal IAsyncEnumerable<List<T1>> ToListChunkPrivateAsyncEnumerable(int chunkS
         }
         public IAsyncEnumerable<List<T1>> ToChunkAsyncEnumerable(int size)
         {
+            if (_diymemexpWithTempQuery != null && _diymemexpWithTempQuery is WithTempQueryParser withTempQueryParser)
+            {
+                if (withTempQueryParser._outsideTable[0] != _tables[0])
+                {
+                    var tps = _tables.Select(a =>
+                    {
+                        var tp = Expression.Parameter(a.Table.Type, a.Alias);
+                        a.Parameter = tp;
+                        return tp;
+                    }).ToArray();
+                    return this.InternalToChunkAsyncEnumerable<T1>(tps[0], size);
+                }
+                var af = withTempQueryParser._insideSelectList[0].InsideAf;
+                return new LocalAsyncEnumerable<T1>
+                {
+                    _GetAsyncEnumerator = (cancellationToken) => this.ToListMrChunkPrivateAsyncEnumerable<T1>(size, this.ToSql(af.field), af, cancellationToken)
+                };
+            }
             if (_selectExpression != null) throw new ArgumentException(CoreErrorStrings.Before_Chunk_Cannot_Use_Select);
             return this.ToListChunkPrivateAsyncEnumerable(size, this.GetAllFieldExpressionTreeLevel2(), null);
         }