PostgreSQL、Greenplum DML合并操作 最佳实践

6 minute read

背景

在很多场景中会涉及到数据的合并,比如

1. 某业务系统的总用户有1亿,每天的活跃用户有100万,新增用户10万,每天需要将新增、活跃用户的数据(比如他们的余额变化、等等)合并到数据仓库的用户信息表。

2. 物化视图,某个表被用户不断的增、删、改。需要将这个表(基表)的某些字段或者某部分数据提取到一个物化视图中。这个物化视图不需要对每一笔基表的DML都实施操作,比如对单条记录的操作,合并成一次操作。

3. 数据同步,将OLTP的数据,同步到OLAP系统,由于OLAP系统的事务处理能力没有TP系统强,所以也必须采用合并的方法,同一条记录被多次更新时,需要将多次更新合并成一次更新。

4. 基于REDO日志的逻辑数据复制,优化手段除了并行复制,还有一种就是合并复制。

不管是哪种数据合并,被合并的表最好是有主键的,本文也假设有主键来处理。否则会增加复杂度(需要使用整行记录来区分),而且整行记录有一个缺陷,例如根据行号定位重复记录中的一条,这样变更后,合并时可能会出错。

数据合并的方法

对于以上几种情况,比较复杂的是逻辑数据复制,它可能涉及到任意操作,单个KEY可能被删除,后续这个KEY又被插入、多次更新的情况。

同时还需要考虑事务一致性的问题,每一次合并操作都需要保证一致性。例如基于REDO的逻辑复制,对于未结束的事务产生的REDO,不能参与合并。

保证单个KEY,在合并时只操作一次,同时确保未结束的事务不参与合并。

pic

REDO要素

table : 库\schema\表名

old : 主键值

new : 新插入的值 、 被变更的字段变更后的值

tag : insert 、 update 、 delete 、 truncate

例子

以逻辑复制为例,分解一下数据合并的过程。

创建测试表

create table tbl (pk1 int, pk2 int, c1 int, c2 text, crt_time timestamp, primary key(pk1,pk2));  

产生一些DML

...忽略中间部分, 假设忽略的这部分已经同步到目标表...  
  
delete from tbl where pk1=2 and pk2=2;   -- 说明目标表已经存在pk1=2,pk2=2的记录  
insert into tbl (pk1,pk2,c1,c2,crt_time) values (2,2,2,'test22','2017-02-14');  
insert into tbl (pk1,pk2,c1,c2,crt_time) values (1,1,2,'test23','2017-02-14');  
insert into tbl (pk1,pk2,c1,c2,crt_time) values (1,2,3,'test34','2017-02-14');  
  
update tbl set c2='new', crt_time=null where pk1=1 and pk2=2;  
  
delete from tbl where pk1=1 and pk2=1;  
  
insert into tbl (pk1,pk2,c1,c2,crt_time) values (1,1,5,'test56','2017-02-14');  
  
update tbl set c2='new11', crt_time=null where pk1=1 and pk2=1;  
update tbl set c2='new12', crt_time='2017-02-15' where pk1=1 and pk2=2;  

将以上DML转换为REDO要素如下(通常以下信息可以在数据库的REDO日志得到)

old: pk1=2,pk2=2  
new: null  
tag: delete  
  
old: null  
new: pk1=2,pk2=2,c1=2,c2='test22',crt_time='2017-02-14'  
tag: insert  
  
old: null  
new: pk1=1,pk2=1,c1=2,c2='test23',crt_time='2017-02-14'  
tag: insert  
  
old: null  
new: pk1=1,pk2=2,c1=3,c2='test34',crt_time='2017-02-14'  
tag: insert  
  
old: pk1=1,pk2=2  
new: c2='new',crt_time=null  
tag: update  
  
old: pk1=1,pk2=1  
new: null  
tag: delete  
  
old: null  
new: pk1=1,pk2=1,c1=5,c2='test56',crt_time='2017-02-14'  
tag: insert  
  
old: pk1=1,pk2=1  
new: c2='new11',crt_time=null  
tag: update  
  
old: pk1=1,pk2=2  
new: c2='new12',crt_time='2017-02-15'  
tag: update  

合并过程,对已提交的记录,按PK进行分组,按执行先后顺序排序

delete from tbl where pk1=2 and pk2=2;   -- 说明目标表已经存在pk1=2,pk2=2的记录  
insert into tbl (pk1,pk2,c1,c2,crt_time) values (2,2,2,'test22','2017-02-14');  
  
