标签

PostgreSQL , 构造测试数据 , 任意字段组合AND,OR查询 , 指定结果集大小

---

背景

在进行一些实际的POC测试时，需要根据业务提出的需求构造数据，比如按照任意字段数组合 AND\OR 条件，指定返回结果条数，构造测试数据。

需求

表记录数A

表字段数B

1、N个字段等值OR，命中M条记录

（两个条件无法同时满足）

2、X个字段等值AND，命中Y条记录

字段取值空间如何计算？

构造算法

1、N个字段等值OR，命中M条记录

单个字段单个VALUE的记录数 = M/N

单个字段取值个数 = A/(M/N)

2、X个字段等值AND，命中Y条记录

（仅适用于完全离散分布，优化器里最难估算的也是多个字段AND的选择性，所以PG 10增加了多列统计信息）

X个字段的总取值空间 = A/Y

单个字段的取值空间 = X_/(A/Y) (开X根)

例子

1、表记录数1000万

2、表字段数64

字段取值空间如何计算？

1、16个字段等值OR，命中1000条记录

单个字段取值个数 = 10000000/(1000/16.0) = 160000

1、建表，64字段，根据要求填入每个字段的取值范围

do language plpgsql $$  declare    sql text := 'create table test1 (id int, ';  begin    for i in 1..64 loop      sql := sql||' c'||i||' int default random()*160000,';  -- 单个字段取值空间    end loop;    sql := rtrim(sql,',');    sql := sql||')';    execute sql;  end;  $$;

根据前面提供的需求，写入1000万记录

insert into test1 select generate_series(1,10000000);

根据要求生成查询SQL，16个字段组合OR

do language plpgsql $$  declare    sql text := 'select count(*) from test1 where ';  begin    for i in 1..16 loop      sql := sql||' c'||i||' ='||(random()*160000)::int||' or';   -- 16个字段 or 查询     end loop;    sql := rtrim(sql,'or');    raise notice '%', sql;  end;  $$;

生成SQL

select count(*) from test1 where    c1 =143477 or c2 =153395 or c3 =102052 or c4 =151143 or c5 =129060 or   c6 =87519 or c7 =148787 or c8 =123117 or c9 =126622 or c10 =118215 or   c11 =134245 or c12 =53791 or c13 =151020 or c14 =53076 or c15 =143204 or c16 =51640 ;

SQL实际返回数

count   -------     905  (1 row)

与算法预期基本一致(1000)。

2、16个字段等值AND，命中20条记录

单个字段的取值空间 = 16_/(10000000/20) = 2.27

1、根据算法，得到取值空间，创建测试表

do language plpgsql $$  declare    sql text := 'create table test2 (id int, ';  begin    for i in 1..64 loop      sql := sql||' c'||i||' int default random()*1,';  -- 单个字段取值空间    end loop;    sql := rtrim(sql,',');    sql := sql||')';    execute sql;  end;  $$;

写入1000万数据

insert into test2 select generate_series(1,10000000);

生成测试SQL，16个字段，OR查询

do language plpgsql $$  declare    sql text := 'select count(*) from test2 where ';  begin    for i in 1..16 loop      sql := sql||' c'||i||' ='||(random()*1)::int||' and';  -- 16个字段 and 查询     end loop;    sql := rtrim(sql,'and');    raise notice '%', sql;  end;  $$;

生成SQL

select count(*) from test2 where  c1 =1 and c2 =0 and c3 =0 and c4 =1 and   c5 =1 and c6 =1 and c7 =0 and c8 =1 and c9 =0 and c10 =0 and c11 =0 and   c12 =0 and c13 =0 and c14 =0 and c15 =1 and c16 =0;

SQL实际返回数

count   -------     154  (1 row)

与算法预期基本一致(取值范围作了取舍2.27，降到了2)。

扩展问题