5

我正在使用 PostgreSQL 9.2.4。

postgres=# 选择版本（）；

                           version
-------------------------------------------------------------
 PostgreSQL 9.2.4, compiled by Visual C++ build 1600, 64-bit
(1 row)

sqlfiddle 链接

我的查询安全地执行插入。我需要的是我的函数应该返回除了 void 数据类型之外的东西。像 text("inserted into table") 或 integer(0-false,1-true) 之类的东西，对我验证它是否被插入很有用？

我需要一个函数的语法，它在插入完成时返回一个integer或一个。text用于验证目的。有没有办法解决这个问题？

4

3 回答 3

5

你可能需要什么

很可能您需要一个函数返回text，另一个函数返回，integer或者返回一个boolean表示成功的函数。所有这些都是微不足道的，我将向您推荐关于 SO 类似问题的优秀手册CREATE FUNCTION或代码示例。

你实际上问了什么

如何编写一个返回文本或整数值的函数？

...从某种意义上说，我们有一个返回类型是textor integer。不像建议的那样微不足道，但也不是不可能的。关键词是：多态类型。

建立在这张简单的桌子上：

CREATE TABLE tbl(
  tbl_id int,
  txt    text,
  nr     int
);

此函数根据输入类型返回整数或文本（或任何其他类型，如果您允许的话）。

CREATE FUNCTION f_insert_data(_id int, _data anyelement, OUT _result anyelement)
  RETURNS anyelement AS
$func$
BEGIN

CASE pg_typeof(_data) 
WHEN 'text'::regtype THEN
    INSERT INTO tbl(tbl_id, txt) VALUES(_id, _data)
    RETURNING txt
    INTO _result;

WHEN 'integer'::regtype THEN
    INSERT INTO tbl(tbl_id, nr) VALUES(_id, _data)
    RETURNING nr
    INTO _result;

ELSE
    RAISE EXCEPTION 'Unexpected data type: %', pg_typeof(_data)::text;
END CASE;

END
$func$
LANGUAGE plpgsql;

称呼：

SELECT f_insert_data(1, 'foo'::text);  -- explicit cast needed.
SELECT f_insert_data(1, 7);

简单案例

一个函数返回TRUE/FALSE表示是否插入了一行，只有一个不同类型的输入参数：

CREATE FUNCTION f_insert_data2(_id int, _data anyelement)
  RETURNS boolean AS
$func$
BEGIN

CASE pg_typeof(_data)
WHEN 'text'::regtype THEN
   INSERT INTO tbl(tbl_id, txt) VALUES(_id, _data);

WHEN 'integer'::regtype THEN
   INSERT INTO tbl(tbl_id, nr) VALUES(_id, _data);

ELSE
   RAISE EXCEPTION 'Unexpected data type: >>%<<', pg_typeof(_data)::text;
END CASE;

IF FOUND THEN RETURN TRUE;
ELSE RETURN FALSE;
END IF;

END
$func$
LANGUAGE plpgsql;

在大多数情况下，输入类型可以替换为text参数，该参数可以转换为任何其他类型。

于 2013-07-11T23:16:14.133 回答
3

听起来你正在通过创造一个更大的问题来解决一个问题。

你根本不需要这个功能。通过检查每个 DML 查询返回的受影响行数，在客户端执行此操作，或使用INSERT ... RETURNING.

你没有提到你的客户端语言，所以这里是如何在 Python 中使用psycopg2. 相同的方法适用于具有语法变化的其他语言。

#!/usr/bin/env python
import psycopg2

# Connect to the db
conn = psycopg2.connect("dbname=regress")
curs = conn.cursor()

# Set up the table to use
curs.execute("""
DROP TABLE IF EXISTS so17587735;

CREATE TABLE so17587735 (
    id serial primary key,
    blah text not null
);
""");

# Approach 1: Do the insert and check the rowcount:
curs.execute("""
INSERT INTO so17587735(blah) VALUES ('whatever');
""");
if curs.rowcount != 1:
    raise Exception("Argh, insert affected zero rows, wtf?")
print("Inserted {0} rows as expected".format(curs.rowcount))

# Approach 2: Use RETURNING
curs.execute("""
INSERT INTO so17587735(blah) VALUES ('bored') RETURNING id;
""");
returned_rows = curs.fetchall();
if len(returned_rows) != 1:
    raise Exception("Got unexpected row count {0} from INSERT".format(len(returned_rows)))

print("Inserted row id is {0}".format(returned_rows[0][0]))

在调用 PL/PgSQL 的情况下，您INSERT可以使用命令、变量或. 使用：GET DIAGNOSTICSFOUNDRETURN QUERY EXECUTE INSERT ... RETURNING ...GET DIAGNOSTICS

CREATE OR REPLACE FUNCTION blah() RETURNS void AS $$
DECLARE
    inserted_rows integer;
BEGIN
    INSERT INTO some_table VALUES ('whatever');
    GET DIAGNOSTICS inserted_rows = ROW_COUNT;
    IF inserted_rows <> 1 THEN
        RAISE EXCEPTION 'Failed to insert rows; expected 1 row, got %', inserted_rows;
    END IF;
END;
$$ LANGUAGE plpgsql VOLATILE;

或者如果您必须返回值并且出于某种原因必须使用 PL/PgSQL：

CREATE OR REPLACE FUNCTION blah() RETURNS SETOF integer AS $$
BEGIN
    RETURN QUERY EXECUTE INSERT INTO some_table VALUES ('whatever') RETURNING id;
END;
$$ LANGUAGE plpgsql VOLATILE;

（假设关键是id）

这与以下内容相同：

CREATE OR REPLACE FUNCTION blah() RETURNS SETOF integer AS $$
INSERT INTO some_table VALUES ('whatever') RETURNING id;
$$ LANGUAGE sql;

要不就

INSERT INTO some_table VALUES ('whatever') RETURNING id;

换句话说：为什么将它包装在一个函数中？这没有意义。只需使用RETURNING或使用客户端驱动程序的受影响行数检查客户端的行数INSERT。

于 2013-07-11T14:13:44.027 回答
0

一个函数只能返回一种类型。在您的情况下，您可以创建一个包含两个字段、一个整数和一个文本的复合类型，然后返回它。

于 2013-07-11T13:20:50.420 回答