1

当尝试将JSONB_ARRAY_ELEMENTS调用的结果分配给VARCHAR 数组（供以后在SELECT ... WHERE IN (UNNEST(...))语句中使用）时，以下存储函数：

CREATE OR REPLACE FUNCTION test1(
        IN in_sids jsonb,
        OUT out_uid integer)
        RETURNS integer AS
$func$
DECLARE
        sids varchar[];
        uids integer[];
BEGIN
        sids := (SELECT x->>'sid' FROM JSONB_ARRAY_ELEMENTS(in_sids) x);
        -- uids := (SELECT uid FROM social WHERE sid IN (UNNEST(sids)));

        RAISE NOTICE 'sids = %', sids;
        RAISE NOTICE 'uids = %', uids;

        SELECT 1;
END
$func$ LANGUAGE plpgsql;

不幸的是打印错误：

# select test1('[{"sid":"aaa"},{"sid":"bbb"}]'::jsonb);

ERROR:  more than one row returned by a subquery used as an expression
CONTEXT:  SQL statement "SELECT (SELECT x->>'sid' FROM JSONB_ARRAY_ELEMENTS(in_sids) x)"
PL/pgSQL function test1(jsonb) line 6 at assignment

所以我试图通过使用ARRAY_AGG来修复分配-

sids := SELECT ARRAY_AGG(SELECT x->>'sid' FROM JSONB_ARRAY_ELEMENTS(in_sids) x);

但得到语法错误：

ERROR:  syntax error at or near "SELECT"
LINE 10:         sids := SELECT ARRAY_AGG(SELECT x->>'sid' FROM JSONB...
                         ^

请问如何将JSONB_ARRAY_ELEMENTS结果存储到数组中？

更新：

我听从了尼克的建议（谢谢），但现在卡在下一步：

CREATE OR REPLACE FUNCTION test1(
        IN in_sids jsonb,
        OUT out_uid integer)
        RETURNS integer AS
$func$
DECLARE
        sids varchar[];
        uids integer[];
BEGIN
        sids := (SELECT ARRAY_AGG(x->>'sid') FROM JSONB_ARRAY_ELEMENTS(in_sids) x);
        uids := (SELECT uid FROM social WHERE sid IN (UNNEST(sids)));

        SELECT 1;
END
$func$ LANGUAGE plpgsql;

尝试在SELECT ... WHERE IN ...语句中使用数组时：

# select test1('[{"sid":"aaa"},{"sid":"bbb"}]'::jsonb);
ERROR:  argument of IN must not return a set
LINE 1: SELECT (SELECT uid FROM social WHERE sid IN (UNNEST(si...
                                              ^
QUERY:  SELECT (SELECT uid FROM social WHERE sid IN (UNNEST(sids)))
CONTEXT:  PL/pgSQL function test1(jsonb) line 7 at assignment
4

1 回答 1

1

你很亲近...

sids := (SELECT ARRAY_AGG(x->>'sid') FROM JSONB_ARRAY_ELEMENTS(in_sids) x);

为了使用该expression IN (subquery)构造，您需要编写一个自包含的SELECT语句：

uids := (SELECT ARRAY_AGG(uid) FROM social WHERE sid IN (SELECT UNNEST(sids)));

或者，您可以使用以下方法检查值是否在数组中ANY：

uids := (SELECT ARRAY_AGG(uid) FROM social WHERE sid = ANY(sids));

如果您不使用该sids数组进行稍后的计算，您可以将它们组合成一个查询：

uids := (
  SELECT array_agg(uid)
  FROM social
  JOIN JSONB_ARRAY_ELEMENTS(in_sids) x ON
    sid = x->>'sid'
);
于 2016-02-29T12:13:40.083 回答