同时对"FOR UPDATE"进行go测试

在测试含有 "FOR UPDATE" 锁的过程时，使用 Go 进行测试存在困难。本文讨论了在读取未提交隔离级别下使用 "FOR UPDATE" 时，尝试插入相同行会导致超时。尽管 SQL 标准定义了读取未提交隔离级别，但在 PostgreSQL 中被视为读取已提交。因此，第二个并发事务无法看到第一个事务中插入的数据，导致测试失败。本文提供了解决方案，建议在测试前设置一个较短的超时，以检查第二个事务是否超时，从而表明存在锁。此外，还讨论了如何通过回滚事务或恢复备份来清除数据库中的测试数据。

我有一个带有 select for update 锁的过程。我想同时从 go 中测试它，以确保锁确实存在。

我正在使用这个：

create table if not exists person (
  name varchar primary key
);
insert into person values ('john');

create table if not exists tickets (
  name varchar primary key references person,
  amount integer not null
);

create or replace procedure sp (_name varchar, _amount integer) as
$$
begin
  -- acquire a lock on person row
  perform name from person where name = _name for update;
  insert into tickets values(_name, _amount);
end
$$ language plpgsql;

这是我目前可以提供的漂亮转储示例，但它表明必须获取锁才能对 sp 调用进行排队。

func TestInsert(t *testing.T) {
    tx, err := db.Begin() // Read Committed level tx
    defer tx.Rollback()

    insertPersonFixtures(tx) // Using this tx to fill database with test data needed by testing SP

    ready1 := make(chan struct{})
    ready2 := make(chan struct{})
    done := make(chan struct{})

    go func() {
      // Must see `prepareSomeData` data in database??
      tx, err := db.BeginTx(context.Background(), &sql.TxOptions{Isolation: sql.LevelReadUncommitted})
      defer tx.Rollback()
      tx.Exec("CALL sp('john', 10)")
      ready1 <- struct{}{}
      <-ready2
      done <- struct{}{}
    }()

    go func() {
      <-ready1
      ctx, cancel := context.WithTimeout(context.Background(), time.Second)
      defer cancel()

      // Must see `prepareSomeData` data in database??
      tx, err := db.BeginTx(context.Background(), &sql.TxOptions{Isolation: sql.LevelReadUncommitted})
      defer tx.Rollback()
      tx.ExecContext(ctx, "CALL sp(`john`, 20)")
      if err == nil {
            t.Error("No lock", err)
        }
      ready2 <- struct{}{}
    }()
    <-done
}

另外，我希望 2 个 goroutine 能够看到第一个事务中填充的测试数据，但是 sp() 失败，因为它看不到数据，这很奇怪，因为读取未提交级别意味着它可以看到脏数据（通过 insertpersonfixtures）。

@brits 的 a：

sql 标准定义了一个附加级别：read uncommitted。在 postgresql 中，read uncommitted 被视为 read committed。

这段代码有什么问题，或者也许这种测试 rdbms 锁的方法不好？或者我误解了隔离级别？在我的示例中，我希望第二个 goroutine 超时并发出信号表明存在锁。

是否有一种直接的方法可以在每次测试后清理数据库？我不确定对每个表运行 truncate 是否容易。显然，在相同的读取已提交和未提交以及缺乏适当的嵌套事务的白痴之后，一切都变得一团糟。

解决方案

来自PostgreSQL docs：

SQL 标准定义了一个附加级别：READ UNCOMMITTED。在 PostgreSQL READ UNCOMMITTED 被视为 READ COMMITTED。

因此，您所看到的情况似乎是可以预料的（但是，由于您没有提供有关 sp() 的太多信息，因此很难发表评论 - 提供 Minimal, Reproducible Example 可能会得到更好的答案）。

请注意，tx.Exec("CAL sp()") 中有一个拼写错误 - 检查从各种数据库调用返回的错误将改进此测试用例。

更新后的其他信息：

tx.ExecContext(ctx, "CALL sp(`john`, 20)") 应该是 err = tx.ExecContext(ctx, "CALL sp(`john`, 20)") （否则你正在检查开始事务返回的错误 - 最好检查两者）。

您可以在执行第二次更新之前通过设置较短的超时（例如一秒 - 类似 tx.ExecContext(ctx, "SET statements_timeout = 1000)) 来执行测试，然后检查调用是否失败。这将起作用因为第一个事务将保持锁定，直到第二个事务提交/回滚。

有没有一种简单的方法可以在每次测试后清理数据库？

这取决于您的要求；在这种情况下，回滚事务将删除您的测试数据。截断工作正常，但通常您会希望保留一些测试数据，因此它并不理想。我通常恢复备份或使用 Docker 容器（数据作为构建的一部分恢复）。

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~