From 341ecc550f758069f0d3b4c1337e43015fff6725 Mon Sep 17 00:00:00 2001
From: xuejiaming <326308290@qq.com>
Date: Fri, 7 Jul 2023 21:47:15 +0800
Subject: [PATCH] track bug

---
 .../DefaultShardingTrackQueryExecutor.cs             | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/ShardingCore/Sharding/ShardingExecutors/DefaultShardingTrackQueryExecutor.cs b/src/ShardingCore/Sharding/ShardingExecutors/DefaultShardingTrackQueryExecutor.cs
index c957151f..7bbcf4a6 100644
--- a/src/ShardingCore/Sharding/ShardingExecutors/DefaultShardingTrackQueryExecutor.cs
+++ b/src/ShardingCore/Sharding/ShardingExecutors/DefaultShardingTrackQueryExecutor.cs
@@ -62,7 +62,8 @@ namespace ShardingCore.Sharding.ShardingExecutors
             {
                 if (queryCompilerContext is IMergeQueryCompilerContext mergeQueryCompilerContext)
                 {
-                    return _shardingQueryExecutor.Execute<TResult>(mergeQueryCompilerContext);
+                    var mergeResult = _shardingQueryExecutor.Execute<TResult>(mergeQueryCompilerContext);
+                    return ResultTrackExecute(mergeResult, queryCompilerContext, TrackEnumerable, Track);
                 }
                 throw new ShardingCoreNotFoundException(queryCompilerContext.GetQueryExpression().ShardingPrint());
             }
@@ -119,7 +120,8 @@ namespace ShardingCore.Sharding.ShardingExecutors
             {
                 if (queryCompilerContext is IMergeQueryCompilerContext mergeQueryCompilerContext)
                 {
-                    return _shardingQueryExecutor.ExecuteAsync<TResult>(mergeQueryCompilerContext);
+                    var mergeResult = _shardingQueryExecutor.ExecuteAsync<TResult>(mergeQueryCompilerContext);
+                    return ResultTrackExecute(mergeResult, queryCompilerContext, TrackAsyncEnumerable, TrackAsync);
                 }
                 throw new ShardingCoreNotFoundException(queryCompilerContext.GetQueryExpression().ShardingPrint());
             }
@@ -141,7 +143,8 @@ namespace ShardingCore.Sharding.ShardingExecutors
             {
                 if (queryCompilerContext is IMergeQueryCompilerContext mergeQueryCompilerContext)
                 {
-                    return _shardingQueryExecutor.ExecuteAsync<IAsyncEnumerable<TResult>>(mergeQueryCompilerContext);
+                    var mergeResult= _shardingQueryExecutor.ExecuteAsync<IAsyncEnumerable<TResult>>(mergeQueryCompilerContext);
+                    return ResultTrackExecute(mergeResult, queryCompilerContext, TrackEnumerable, Track);
                 }
                 throw new ShardingCoreNotFoundException(queryCompilerContext.GetQueryExpression().ShardingPrint());
             }
@@ -159,7 +162,8 @@ namespace ShardingCore.Sharding.ShardingExecutors
             {
                 if (queryCompilerContext is IMergeQueryCompilerContext mergeQueryCompilerContext)
                 {
-                    return _shardingQueryExecutor.ExecuteAsync<Task<TResult>>(mergeQueryCompilerContext);
+                    var mergeResult= _shardingQueryExecutor.ExecuteAsync<Task<TResult>>(mergeQueryCompilerContext);
+                    return ResultTrackExecute(mergeResult, queryCompilerContext, TrackEnumerable, TrackAsync);
                 }
                 throw new ShardingCoreNotFoundException(queryCompilerContext.GetQueryExpression().ShardingPrint());
             }