强制立即返回新时间戳而不提交事务

Question

我知道，在 PostgreSQL 中，返回值NOW()是事务开始时间戳。因此，如果您NOW()在同一事务中多次使用，它将始终返回相同的值。

这在我的书中是很好的。

但是我在客户端应用程序中的单元测试方面遇到了一个小问题，并且希望能够告诉 PostgreSQL 如果可能的话（暂时）禁用它。

我不想（不能）使用另一个时间戳函数的原因是clock_timestamp()因为函数调用NOW()位于触发器内部，并且在生产代码中我想要“事务开始”行为。

但在我的单元测试中，我正在修补 API 级“提交”功能，这样我就不会在测试期间意外地将真实数据提交到数据库（不用担心，我在测试期间不使用生产数据库）。因此，在单元测试期间，commit永远不会访问数据库，因此我没有获得新的事务时间戳。

数据库使用时态表，如果时间戳发生更改，新条目仅会附加到历史表中，以确保每个事务仅合并历史表中的一个条目。

但是，当测试时态表行为时，这会导致历史表中不会出现任何条目。时态表触发器的关键片段是这样的：

new_validity_period = tstzrange(
    lower(OLD.validity_period),
    NOW(),
    '[)'
);
IF isempty(new_validity_period) THEN
    RAISE DEBUG 'New entry % will not introduce a new history item', OLD;
    RETURN OLD;
END IF;

因此，当我在单元测试中执行“插入”操作，且交易时间为“2020-01-01 01:02:03”时，该条目的有效期将为[2020-01-01 01:02:02,)。如果仍然在同一个单元测试中，我删除该条目（并测试它是否出现在历史表中），则该操作发生在同一个 TX 中，上面的代码将如下所示：

new_validity_perion = tstzrange(
    '2020-01-01 01:02:03', -- the lower-bound of the 'OLD' row
    '2020-01-01 01:02:03', -- the result of 'NOW()'
    '[)'
)
-- resulting in an empty range because the two timestamps are identical
IF isempty(new_validity_periond) THEN  -- <- Resulting to TRUE
    ...
    RETURN OLD;  -- returning here, not continuing to store the history entry
END IF;
-- code below here is skipped

有没有办法配置 PG在调用时始终返回 wall-time NOW()？测试是针对 docker 容器运行的，因此我可以完全控制服务器配置。但如果我可以通过 SQL 命令设置配置来更改行为，那就更好了，这样我只能更改时态表测试的行为。

如果这是不可能的，我将需要使用不同的数据库会话/事务设置来进行这些测试。这也很好，但我想知道我是否可以控制这种 PG 行为。

附录：单元测试事务隔离

这是我用来隔离用户代码中的提交调用的代码：

new_validity_period = tstzrange(
    lower(OLD.validity_period),
    NOW(),
    '[)'
);
IF isempty(new_validity_period) THEN
    RAISE DEBUG 'New entry % will not introduce a new history item', OLD;
    RETURN OLD;
END IF;

Answer 1

我认为这是值得怀疑的做法。测试与生产环境的任何不同都会增加测试错误的危险。

不管怎样，你不能使用now()，而是使用不同的函数，比如mytimestamp()。在您的测试系统上，该函数定义为

CREATE FUNCTION mytimestamp() RETURNS timestamp with time zone
   LANGUAGE sql VOLATILE AS
'SELECT clock_timestamp()';

在生产数据库中，您使用

CREATE FUNCTION mytimestamp() RETURNS timestamp with time zone
   LANGUAGE sql VOLATILE AS
'SELECT current_timestamp';

然后你就会得到你想要的。唯一的区别是current_timestamp（或now()，这是同一件事） is STABLE， not VOLATILE，这可能会使查询的行为不同（或者如果函数是内联的，则不会）。但这正是我一开始就警告过你的事情。