For循环中的Postgresql更新

时间:2014-08-26 13:48:58

标签: postgresql for-loop sql-update

我对postgresql足够新,我在使用for循环更新表中的空值列时遇到问题。我正在努力的表是巨大的,所以为了简洁,我将给出一个更小的例子,它应该得到重点。请参考下表

+----+----------+----------+
| id  |  A  | B  |  C       |
+----+----------+----------+
| a   | 1   | 0  |  NULL    |
| b   | 1   | 1  |  NULL    |
| c   | 2   | 4  |  NULL    |
| a   | 3   | 2  |  NULL    |
| c   | 2   | 3  |  NULL    |
| d   | 4   | 2  |  NULL    |
+----+----------+----------+

我想编写一个for循环,遍历所有行并执行一些操作 在a和b列中的值,然后在c中插入一个新值。 例如,在id = a的情况下,更新表集合C = A * B,或者id = d设置C = A + B等。这样就会给我一个像

这样的表
+----+----------+----------+
| id  |  A  | B  |  C       |
+----+----------+----------+
| a   | 1   | 0  |  0       |
| b   | 1   | 1  |  NULL    |
| c   | 2   | 4  |  NULL    |
| a   | 3   | 2  |  6       |
| c   | 2   | 3  |  NULL    |
| d   | 4   | 2  |  6       |
+----+----------+----------+ 

所以最终我想循环遍历表的所有行并根据“id”列中的值更新列C.我写的函数(没有给出任何错误但也没有更新任何内容)看起来像这样......

-- DROP FUNCTION some_function();

CREATE OR REPLACE FUNCTION some_function()
RETURNS void AS
$BODY$
DECLARE
--r integer; not too sure if this needs to be declared or not
result int;

BEGIN

FOR r IN select * from 'table_name' 
LOOP
select(
case
when id = 'a'  THEN B*C
when id = 'd'  THEN B+C
end) 
into result;

update table set C = result 
WHERE id = '';
END LOOP;
RETURN;

END
$BODY$
LANGUAGE plpgsql

我确定我有一些愚蠢的错过,可能在我身边,回归......在这种情况下无效。但是,由于我只想更新现有的行,我是否需要返回任何内容?这可能比使用循环更简单的方法,但我想让它使用这种方法。 如果有人能指出我正确的方向或指出任何明显我做错的事情,我会非常感激。 提前致谢。

1 个答案:

答案 0 :(得分:7)

不需要循环或函数,可以使用单个update语句完成:

update table_name
  set c = case 
            when id = 'a' then a*b
            when id = 'd' then a+b
            else c -- don't change anything
          end;

SQLFiddle:http://sqlfiddle.com/#!15/b65cb/2

你的功能没有做任何事情的原因是:

update table set C = result 
WHERE id = '';

id列中没有包含空字符串的行。您的函数似乎也使用了错误的公式:when id = 'a' THEN B*C我想这应该是:then a*b。最初CNULLb*c也会产生空值。所以即使如果你的循环中的更新会找到一行,它也会将其更新为NULL

您还从光标中错误地检索值。

如果你真的,真的想在循环中效率低下,你的函数应该看起来像这样(未经测试!):

CREATE OR REPLACE FUNCTION some_function()
  RETURNS void AS
$BODY$
DECLARE
  result int;
BEGIN
  -- r is a structure that contains an element for each column in the select list
  FOR r IN select * from table_name
  LOOP
    if r.id = 'a' then 
      result := r.a * r.b;
    end if;
    if r.id = 'b' then 
      result := r.a + r.b;
    end if;

    update table 
      set C = result 
    WHERE id = r.id; -- note the where condition that uses the value from the record variable
  END LOOP;
END
$BODY$
LANGUAGE plpgsql

但是又一次:如果你的桌子是"巨大的"正如你所说,循环是一个非常糟糕的解决方案。关系数据库用于处理"集"数据的。逐行处理是一种反模式,几乎总是会有糟糕的性能。

或者反过来说:做基于集合的操作(比如我的单update示例)总是更好的选择。