TiDB Lightning 错误处理功能

从 TiDB 5.4.0 开始，你可以配置 TiDB Lightning 以跳过诸如无效类型转换、唯一键冲突等错误，让导入任务持续进行，就如同出现错误的行数据不存在一样。你可以依据生成的报告，手动修复这些错误。该功能适用于以下场景：

  • 要导入的数据有少许错误
  • 手动定位错误比较困难
  • 如果遇到错误就重启 TiDB Lightning，代价太大

本文介绍了类型错误处理功能 (lightning.max-error) 和重复问题处理功能 (tikv-importer.duplicate-resolution) 的使用方法，以及保存这些错误的数据库 (lightning.task-info-schema-name)，并提供了一个示例。

类型错误 (Type error)

你可以通过修改配置项 lightning.max-error 来增加数据类型相关的容错数量。如果设置为 N，那么 TiDB Lightning 允许数据源中出现 N 个错误，而且会跳过这些错误，一旦超过这个错误数就会退出。默认值为 0，表示不允许出现错误。

这些错误会被记录到数据库中。在导入完成后，你可以查看数据库中的数据，手动进行处理。请参见错误报告。

  1. [lightning]
  2. max-error = 0

该配置对下列错误有效：

  • 无效值。例如：在 INT 列设置了 'Text'
  • 数字溢出。例如：在 TINYINT 列设置了 500
  • 字符串溢出。例如: 在 VARCHAR(5) 列中设置了'非常长的文字'
  • 零日期时间，如 '0000-00-00' 和 '2021-12-00'
  • 在 NOT NULL 列中设置了 NULL
  • 生成的列表达式求值失败
  • 列计数不匹配。行中数值的数量和列的数量不一致
  • on-duplicate = "error" 时，TiDB 后端的唯一键/主键冲突
  • 其他 SQL 错误

下列错误是致命错误，不能通过配置 max-error 跳过：

  • 原始 CSV、SQL 或者 Parquet 文件中的语法错误，例如未闭合的引号
  • I/O、网络、或系统权限错误

在 Physical Import Mode 下，唯一键/主键的冲突是单独处理的。相关内容将在接下来的章节进行介绍。

错误报告

如果 TiDB Lightning 在运行过程中收集到报错的记录，则在退出时会同时在终端和日志中输出各个类型报错数量的统计信息。

  • 输出在终端的报错统计如下表所示：

    | # | ERROR TYPE | ERROR COUNT | ERROR DATA TABLE | | - | —- | —- | ——— | | 1 | Data Type | 1000 | lightning_task_info.type_error_v1 |

  • 输出在 TiDB Lightning 的 log 文件的结尾如下：

    1.   [2022/03/13 05:33:57.736 +08:00] [WARN] [errormanager.go:459] ["Detect 1000 data type errors in total, please refer to table `lightning_task_info`.`type_error_v1` for more details"]

所有错误都会写入下游 TiDB 集群 lightning_task_info 数据库中的表中。在导入完成后，如果收集到报错的数据，你可以根据数据库中记录的内容，手动进行处理。

你可以使用 lightning.task-info-schema-name 配置更改数据库名称。

  1. [lightning]
  2. task-info-schema-name = 'lightning_task_info'

在此数据库中，TiDB Lightning 创建了 3 个表：

  1. CREATE TABLE syntax_error_v1 (
  2.     task_id     bigint NOT NULL,
  3.     create_time datetime(6) NOT NULL DEFAULT now(6),
  4.     table_name  varchar(261) NOT NULL,
  5.     path        varchar(2048) NOT NULL,
  6.     offset      bigint NOT NULL,
  7.     error       text NOT NULL,
  8.     context     text
  9. );
  10. CREATE TABLE type_error_v1 (
  11.     task_id     bigint NOT NULL,
  12.     create_time datetime(6) NOT NULL DEFAULT now(6),
  13.     table_name  varchar(261) NOT NULL,
  14.     path        varchar(2048) NOT NULL,
  15.     offset      bigint NOT NULL,
  16.     error       text NOT NULL,
  17.     row_data    text NOT NULL
  18. );
  19. CREATE TABLE conflict_error_v1 (
  20.     task_id     bigint NOT NULL,
  21.     create_time datetime(6) NOT NULL DEFAULT now(6),
  22.     table_name  varchar(261) NOT NULL,
  23.     index_name  varchar(128) NOT NULL,
  24.     key_data    text NOT NULL,
  25.     row_data    text NOT NULL,
  26.     raw_key     mediumblob NOT NULL,
  27.     raw_value   mediumblob NOT NULL,
  28.     raw_handle  mediumblob NOT NULL,
  29.     raw_row     mediumblob NOT NULL,
  30.     KEY (task_id, table_name)
  31. );

