9

我在 PostgreSQL 8.4 中有一个存储过程,它根据作为参数传入的整数值调用另一个存储过程。调用这些存储过程以便它们应该返回与一个整数列的关系。我遇到的问题是外部存储过程总是返回具有正确行数但所有 id 为 NULL 的关系。

这是简化为最简单形式的存储过程:

CREATE OR REPLACE FUNCTION spa(count integer) 
RETURNS TABLE (id integer) AS $$
BEGIN
    RETURN QUERY SELECT generate_series(1, count);
END;
$$ LANGUAGE plpgsql;

CREATE OR REPLACE FUNCTION spb(count integer) 
RETURNS TABLE (id integer) AS $$
BEGIN
    RETURN QUERY SELECT generate_series(1, count);
END;
$$ LANGUAGE plpgsql;

CREATE OR REPLACE FUNCTION conditional_relation_return(objectType integer, count integer) 
RETURNS TABLE (id integer) AS $$
BEGIN
    IF objectType = 1 THEN
        RETURN QUERY SELECT id FROM spa(count);
    ELSIF objectType = 2 OR objectType = 3 THEN
        RETURN QUERY SELECT id FROM spb(count);
    END IF;

END;
$$ LANGUAGE plpgsql;

如果你称之为:

# select * from conditional_relation_return(1, 2);
 id 
----


(2 rows)

或者更具体地说:

# select count(*) from conditional_relation_return(1, 2) where id is null;
 count 
-------
     2
(1 row)

但是,如果您调用其中一个引用的存储过程,您会得到正确的结果:

# select * from spa(2);
 id 
----
  1
  2
(2 rows)

那么为什么 conditional_relation_return 返回所有的 NULL 呢?

4

1 回答 1

13

idspa 与 out 参数 id ( RETURNS TABLE (id integer))冲突。Postgresql 8.4 没有抱怨,它从参数 id 中选择 id 而不是更理智的一个(spa 的 id)。

Postgresql 9.1 对您的原始代码的抱怨:

ERROR:  column reference "id" is ambiguous
LINE 1: SELECT id FROM spa(count)
               ^
DETAIL:  It could refer to either a PL/pgSQL variable or a table column.
QUERY:  SELECT id FROM spa(count)
CONTEXT:  PL/pgSQL function "conditional_relation_return" line 4 at RETURN QUERY

要修复它,请完全限定查询中的 id:

CREATE OR REPLACE FUNCTION conditional_relation_return(
    objectType integer, count integer) 
RETURNS TABLE (id integer) AS $$
BEGIN
    IF objectType = 1 THEN
        RETURN QUERY SELECT x.id FROM spa(count) as x;
    ELSIF objectType = 2 OR objectType = 3 THEN
        RETURN QUERY SELECT x.id FROM spb(count) as x;
    END IF;

END;
$$ LANGUAGE plpgsql;

输出:

test=# select * from conditional_relation_return(1, 2);
 id 
----
  1
  2
(2 rows)

Postgresql 尊重您从RETURNS TABLE. 它仍然x.id插入id到您的RETURNS TABLE. 因此,即使您决定重命名RETURNS TABLE返回列的名称,它仍会x.id使用该名称,例如

CREATE OR REPLACE FUNCTION conditional_relation_return(
    objectType integer, count integer) 
RETURNS TABLE (hahah integer) AS $$
BEGIN
    IF objectType = 1 THEN
        RETURN QUERY SELECT x.id FROM spa(count) as x;
    ELSIF objectType = 2 OR objectType = 3 THEN
        RETURN QUERY SELECT x.id FROM spb(count) as x;
    END IF;

END;
$$ LANGUAGE plpgsql;

输出:

test=# select * from conditional_relation_return(1, 2);
 hahah 
-------
     1
     2
(2 rows)

注意hahah

于 2012-07-14T00:47:46.530 回答