Autotest
/
sharding
mirror of https://gitee.com/dotnetchina/sharding-core.git
1
0
Fork 0
Code Issues Releases Wiki Activity

track bug 跨库也支持track

This commit is contained in:
xuejiaming 2023-07-07 21:56:59 +08:00
parent 341ecc550f
commit 80a8041059
3 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/ShardingCore/Sharding/ShardingExecutors/DefaultShardingTrackQueryExecutor.cs
View File

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

6
src/ShardingCore/Sharding/StreamMergeContext.cs
View File

@ -278,7 +278,13 @@ namespace ShardingCore.Sharding
public bool IsUseShardingTrack(Type entityType) public bool IsUseShardingTrack(Type entityType)
{ {
if (!IsParallelQuery()) if (!IsParallelQuery())
{
if (IsCrossDataSource)
{
return true;
}
return false; return false;
}
return QueryTrack() && _trackerManager.EntityUseTrack(entityType); return QueryTrack() && _trackerManager.EntityUseTrack(entityType);
} }