MdpDbPullExecutor.cs 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633
  1. using System.Data;
  2. using System.Globalization;
  3. using System.Text.Json;
  4. using System.Text.Json.Serialization;
  5. using Admin.NET.Plugin.AiDOP.Entity.DataPlatform;
  6. using SqlSugar;
  7. namespace Admin.NET.Plugin.AiDOP.DataPlatform.Executors;
  8. /// <summary>
  9. /// 方式甲:从 mdp_source DB 源按 mdp_entity.source_table_name 增量抽数 → target_table_name(贴源)。
  10. /// </summary>
  11. public sealed class MdpDbPullExecutor : IMdpSourcePullExecutor, ITransient
  12. {
  13. public string SupportedType => "DB_SYNC";
  14. /// <summary>
  15. /// 贴源 raw_data 的时间统一按 MySQL 字面量格式输出;各域转换层用
  16. /// STR_TO_DATE(..., '%Y-%m-%d %H:%i:%s.%f') 解析,ISO 8601 的 T 分隔符会解析失败。
  17. /// </summary>
  18. private static readonly JsonSerializerOptions RawDataJsonOptions = new()
  19. {
  20. Converters =
  21. {
  22. new MysqlLiteralDateTimeConverter(),
  23. new MysqlLiteralDateTimeOffsetConverter()
  24. }
  25. };
  26. private readonly MdpSourceScopeFactory _scopeFactory;
  27. private readonly ISqlSugarClient _db;
  28. private readonly MdpStagingWriter _writer;
  29. public MdpDbPullExecutor(MdpSourceScopeFactory scopeFactory, ISqlSugarClient db, MdpStagingWriter writer)
  30. {
  31. _scopeFactory = scopeFactory;
  32. _db = db;
  33. _writer = writer;
  34. }
  35. public async Task<MdpPullResult> PullAsync(MdpSource source, MdpEntity entity, MdpPullContext ctx, CancellationToken cancellationToken = default)
  36. {
  37. if (string.IsNullOrWhiteSpace(entity.SourceTableName))
  38. throw new InvalidOperationException($"实体 {entity.EntityCode} 未配置 source_table_name");
  39. if (string.IsNullOrWhiteSpace(entity.TargetTableName))
  40. throw new InvalidOperationException($"实体 {entity.EntityCode} 未配置 target_table_name");
  41. var startedAt = DateTime.Now;
  42. var scope = await _scopeFactory.GetScopeAsync(source.SourceCode, cancellationToken);
  43. var batchSize = entity.BatchSize > 0 ? entity.BatchSize : 1000;
  44. var isSqlServer = string.Equals(source.DbType, "SQLSERVER", StringComparison.OrdinalIgnoreCase)
  45. || string.Equals(source.DbType, "MSSQL", StringComparison.OrdinalIgnoreCase);
  46. var useKeyset = ctx.UseKeysetCursor
  47. && !string.IsNullOrWhiteSpace(ctx.CursorColumn)
  48. && !string.IsNullOrWhiteSpace(ctx.TieBreakerColumn);
  49. string? tenantCol = null;
  50. string? factoryCol = null;
  51. if (ctx.TenantId > 0)
  52. {
  53. var cols = await TryGetSourceColumnsAsync(scope, entity.SourceTableName!, isSqlServer, cancellationToken);
  54. tenantCol = FindColumn(cols, "tenant_id", "TenantId", "TenantID");
  55. if (ctx.FactoryId > 0)
  56. factoryCol = FindColumn(cols, "factory_id", "FactoryId");
  57. }
  58. var sql = useKeyset
  59. ? BuildKeysetSelectSql(entity, isSqlServer, batchSize, ctx, tenantCol, factoryCol)
  60. : BuildSelectSql(entity, isSqlServer, batchSize, ctx, tenantCol, factoryCol);
  61. var parameters = new List<SugarParameter>();
  62. if (NeedsTenantParameter(tenantCol, factoryCol))
  63. parameters.Add(new SugarParameter("@scopeTenantId", ctx.TenantId));
  64. if (factoryCol != null)
  65. parameters.Add(new SugarParameter("@scopeFactoryId", ctx.FactoryId));
  66. if (useKeyset)
  67. {
  68. AddKeysetParameters(parameters, ctx);
  69. }
  70. else
  71. {
  72. if (ctx.WindowFrom.HasValue)
  73. parameters.Add(new SugarParameter("@windowFrom", ctx.WindowFrom.Value));
  74. if (!ctx.FullRefresh
  75. && !string.Equals(ctx.SyncWindowType, "FULL", StringComparison.OrdinalIgnoreCase)
  76. && !string.IsNullOrWhiteSpace(entity.IncrColumn)
  77. && !string.IsNullOrWhiteSpace(entity.LastCursor)
  78. && !string.Equals(ctx.SyncWindowType, "ROLLING", StringComparison.OrdinalIgnoreCase))
  79. parameters.Add(new SugarParameter("@cursor", entity.LastCursor));
  80. // ROLLING:用窗口下界;若已有水位且水位晚于下界,则进一步用水位收窄
  81. if (!ctx.FullRefresh
  82. && string.Equals(ctx.SyncWindowType, "ROLLING", StringComparison.OrdinalIgnoreCase)
  83. && !string.IsNullOrWhiteSpace(entity.IncrColumn)
  84. && !string.IsNullOrWhiteSpace(entity.LastCursor)
  85. && ctx.WindowFrom.HasValue
  86. && DateTime.TryParse(entity.LastCursor, out var cursorDt)
  87. && cursorDt > ctx.WindowFrom.Value)
  88. parameters.Add(new SugarParameter("@cursor", entity.LastCursor));
  89. }
  90. var table = await scope.Ado.GetDataTableAsync(sql, parameters);
  91. cancellationToken.ThrowIfCancellationRequested();
  92. var written = 0;
  93. string? maxCursor = entity.LastCursor;
  94. var now = DateTime.Now;
  95. string? lastKeysetCursor = null;
  96. string? lastKeysetTie = null;
  97. foreach (DataRow row in table.Rows)
  98. {
  99. cancellationToken.ThrowIfCancellationRequested();
  100. var dict = new Dictionary<string, object?>(StringComparer.OrdinalIgnoreCase);
  101. foreach (DataColumn col in table.Columns)
  102. dict[col.ColumnName] = row[col] == DBNull.Value ? null : row[col];
  103. var sourceRowId = ResolveSourceRowId(dict);
  104. var rawJson = JsonSerializer.Serialize(dict, RawDataJsonOptions);
  105. if (useKeyset)
  106. {
  107. lastKeysetCursor = FormatCursorValue(dict, ctx.CursorColumn!);
  108. lastKeysetTie = FormatCursorValue(dict, ctx.TieBreakerColumn!);
  109. if (!string.IsNullOrEmpty(lastKeysetTie))
  110. maxCursor = EncodeKeysetCursor(lastKeysetCursor, lastKeysetTie);
  111. }
  112. else if (!string.IsNullOrWhiteSpace(entity.IncrColumn) && dict.TryGetValue(entity.IncrColumn, out var incrVal) && incrVal != null)
  113. {
  114. var cursor = incrVal is DateTime dt ? dt.ToString("yyyy-MM-dd HH:mm:ss.fff") : incrVal.ToString();
  115. if (!string.IsNullOrEmpty(cursor) && (maxCursor == null || string.CompareOrdinal(cursor, maxCursor) > 0))
  116. maxCursor = cursor;
  117. }
  118. written += await _writer.UpsertAsync(
  119. source, entity, entity.SourceTableName!, dict, rawJson, sourceRowId, ctx);
  120. }
  121. if (useKeyset && !string.IsNullOrEmpty(lastKeysetTie))
  122. {
  123. ctx.CursorValue = lastKeysetCursor;
  124. ctx.TieBreakerValue = lastKeysetTie;
  125. }
  126. if (!ctx.DeferCursorPersist
  127. && !string.IsNullOrEmpty(maxCursor)
  128. && maxCursor != entity.LastCursor)
  129. {
  130. await _db.Updateable<MdpEntity>()
  131. .SetColumns(x => new MdpEntity
  132. {
  133. LastCursor = maxCursor,
  134. LastSyncTo = now,
  135. UpdateTime = now
  136. })
  137. .Where(x => x.Id == entity.Id)
  138. .ExecuteCommandAsync(cancellationToken);
  139. }
  140. // 零行假成功守卫:本页 0 行时,用同谓词、去作用域的探针确认源侧是否本就无数据。
  141. // 源侧有数据而作用域后 0 行 ⇒ 全量被租户/工厂作用域过滤掉,不得记为无保留的 SUCCESS。
  142. var scopeMismatch = IsScopeMismatch(
  143. table.Rows.Count, ctx, tenantCol, factoryCol,
  144. sourceHasRowsOutsideScope: ShouldProbeScope(table.Rows.Count, ctx, tenantCol, factoryCol)
  145. && await SourceHasRowsOutsideScopeAsync(scope, entity, ctx, isSqlServer, useKeyset, parameters, tenantCol, cancellationToken));
  146. var diagnostic = scopeMismatch
  147. ? $"{ScopeMismatchCode}: 源表 {entity.SourceTableName} 在 tenantId={ctx.TenantId}/factoryId={ctx.FactoryId} 作用域外仍有数据,"
  148. + "但作用域过滤后 0 行;请核对该实体的工厂口径(factory_id 是否被写成 tenant_id 或真实工厂号与作用域不一致)。"
  149. : null;
  150. await WriteSyncLogAsync(source, entity, ctx, startedAt, table.Rows.Count, written, diagnostic,
  151. scopeMismatch ? SyncStatusPartial : SyncStatusSuccess);
  152. var windowHint = useKeyset
  153. ? (ctx.NullTimePhase ? "KEYSET_NULL" : "KEYSET")
  154. : (ctx.SyncWindowType ?? (ctx.FullRefresh ? "FULL" : "INCR"));
  155. return new MdpPullResult
  156. {
  157. RowsPulled = table.Rows.Count,
  158. RowsWritten = written,
  159. NewCursor = maxCursor,
  160. Message = (scopeMismatch ? $"{ScopeMismatchCode} window=" : "OK window=") + windowHint
  161. + (ctx.WindowFrom.HasValue ? $" from={ctx.WindowFrom:yyyy-MM-dd HH:mm:ss}" : "")
  162. + (ctx.BootstrapFrom.HasValue ? $" bootstrapFrom={ctx.BootstrapFrom:yyyy-MM-dd HH:mm:ss}" : "")
  163. };
  164. }
  165. internal const string SyncStatusSuccess = "SUCCESS";
  166. /// <summary>mdp_sync_log.status 既有 enum 成员('RUNNING','SUCCESS','PARTIAL','FAILED'),无需改表。</summary>
  167. internal const string SyncStatusPartial = "PARTIAL";
  168. internal const string ScopeMismatchCode = "SCOPE_MISMATCH";
  169. /// <summary>
  170. /// 是否需要跑作用域探针。仅在 ①本页 0 行 ②确实发射了作用域谓词 ③非 OFFSET 续页 时才探。
  171. /// 第 ③ 条:ctx.Offset&gt;0 说明本轮前一页已经拉满了作用域内的数据,作用域显然没有过滤掉全部,
  172. /// 此时的 0 行是正常翻页到底,探针(不带 OFFSET)会误报。
  173. /// </summary>
  174. internal static bool ShouldProbeScope(int rowsRead, MdpPullContext ctx, string? tenantCol, string? factoryCol)
  175. => rowsRead == 0 && ctx.Offset <= 0 && NeedsTenantParameter(tenantCol, factoryCol);
  176. /// <summary>
  177. /// 最终判定:本页 0 行且探针证明"去掉作用域后源侧仍有数据" ⇒ 全量被作用域过滤,记 PARTIAL。
  178. /// 源侧本就无数据(合法空源 / 增量无新数据)时 <paramref name="sourceHasRowsOutsideScope"/> 为 false ⇒ 正常 SUCCESS。
  179. /// </summary>
  180. internal static bool IsScopeMismatch(
  181. int rowsRead, MdpPullContext ctx, string? tenantCol, string? factoryCol, bool sourceHasRowsOutsideScope)
  182. => ShouldProbeScope(rowsRead, ctx, tenantCol, factoryCol) && sourceHasRowsOutsideScope;
  183. private static async Task<bool> SourceHasRowsOutsideScopeAsync(
  184. ISqlSugarClient scope, MdpEntity entity, MdpPullContext ctx,
  185. bool isSqlServer, bool useKeyset, List<SugarParameter> parameters,
  186. string? tenantCol, CancellationToken cancellationToken)
  187. {
  188. cancellationToken.ThrowIfCancellationRequested();
  189. try
  190. {
  191. var keepTenant = !string.IsNullOrWhiteSpace(tenantCol);
  192. var probeSql = BuildScopeProbeSql(entity, isSqlServer, ctx, useKeyset, tenantCol);
  193. var probe = await scope.Ado.GetDataTableAsync(probeSql, StripScopeParameters(parameters, keepTenant));
  194. return probe.Rows.Count > 0;
  195. }
  196. catch
  197. {
  198. // 探针纯诊断用途,任何失败都不得影响主流程,按"无异常"处理。
  199. return false;
  200. }
  201. }
  202. /// <summary>
  203. /// 探针复用正式查询的参数,剔除探针 SQL 不再引用的作用域参数。
  204. /// <para><paramref name="keepTenantParameter"/> 必须与
  205. /// <see cref="BuildScopeProbeSql"/> 是否发射了租户谓词保持一致:
  206. /// 探针带租户谓词时要留下 <c>@scopeTenantId</c>,否则参数会缺失;
  207. /// 不带时要剔除,否则会多绑一个 SQL 里不存在的参数。</para>
  208. /// <para><c>@scopeFactoryId</c> 恒剔除 —— 探针的全部意义就是"去掉工厂口径再看一眼"。</para>
  209. /// </summary>
  210. internal static List<SugarParameter> StripScopeParameters(
  211. List<SugarParameter> parameters, bool keepTenantParameter = false)
  212. => parameters
  213. .Where(p => !IsStrippedScopeParameter(p.ParameterName, keepTenantParameter))
  214. .ToList();
  215. private static bool IsStrippedScopeParameter(string? name, bool keepTenantParameter)
  216. {
  217. var n = (name ?? "").TrimStart('@', ':', '?');
  218. if (string.Equals(n, "scopeFactoryId", StringComparison.OrdinalIgnoreCase)) return true;
  219. return !keepTenantParameter
  220. && string.Equals(n, "scopeTenantId", StringComparison.OrdinalIgnoreCase);
  221. }
  222. private static string RequireSourceTable(MdpEntity entity)
  223. {
  224. var table = entity.SourceTableName!.Trim();
  225. // 仅允许简单标识符/schema.table,防注入
  226. if (!System.Text.RegularExpressions.Regex.IsMatch(table, @"^[A-Za-z0-9_\.\[\]]+$"))
  227. throw new InvalidOperationException($"非法 source_table_name:{table}");
  228. return table;
  229. }
  230. /// <summary>增量/窗口谓词(不含租户/工厂作用域),供正式查询与作用域探针复用。</summary>
  231. private static List<string> BuildIncrementalPredicates(MdpEntity entity, bool isSqlServer, MdpPullContext ctx, out string orderBy)
  232. {
  233. var predicates = new List<string>();
  234. if (!string.IsNullOrWhiteSpace(entity.IncrColumn))
  235. {
  236. var incr = entity.IncrColumn.Trim();
  237. if (!System.Text.RegularExpressions.Regex.IsMatch(incr, @"^[A-Za-z0-9_]+$"))
  238. throw new InvalidOperationException($"非法 incr_column:{incr}");
  239. var incrExpr = isSqlServer ? incr : $"`{incr}`";
  240. orderBy = incrExpr;
  241. var isRolling = string.Equals(ctx.SyncWindowType, "ROLLING", StringComparison.OrdinalIgnoreCase);
  242. if (!ctx.FullRefresh && isRolling && ctx.WindowFrom.HasValue)
  243. predicates.Add($"{incrExpr} >= @windowFrom");
  244. var useCursor = !ctx.FullRefresh
  245. && !string.IsNullOrWhiteSpace(entity.LastCursor)
  246. && (!isRolling
  247. || (ctx.WindowFrom.HasValue
  248. && DateTime.TryParse(entity.LastCursor, out var cursorDt)
  249. && cursorDt > ctx.WindowFrom.Value));
  250. if (useCursor)
  251. predicates.Add($"{incrExpr} > @cursor");
  252. }
  253. else
  254. {
  255. orderBy = isSqlServer ? "(SELECT NULL)" : "1";
  256. }
  257. return predicates;
  258. }
  259. private static string BuildSelectSql(MdpEntity entity, bool isSqlServer, int batchSize, MdpPullContext ctx, string? tenantCol = null, string? factoryCol = null)
  260. {
  261. var table = RequireSourceTable(entity);
  262. var predicates = BuildIncrementalPredicates(entity, isSqlServer, ctx, out var orderBy);
  263. AppendScopePredicates(predicates, isSqlServer, tenantCol, factoryCol);
  264. var where = predicates.Count > 0 ? " WHERE " + string.Join(" AND ", predicates) : "";
  265. var offset = ctx.Offset > 0 ? ctx.Offset : 0;
  266. if (isSqlServer)
  267. {
  268. // SQL Server:OFFSET 需 ORDER BY;无 incr 时用稳定键兜底
  269. if (string.IsNullOrWhiteSpace(entity.IncrColumn))
  270. orderBy = "(SELECT NULL)";
  271. return $"SELECT * FROM {table}{where} ORDER BY {orderBy} OFFSET {offset} ROWS FETCH NEXT {batchSize} ROWS ONLY";
  272. }
  273. return offset > 0
  274. ? $"SELECT * FROM {table}{where} ORDER BY {orderBy} LIMIT {batchSize} OFFSET {offset}"
  275. : $"SELECT * FROM {table}{where} ORDER BY {orderBy} LIMIT {batchSize}";
  276. }
  277. /// <summary>keyset 谓词(不含租户/工厂作用域),供正式查询与作用域探针复用。</summary>
  278. private static List<string> BuildKeysetPredicates(bool isSqlServer, MdpPullContext ctx, out string orderBy)
  279. {
  280. var cursorCol = RequireIdent(ctx.CursorColumn, "CursorColumn");
  281. var tieCol = RequireIdent(ctx.TieBreakerColumn, "TieBreakerColumn");
  282. var cursorExpr = isSqlServer ? cursorCol : $"`{cursorCol}`";
  283. var tieExpr = isSqlServer ? tieCol : $"`{tieCol}`";
  284. var predicates = new List<string>();
  285. if (ctx.NullTimePhase)
  286. {
  287. predicates.Add($"{cursorExpr} IS NULL");
  288. if (!string.IsNullOrWhiteSpace(ctx.TieBreakerValue))
  289. predicates.Add($"{tieExpr} > @tieBreaker");
  290. }
  291. else
  292. {
  293. predicates.Add($"{cursorExpr} IS NOT NULL");
  294. if (ctx.BootstrapFrom.HasValue)
  295. predicates.Add($"{cursorExpr} >= @bootstrapFrom");
  296. // keyset lower bound
  297. predicates.Add($"""
  298. (
  299. @cursorTime IS NULL
  300. OR {cursorExpr} > @cursorTime
  301. OR ({cursorExpr} = @cursorTime AND {tieExpr} > @tieBreaker)
  302. )
  303. """);
  304. if (!string.IsNullOrWhiteSpace(ctx.UpperCursorValue))
  305. {
  306. predicates.Add($"""
  307. (
  308. {cursorExpr} < @upperTime
  309. OR ({cursorExpr} = @upperTime AND {tieExpr} <= @upperTie)
  310. )
  311. """);
  312. }
  313. }
  314. orderBy = ctx.NullTimePhase ? tieExpr : $"{cursorExpr}, {tieExpr}";
  315. return predicates;
  316. }
  317. private static string BuildKeysetSelectSql(MdpEntity entity, bool isSqlServer, int batchSize, MdpPullContext ctx, string? tenantCol = null, string? factoryCol = null)
  318. {
  319. var table = RequireSourceTable(entity);
  320. var predicates = BuildKeysetPredicates(isSqlServer, ctx, out var orderBy);
  321. AppendScopePredicates(predicates, isSqlServer, tenantCol, factoryCol);
  322. var where = " WHERE " + string.Join(" AND ", predicates);
  323. if (isSqlServer)
  324. return $"SELECT * FROM {table}{where} ORDER BY {orderBy} OFFSET 0 ROWS FETCH NEXT {batchSize} ROWS ONLY";
  325. return $"SELECT * FROM {table}{where} ORDER BY {orderBy} LIMIT {batchSize}";
  326. }
  327. /// <summary>
  328. /// 作用域探针:与正式查询同一组增量/窗口/keyset 谓词,**保留租户谓词、只去掉工厂谓词**。
  329. /// 正式查询 0 行而探针有行 ⇒ 差异只可能来自工厂口径,可据此判定"本租户的数据被工厂过滤掉了"。
  330. /// 只做存在性判断(TOP 1 / LIMIT 1),不取数据。
  331. ///
  332. /// <para><b>租户谓词必须保留</b>。此前两个作用域谓词一起剥掉,探针退化成
  333. /// <c>SELECT 1 FROM 源表 LIMIT 1</c> —— 在多租户共享同一张源表的模型下,
  334. /// 它对**任何**租户都会命中别的租户的行,于是每一个合法空租户都被误报成 SCOPE_MISMATCH。
  335. /// 实测该误报在共享库上累计 291 条,且全部落在源表本就 0 行的租户上,
  336. /// 而真正被工厂谓词滤空的那两个租户反而一条都没报出来 —— 信号方向是反的。</para>
  337. ///
  338. /// <para>探针要回答的是「同一租户内,是否因工厂口径而取空」,
  339. /// 不是「整张表有没有数据」。丢掉租户维度,这个问题就问不出来了。</para>
  340. /// </summary>
  341. internal static string BuildScopeProbeSql(
  342. MdpEntity entity, bool isSqlServer, MdpPullContext ctx, bool useKeyset, string? tenantCol = null)
  343. {
  344. var table = RequireSourceTable(entity);
  345. var predicates = useKeyset
  346. ? BuildKeysetPredicates(isSqlServer, ctx, out _)
  347. : BuildIncrementalPredicates(entity, isSqlServer, ctx, out _);
  348. if (!string.IsNullOrWhiteSpace(tenantCol))
  349. predicates.Add($"{QuoteIdent(tenantCol, isSqlServer)} = @scopeTenantId");
  350. var where = predicates.Count > 0 ? " WHERE " + string.Join(" AND ", predicates) : "";
  351. return isSqlServer
  352. ? $"SELECT TOP 1 1 AS probe FROM {table}{where}"
  353. : $"SELECT 1 AS probe FROM {table}{where} LIMIT 1";
  354. }
  355. private static void AddKeysetParameters(List<SugarParameter> parameters, MdpPullContext ctx)
  356. {
  357. if (ctx.BootstrapFrom.HasValue)
  358. parameters.Add(new SugarParameter("@bootstrapFrom", ctx.BootstrapFrom.Value));
  359. if (ctx.NullTimePhase)
  360. {
  361. parameters.Add(new SugarParameter("@tieBreaker",
  362. string.IsNullOrWhiteSpace(ctx.TieBreakerValue) ? "0" : ctx.TieBreakerValue));
  363. return;
  364. }
  365. object cursorTime = string.IsNullOrWhiteSpace(ctx.CursorValue)
  366. ? DBNull.Value
  367. : (object)ctx.CursorValue!;
  368. parameters.Add(new SugarParameter("@cursorTime", cursorTime));
  369. parameters.Add(new SugarParameter("@tieBreaker",
  370. string.IsNullOrWhiteSpace(ctx.TieBreakerValue) ? "0" : ctx.TieBreakerValue));
  371. if (!string.IsNullOrWhiteSpace(ctx.UpperCursorValue))
  372. {
  373. parameters.Add(new SugarParameter("@upperTime", ctx.UpperCursorValue));
  374. parameters.Add(new SugarParameter("@upperTie",
  375. string.IsNullOrWhiteSpace(ctx.UpperTieBreakerValue) ? "0" : ctx.UpperTieBreakerValue));
  376. }
  377. }
  378. /// <summary>
  379. /// 租户/工厂作用域谓词。工厂维度走 <see cref="MdpFactoryScope"/> 的归一语义
  380. /// (factory_id 为 NULL / &lt;=0 / 等于 tenant_id 一律视作"无真实工厂"→默认工厂 1),
  381. /// 与 <see cref="MdpStagingWriter"/> 写入侧完全一致;真实工厂号仍逐值隔离。
  382. /// </summary>
  383. internal static void AppendScopePredicates(List<string> predicates, bool isSqlServer, string? tenantCol, string? factoryCol)
  384. {
  385. if (!string.IsNullOrWhiteSpace(tenantCol))
  386. predicates.Add($"{QuoteIdent(tenantCol, isSqlServer)} = @scopeTenantId");
  387. if (!string.IsNullOrWhiteSpace(factoryCol))
  388. {
  389. predicates.Add(MdpFactoryScope.SqlPredicate(
  390. QuoteIdent(factoryCol, isSqlServer), "@scopeTenantId", "@scopeFactoryId"));
  391. }
  392. }
  393. /// <summary>
  394. /// 工厂谓词引用 @scopeTenantId,故只要发射了任一作用域谓词就必须绑定该参数
  395. /// (源表只有 factory_id 而无 tenant_id 时,旧逻辑会漏绑导致 SQL 参数缺失)。
  396. /// </summary>
  397. internal static bool NeedsTenantParameter(string? tenantCol, string? factoryCol)
  398. => !string.IsNullOrWhiteSpace(tenantCol) || !string.IsNullOrWhiteSpace(factoryCol);
  399. private static string QuoteIdent(string name, bool isSqlServer) => isSqlServer ? name : $"`{name}`";
  400. private static string? FindColumn(IReadOnlyCollection<string> columns, params string[] names)
  401. {
  402. foreach (var n in names)
  403. {
  404. var hit = columns.FirstOrDefault(c => string.Equals(c, n, StringComparison.OrdinalIgnoreCase));
  405. if (hit != null) return hit;
  406. }
  407. return null;
  408. }
  409. private static async Task<IReadOnlyCollection<string>> TryGetSourceColumnsAsync(
  410. ISqlSugarClient scope, string table, bool isSqlServer, CancellationToken cancellationToken)
  411. {
  412. try
  413. {
  414. var simple = table.Contains('.') ? table.Split('.').Last().Trim('[', ']') : table.Trim('[', ']');
  415. if (!System.Text.RegularExpressions.Regex.IsMatch(simple, @"^[A-Za-z0-9_]+$"))
  416. return Array.Empty<string>();
  417. if (isSqlServer)
  418. {
  419. var rows = await scope.Ado.SqlQueryAsync<string>(
  420. "SELECT name FROM sys.columns WHERE object_id = OBJECT_ID(@t)",
  421. new SugarParameter("@t", simple));
  422. return rows ?? new List<string>();
  423. }
  424. var mysql = await scope.Ado.SqlQueryAsync<string>(
  425. """
  426. SELECT COLUMN_NAME FROM information_schema.COLUMNS
  427. WHERE TABLE_SCHEMA=DATABASE() AND TABLE_NAME=@t
  428. """,
  429. new SugarParameter("@t", simple));
  430. return mysql ?? new List<string>();
  431. }
  432. catch
  433. {
  434. return Array.Empty<string>();
  435. }
  436. }
  437. private static string RequireIdent(string? name, string label)
  438. {
  439. var v = (name ?? "").Trim();
  440. if (!System.Text.RegularExpressions.Regex.IsMatch(v, @"^[A-Za-z0-9_]+$"))
  441. throw new InvalidOperationException($"非法 {label}:{name}");
  442. return v;
  443. }
  444. private static string? FormatCursorValue(Dictionary<string, object?> dict, string column)
  445. {
  446. if (!dict.TryGetValue(column, out var raw) || raw == null || raw is DBNull)
  447. return null;
  448. return raw switch
  449. {
  450. DateTime dt => dt.ToString("yyyy-MM-dd HH:mm:ss.fff", CultureInfo.InvariantCulture),
  451. DateTimeOffset dto => dto.ToString("yyyy-MM-dd HH:mm:ss.fff", CultureInfo.InvariantCulture),
  452. _ => raw.ToString()
  453. };
  454. }
  455. internal static string EncodeKeysetCursor(string? cursor, string tieBreaker)
  456. => System.Text.Json.JsonSerializer.Serialize(new KeysetCursorDto
  457. {
  458. Cursor = cursor,
  459. TieBreaker = tieBreaker
  460. });
  461. internal static bool TryDecodeKeysetCursor(string? raw, out string? cursor, out string? tieBreaker)
  462. {
  463. cursor = null;
  464. tieBreaker = null;
  465. if (string.IsNullOrWhiteSpace(raw))
  466. return false;
  467. try
  468. {
  469. var dto = System.Text.Json.JsonSerializer.Deserialize<KeysetCursorDto>(raw);
  470. if (dto == null || string.IsNullOrWhiteSpace(dto.TieBreaker))
  471. return false;
  472. cursor = dto.Cursor;
  473. tieBreaker = dto.TieBreaker;
  474. return true;
  475. }
  476. catch
  477. {
  478. return false;
  479. }
  480. }
  481. private sealed class KeysetCursorDto
  482. {
  483. public string? Cursor { get; set; }
  484. public string? TieBreaker { get; set; }
  485. }
  486. private static string ResolveSourceRowId(Dictionary<string, object?> dict)
  487. {
  488. // InvTransHist 等表常有空字符串 ID,若优先取 ID 会导致 source_row_id="" 唯一键互相覆盖。
  489. // 跳过 null/空白后,再落到 RecID 等真实键。
  490. foreach (var key in new[]
  491. {
  492. "RecID", "RecId", "recid", "Recid",
  493. "id", "Id", "ID",
  494. "noid", "billno", "BillNo", "djbh", "Djbh"
  495. })
  496. {
  497. if (!dict.TryGetValue(key, out var v) || v is null || v is DBNull)
  498. continue;
  499. var s = v.ToString();
  500. if (!string.IsNullOrWhiteSpace(s))
  501. return s;
  502. }
  503. return Guid.NewGuid().ToString("N");
  504. }
  505. private async Task WriteSyncLogAsync(MdpSource source, MdpEntity entity, MdpPullContext ctx, DateTime startedAt, int pulled, int written, string? error, string? status = null)
  506. {
  507. try
  508. {
  509. var endedAt = DateTime.Now;
  510. var syncType = ctx.FullRefresh || string.Equals(ctx.SyncWindowType, "FULL", StringComparison.OrdinalIgnoreCase)
  511. ? "FULL"
  512. : "INCR";
  513. await _db.Ado.ExecuteCommandAsync(@"
  514. INSERT INTO mdp_sync_log
  515. (tenant_id, entity_id, source_code, entity_name, sync_batch_id, sync_type, trigger_type,
  516. sync_start, sync_end, duration_ms, rows_read, rows_written, status, error_message)
  517. VALUES
  518. (@tenantId, @entityId, @sourceCode, @entityName, @batchId, @syncType, 'AUTO',
  519. @syncStart, @syncEnd, @durationMs, @rowsRead, @rowsWritten, @status, @error)",
  520. new SugarParameter("@tenantId", ctx.TenantId),
  521. new SugarParameter("@entityId", entity.Id),
  522. new SugarParameter("@sourceCode", source.SourceCode),
  523. new SugarParameter("@entityName", string.IsNullOrWhiteSpace(entity.EntityName) ? entity.EntityCode : entity.EntityName),
  524. new SugarParameter("@batchId", ctx.BatchId),
  525. new SugarParameter("@syncType", syncType),
  526. new SugarParameter("@syncStart", startedAt),
  527. new SugarParameter("@syncEnd", endedAt),
  528. new SugarParameter("@durationMs", (int)(endedAt - startedAt).TotalMilliseconds),
  529. new SugarParameter("@rowsRead", pulled),
  530. new SugarParameter("@rowsWritten", written),
  531. new SugarParameter("@status", ResolveSyncStatus(status, error)),
  532. new SugarParameter("@error", error));
  533. }
  534. catch
  535. {
  536. // 日志表结构可能与种子不一致时不阻断主流程
  537. }
  538. }
  539. /// <summary>
  540. /// 显式 status 优先(用于 PARTIAL 等非成功但非失败的场景);未指定时沿用"有 error 即 FAILED"。
  541. /// 注意:SCOPE_MISMATCH 走 status=PARTIAL + error_message 诊断,而不是 FAILED——
  542. /// 它是"读到 0 行且可疑"的告警,不应把正常调度打成失败。
  543. /// </summary>
  544. internal static string ResolveSyncStatus(string? status, string? error)
  545. {
  546. if (!string.IsNullOrWhiteSpace(status)) return status!;
  547. return string.IsNullOrEmpty(error) ? SyncStatusSuccess : "FAILED";
  548. }
  549. private sealed class MysqlLiteralDateTimeConverter : JsonConverter<DateTime>
  550. {
  551. public override DateTime Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options)
  552. => reader.GetDateTime();
  553. public override void Write(Utf8JsonWriter writer, DateTime value, JsonSerializerOptions options)
  554. => writer.WriteStringValue(value.ToString("yyyy-MM-dd HH:mm:ss.ffffff", CultureInfo.InvariantCulture));
  555. }
  556. private sealed class MysqlLiteralDateTimeOffsetConverter : JsonConverter<DateTimeOffset>
  557. {
  558. public override DateTimeOffset Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options)
  559. => reader.GetDateTimeOffset();
  560. public override void Write(Utf8JsonWriter writer, DateTimeOffset value, JsonSerializerOptions options)
  561. => writer.WriteStringValue(value.ToString("yyyy-MM-dd HH:mm:ss.ffffff", CultureInfo.InvariantCulture));
  562. }
  563. }