一个简单的pg/plsql循环示例

Question

我们正在重新设计postgres中的表模式.它曾经有两个表partner和advertiser名称的链接表,partner_advertiser因为我们假设合作伙伴和广告商之间有很多关系.进行更改,使得advertiser只有一个partner,因此partner将具有一个可能的关系advertiser.

如何在不丢失先前信息的情况下进行更改？必须使用链接器表数据来填充新架构设计的映射.这是我的初始代码:

BEGIN

FOR r IN SELECT partnerid, advertiserid from partner_advertiser
    LOOP
    NEXT r;
    UPDATE advertiser SET partnerid = r.partnerid WHERE id = r.advertiserid 
    END LOOP;

END

顺便说一下,我自己没有做过任何pg/plsql.所以,如果我有任何基本步骤,请给我一个提醒.

Answer 1

您可以使用:

UPDATE advertiser a SET partnerid = r.partnerid
FROM partner_advertiser r
WHERE a.id = r.advertiserid

通常,像这样的简单关系转换永远不需要循环.但是,如果你真的需要一个,请查看http://www.postgresql.org/docs/9.0/interactive/plpgsql-control-structures.html

一个额外的注意事项:如果advertiserid不是唯一的话,任何转换都会明显丢失数据,因此你应该首先运行类似的东西

SELECT count(*), advertiserid FROM partner_advertiser
GROUP BY advertiserid HAVING COUNT(*) > 1

如果返回任何行,则需要手动修复.

Answer 2

这也有效.使用简单的plpgsql代码段:

CREATE OR REPLACE FUNCTIOn migratePartnerAdvertiser() RETURNS int4 AS '
DECLARE r RECORD;

BEGIN
    FOR r IN SELECT * from partner_advertiser LOOP
            UPDATE advertiser SET partnerId = r.partnerId WHERE id = r.advertiserId; 
    END LOOP;
return 1;
END;
' LANGUAGE plpgsql;


SELECT migratePartnerAdvertiser() as output;