postgresql创建分区表的两种方式

postgresql按天创建分区表

postgresql创建分区表由两种方式，一种是通过继承主表的方式创建，另一种通过partition of的方式创建。推荐使用partition of的方式创建，按照继承的方式创建的分区表在查询时分区字段不支持动态的条件查询。比如查询以下按照继承的方式创建的表时：select * from tbl_partition where gather_time > now() and gather_time < now() + interval 1；仍然会进行全表扫描。如果想扫描固定分区只能：select * from tbl_partition where gather_time = ‘20210101’；where条件要跟确定的值才行。

通过继承的方式自动按天创建分区表

1. 创建主表：

CREATE TABLE tbl_partition
(
  id text,
  name text,
  data numeric,
  gather_time timestamp
);

2. 创建可重用的触发器函数

CREATE OR REPLACE FUNCTION auto_insert_into_tbl_partition()
  RETURNS trigger AS
$BODY$
DECLARE
    id	text ;
    time_column_name 	text ;			-- 父表中用于分区的时间字段的名称[必须首先初始化!!]
    curMM 		varchar(8);		-- 'YYYYMM'字串,用做分区子表的后缀
    isExist 		boolean;		-- 分区子表,是否已存在
    startTime 		text;
    endTime		text;
    strSQL  		text;
    
BEGIN
    -- 调用前,必须首先初始化(时间字段名):time_column_name [直接从调用参数中获取!!]
    time_column_name := TG_ARGV[0];
    id := TG_ARGV[1];
    -- 判断对应分区表 是否已经存在?
    EXECUTE 'SELECT $1.'||time_column_name INTO strSQL USING NEW;
    curMM := to_char( strSQL::timestamp , 'YYYYMMDD' );
    select count(*) INTO isExist from pg_class where relname = (TG_RELNAME||'_'||curMM);
 
    -- 若不存在, 则插入前需 先创建子分区
    IF ( isExist = false ) THEN  
        -- 创建子分区表
        startTime := curMM||' 00:00:00.000';
        endTime := to_char( startTime::timestamp + interval '1 day', 'YYYY-MM-DD HH24:MI:SS.MS');
        strSQL := 'CREATE TABLE IF NOT EXISTS '||TG_RELNAME||'_'||curMM||
                  ' ( CHECK('||time_column_name||'>='''|| startTime ||''' AND '
                             ||time_column_name||'< '''|| endTime ||''' )
                          ) INHERITS ('||TG_RELNAME||') ;'  ;  
        EXECUTE strSQL;
 
        -- 创建索引
        strSQL := 'CREATE INDEX '||TG_RELNAME||'_'||curMM||'_INDEX_'||id||' ON '
                  ||TG_RELNAME||'_'||curMM||' ('||id||');' ;
        EXECUTE strSQL;
       
    END IF;
 
    -- 插入数据到子分区!
    strSQL := 'INSERT INTO '||TG_RELNAME||'_'||curMM||' SELECT $1.*' ;
    EXECUTE strSQL USING NEW;
    RETURN NULL; 
END
$BODY$
  LANGUAGE plpgsql;

3. 为主表创建触发器

CREATE TRIGGER insert_tbl_partition
  BEFORE INSERT
  ON tbl_partition
  FOR EACH ROWid
  EXECUTE PROCEDURE auto_insert_into_tbl_partition('gather_time ','id');

通过partition of 的方式创建（推荐）

1. 创建分区函数

 CREATE OR REPLACE FUNCTION "table_partitions_add_func" (p_tablename text, p_schema text, p_date_start text, p_step integer)  RETURNS void
  VOLATILE
AS $body$
declare
v_cnt int;
v_schema_name varchar(500);
v_table_name varchar(500);
v_curr_limit varchar(50);
v_steps int;
v_exec_sql text;
v_partition_name text;
v_date_start text;
begin
    v_schema_name = p_schema;
    v_table_name = p_tablename;
    v_date_start = p_date_start;
    v_steps = p_step;
    v_exec_sql = ' ';
    for i in 0 .. v_steps loop
	  v_curr_limit = to_char(to_date(v_date_start,'yyyy-mm-dd') + i,'yyyy-mm-dd');
	  v_partition_name = to_char(to_date(v_date_start,'yyyy-mm-dd')  + i, 'yyyymmdd');
	  SELECT count(1) into v_cnt from pg_tables where schemaname=v_schema_name and tablename = v_table_name||'_'||v_partition_name;
	  if v_cnt = 0 then
	    v_exec_sql = v_exec_sql||' create table '||v_schema_name||'.'||v_table_name||'_'||v_partition_name||' partition of '||v_table_name||' for values FROM ('''||v_curr_limit||' 00:00:00'||''') TO ('''||v_curr_limit||' 23:59:59'||''');';
          end if;	
    end loop;
    execute v_exec_sql;
end;
$body$ LANGUAGE plpgsql

参数说明 ：
p_date_start : 开始日期
p_step ：执行一次创建的分区数