部分列更新

概述

要实现部分列更新,在Doris中可以使用Unique Key模型或Aggregate Key模型

Unique Key模型

Doris Unique Key表默认的数据写入语义是整行Upsert,在2.0版本之前,用户想要更新某些行的一部分字段,只能通过UPDATE命令,但是UPDATE命令由于读写事务的锁粒度原因,并不适合高频的数据写入场景。因此我们在2.0版本引入了Unique Key模型的部分列更新支持

注意:

  1. 2.0.0版本仅在Unique Key的Merge-on-Write实现中支持了部分列更新能力
  2. 2.0.2版本支持使用INSERT INTO进行部分列更新
  3. 2.1.0版本将支持更为灵活的列更新,见下文“使用限制”部分的说明

Aggregate Key模型

Aggregate Key表主要在预聚合场景使用而非数据更新的场景使用,但也可以通过将聚合函数设置为REPLACE_IF_NOT_NULL来实现列更新效果

适用场景

  • 实时的动态列更新,需要在表中实时的高频更新某些字段值。例如T+1生成的用户标签表中有一些关于用户最新行为信息的字段需要实时的更新,以实现广告/推荐等系统能够据其进行实时的分析和决策
  • 将多张源表拼接成一张大宽表
  • 数据修正

基本原理

关于Unique Key模型和Aggregate Key模型的原理,可以主要参考数据模型的介绍

Unique Key 模型

Unique Key模型目前仅支持在Merge-on-Write实现上进行列更新

用户通过正常的导入方式将一部分列的数据写入Doris的Memtable,此时Memtable中并没有整行数据,在Memtable下刷的时候,会查找历史数据,用历史数据补齐一整行,并写入数据文件中,同时将历史数据文件中相同key的数据行标记删除

当出现并发导入时,Doris会利用MVCC机制来保证数据的正确性。如果两批数据导入都更新了一个相同key的不同列,则其中系统版本较高的导入任务会在版本较低的导入任务成功后,使用版本较低的导入任务写入的相同key的数据行重新进行补齐

Aggregate Key模型

将聚合函数设置为REPLACE_IF_NOT_NULL即可实现部分列更新的支持,详细用法参考下文示例

并发写入和数据可见性

部分列更新支持高频的并发写入,写入成功后数据即可见,系统自动通过MVCC机制来保证并发写入的数据正确性

性能

使用建议:

  1. 对写入性能要求较高,查询性能要求较低的用户,建议使用Aggregate Key模型
  2. 对查询性能要求较高,对写入性能要求不高(例如数据的写入和更新基本都在凌晨低峰期完成),或者写入频率不高的用户,建议使用Unique Key模型merge-on-write实现

Unique Key模型Merge-on-Write实现

由于Merge-on-Write实现需要在数据写入的时候,进行整行数据的补齐,以保证最优的查询性能,因此使用Merge-on-Write实现进行部分列更新会有较为明显的导入性能下降。

写入性能优化建议:

  1. 使用配备了NVMe的SSD,或者极速SSD云盘。因为补齐数据时会大量的读取历史数据,产生较高的读IOPS,以及读吞吐
  2. 开启行存将能够大大减少补齐数据时产生的IOPS,导入性能提升明显,用户可以在建表时通过如下property来开启行存:
  1. "store_row_column" = "true"

Aggregate Key模型

Aggregate Key模型在写入过程中不做任何额外处理,所以写入性能不受影响,与普通的数据导入相同。但是在查询时进行聚合的代价较大,典型的聚合查询性能相比Unique Key模型的Merge-on-Write实现会有5-10倍的下降。

使用方式及示例

Unique Key模型

建表

建表时需要指定如下property,以开启Merge-on-Write实现

  1. enable_unique_key_merge_on_write = true

StreamLoad/BrokerLoad/RoutineLoad

如果使用的是StreamLoad/BrokerLoad/RoutineLoad,在导入时添加如下header

  1. partial_columns:true

同时在columns中指定要导入的列(必须包含所有key列,不然无法更新)

