0

我有一个基于 Web 的系统,它有几个表(postgres/pgsql),它们包含多对多的关系,例如;表 x column_id1 smallint FK column_id2 smallint FK

在这种情况下,更新是基于 column_id2

首先要更新这些记录,我们将运行以下函数;

-- edited to protect the innocent
CREATE FUNCTION associate_id1_with_id2(integer[], integer) RETURNS integer
AS $_$
DECLARE
    a alias for $1;
    b alias for $2;
    i integer;
BEGIN
    delete from tablex where user_id = b;
    FOR i IN array_lower(a,1) .. array_upper(a,1) LOOP
        INSERT INTO tablex (
            column_id2,
            column_id1)
        VALUES (
            b,
            a[i]);
    end loop;
RETURN i;
END;
$_$ 
LANGUAGE plpgsql;

这看起来很草率,现在加上审计,它真的显示出来了。

我现在要做的只是删除并插入必要的行。

我一直在尝试以下各种形式但没有运气

CREATE OR REPLACE FUNCTION associate_id1_with_id2(integer[], integer) RETURNS integer
AS $_$
DECLARE
    a alias for $1;
    b alias for $2;
c varchar;
    i integer;
BEGIN
    c = array_to_string($1,',');

    INSERT INTO tablex (
            column_id2,
            column_id1)
            (
    SELECT column_id2, column_id1
    FROM tablex
            WHERE column_id2 = b
    AND column_id1 NOT IN (c)
    );

    DELETE FROM tablex
    WHERE column_id2 = b
    AND column_id1 NOT IN (c);

RETURN i;
END;
$_$
LANGUAGE plpgsql;

根据我正在尝试的函数版本,当前版本存在各种错误,例如显式类型转换(我猜它不喜欢 c 是 varchar?)。

首先,我的方法是正确的还是有更优雅的解决方案,因为有几个表需要这种类型的处理?如果不能,请您指出正确的方向吗?

如果这是正确的方法,您能否协助对 where 子句的 NOT IN 部分进行数组转换?

4

1 回答 1

1

而不是array_to_string,使用unnest将数组转换为一组行(就像它是一个表一样),并且可以使用 vanilla SQL 解决该问题:

INSERT INTO tablex(column_id1,column_id2)
 select ai,b from unnest(a) as ai where not exists
  (select 1 from tablex where column_id1=ai and column_id2=b);

DELETE FROM tablex
where column_id2=b and column_id1 not in
  (select ai from unnest(a) as ai);
于 2012-06-12T00:36:18.590 回答