CREATE OR REPLACE FUNCTION auto_insert_into_tbl_partition()
RETURNS trigger AS
$BODY$
DECLARE
time_column_name text ; -- 时间字段,用时间判断日期以此分表,属于必填的
curMMDD varchar(16); -- 'YYYYMMDD'格式的字串,用做分区子表的后缀
isExist boolean; -- 判断分区子表是否已存在
strSQL text;
BEGIN
-- 从调用参数中获取time_column_name
time_column_name := TG_ARGV[0];
-- 判断对应分区表是否已经存在
EXECUTE 'SELECT $1.'||time_column_name INTO strSQL USING NEW;
curMMDD := to_char(strSQL::timestamp,'YYYYMMDD');
select count(1) INTO isExist from pg_class where relname = ('t_audit_'||curMMDD);
-- 若不存在先创建子分区再插入
IF (isExist = false) THEN
-- 创建子分区表继承父表
strSQL := 'CREATE TABLE IF NOT EXISTS t_audit_'||curMMDD||'() INHERITS (t_audit);';
EXECUTE strSQL;
-- 创建索引
strSQL := 'CREATE INDEX t_audit_'||curMMDD||'_INDEX_'||time_column_name||' ON t_audit_'||curMMDD||' ('||time_column_name||');' ;
EXECUTE strSQL;
END IF;
-- 插入数据到子分区!
strSQL := 'INSERT INTO t_audit_'||curMMDD||' SELECT $1.*';
EXECUTE strSQL USING NEW;
RETURN NULL;
END
$BODY$
LANGUAGE plpgsql;
-- 创建触发器,insert操作时触发
CREATE TRIGGER insert_tbl_partition_trigger
BEFORE INSERT
ON t_audit
FOR EACH ROW
EXECUTE PROCEDURE auto_insert_into_tbl_partition('time');
整体就是这段代码,就可以完成一个简易的按天分表的操作。