MdpDbPullExecutor.cs 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610
  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, 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. CancellationToken cancellationToken)
  187. {
  188. cancellationToken.ThrowIfCancellationRequested();
  189. try
  190. {
  191. var probeSql = BuildScopeProbeSql(entity, isSqlServer, ctx, useKeyset);
  192. var probe = await scope.Ado.GetDataTableAsync(probeSql, StripScopeParameters(parameters));
  193. return probe.Rows.Count > 0;
  194. }
  195. catch
  196. {
  197. // 探针纯诊断用途,任何失败都不得影响主流程,按"无异常"处理。
  198. return false;
  199. }
  200. }
  201. /// <summary>探针复用正式查询的参数,但必须剔除作用域参数(探针 SQL 不再引用它们)。</summary>
  202. internal static List<SugarParameter> StripScopeParameters(List<SugarParameter> parameters)
  203. => parameters
  204. .Where(p => !IsScopeParameter(p.ParameterName))
  205. .ToList();
  206. private static bool IsScopeParameter(string? name)
  207. {
  208. var n = (name ?? "").TrimStart('@', ':', '?');
  209. return string.Equals(n, "scopeTenantId", StringComparison.OrdinalIgnoreCase)
  210. || string.Equals(n, "scopeFactoryId", StringComparison.OrdinalIgnoreCase);
  211. }
  212. private static string RequireSourceTable(MdpEntity entity)
  213. {
  214. var table = entity.SourceTableName!.Trim();
  215. // 仅允许简单标识符/schema.table,防注入
  216. if (!System.Text.RegularExpressions.Regex.IsMatch(table, @"^[A-Za-z0-9_\.\[\]]+$"))
  217. throw new InvalidOperationException($"非法 source_table_name:{table}");
  218. return table;
  219. }
  220. /// <summary>增量/窗口谓词(不含租户/工厂作用域),供正式查询与作用域探针复用。</summary>
  221. private static List<string> BuildIncrementalPredicates(MdpEntity entity, bool isSqlServer, MdpPullContext ctx, out string orderBy)
  222. {
  223. var predicates = new List<string>();
  224. if (!string.IsNullOrWhiteSpace(entity.IncrColumn))
  225. {
  226. var incr = entity.IncrColumn.Trim();
  227. if (!System.Text.RegularExpressions.Regex.IsMatch(incr, @"^[A-Za-z0-9_]+$"))
  228. throw new InvalidOperationException($"非法 incr_column:{incr}");
  229. var incrExpr = isSqlServer ? incr : $"`{incr}`";
  230. orderBy = incrExpr;
  231. var isRolling = string.Equals(ctx.SyncWindowType, "ROLLING", StringComparison.OrdinalIgnoreCase);
  232. if (!ctx.FullRefresh && isRolling && ctx.WindowFrom.HasValue)
  233. predicates.Add($"{incrExpr} >= @windowFrom");
  234. var useCursor = !ctx.FullRefresh
  235. && !string.IsNullOrWhiteSpace(entity.LastCursor)
  236. && (!isRolling
  237. || (ctx.WindowFrom.HasValue
  238. && DateTime.TryParse(entity.LastCursor, out var cursorDt)
  239. && cursorDt > ctx.WindowFrom.Value));
  240. if (useCursor)
  241. predicates.Add($"{incrExpr} > @cursor");
  242. }
  243. else
  244. {
  245. orderBy = isSqlServer ? "(SELECT NULL)" : "1";
  246. }
  247. return predicates;
  248. }
  249. private static string BuildSelectSql(MdpEntity entity, bool isSqlServer, int batchSize, MdpPullContext ctx, string? tenantCol = null, string? factoryCol = null)
  250. {
  251. var table = RequireSourceTable(entity);
  252. var predicates = BuildIncrementalPredicates(entity, isSqlServer, ctx, out var orderBy);
  253. AppendScopePredicates(predicates, isSqlServer, tenantCol, factoryCol);
  254. var where = predicates.Count > 0 ? " WHERE " + string.Join(" AND ", predicates) : "";
  255. var offset = ctx.Offset > 0 ? ctx.Offset : 0;
  256. if (isSqlServer)
  257. {
  258. // SQL Server:OFFSET 需 ORDER BY;无 incr 时用稳定键兜底
  259. if (string.IsNullOrWhiteSpace(entity.IncrColumn))
  260. orderBy = "(SELECT NULL)";
  261. return $"SELECT * FROM {table}{where} ORDER BY {orderBy} OFFSET {offset} ROWS FETCH NEXT {batchSize} ROWS ONLY";
  262. }
  263. return offset > 0
  264. ? $"SELECT * FROM {table}{where} ORDER BY {orderBy} LIMIT {batchSize} OFFSET {offset}"
  265. : $"SELECT * FROM {table}{where} ORDER BY {orderBy} LIMIT {batchSize}";
  266. }
  267. /// <summary>keyset 谓词(不含租户/工厂作用域),供正式查询与作用域探针复用。</summary>
  268. private static List<string> BuildKeysetPredicates(bool isSqlServer, MdpPullContext ctx, out string orderBy)
  269. {
  270. var cursorCol = RequireIdent(ctx.CursorColumn, "CursorColumn");
  271. var tieCol = RequireIdent(ctx.TieBreakerColumn, "TieBreakerColumn");
  272. var cursorExpr = isSqlServer ? cursorCol : $"`{cursorCol}`";
  273. var tieExpr = isSqlServer ? tieCol : $"`{tieCol}`";
  274. var predicates = new List<string>();
  275. if (ctx.NullTimePhase)
  276. {
  277. predicates.Add($"{cursorExpr} IS NULL");
  278. if (!string.IsNullOrWhiteSpace(ctx.TieBreakerValue))
  279. predicates.Add($"{tieExpr} > @tieBreaker");
  280. }
  281. else
  282. {
  283. predicates.Add($"{cursorExpr} IS NOT NULL");
  284. if (ctx.BootstrapFrom.HasValue)
  285. predicates.Add($"{cursorExpr} >= @bootstrapFrom");
  286. // keyset lower bound
  287. predicates.Add($"""
  288. (
  289. @cursorTime IS NULL
  290. OR {cursorExpr} > @cursorTime
  291. OR ({cursorExpr} = @cursorTime AND {tieExpr} > @tieBreaker)
  292. )
  293. """);
  294. if (!string.IsNullOrWhiteSpace(ctx.UpperCursorValue))
  295. {
  296. predicates.Add($"""
  297. (
  298. {cursorExpr} < @upperTime
  299. OR ({cursorExpr} = @upperTime AND {tieExpr} <= @upperTie)
  300. )
  301. """);
  302. }
  303. }
  304. orderBy = ctx.NullTimePhase ? tieExpr : $"{cursorExpr}, {tieExpr}";
  305. return predicates;
  306. }
  307. private static string BuildKeysetSelectSql(MdpEntity entity, bool isSqlServer, int batchSize, MdpPullContext ctx, string? tenantCol = null, string? factoryCol = null)
  308. {
  309. var table = RequireSourceTable(entity);
  310. var predicates = BuildKeysetPredicates(isSqlServer, ctx, out var orderBy);
  311. AppendScopePredicates(predicates, isSqlServer, tenantCol, factoryCol);
  312. var where = " WHERE " + string.Join(" AND ", predicates);
  313. if (isSqlServer)
  314. return $"SELECT * FROM {table}{where} ORDER BY {orderBy} OFFSET 0 ROWS FETCH NEXT {batchSize} ROWS ONLY";
  315. return $"SELECT * FROM {table}{where} ORDER BY {orderBy} LIMIT {batchSize}";
  316. }
  317. /// <summary>
  318. /// 作用域探针:与正式查询完全同一组增量/窗口/keyset 谓词,唯独**去掉**租户/工厂作用域谓词。
  319. /// 正式查询 0 行而探针有行 ⇒ 差异只可能来自作用域过滤,可据此判定"全量被作用域过滤掉"。
  320. /// 只做存在性判断(TOP 1 / LIMIT 1),不取数据。
  321. /// </summary>
  322. internal static string BuildScopeProbeSql(MdpEntity entity, bool isSqlServer, MdpPullContext ctx, bool useKeyset)
  323. {
  324. var table = RequireSourceTable(entity);
  325. var predicates = useKeyset
  326. ? BuildKeysetPredicates(isSqlServer, ctx, out _)
  327. : BuildIncrementalPredicates(entity, isSqlServer, ctx, out _);
  328. var where = predicates.Count > 0 ? " WHERE " + string.Join(" AND ", predicates) : "";
  329. return isSqlServer
  330. ? $"SELECT TOP 1 1 AS probe FROM {table}{where}"
  331. : $"SELECT 1 AS probe FROM {table}{where} LIMIT 1";
  332. }
  333. private static void AddKeysetParameters(List<SugarParameter> parameters, MdpPullContext ctx)
  334. {
  335. if (ctx.BootstrapFrom.HasValue)
  336. parameters.Add(new SugarParameter("@bootstrapFrom", ctx.BootstrapFrom.Value));
  337. if (ctx.NullTimePhase)
  338. {
  339. parameters.Add(new SugarParameter("@tieBreaker",
  340. string.IsNullOrWhiteSpace(ctx.TieBreakerValue) ? "0" : ctx.TieBreakerValue));
  341. return;
  342. }
  343. object cursorTime = string.IsNullOrWhiteSpace(ctx.CursorValue)
  344. ? DBNull.Value
  345. : (object)ctx.CursorValue!;
  346. parameters.Add(new SugarParameter("@cursorTime", cursorTime));
  347. parameters.Add(new SugarParameter("@tieBreaker",
  348. string.IsNullOrWhiteSpace(ctx.TieBreakerValue) ? "0" : ctx.TieBreakerValue));
  349. if (!string.IsNullOrWhiteSpace(ctx.UpperCursorValue))
  350. {
  351. parameters.Add(new SugarParameter("@upperTime", ctx.UpperCursorValue));
  352. parameters.Add(new SugarParameter("@upperTie",
  353. string.IsNullOrWhiteSpace(ctx.UpperTieBreakerValue) ? "0" : ctx.UpperTieBreakerValue));
  354. }
  355. }
  356. /// <summary>
  357. /// 租户/工厂作用域谓词。工厂维度走 <see cref="MdpFactoryScope"/> 的归一语义
  358. /// (factory_id 为 NULL / &lt;=0 / 等于 tenant_id 一律视作"无真实工厂"→默认工厂 1),
  359. /// 与 <see cref="MdpStagingWriter"/> 写入侧完全一致;真实工厂号仍逐值隔离。
  360. /// </summary>
  361. internal static void AppendScopePredicates(List<string> predicates, bool isSqlServer, string? tenantCol, string? factoryCol)
  362. {
  363. if (!string.IsNullOrWhiteSpace(tenantCol))
  364. predicates.Add($"{QuoteIdent(tenantCol, isSqlServer)} = @scopeTenantId");
  365. if (!string.IsNullOrWhiteSpace(factoryCol))
  366. {
  367. predicates.Add(MdpFactoryScope.SqlPredicate(
  368. QuoteIdent(factoryCol, isSqlServer), "@scopeTenantId", "@scopeFactoryId"));
  369. }
  370. }
  371. /// <summary>
  372. /// 工厂谓词引用 @scopeTenantId,故只要发射了任一作用域谓词就必须绑定该参数
  373. /// (源表只有 factory_id 而无 tenant_id 时,旧逻辑会漏绑导致 SQL 参数缺失)。
  374. /// </summary>
  375. internal static bool NeedsTenantParameter(string? tenantCol, string? factoryCol)
  376. => !string.IsNullOrWhiteSpace(tenantCol) || !string.IsNullOrWhiteSpace(factoryCol);
  377. private static string QuoteIdent(string name, bool isSqlServer) => isSqlServer ? name : $"`{name}`";
  378. private static string? FindColumn(IReadOnlyCollection<string> columns, params string[] names)
  379. {
  380. foreach (var n in names)
  381. {
  382. var hit = columns.FirstOrDefault(c => string.Equals(c, n, StringComparison.OrdinalIgnoreCase));
  383. if (hit != null) return hit;
  384. }
  385. return null;
  386. }
  387. private static async Task<IReadOnlyCollection<string>> TryGetSourceColumnsAsync(
  388. ISqlSugarClient scope, string table, bool isSqlServer, CancellationToken cancellationToken)
  389. {
  390. try
  391. {
  392. var simple = table.Contains('.') ? table.Split('.').Last().Trim('[', ']') : table.Trim('[', ']');
  393. if (!System.Text.RegularExpressions.Regex.IsMatch(simple, @"^[A-Za-z0-9_]+$"))
  394. return Array.Empty<string>();
  395. if (isSqlServer)
  396. {
  397. var rows = await scope.Ado.SqlQueryAsync<string>(
  398. "SELECT name FROM sys.columns WHERE object_id = OBJECT_ID(@t)",
  399. new SugarParameter("@t", simple));
  400. return rows ?? new List<string>();
  401. }
  402. var mysql = await scope.Ado.SqlQueryAsync<string>(
  403. """
  404. SELECT COLUMN_NAME FROM information_schema.COLUMNS
  405. WHERE TABLE_SCHEMA=DATABASE() AND TABLE_NAME=@t
  406. """,
  407. new SugarParameter("@t", simple));
  408. return mysql ?? new List<string>();
  409. }
  410. catch
  411. {
  412. return Array.Empty<string>();
  413. }
  414. }
  415. private static string RequireIdent(string? name, string label)
  416. {
  417. var v = (name ?? "").Trim();
  418. if (!System.Text.RegularExpressions.Regex.IsMatch(v, @"^[A-Za-z0-9_]+$"))
  419. throw new InvalidOperationException($"非法 {label}:{name}");
  420. return v;
  421. }
  422. private static string? FormatCursorValue(Dictionary<string, object?> dict, string column)
  423. {
  424. if (!dict.TryGetValue(column, out var raw) || raw == null || raw is DBNull)
  425. return null;
  426. return raw switch
  427. {
  428. DateTime dt => dt.ToString("yyyy-MM-dd HH:mm:ss.fff", CultureInfo.InvariantCulture),
  429. DateTimeOffset dto => dto.ToString("yyyy-MM-dd HH:mm:ss.fff", CultureInfo.InvariantCulture),
  430. _ => raw.ToString()
  431. };
  432. }
  433. internal static string EncodeKeysetCursor(string? cursor, string tieBreaker)
  434. => System.Text.Json.JsonSerializer.Serialize(new KeysetCursorDto
  435. {
  436. Cursor = cursor,
  437. TieBreaker = tieBreaker
  438. });
  439. internal static bool TryDecodeKeysetCursor(string? raw, out string? cursor, out string? tieBreaker)
  440. {
  441. cursor = null;
  442. tieBreaker = null;
  443. if (string.IsNullOrWhiteSpace(raw))
  444. return false;
  445. try
  446. {
  447. var dto = System.Text.Json.JsonSerializer.Deserialize<KeysetCursorDto>(raw);
  448. if (dto == null || string.IsNullOrWhiteSpace(dto.TieBreaker))
  449. return false;
  450. cursor = dto.Cursor;
  451. tieBreaker = dto.TieBreaker;
  452. return true;
  453. }
  454. catch
  455. {
  456. return false;
  457. }
  458. }
  459. private sealed class KeysetCursorDto
  460. {
  461. public string? Cursor { get; set; }
  462. public string? TieBreaker { get; set; }
  463. }
  464. private static string ResolveSourceRowId(Dictionary<string, object?> dict)
  465. {
  466. // InvTransHist 等表常有空字符串 ID,若优先取 ID 会导致 source_row_id="" 唯一键互相覆盖。
  467. // 跳过 null/空白后,再落到 RecID 等真实键。
  468. foreach (var key in new[]
  469. {
  470. "RecID", "RecId", "recid", "Recid",
  471. "id", "Id", "ID",
  472. "noid", "billno", "BillNo", "djbh", "Djbh"
  473. })
  474. {
  475. if (!dict.TryGetValue(key, out var v) || v is null || v is DBNull)
  476. continue;
  477. var s = v.ToString();
  478. if (!string.IsNullOrWhiteSpace(s))
  479. return s;
  480. }
  481. return Guid.NewGuid().ToString("N");
  482. }
  483. private async Task WriteSyncLogAsync(MdpSource source, MdpEntity entity, MdpPullContext ctx, DateTime startedAt, int pulled, int written, string? error, string? status = null)
  484. {
  485. try
  486. {
  487. var endedAt = DateTime.Now;
  488. var syncType = ctx.FullRefresh || string.Equals(ctx.SyncWindowType, "FULL", StringComparison.OrdinalIgnoreCase)
  489. ? "FULL"
  490. : "INCR";
  491. await _db.Ado.ExecuteCommandAsync(@"
  492. INSERT INTO mdp_sync_log
  493. (tenant_id, entity_id, source_code, entity_name, sync_batch_id, sync_type, trigger_type,
  494. sync_start, sync_end, duration_ms, rows_read, rows_written, status, error_message)
  495. VALUES
  496. (@tenantId, @entityId, @sourceCode, @entityName, @batchId, @syncType, 'AUTO',
  497. @syncStart, @syncEnd, @durationMs, @rowsRead, @rowsWritten, @status, @error)",
  498. new SugarParameter("@tenantId", ctx.TenantId),
  499. new SugarParameter("@entityId", entity.Id),
  500. new SugarParameter("@sourceCode", source.SourceCode),
  501. new SugarParameter("@entityName", string.IsNullOrWhiteSpace(entity.EntityName) ? entity.EntityCode : entity.EntityName),
  502. new SugarParameter("@batchId", ctx.BatchId),
  503. new SugarParameter("@syncType", syncType),
  504. new SugarParameter("@syncStart", startedAt),
  505. new SugarParameter("@syncEnd", endedAt),
  506. new SugarParameter("@durationMs", (int)(endedAt - startedAt).TotalMilliseconds),
  507. new SugarParameter("@rowsRead", pulled),
  508. new SugarParameter("@rowsWritten", written),
  509. new SugarParameter("@status", ResolveSyncStatus(status, error)),
  510. new SugarParameter("@error", error));
  511. }
  512. catch
  513. {
  514. // 日志表结构可能与种子不一致时不阻断主流程
  515. }
  516. }
  517. /// <summary>
  518. /// 显式 status 优先(用于 PARTIAL 等非成功但非失败的场景);未指定时沿用"有 error 即 FAILED"。
  519. /// 注意:SCOPE_MISMATCH 走 status=PARTIAL + error_message 诊断,而不是 FAILED——
  520. /// 它是"读到 0 行且可疑"的告警,不应把正常调度打成失败。
  521. /// </summary>
  522. internal static string ResolveSyncStatus(string? status, string? error)
  523. {
  524. if (!string.IsNullOrWhiteSpace(status)) return status!;
  525. return string.IsNullOrEmpty(error) ? SyncStatusSuccess : "FAILED";
  526. }
  527. private sealed class MysqlLiteralDateTimeConverter : JsonConverter<DateTime>
  528. {
  529. public override DateTime Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options)
  530. => reader.GetDateTime();
  531. public override void Write(Utf8JsonWriter writer, DateTime value, JsonSerializerOptions options)
  532. => writer.WriteStringValue(value.ToString("yyyy-MM-dd HH:mm:ss.ffffff", CultureInfo.InvariantCulture));
  533. }
  534. private sealed class MysqlLiteralDateTimeOffsetConverter : JsonConverter<DateTimeOffset>
  535. {
  536. public override DateTimeOffset Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options)
  537. => reader.GetDateTimeOffset();
  538. public override void Write(Utf8JsonWriter writer, DateTimeOffset value, JsonSerializerOptions options)
  539. => writer.WriteStringValue(value.ToString("yyyy-MM-dd HH:mm:ss.ffffff", CultureInfo.InvariantCulture));
  540. }
  541. }