16

这已经在这里和这里多次询问，但没有一个答案适合我的情况，因为我不想在 PL/PgSQL 函数中执行我的更新语句并使用GET DIAGNOSTICS integer_var = ROW_COUNT.

我必须在原始 SQL 中执行此操作。

例如，在 MS SQL SERVER 中，我们有 @@ROWCOUNT ，可以像下面这样使用：

UPDATE <target_table> 
SET Proprerty0 = Value0
WHERE <predicate>;
SELECT <computed_value_columns> 
 FROM <target> 
 WHERE @@ROWCOUNT > 0;

在数据库的一次往返中，我知道更新是否成功并返回计算值。

可以使用什么来代替 '@@ROWCOUNT' ？有人可以确认目前这实际上是不可能的吗？

提前致谢。

编辑 1：我确认我需要使用原始 SQL（我在原始描述中写了“原始 plpgsql”）。

为了使我的问题更清楚，请考虑更新语句仅影响一行并考虑乐观并发：

  1. 客户SELECT首先做了一个声明。

  2. 他构建了 UPDATE 并知道哪些数据库计算列将包含在 SELECT 子句中。除其他外，谓词包括每次更新行时计算的时间戳。

  3. 所以，如果我们返回了 1 行，那么一切正常。如果没有返回任何行，那么我们知道之前有更新，客户端可能需要在再次尝试更新子句之前刷新数据。这就是为什么我们需要在返回计算列之前知道有多少行受更新语句影响。如果更新失败，则不应返回任何行。

4

2 回答 2

23

您所描述的形式目前无法实现您想要的，但我认为您可以使用UPDATE ... RETURNING. 参见UPDATE ... RETURNING手册。

UPDATE <target_table> 
SET Proprerty0 = Value0
WHERE <predicate>
RETURNING Property0;

很难确定，因为您提供的示例是如此抽象以至于毫无意义。

您还可以使用 wCTE，它允许更复杂的情况：

WITH updated_rows AS (
    UPDATE <target_table> 
    SET Proprerty0 = Value0
    WHERE <predicate>
    RETURNING row_id, Property0
)
SELECT row_id, some_computed_value_from_property
FROM updated_rows;

请参阅公用表表达式（WITH查询）和depesz 关于 wCTE 的文章。

更新基于问题中的一些附加细节，这是一个使用的演示UPDATE ... RETURNING：

CREATE TABLE upret_demo(
  id serial primary key,
  somecol text not null,
  last_updated timestamptz
);

INSERT INTO upret_demo (somecol, last_updated) VALUES ('blah',current_timestamp);

UPDATE upret_demo
SET
  somecol = 'newvalue',
  last_updated = current_timestamp
WHERE last_updated = '2012-12-03 19:36:15.045159+08'    -- Change to your timestamp
RETURNING 
  somecol || '_computed' AS a,
  'totally_new_computed_column' AS b;

第一次运行时输出：

         a         |              b              
-------------------+-----------------------------
 newvalue_computed | totally_new_computed_column
(1 row)

再次运行时，它将不起作用并且不返回任何行。

如果您在结果集中有更复杂的计算要做，您可以使用 wCTE，这样您就可以 JOIN 更新的结果并执行其他复杂的事情。

WITH upd_row AS (
  UPDATE upret_demo SET 
    somecol = 'newvalue',
    last_updated = current_timestamp
  WHERE last_updated = '2012-12-03 19:36:15.045159+08'
  RETURNING id, somecol, last_updated
)
SELECT
  'row_'||id||'_'||somecol||', updated '||last_updated AS calc1,
 repeat('x',4) AS calc2
FROM upd_row;

换句话说：UPDATE ... RETURNING直接使用 来生成计算的行，或者在可写的 CTE 中用于更复杂的情况。

于 2012-11-30T15:36:28.977 回答
0

一般来说，这个问题的答案取决于所使用的驱动程序的类型。

如果应用程序使用 libpq，PQcmdTuples() 函数会执行所需的操作。libpq 之上的其他库需要在此函数之上有一些包装器。

对于 JDBC，Statement.executeUpdate() 方法似乎可以胜任。

ODBC 为类似目的提供了 SQLRowCount() 函数。

于 2021-02-13T17:47:34.113 回答