type_error_v1 记录由 max-error 配置项管理的所有类型错误 (Type error)。每个错误一行。

conflict_error_v1 记录所有后端中的唯一键/主键冲突，每对冲突有两行。

列名 语法 类型 冲突 说明
task_id ✓ ✓ ✓ 生成此错误的 TiDB Lightning 任务 ID
create_time ✓ ✓ ✓ 记录错误的时间
table_name ✓ ✓ ✓ 包含错误的表的名称，格式为 '`db`.`tbl`'
path ✓ ✓ 包含错误文件的路径
offset ✓ ✓ 文件中发现错误的字节位置
error ✓ ✓ 错误信息
context ✓ 围绕错误的文本
index_name ✓ 冲突中唯一键的名称。主键冲突为 'PRIMARY'
key_data ✓ 导致错误的行的格式化键句柄。该内容仅供人参考，机器不可读
row_data ✓ ✓ 导致错误的格式化行数据。该内容仅供人参考，机器不可读
raw_key ✓ 冲突的 KV 对的键
raw_value ✓ 冲突的 KV 对的值
raw_handle ✓ 冲突行的行句柄
raw_row ✓ 冲突行的编码值

注意：

错误报告记录的是文件偏移量，不是行号或列号，因为行号或列号的获取效率很低。你可以使用下列命令在字节位实现快速跳转（以 183 为例）：

  • shell：输出前面几行

    1.   head -c 183 file.csv | tail

  • shell，输出后面几行

    1.   tail -c +183 file.csv | head

  • vim：:goto 183 或 183go

示例

