如何使用go-redis/redis包在Go中创建Redis事务？

Question

我想使用 MULTI 和 EXEC 执行多个redis命令来实现事务，这样如果出现问题，我就可以使用 DISCARD 来撤消它。

我一直在寻找如何使用 go-redis/redis 包执行redis事务的示例，但没有找到任何内容。

我还查看了 此处 的文档，但没有找到任何与使用该包进行redis事务 例如这样 相关的内容。或许我在文档中遗漏了某些内容，因为你知道godoc只是用单行来解释大多数包中的每个函数。

尽管我找到了一些使用其他Go Redis库执行redis事务的示例，但我不想修改我的程序以使用另一个库，因为将整个应用程序移植到另一个库的工作量将会很大。

有人能帮我使用 go-redis/redis 包来实现吗？

- Kebeng

---

该包中的事务被称为Tx。 - JimB

我知道。但是如何创建tx对象？ 我没有找到任何返回tx的函数。 - Kebeng

Answer 1

当您使用Client.Watch时，会为事务获得一个Tx值。

err := client.Watch(func(tx *redis.Tx) error {
    n, err := tx.Get(key).Int64()
    if err != nil && err != redis.Nil {
        return err
    }

    _, err = tx.Pipelined(func(pipe *redis.Pipeline) error {
        pipe.Set(key, strconv.FormatInt(n+1, 10), 0)
        return nil
    })
    return err
}, key)

- JimB

---

哇，我在文档中没有看到那个。Tx是Watch函数内匿名函数的参数，所以我错过了它。谢谢啊。 - Kebeng

此答案中的代码将生成以下 Redis 命令：WATCH key GET key SET key UNWATCH，这不是 Redis 网站上关于事务的解释 https://redis.io/topics/transactions#cas。为了使用 MULTI / EXEC 包装 SET，请改用 tx.TxPipelined 而非 tx.Pipelined。 - D.C. Joo

Answer 2

你可以在这里找到一个创建Redis事务的示例：

代码：

pipe := rdb.TxPipeline()

incr := pipe.Incr("tx_pipeline_counter")
pipe.Expire("tx_pipeline_counter", time.Hour)

// Execute
//
//     MULTI
//     INCR pipeline_counter
//     EXPIRE pipeline_counts 3600
//     EXEC
//
// using one rdb-server roundtrip.
_, err := pipe.Exec()
fmt.Println(incr.Val(), err)

输出：

1 <nil>

如果您喜欢使用watch（乐观锁定）

您可以在此处看到一个示例

代码：

const routineCount = 100

// Transactionally increments key using GET and SET commands.
increment := func(key string) error {
    txf := func(tx *redis.Tx) error {
        // get current value or zero
        n, err := tx.Get(key).Int()
        if err != nil && err != redis.Nil {
            return err
        }

        // actual opperation (local in optimistic lock)
        n++

        // runs only if the watched keys remain unchanged
        _, err = tx.TxPipelined(func(pipe redis.Pipeliner) error {
            // pipe handles the error case
            pipe.Set(key, n, 0)
            return nil
        })
        return err
    }

    for retries := routineCount; retries > 0; retries-- {
        err := rdb.Watch(txf, key)
        if err != redis.TxFailedErr {
            return err
        }
        // optimistic lock lost
    }
    return errors.New("increment reached maximum number of retries")
}

var wg sync.WaitGroup
wg.Add(routineCount)
for i := 0; i < routineCount; i++ {
    go func() {
        defer wg.Done()

        if err := increment("counter3"); err != nil {
            fmt.Println("increment error:", err)
        }
    }()
}
wg.Wait()

n, err := rdb.Get("counter3").Int()
fmt.Println("ended with", n, err)

输出：

ended with 100 <nil>

- Brachi