postgresql 按小時(shí)分表（含觸發(fā)器）的實(shí)現(xiàn)方式

2020-06-13 15:01:06 來源：互聯(lián)網(wǎng)作者：心機(jī)雅 人氣： 次閱讀 16 條評(píng)論

文章主要介紹了postgresql 按小時(shí)分表（含觸發(fā)器）的實(shí)現(xiàn)方式，本文給大家介紹的非常詳細(xì)，需要的朋友可以參考下...

本人后端開發(fā)，因?yàn)闃I(yè)務(wù)需求需要使用分表方式進(jìn)行數(shù)據(jù)存儲(chǔ)。結(jié)合網(wǎng)上的資料最后使用的以下方式：

CREATE OR REPLACE FUNCTION auto_insert_into_tbl_partition()
 RETURNS trigger AS
$BODY$
DECLARE
  time_column_name   text ;      -- 父表中用于分區(qū)的時(shí)間字段的名稱[必須首先初始化!!]
  curMM     varchar(16);    -- 'YYYYMM'字串,用做分區(qū)子表的后綴
  isExist     boolean;    -- 分區(qū)子表,是否已存在
  strSQL     text;
   
BEGIN
  -- 調(diào)用前,必須首先初始化(時(shí)間字段名):time_column_name [直接從調(diào)用參數(shù)中獲取!!]
  time_column_name := TG_ARGV[0];
   
  -- 判斷對(duì)應(yīng)分區(qū)表 是否已經(jīng)存在?
  EXECUTE 'SELECT $1.'||time_column_name INTO strSQL USING NEW;
  curMM := to_char( strSQL::timestamp , 'YYYYMMDDHH' );
  select count(1) INTO isExist from pg_class where relname = ('t_audit_'||curMM);
  
  -- 若不存在, 則插入前需 先創(chuàng)建子分區(qū)
  IF ( isExist = false ) THEN 
    -- 創(chuàng)建子分區(qū)表
    strSQL := 'CREATE TABLE IF NOT EXISTS t_audit_'||curMM||'() INHERITS (t_audit);' ; 
    EXECUTE strSQL;
    -- 創(chuàng)建索引
    strSQL := 'CREATE INDEX t_audit_'||curMM||'_INDEX_'||time_column_name||' ON t_audit_'||curMM||' ('||time_column_name||');' ;
    EXECUTE strSQL;
  END IF;
  
  -- 插入數(shù)據(jù)到子分區(qū)!
  strSQL := 'INSERT INTO t_audit_'||curMM||' SELECT $1.*' ;
  EXECUTE strSQL USING NEW;
  RETURN NULL; 
END
$BODY$
 LANGUAGE plpgsql;

我是按照小時(shí)對(duì)固定的一張表進(jìn)行分表的，其實(shí)可以寫活就是加個(gè)變量，拼sql的時(shí)候拼接進(jìn)去。這個(gè)就是寫個(gè)函數(shù)作為觸發(fā)器的回調(diào)函數(shù)。就先判斷一下這個(gè)表有沒有，有就直接插，沒有就建表再插。

CREATE TRIGGER insert_tbl_partition_trigger
 BEFORE INSERT
 ON t_audit
 FOR EACH ROW
 EXECUTE PROCEDURE auto_insert_into_tbl_partition('time');

建一個(gè)觸發(fā)器，在父表的插入數(shù)據(jù)的時(shí)候執(zhí)行前面的那個(gè)回調(diào)函數(shù)。

ps：下面看下postgresql 表觸發(fā)器

1、先建一個(gè)函數(shù)，用來執(zhí)行觸發(fā)器啟動(dòng)后要執(zhí)行的腳本

CREATE OR REPLACE FUNCTION "public"."trigger_day_aqi"()
 RETURNS "pg_catalog"."trigger" AS $BODY$
BEGIN
  --日均值表，沒有o3,小時(shí)值表，沒有o3_8h
   NEW.so2iaqi=DAY_SO2_AQI(NEW.so2);
     NEW.no2iaqi=DAY_NO2_AQI(NEW.no2);
     NEW.coiaqi=DAY_CO_AQI(NEW.co);
     NEW.o3_8hiaqi=O3_8_AQI(NEW.o3_8h);
     NEW.pm10iaqi=PM10_AQI(NEW.pm10);
     NEW.pm25iaqi=PM25_AQI(NEW.pm25);
     NEW.aqi=NEW.coiaqi;
     NEW.primarypol='CO';
    IF NEW.aqi<NEW.no2iaqi  then  NEW.aqi=NEW.no2iaqi;  NEW.primarypol='NO2';  end if;
    IF NEW.aqi<NEW.so2iaqi  then  NEW.aqi=NEW.so2iaqi;  NEW.primarypol='SO2';  end if;
    IF NEW.aqi<NEW.o3_8hiaqi then  NEW.aqi=NEW.o3_8hiaqi;  NEW.primarypol='O3_8H'; end if;
    IF NEW.aqi<NEW.pm10iaqi then  NEW.aqi=NEW.pm10iaqi;  NEW.primarypol='PM10';  end if;
    IF NEW.aqi<NEW.pm25iaqi then  NEW.aqi=NEW.pm25iaqi;  NEW.primarypol='PM2.5'; end if; 
    IF NEW.aqi<=50      then  NEW.primarypol='-';   end if;
     NEW.aqilevel=getRank(NEW.aqi);
  RETURN NEW;
END;
$BODY$
 LANGUAGE plpgsql VOLATILE
 COST 100

2、建表的觸發(fā)器，

CREATE TRIGGER gk_site_day_insert BEFORE INSERT ON gk_site_day
  FOR EACH ROW EXECUTE PROCEDURE trigger_day_aqi();

總結(jié)

以上所述是小編給大家介紹的postgresql 按小時(shí)分表（含觸發(fā)器）的實(shí)現(xiàn)方式，希望對(duì)大家有所幫助。