在该示例中，我们准备了一个包含一些已知错误的数据源。以下是处理这些错误的具体步骤：

  1. 准备数据库和表结构：

    1.  mkdir example && cd example
    2.  echo 'CREATE SCHEMA example;' > example-schema-create.sql
    3.  echo 'CREATE TABLE t(a TINYINT PRIMARY KEY, b VARCHAR(12) NOT NULL UNIQUE);' > example.t-schema.sql

  2. 准备数据：

    1.  cat <<EOF > example.t.1.sql
    2.      INSERT INTO t (a, b) VALUES
    3.      (0, NULL),              -- 列不为空
    4.      (1, 'one'),
    5.      (2, 'two'),
    6.      (40, 'forty'),          -- 与下面的 `40` 冲突
    7.      (54, 'fifty-four'),     -- 与下面的 `'fifty-four'` 冲突
    8.      (77, 'seventy-seven'),  -- 字符串长度超过 12 个字符
    9.      (600, 'six hundred'),   -- 数字超出了 TINYINT 数据类型支持的范围
    10.      (40, 'fourty'),         -- 与上面的 `40` 冲突
    11.      (42, 'fifty-four');     -- 与上面的 `'fifty-four'` 冲突
    12.  EOF

  3. 配置 TiDB Lightning，启用严格 SQL 模式，使用 Local 后端模式进行导入，通过删除解决重复项，并最多跳过 10 个错误：

    1.  cat <<EOF > config.toml
    2.      [lightning]
    3.      max-error = 10
    4.      [tikv-importer]
    5.      backend = 'local'
    6.      sorted-kv-dir = '/tmp/lightning-tmp/'
    7.      duplicate-resolution = 'remove'
    8.      [mydumper]
    9.      data-source-dir = '.'
    10.      [tidb]
    11.      host = '127.0.0.1'
    12.      port = 4000
    13.      user = 'root'
    14.      password = ''
    15.      sql-mode = 'STRICT_TRANS_TABLES,NO_ZERO_IN_DATE,NO_ZERO_DATE'
    16.  EOF

  4. 运行 TiDB Lightning。因为已跳过所有错误，该命令执行完会成功退出：

    1.  tiup tidb-lightning -c config.toml

  5. 验证导入的表仅包含两个正常行：

    1.  $ mysql -u root -h 127.0.0.1 -P 4000 -e 'select * from example.t'
    2.  +---+-----+
    3.  | a | b   |
    4.  +---+-----+
    5.  | 1 | one |
    6.  | 2 | two |
    7.  +---+-----+

  6. 检查 type_error_v1 表是否捕获了涉及类型转换的三行：

    1.  $ mysql -u root -h 127.0.0.1 -P 4000 -e 'select * from lightning_task_info.type_error_v1;' -E
    2.  *************************** 1. row ***************************
    3.      task_id: 1635888701843303564
    4.  create_time: 2021-11-02 21:31:42.620090
    5.   table_name: `example`.`t`
    6.         path: example.t.1.sql
    7.       offset: 46
    8.        error: failed to cast value as varchar(12) CHARACTER SET utf8mb4 COLLATE utf8mb4_bin for column `b` (#2): [table:1048]Column 'b' cannot be null
    9.     row_data: (0,NULL)
    10.  *************************** 2. row ***************************
    11.      task_id: 1635888701843303564
    12.  create_time: 2021-11-02 21:31:42.627496
    13.   table_name: `example`.`t`
    14.         path: example.t.1.sql
    15.       offset: 183
    16.        error: failed to cast value as varchar(12) CHARACTER SET utf8mb4 COLLATE utf8mb4_bin for column `b` (#2): [types:1406]Data Too Long, field len 12, data len 13
    17.     row_data: (77,'seventy-seven')
    18.  *************************** 3. row ***************************
    19.      task_id: 1635888701843303564
    20.  create_time: 2021-11-02 21:31:42.629929
    21.   table_name: `example`.`t`
    22.         path: example.t.1.sql
    23.       offset: 253
    24.        error: failed to cast value as tinyint(4) for column `a` (#1): [types:1690]constant 600 overflows tinyint
    25.     row_data: (600,'six hundred')

  7. 检查 conflict_error_v1 表是否捕获了具有唯一键/主键冲突的四行：

    1.  $ mysql -u root -h 127.0.0.1 -P 4000 -e 'select * from lightning_task_info.conflict_error_v1;' --binary-as-hex -E
    2.  *************************** 1. row ***************************
    3.      task_id: 1635888701843303564
    4.  create_time: 2021-11-02 21:31:42.669601
    5.   table_name: `example`.`t`
    6.   index_name: PRIMARY
    7.     key_data: 40
    8.     row_data: (40, "forty")
    9.      raw_key: 0x7480000000000000C15F728000000000000028
    10.    raw_value: 0x800001000000020500666F727479
    11.   raw_handle: 0x7480000000000000C15F728000000000000028
    12.      raw_row: 0x800001000000020500666F727479
    13.  *************************** 2. row ***************************
    14.      task_id: 1635888701843303564
    15.  create_time: 2021-11-02 21:31:42.674798
    16.   table_name: `example`.`t`
    17.   index_name: PRIMARY
    18.     key_data: 40
    19.     row_data: (40, "fourty")
    20.      raw_key: 0x7480000000000000C15F728000000000000028
    21.    raw_value: 0x800001000000020600666F75727479
    22.   raw_handle: 0x7480000000000000C15F728000000000000028
    23.      raw_row: 0x800001000000020600666F75727479
    24.  *************************** 3. row ***************************
    25.      task_id: 1635888701843303564
    26.  create_time: 2021-11-02 21:31:42.680332
    27.   table_name: `example`.`t`
    28.   index_name: b
    29.     key_data: 54
    30.     row_data: (54, "fifty-four")
    31.      raw_key: 0x7480000000000000C15F6980000000000000010166696674792D666FFF7572000000000000F9
    32.    raw_value: 0x0000000000000036
    33.   raw_handle: 0x7480000000000000C15F728000000000000036
    34.      raw_row: 0x800001000000020A0066696674792D666F7572
    35.  *************************** 4. row ***************************
    36.      task_id: 1635888701843303564
    37.  create_time: 2021-11-02 21:31:42.681073
    38.   table_name: `example`.`t`
    39.   index_name: b
    40.     key_data: 42
    41.     row_data: (42, "fifty-four")
    42.      raw_key: 0x7480000000000000C15F6980000000000000010166696674792D666FFF7572000000000000F9
    43.    raw_value: 0x000000000000002A
    44.   raw_handle: 0x7480000000000000C15F72800000000000002A
    45.      raw_row: 0x800001000000020A0066696674792D666F7572