合并后,目标表应该仅仅执行如下SQL  
update tbl set c1=2,c2='test22',crt_time='2017-02-14' where pk1=2 and pk2=2;   
insert into tbl (pk1,pk2,c1,c2,crt_time) values (1,1,2,'test23','2017-02-14');  
delete from tbl where pk1=1 and pk2=1;  
insert into tbl (pk1,pk2,c1,c2,crt_time) values (1,1,5,'test56','2017-02-14');  
update tbl set c2='new11', crt_time=null where pk1=1 and pk2=1;  
  
合并后,目标表应该仅仅执行如下SQL  
insert into tbl (pk1,pk2,c1,c2,crt_time) values (1,1,5,'new11',null);  
insert into tbl (pk1,pk2,c1,c2,crt_time) values (1,2,3,'test34','2017-02-14');  
update tbl set c2='new', crt_time=null where pk1=1 and pk2=2;  
update tbl set c2='new12', crt_time='2017-02-15' where pk1=1 and pk2=2;  
  
合并后,目标表应该仅仅执行如下SQL  
insert into tbl (pk1,pk2,c1,c2,crt_time) values (1,2,3,'new12','2017-02-15');  

如果涉及到PK的变更,需要将其分解为delete和insert两条

例如  
old: pk1=1,pk2=1  
new: pk1=1,pk2=3,c2='new11',crt_time=null  
tag: update  
  
需要分解为  
delete from tbl where pk1=1,pk2=1;  
insert into tbl (pk1,pk2,c2,crt_time) values (1,3,'new11',null);  

OLAP数据合并更新、删除的例子1

逻辑复制的合并相对来说比较复杂,但是PostgreSQL是一个功能强大的数据库,它支持窗口查询,编程能力强大的plpgsql函数语言(还有python, java, perl等数据库函数语言),使用SQL还是比较方便的可以完成以上合并的。

除了逻辑复制,在OLAP中也经常要用到合并更新,主要的目的是减少OLAP系统SQL的执行次数(因为OLAP系统并不是为TP业务设计,而是为批处理或大量运算设计的,多次SQL如果能合并成一次的话,可以大幅提升效率)

下面就以更新的合并为例,简单的讲解一下数据合并的例子。

比如一张表有1亿记录,每天要更新其中的10万条记录。我们要做的是将10万条UPDATE语句,合并成一条UPDATE语句。

合并方法

1. 首先将更新语句转换为数据,插入一张临时表

2. 然后使用join update来更新目标表

过程如下

创建一个生产表(目标表,必须有PK),假设它有1亿(为了演示,仅使用100万记录)用户数据。

create table prod(id int primary key, c1 int, info text, crt_time timestamp, mod_time timestamp);    

创建一张临时表,用来存储合并前的DML,表结构如下,需要包含一个新增的序列PK,以及目标表的所有字段,以及每个字段对应的SET位(表示该字段是否被更新)

我们这里假设一条记录,可能被多次更新。

create table tmp1  
(  
  pk serial8 primary key,   -- 标记插入顺序  
  id int, c1 int, info text, crt_time timestamp, mod_time timestamp, -- 更新后的值  
  set_id boolean, set_c1 boolean, set_info boolean, set_crt_time boolean, set_mod_time boolean -- 被更新的字段  
);  

插入100万数据到prod表

insert into prod select generate_series(1,1000000), 1, 'test', now(), null;  

TP系统中的UPDATE语句,我们将它转换为目标值,插入临时表

insert into tmp1 (id,c1,info,crt_time,mod_time,set_id,set_c1,set_info,set_crt_time,set_mod_time)  
  select random()*10000, 2, null, null, clock_timestamp(), true,true,false,false,true from generate_series(1,10000);  -- c1=2, mod_time=clock_timestamp()  
  
insert into tmp1 (id,c1,info,crt_time,mod_time,set_id,set_c1,set_info,set_crt_time,set_mod_time)  
  select random()*10000, 3, 'new', null, clock_timestamp(), true,true,true,false,true from generate_series(1,10000);  -- c1=3, info='new', mod_time=clock_timestamp()  
  
insert into tmp1 (id,c1,info,crt_time,mod_time,set_id,set_c1,set_info,set_crt_time,set_mod_time)  
  select random()*10000, null, 'new1', null, clock_timestamp(), true,false,true,false,true from generate_series(1,10000);  -- info='new1', mod_time=clock_timestamp()  
  
insert into tmp1 (id,c1,info,crt_time,mod_time,set_id,set_c1,set_info,set_crt_time,set_mod_time)  
  select random()*10000, 5, null, null, clock_timestamp(), true,true,true,false,true from generate_series(1,10000);  -- c1=5, info=null, mod_time=clock_timestamp()  

true大于false,将用于多条记录的合并

postgres=# select true > false;  
 ?column?   
----------  
 t  
(1 row)  

如果一条记录被多次UPDATE,需要将多个UPDATE合并为一个UPDATE

