pgsql 创建分区表

最新推荐文章于 2024-04-26 20:19:53 发布

IT界的小菜鸡

最新推荐文章于 2024-04-26 20:19:53 发布

阅读量628

点赞数

文章标签：数据库 java 服务器

本文链接：https://blog.csdn.net/csdnmuyi/article/details/130586951

版权

CREATE OR REPLACE FUNCTION "public"."auto_create_result_record_table_partition"()
  RETURNS "pg_catalog"."trigger" AS $BODY$
DECLARE
    time_column_name 	text ;			-- 父表中用于分区的时间字段的名称[必须首先初始化!!]
    curMM 		varchar(6);		-- 'YYYYMM'字串,用做分区子表的后缀
    cur_MM 		varchar(8);		-- 'YYYY-MM'字串,判断范围
    isExist 		boolean;		-- 分区子表,是否已存在
    strSQL  		text;
    
BEGIN
    -- 调用前,必须首先初始化(时间字段名):time_column_name [直接从调用参数中获取!!]
    time_column_name := TG_ARGV[0];
   
    -- 判断对应分区表 是否已经存在?
    EXECUTE 'SELECT $1.'||time_column_name INTO strSQL USING NEW;
	  cur_MM := ''||strSQL;
 
	
    curMM := replace(strSQL, '-', '');
    select count(*) INTO isExist from pg_class where relname = (TG_RELNAME||'_'||curMM);
 
    -- 若不存在, 则插入前需 先创建子分区
    IF ( isExist = false ) THEN  
        -- 创建子分区表
				   
        strSQL := 'CREATE TABLE IF NOT EXISTS '||TG_RELNAME||'_'||curMM|| 
                 ' ( CHECK('||time_column_name||'='''|| cur_MM||'''))' ||'INHERITS ('||TG_RELNAME||');'  ;   
									
        EXECUTE strSQL;

			 	strSQL := 'ALTER TABLE ' ||TG_RELNAME||'_'||curMM|| ' ADD PRIMARY KEY( id )';
				
				EXECUTE strSQL;
				
				strSQL := 'CREATE INDEX  ' ||TG_RELNAME||'_'||curMM|| '_create_time'||' ON ' ||TG_RELNAME||'_'||curMM|| ' (create_time)';
			
				EXECUTE strSQL;
    END IF;
 
    -- 插入数据到子分区!
    strSQL := 'INSERT INTO '||TG_RELNAME||'_'||curMM||' SELECT $1.*' ;
    EXECUTE strSQL USING NEW;

    RETURN NULL; 
END
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100