PostgreSql递归包含层级level

前端之家收集整理的这篇文章主要介绍了PostgreSql递归包含层级level前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

当在用oracle做递归时,往往会用到一个虚列level,表示当前的节点是第几层,现在Postgresql的递归语法中(with recursive)中不包含层级列,当然你安装了Postgresql的扩展也是可以的,就是安装cross这个脚本,具体在postgresql安装目录项的share/extension下的tablefunc.sql,脚本。

由于在我实际的项目中不允许安装哪些脚本,因此只能先自定义一个函数,具体代码如下:

首先,创建自定义类型

CREATE TYPE public.type_custom_recursive AS (
keyid TEXT,
branch TEXT,
parent_keyid TEXT,
level TEXT
);

再创建函数

CREATE OR REPLACE FUNCTION public.custom_recursive (
table_name varchar,
keyid text,
parent_keyid text,
keyid_value text,
level integer,
split_value varchar
)
RETURNS SETOF public.type_custom_recursive AS
$body$
declare
test_ type_custom_recursive;
sql_text VARCHAR(1000);


BEGIN
-- i_id as keyid,id as branch,level as level
if $5 = 0 THEN
sql_text := 'select i_id::text as keyid,id::text as branch,i_parent_id::text as parent_keyid,level::text as level from (
with recursive temp as(
select '||$2||' as i_id,'||$2||'::text as id,'||$3||' from '|| $1 ||' where '||$2||' = cast('||$4||
'as int) union all
select b.'||$2||',cast (a.id || '''||split_value||''' || b.'||$2||' as text) as id,b.'||$3||' from temp a,'||$1||' b
where a.'||$2||' = b.'||$3||')select i_id,id,f_char_in_string(id,'''||$6||''')+1 as level,i_parent_id from temp
) t order by i_id';

else
sql_text := 'select i_id::text as keyid,i_parent_id from temp
) t where level = '|| $5 ||' order by i_id';
end if;


for test_ in execute sql_text loop
return next test_;
end loop;
end;
$body$
LANGUAGE 'plpgsql'
;

说明:

table_name varchar,--需要递归的表名
keyid text,--id
parent_keyid text,--父id
keyid_value text,--id的值
level integer,--层级 0的话显示当前id下的所有节点
split_value varchar -- 分隔符比如"->"

注意这里的id默认都是为整型的,如果id是varchar型那么需要稍作修改的。

下面做个小例:

select * fromcustom_recursive('organization','i_id','i_parent_id','1','->');

输出

keyid,branch,parentkeyid,level

1 ,1,null,1

2,1->2,1,2

3,1->3,1,2

原文链接:https://www.f2er.com/postgresql/195522.html

猜你在找的Postgre SQL相关文章