主页 / dba / 问题 / 291855
Accepted
exhuma
Asked: 2021-05-20 05:14:26 +0800 CST

现在强制返回新的时间戳而不提交事务

  • 772

我知道，在 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 配置为在调用时始终返回挂墙时间NOW()？测试是针对 docker-container 运行的，因此我可以完全控制服务器配置。但是，如果我可以通过 SQL 命令设置配置来更改行为会更好，所以我只能更改临时表测试的行为。

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

附录：单元测试事务隔离

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

@fixture
def rb_session():
    """
    Returns a session which will always be rolled back.
    """

    engine = create_engine(Configurations.getenv("IPBASE_DATABASE_DSN"))

    # Get a *specific* connection from the pool
    connection = engine.connect()
    # Explicitly start a new connection on the connection we got
    # This makes the normal "session.begin()" and "session.commit()"
    # calls in SQLAlchemy no-ops as there is already a transaction
    # in progress.
    transaction = connection.begin()
    # Ensure we use the "primed" connection for all our SQLAlchemy
    # session needs in our unit-tests.
    session = Session(bind=connection)
    try:
        yield session
    finally:
        transaction.rollback()
        session.close()
        connection.close()
postgresql timestamp

1 个回答

  1. Best Answer

    我认为这是有问题的做法。您的测试与生产环境不同的任何方式都会增加测试错误事物的危险。

    无论如何，您不能使用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(or now()，这是同一件事) is STABLE， not VOLATILE，这可能会使查询的行为不同（或者不是，如果函数是内联的）。但这正是我一开始就警告过你的那种事情。

    • 4

相关问题