如何从 PostgreSQL 函数正确返回 SELECT 查询结果？-mysql教程-PHP中文网

如何从 PostgreSQL 函数正确返回 SELECT 查询结果？

Susan Sarandon

发布： 2025-01-17 10:32:14

原创

120 人浏览过

How to Correctly Return SELECT Query Results from a PostgreSQL Function?

在 PostgreSQL 函数中返回 SELECT 查询结果

在 PostgreSQL 中，使用函数返回 SELECT 查询的结果需要理解合适的语法和返回类型。我们将检查一个提供的函数，并指导您如何正确检索查询结果。

问题中的函数 wordFrequency 目前缺少正确的返回语句。为了解决这个问题，我们将使用 RETURN QUERY 语法。

修正后的函数：

<code class="language-sql">CREATE OR REPLACE FUNCTION wordFrequency(_max_tokens integer)
RETURNS SETOF RECORD AS $$
BEGIN
  RETURN QUERY
  SELECT text, count(*), 100.0 / _max_tokens * count(*) AS ratio
  FROM (
    SELECT text
    FROM token
    WHERE chartype = 'ALPHABETIC'
    LIMIT _max_tokens
  ) AS tokens
  GROUP BY text
  ORDER BY count DESC;
END;
$$ LANGUAGE plpgsql;</code>

登录后复制

关键点：

RETURN QUERY 语法允许您直接返回 SELECT 查询的结果。
SETOF RECORD 指示函数将返回一组记录。
函数体中的输出参数（例如 text 和 count(*)）必须与查询中的列名匹配。我们添加了 AS ratio 为第三个列名提供一个明确的名称，避免歧义。同时将 100 改为 100.0，避免整数除法导致精度损失。

调用函数：

<code class="language-sql">SELECT * FROM wordFrequency(123);</code>

登录后复制

此外，使用 RETURNS TABLE 显式定义返回类型比返回泛型记录更实用，因为它消除了每次函数调用都需要指定列定义列表的需要。例如：

<code class="language-sql">CREATE OR REPLACE FUNCTION wordFrequency(_max_tokens integer)
RETURNS TABLE (text TEXT, cnt BIGINT, ratio NUMERIC) AS $$
BEGIN
  RETURN QUERY
  SELECT text, count(*), 100.0 / _max_tokens * count(*)
  FROM (
    SELECT text
    FROM token
    WHERE chartype = 'ALPHABETIC'
    LIMIT _max_tokens
  ) AS tokens
  GROUP BY text
  ORDER BY count DESC;
END;
$$ LANGUAGE plpgsql;</code>

登录后复制

重要注意事项：