PostgreSQL自定义函数处理特殊需求

来源：互联网发布：全境封锁网络优化编辑：程序博客网时间：2024/06/03 19:48

工作中，在某些特殊需求下，基本的SQL语句已经不能满足业务需求，需要使用函数来处理。

1. 需要新建sequence。由于部分表已经有数据，这时，新建的sequence的起始位置不能从1开始。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
CREATEOR REPLACE FUNCTION create_sequence(sequence_name charactervarying, table_name charactervarying, column_name charactervarying)
  --sequence_name 是需要新建的sequence名称
  --table_name 是sequence作用的表
  --column_name 是统计当前表的数据量的列值
  RETURNSbigint AS
$BODY$
    DECLARE
    startNum int8;
    createsequence"varchar":= '';
    selectsequence"varchar":= '';
 
    BEGIN
        --先获取当面表中已经保存了多少数据
        selectsequence := 'select COALESCE(max('||column_name||'), 0) + 1 FROM '|| table_name ||'';
        --将查询出来的结果放入startNum中
    executeselectsequence intostartNum;
        --声明 创建 sequence的语句。
        createsequence := 'CREATE SEQUENCE '||sequence_name||'
     INCREMENT 1
     MINVALUE 1
     MAXVALUE 9223372036854775807
     START '|| startNum ||'
     CACHE 1';
    --执行创建sequence的SQL语句
    executecreatesequence;
      
        RETURNstartNum;
    END;
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100;
ALTERFUNCTION create_sequence(charactervarying,charactervarying,charactervarying) OWNER TOpostgres;

2. 需求，需要将这两行数据每列相加。

初始数据：

结果为<喎�"/kf/ware/vc/" target="_blank" class="keylink">vcD4KPHA+PGltZyBzcmM9"/uploadfile/Collfiles/20141219/2014121909224898.jpg" alt="\">

2.1 先将将string转换为数组。如1,2,3,4,5转换为 [1,2,3,4,5]

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
CREATEOR REPLACE FUNCTION tools_str2array(_originstr text)
  RETURNSinteger[]AS
$BODY$
declare_cindex INTEGER;
declare_arrIndex INTEGER;
DECLARE_arr_str int4[];
DECLARE_tmp_str VARCHAR(10);
DECLARE_delimeter VARCHAR(1);
BEGIN
_arrIndex:=1;
_cindex:=1;
_delimeter:=',';
 
--_strres:=_strres||'原始字符串是：'||_originStr;
 
while _cindex<"length"(_originStr) loop
--_strres :=_strres||'【这个是什么？】'||split_part(_originStr, _delimeter, _arrIndex);
_tmp_str:=split_part(_originStr, _delimeter, _arrIndex);
if"character_length"(_tmp_str)<1then
exit;
endif;
_arr_str:=_arr_str||CAST(_tmp_strasint4);
_arrIndex:=_arrIndex+1;
ENDloop;
return_arr_str;
end;
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100;
ALTERFUNCTION tools_str2array(text) OWNER TOpostgres;

2.2. 新增聚集函数。

?
1
2
3
4
5
6
CREATEAGGREGATE hot_map_test(
BASETYPE = TEXT,--输入值
SFUNC = csk_test_start,--最初执行的函数
STYPE = TEXT[],--输出类型
FINALFUNC = count_hot_map--最后执行的函数
);

2.3 新建计算函数计算结果

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
CREATEAGGREGATE count_test(
BASETYPE = TEXT,
SFUNC = csk_test_start,
STYPE = TEXT[],
FINALFUNC = count_hot_map
);
 
CREATEOR REPLACE FUNCTION csk_test_start(a text[], s text)
  RETURNStext[] AS
$BODY$
BEGIN
RETURNa || s;
END;
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100;
ALTERFUNCTION csk_test_start(text[], text) OWNER TOpostgres;
 
CREATEOR REPLACE FUNCTION count_hot_map(text[])
  RETURNSinteger[]AS
$BODY$
    DECLARE
    startNum int8;
    spilt_sql"varchar":= '';
    selectsequence"varchar":= '';
    temp_array int4[];
    result_array int4[];
 
    BEGIN
        forindex in 1..array_length($1 , 1) loop
            spilt_sql = 'select tools_str2Array('''||$1[index]||''')';
            executespilt_sql intotemp_array;
             
            forarray_items_index in1..array_length(temp_array, 1) loop
                if result_array[array_items_index] isnull then
                   result_array[array_items_index] = 0;
                endif;  
    
                result_array[array_items_index] = result_array[array_items_index] + temp_array[array_items_index];
            endloop;
        endloop;
    RETURNresult_array;
    END;
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100;
ALTERFUNCTION count_hot_map(text[]) OWNER TOpostgres;

0 0