PostgreSQL 9.3 自定义聚合函数实现多行数据合并成

常见的一种需求,如下图(1):

这里写图片描述


目前需要按右边的factor_code进行分组,将左边的behavior_code这一列通过指定分隔符连接起来,比如通过<br /> 来连接,理想的效果应当是如下图(2)这样:

这里写图片描述


下面就来讨论一下实现方式,假如在别的数据库中来实现,例如MySQL或Oracle,好像没有特别方便和直接的方式,同样的在早期的Postgres中也是一件麻烦的事情,下面先看一下Postgres早期版本的解决方案~

postgres 8.x的解决方案

没错,同标题一样,在postgres 8.x的版本中有几种解决办法,可以通过内置的数组函数array_to_string,也可以通过自定义function结合自定义聚合函数来实现,下面先看一下第一种。

array_to_string

通过嵌套子查询的方式来完成,缺点是写法是略微麻烦,而且SQL层次不清晰,效率也不高,下面看一下SQL:

select array_to_string (array(select behavior_code from t_evaluation_behavior t2 where t1.factor_code = t2.factor_code), '<br />') as behavior_code, factor_code from t_evaluation_behavior t1 group by factor_code order by factor_code;

运行效果如图(2)一致,接下来再看一下第二种解决方案,通过自定义function结合自定义聚合函数来实现。

创建自定义聚合函数(CREATE AGGREGATE)

这种方式使用简便,就是第一次写起来略微麻烦一点。思路和上一种一致,同样是间接的利用了array_to_string函数,只不过是在一个自定义的function中去做了。关于自定义聚合函数,我们可以在官方文档中的CREATE ARRGREAGTE 章节中看到,在postgres 8.x和9.x中基本是一致的,仅有略微差别,如下图(3)所示:

这里写图片描述


由于我本地的postgres是9.3,所以在此就着重演示一下9.3的CREATE AGGREGATE(关于实现多行数据合并成一列的最简便方式不是这种,稍后在后面说)。
- name
要创建的聚集函数名(可以有模式修饰的)。
- base_type
本聚集函数要处理的基本数据类型。 对于不检查输入类型的聚集来说,这个参数可以声明为”ANY”。 (比如 count(*))。
- sfunc
用于处理源数据列里的每一个输入数据的状态转换函数名称。 它通常是一个双参数的函数,第一个参数的类型是 state_data_type 而第二个参数的类型是 input_data_type. 另外,对于一个不检查输入数据的聚集,该函数只接受一个类型为 state_data_type 的参数。 不管是哪种情况,此函数必须返回一个类型为 state_data_type的值。 这个函数接受当前状态值和当前输入数据条目,而返回下个状态值。
- state_data_type
聚集的状态值的数据类型。
- ffunc
在转换完所有输入域/字段后调用的最终处理函数。它计算聚集的结果。 此函数必须接受一个类型为 state_data_type 的参数。 聚集的输出数据类型被定义为此函数的返回类型。 如果没有声明 ffunc 则使用聚集结果的状态值作为聚集的结果,而输出类型为 state_data_type
- initial_condition
状态值的初始设置(值)。它必须是一个数据类型 state_data_type 可以接受的文本常量值。 如果没有声明,状态值初始为 NULL。
- sort_operator
用于 MIN 或者 MAX 类型的聚集的相关的排序操作符。 这个只是一个操作符名(可以有模式修饰)。 这个操作符假设接受和聚集一样的输入数据类型。

OK,看完了所有的参数介绍,我们现在实现自己的聚合函数。

准备sfunc

这是第一步,sfunc需要我们自定义一个function,根据官方文档的描述,sfunc是一个状态转换函数,下面看一下文档中的这一段话:

PostgreSQL创建一个类型为stype的临时变量。它保存这个聚集的当前内部状态。对于每个输入数据条目,都调用状态转换函数计算内部状态值的新数值。 在处理完所有数据后,调用一次最终处理函数以计算聚集的返回值。如果没有最终处理函数, 则将最后的状态值当做返回值。

OK,根据官方文档的描述需要两个参数,一个是internal-state,一个是next-data-values。下面是sfunc的代码:

CREATE FUNCTION "public"."NewProc"(aa _text, s text) RETURNS "pg_catalog"."_text" AS $BODY$ BEGIN RETURN array_append(aa, s); END; $BODY$ LANGUAGE 'plpgsql' VOLATILE COST 100 ; ALTER FUNCTION "public"."NewProc"(aa _text, s text) OWNER TO "postgres";

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/776c2f68fb2091b93162d387c45c159b.html