用到了窗口查询,以目标表的PK为分组,按不同字段的set位优先取true的最后一条值,以及它的set状态。

select id, c1, set_c1, info, set_info, crt_time, set_crt_time, mod_time, set_mod_time from  
(  
  select   
    row_number() over (partition by id) as rn,  
    id,   
    first_value(c1) over (partition by id order by set_c1 desc, pk desc) c1,   
    first_value(set_c1) over (partition by id order by set_c1 desc, pk desc) set_c1,   
    first_value(info) over (partition by id order by set_info desc, pk desc) info,   
    first_value(set_info) over (partition by id order by set_info desc, pk desc) set_info,   
    first_value(crt_time) over (partition by id order by set_crt_time desc, pk desc) crt_time,   
    first_value(set_crt_time) over (partition by id order by set_crt_time desc, pk desc) set_crt_time,   
    first_value(mod_time) over (partition by id order by set_mod_time desc, pk desc) mod_time,   
    first_value(set_mod_time) over (partition by id order by set_mod_time desc, pk desc) set_mod_time  
  from tmp1  
) t  
where t.rn=1;  

以上就是合并后的数据

更新时,使用case,将字段set位为true的值更新为新的值,false的不变。

update prod set   
  c1=(case when t.set_c1 then t.c1 else prod.c1 end) ,  -- 将字段set位为true的值更新为新的值,false的不变。  
  info=(case when t.set_info then t.info else prod.info end) ,  
  crt_time=(case when t.set_crt_time then t.crt_time else prod.crt_time end) ,  
  mod_time=(case when t.set_mod_time then t.mod_time else prod.mod_time end)   
from   
(  
  select id, c1, set_c1, info, set_info, crt_time, set_crt_time, mod_time, set_mod_time   
  from  
  (  
    select   
      row_number() over (partition by id) as rn,  
      id,   
      first_value(c1) over (partition by id order by set_c1 desc, pk desc) c1,   
      first_value(set_c1) over (partition by id order by set_c1 desc, pk desc) set_c1,   
      first_value(info) over (partition by id order by set_info desc, pk desc) info,   
      first_value(set_info) over (partition by id order by set_info desc, pk desc) set_info,   
      first_value(crt_time) over (partition by id order by set_crt_time desc, pk desc) crt_time,   
      first_value(set_crt_time) over (partition by id order by set_crt_time desc, pk desc) set_crt_time,   
      first_value(mod_time) over (partition by id order by set_mod_time desc, pk desc) mod_time,   
      first_value(set_mod_time) over (partition by id order by set_mod_time desc, pk desc) set_mod_time  
      from tmp1  
  ) t  
  where t.rn=1  
) t  
where prod.id=t.id;  

删除更加简单,只需要将ID记录下来,delete from tbl where id in (…)即可,不再列举。

验证以上合并方法的一致性

使用两张目标表,一张为合并更新(合并更新的数据来源于实时更新的触发器日志),一张为实时更新。

drop table IF EXISTS prod;  
drop table IF EXISTS prod_ck;  
drop table IF EXISTS tmp1;  
  
create table prod(id int primary key, c1 int, info text, crt_time timestamp, mod_time timestamp);    
insert into prod select generate_series(1,1000000), 1, 'test', now(), null;  
  
create table prod_ck(id int primary key, c1 int, info text, crt_time timestamp, mod_time timestamp);    
insert into prod_ck select * from prod;  
  
create table tmp1  
(  
  pk serial8 primary key,   -- 标记插入顺序  
  id int, c1 int, info text, crt_time timestamp, mod_time timestamp, -- 更新后的值  
  set_id boolean, set_c1 boolean, set_info boolean, set_crt_time boolean, set_mod_time boolean -- 被更新的字段  
);    
  
create or replace function f_tg() returns trigger as $$  
declare  
begin  
  insert into tmp1 (id,c1,info,crt_time,mod_time,set_id,set_c1,set_info,set_crt_time,set_mod_time) values  
    (NEW.id, NEW.c1, NEW.info, NEW.crt_time, NEW.mod_time, true, true, true, true, true);  
  return null;  
end;  
$$ language plpgsql strict;  
  
create trigger tg after update on prod_ck for each row execute procedure f_tg();  

使用pgbench,不断更新prod_ck

vi test.sql  
  
\set id random(1,10000)  
\set c1 random(1,1000000)  
update prod_ck set c1=:c1, crt_time=clock_timestamp(), mod_time=null where id=:id;  
update prod_ck set c1=:c1, crt_time=null, mod_time=clock_timestamp() where id=:id+1;  
  
pgbench -M prepared -n -r -P 1 -f ./test.sql -c 32 -j 32 -T 30  
  
  
progress: 3.0 s, 26379.3 tps, lat 1.211 ms stddev 0.660  
progress: 4.0 s, 26339.9 tps, lat 1.213 ms stddev 0.700  