如果使用Flink Connector, 需要添加如下配置:

  1. 'sink.properties.partial_columns' = 'true',

同时在sink.properties.columns中指定要导入的列(必须包含所有key列,不然无法更新)

INSERT INTO

在所有的数据模型中,INSERT INTO 给定一部分列时默认行为都是整行写入,为了防止误用,在Merge-on-Write实现中,INSERT INTO默认仍然保持整行UPSERT的语意,如果需要开启部分列更新的语意,需要设置如下 session variable

  1. set enable_unique_key_partial_update=true

需要注意的是,控制insert语句是否开启严格模式的会话变量enable_insert_strict的默认值为true,即insert语句默认开启严格模式,而在严格模式下进行部分列更新不允许更新不存在的key。所以,在使用insert语句进行部分列更新的时候如果希望能插入不存在的key,需要在enable_unique_key_partial_update设置为true的基础上同时将enable_insert_strict设置为false。

示例

假设 Doris 中存在一张订单表order_tbl,其中 订单id 是 Key 列,订单状态,订单金额是 Value 列。数据状态如下:

订单id订单金额订单状态
1100待付款
  1. +----------+--------------+--------------+
  2. | order_id | order_amount | order_status |
  3. +----------+--------------+--------------+
  4. | 1 | 100 | 待付款 |
  5. +----------+--------------+--------------+
  6. 1 row in set (0.01 sec)

这时候,用户点击付款后,Doris 系统需要将订单id 为 ‘1’ 的订单状态变更为 ‘待发货’。

若使用StreamLoad可以通过如下方式进行更新:

  1. $cat update.csv
  2. 1,待发货
  3. $ curl --location-trusted -u root: -H "partial_columns:true" -H "column_separator:," -H "columns:order_id,order_status" -T /tmp/update.csv http://127.0.0.1:48037/api/db1/order_tbl/_stream_load

若使用INSRT INTO可以通过如下方式进行更新:

  1. set enable_unique_key_partial_update=true;
  2. INSERT INTO order_tbl (order_id, order_status) values (1,'待发货');

更新后结果如下

  1. +----------+--------------+--------------+
  2. | order_id | order_amount | order_status |
  3. +----------+--------------+--------------+
  4. | 1 | 100 | 待发货 |
  5. +----------+--------------+--------------+
  6. 1 row in set (0.01 sec)

Aggregate Key模型

建表

将需要进行列更新的字段对应的聚合函数设置为REPLACE_IF_NOT_NULL

  1. CREATE TABLE `order_tbl` (
  2. `order_id` int(11) NULL,
  3. `order_amount` int(11) REPLACE_IF_NOT_NULL NULL,
  4. `order_status` varchar(100) REPLACE_IF_NOT_NULL NULL
  5. ) ENGINE=OLAP
  6. AGGREGATE KEY(`order_id`)
  7. COMMENT 'OLAP'
  8. DISTRIBUTED BY HASH(`order_id`) BUCKETS 1
  9. PROPERTIES (
  10. "replication_allocation" = "tag.location.default: 1"
  11. );

数据写入

无论是导入任务还是INSERT INTO, 直接写入要更新的字段的数据即可

示例

与前面例子相同,对应的Stream Load命令为(不需要额外的header):

  1. curl --location-trusted -u root: -H "column_separator:," -H "columns:order_id,order_status" -T /tmp/update.csv http://127.0.0.1:48037/api/db1/order_tbl/_stream_load

对应的INSERT INTO语句为(不需要额外设置session variable):

  1. INSERT INTO order_tbl (order_id, order_status) values (1,'待发货');

使用限制

Unique Key模型Merge-on-Write实现

同一批次数据写入任务(无论是导入任务还是INSERT INTO)的所有行只能更新相同的列,如果需要更新不同列的数据,则需要分不同的批次进行写入

Aggregate Key模型

用户无法通过将某个字段由非NULL设置为NULL,写入的NULL值在REPLACE_IF_NOT_NULL聚合函数的处理中会自动忽略