PSQL 存储过程 - 返回多个表值 - SQL

在 PSQL 中，我们可以使用存储过程来实现一些复杂的数据库操作。有时候我们需要在一个存储过程中返回多个表值。这篇文章介绍如何在 PSQL 存储过程中返回多个表值。

准备工作

首先，我们需要创建一些测试数据。假设我们有两个表：users 和 orders。 users 表包含用户的信息， orders 表包含订单的信息。我们要编写一个存储过程，返回所有用户的信息以及他们的订单。

CREATE TABLE users (
  id SERIAL PRIMARY KEY,
  name VARCHAR(50),
  email VARCHAR(50),
  created_at TIMESTAMP NOT NULL DEFAULT now()
);

CREATE TABLE orders (
  id SERIAL PRIMARY KEY,
  user_id INTEGER NOT NULL REFERENCES users(id),
  amount NUMERIC(10,2) NOT NULL,
  created_at TIMESTAMP NOT NULL DEFAULT now()
);

INSERT INTO users (name, email) VALUES
  ('Alice', 'alice@example.com'),
  ('Bob', 'bob@example.com');

INSERT INTO orders (user_id, amount)
  SELECT id, (random() * 100)::numeric(10,2) FROM users;

编写存储过程

现在，我们可以编写一个存储过程，返回所有用户的信息以及他们的订单。

CREATE OR REPLACE FUNCTION get_users_and_orders()
  RETURNS TABLE (id INTEGER, name VARCHAR, email VARCHAR, order_id INTEGER, order_amount NUMERIC, order_created_at TIMESTAMP)
AS $$
BEGIN
  RETURN QUERY SELECT u.id, u.name, u.email, o.id, o.amount, o.created_at
               FROM users u
               INNER JOIN orders o
                 ON u.id = o.user_id;
END;
$$ LANGUAGE plpgsql;

这个存储过程返回一个 TABLE 类型的结果集，包含用户的 id、name、email 以及订单的 id、amount、created_at。我们使用 RETURN QUERY 语句来生成结果集。

查询结果

现在我们可以使用下列语句来查询结果：

SELECT * FROM get_users_and_orders();

返回结果将包括所有用户的信息以及他们的订单。

 id | name  |       email       | order_id | order_amount |  order_created_at  
----+-------+------------------+----------+--------------+---------------------
  1 | Alice | alice@example.com |        1 |        77.58 | 2021-11-03 16:06:18
  2 | Bob   | bob@example.com   |        2 |        74.37 | 2021-11-03 16:06:18
  1 | Alice | alice@example.com |        3 |        98.52 | 2021-11-03 16:06:18
  2 | Bob   | bob@example.com   |        4 |        91.67 | 2021-11-03 16:06:18
(4 rows)

结论

在 PSQL 中，我们可以编写存储过程来实现一些复杂的数据库操作。本篇文章介绍了如何在存储过程中返回多个表值，以及如何查询这些返回值。