将tmp1的更新合并到prod

update prod set   
  c1=(case when t.set_c1 then t.c1 else prod.c1 end) ,  -- 将字段set位为true的值更新为新的值,false的不变。  
  info=(case when t.set_info then t.info else prod.info end) ,  
  crt_time=(case when t.set_crt_time then t.crt_time else prod.crt_time end) ,  
  mod_time=(case when t.set_mod_time then t.mod_time else prod.mod_time end)   
from   
(  
  select id, c1, set_c1, info, set_info, crt_time, set_crt_time, mod_time, set_mod_time   
  from  
  (  
    select   
      row_number() over (partition by id) as rn,  
      id,   
      first_value(c1) over (partition by id order by set_c1 desc, pk desc) c1,   
      first_value(set_c1) over (partition by id order by set_c1 desc, pk desc) set_c1,   
      first_value(info) over (partition by id order by set_info desc, pk desc) info,   
      first_value(set_info) over (partition by id order by set_info desc, pk desc) set_info,   
      first_value(crt_time) over (partition by id order by set_crt_time desc, pk desc) crt_time,   
      first_value(set_crt_time) over (partition by id order by set_crt_time desc, pk desc) set_crt_time,   
      first_value(mod_time) over (partition by id order by set_mod_time desc, pk desc) mod_time,   
      first_value(set_mod_time) over (partition by id order by set_mod_time desc, pk desc) set_mod_time  
      from tmp1  
  ) t  
  where t.rn=1  
) t  
where prod.id=t.id;  
  
UPDATE 10001  

验证合并更新后prod和prod_ck是否一致

postgres=# select sum(hashtext(t.*::text)) from prod t;  
      sum         
----------------  
 -2538529730583  
(1 row)  
  
postgres=# select sum(hashtext(t.*::text)) from prod_ck t;  
      sum         
----------------  
 -2538529730583  
(1 row)  

小结

数据合并的目标是将多条DML语句合并成一条,

包括将单条记录的多次更新、插入、删除合并为一次更新、插入或删除操作,

也包括将多条记录的多次DML合并成一条DML语句。

在数据逻辑复制、TP到AP业务系统的同步、物化视图 等场景有着广泛的应用。

特别是OLAP系统,由于并不是针对TP场景涉及,使用合并操作,可以大幅提升AP系统的操作效率。(Greenplum更新和删除都是表级锁, 效率也一般)

在greenplum单条记录,基于PK的更新速度测试

pgbench -M simple -n -r -f ./test.sql -P 1 -c 4 -j 4 -T 100 -h 127.0.0.1 -p 29999 -U digoal
progress: 1.0 s, 203.0 tps, lat 19.449 ms stddev 29.442
progress: 2.0 s, 290.0 tps, lat 13.739 ms stddev 0.337
progress: 3.0 s, 291.0 tps, lat 13.763 ms stddev 0.627
progress: 4.0 s, 276.0 tps, lat 14.035 ms stddev 3.919
progress: 5.0 s, 280.0 tps, lat 14.791 ms stddev 5.954
progress: 6.0 s, 296.0 tps, lat 13.493 ms stddev 1.720
progress: 7.0 s, 300.0 tps, lat 13.347 ms stddev 1.433

1万次更新需要十几多秒才能完成,而使用合并更新,只需要0.几秒

而插入方面,Greenplum没有表级锁,效率比更新高很多。
^C
pgbench -M simple -n -r -f ./test.sql -P 1 -c 2 -j 2 -T 100
progress: 1.0 s, 11124.9 tps, lat 0.178 ms stddev 0.018
progress: 2.0 s, 10910.9 tps, lat 0.182 ms stddev 0.011
progress: 3.0 s, 10903.3 tps, lat 0.182 ms stddev 0.011
^C
pgbench -M simple -n -r -f ./test.sql -P 1 -c 1 -j 1 -T 100
progress: 1.0 s, 4998.3 tps, lat 0.170 ms stddev 0.018
progress: 2.0 s, 4041.6 tps, lat 0.282 ms stddev 7.163
progress: 3.0 s, 5894.7 tps, lat 0.169 ms stddev 0.006
^C
pgbench -M simple -n -r -f ./test.sql -P 1 -c 8 -j 8 -T 100
progress: 1.0 s, 39995.8 tps, lat 0.198 ms stddev 0.146
progress: 2.0 s, 41325.8 tps, lat 0.192 ms stddev 0.037
progress: 3.0 s, 40775.6 tps, lat 0.195 ms stddev 0.149
progress: 4.0 s, 41638.2 tps, lat 0.191 ms stddev 0.046

Flag Counter

digoal’s 大量PostgreSQL文章入口