在函数中按列类型过滤行

时间:2017-05-19 21:21:41

标签: postgresql plpgsql

现在我有一个通用的通知函数,它在我的数据库中的几个表上创建后触发(另一端的节点进程正在监听通知)。这是我的更新触发器的样子:

CREATE OR REPLACE FUNCTION notify_create() RETURNS trigger
  LANGUAGE plpgsql
  AS $$
BEGIN
  PERFORM pg_notify('update_watchers',
    json_build_object(
      'eventType', 'new',
      'type', TG_TABLE_NAME,
      'payload', row_to_json(NEW)
    )::text
  );
  RETURN NEW;
END;
$$;

问题是,如果NEW太大,这将在几个有限的极端情况下超出8000字节的限制(我很少在表中有一个新项目那么大)。在notify_update函数中,我只是通过列出列名来报告哪些列已更改。这可以在这里工作,但我而不是做的只是让row_to_json从NEW中提取整数类型的条目。

这是因为有时候我通知的是“嘿,实体表中有一个新条目”。新条目可以来自几个不同的表(文档,配置文件等)。在这种情况下,我真的只需要id,因为任何对新值感兴趣的人最终都会在以后获取它。

有时我会通知“嘿,在连接表中有一个新条目”,在这种情况下,我没有id字段,而是有类似documents_id和profiles_id的东西。

可以为每个场景编写一堆不同的notify_create函数。我更喜欢有一个像

这样的东西
row_to_json(NEW.filter(t => typeof t === 'number'))

将plpgsql和javascript表示法混合在一起,但我确信你明白了这一点:只包括那些数字类型的NEW字段

这可能,或者我应该写一堆不同的通知符吗?

1 个答案:

答案 0 :(得分:2)

您可以轻松地消除number以外类型的json对象,例如:

with my_table(int1, text1, int2, date1, float1) as (
values
    (1, 'text1', 100, '2017-01-01'::date, 123.54)
)

select jsonb_object_agg(key, value) filter (where jsonb_typeof(value) = 'number')
from my_table,
jsonb_each(to_jsonb(my_table))


              jsonb_object_agg              
--------------------------------------------
 {"int1": 1, "int2": 100, "float1": 123.54}
(1 row)

下面的函数只留下整数:

create or replace function leave_integers(jdata jsonb)
returns jsonb language sql as $$
    select jsonb_object_agg(key, value) 
    filter (
        where jsonb_typeof(value) = 'number'
        and value::text not like '%.%')
    from jsonb_each(jdata)
$$;

with my_table(int1, text1, int2, date1, float1) as (
values
    (1, 'text1', 100, '2017-01-01'::date, 123.54)
)

select leave_integers(to_jsonb(my_table))
from my_table;

      leave_integers      
--------------------------
 {"int1": 1, "int2": 100}
(1 row) 

替代(更好)解决方案

此函数直接检查Postgres类型,并严格从整数列返回值。

create or replace function integer_columns_to_jsonb(anyelement)
returns jsonb language sql as $$
    select jsonb_object_agg(key, value) 
    from jsonb_each(to_jsonb($1))
    where key in (
        select attname
        from pg_type t
        join pg_attribute on typrelid = attrelid
        where t.oid = pg_typeof($1)
        and atttypid = 'int'::regtype)
$$;

该示例显示该函数消除了leave_integers()错误处理的一些极端情况:

create table my_table (int1 int, int2 int, float1 float, text1 text);
insert into my_table values (1, 2, 3, '4');

select integer_columns_to_jsonb(t), leave_integers(to_jsonb(t))
from my_table t;

 integer_columns_to_jsonb |           leave_integers            
--------------------------+-------------------------------------
 {"int1": 1, "int2": 2}   | {"int1": 1, "int2": 2, "float1": 3}
(1 row)