如何在PostgreSQL中获取聚合的定义/源代码?

时间:2015-01-12 10:16:57

标签: sql postgresql aggregate-functions psql ddl

我发现这个相关的答案很有用:

但是如何在没有GUI客户端的情况下获取CREATE AGGREGATE语句(例如使用psql命令行)?

3 个答案:

答案 0 :(得分:3)

这样的东西,但我不确定这是否涵盖了创建聚合的所有可能方式(它绝对不需要考虑引用的标识符)

SELECT 'create aggregate '||n.nspname||'.'||p.proname||'('||format_type(a.aggtranstype, null)||') (sfunc = '||a.aggtransfn
       ||', stype = '||format_type(a.aggtranstype, null)
       ||case when op.oprname is null then '' else ', sortop = '||op.oprname end 
       ||case when a.agginitval is null then '' else ', initcond = '||a.agginitval end
       ||')' as source
FROM pg_proc p 
  JOIN pg_namespace n ON p.pronamespace = n.oid 
  JOIN pg_aggregate a ON a.aggfnoid = p.oid 
  LEFT JOIN pg_operator op ON op.oid = a.aggsortop 
where p.proname = 'your_aggregate'
  and n.nspname = 'public' --- replace with your schema name  

答案 1 :(得分:1)

生成CREATE AGGREGATE语句的现代版本 - 使用format()并强制转换为对象标识符类型,使其变得简单,并在需要时自动添加双引号和模式限定:

SELECT format('CREATE AGGREGATE %s (SFUNC = %s, STYPE = %s%s%s%s%s)'
            , aggfnoid::regprocedure
            , aggtransfn
            , aggtranstype::regtype
            , ', SORTOP = '    || NULLIF(aggsortop, 0)::regoper
            , ', INITCOND = '  || agginitval
            , ', FINALFUNC = ' || NULLIF(aggfinalfn, 0)
            , CASE WHEN aggfinalextra THEN ', FINALFUNC_EXTRA' END
            --  add more to cover special cases like moving-aggregate etc.
              ) AS ddl_agg
FROM   pg_aggregate
WHERE  aggfnoid = 'my_agg_func'::regproc;  -- name of agg func here

您可以根据需要对汇总功能的名称进行架构限定:'public.my_agg_func'::regproc 和/或添加函数参数以在重载聚合函数的情况下消除歧义:'array_agg(anyarray)'::regprocedure

这不包括移动聚合功能等特殊情况。可以轻松扩展以涵盖当前Postgres版本的所有选项。但是下一个主要版本可能会带来新的选择。

类似于现有pg_get_functiondef()的函数 pg_get_aggregatedef() 可以很好地消除对此自定义查询的需求 - 可能必须适应每个新的主要Postgres版本...

答案 2 :(得分:0)

我的版本使用了一些系统功能

SELECT 
format(
   E'CREATE AGGREGATE %s (\n%s\n);'
   , (pg_identify_object('pg_proc'::regclass, aggfnoid, 0)).identity
   , array_to_string(
      ARRAY[
     format(E'\tSFUNC = %s', aggtransfn::regproc)
     , format(E'\tSTYPE = %s', format_type(aggtranstype, NULL))
     , CASE aggfinalfn WHEN '-'::regproc THEN NULL ELSE format(E'\tFINALFUNC = %s',aggfinalfn::text) END
     , CASE aggsortop WHEN 0 THEN NULL ELSE format(E'\tSORTOP = %s', oprname) END
     , CASE WHEN agginitval IS NULL THEN NULL ELSE format(E'\tINITCOND = %s', agginitval) END
      ]
      , E',\n'
   )
)

FROM pg_aggregate
LEFT JOIN pg_operator ON pg_operator.oid = aggsortop
WHERE aggfnoid = 'regr_r2'::regproc;