golang 实现redis分布式锁
时间: 2023-08-24 14:04:48 浏览: 207
基于redis的分布式锁实现
实现 Redis 分布式锁的基本思路是利用 Redis 的 SETNX 命令(SET if Not eXists)实现。SETNX 命令会在 key 不存在的情况下,将 key 的值设为 value,如果 key 已经存在,则不做任何操作。
以下是一个简单的 Golang 实现 Redis 分布式锁的代码示例:
```go
package redislock
import (
"fmt"
"time"
"github.com/go-redis/redis/v7"
)
type RedisLock struct {
redisClient *redis.Client
key string
value string
expiration time.Duration
}
func NewRedisLock(redisClient *redis.Client, key, value string, expiration time.Duration) *RedisLock {
return &RedisLock{
redisClient: redisClient,
key: key,
value: value,
expiration: expiration,
}
}
func (r *RedisLock) Lock() (bool, error) {
success, err := r.redisClient.SetNX(r.key, r.value, r.expiration).Result()
if err != nil {
return false, err
}
return success, nil
}
func (r *RedisLock) Unlock() error {
err := r.redisClient.Del(r.key).Err()
if err != nil {
return err
}
return nil
}
```
在上面的代码中,NewRedisLock 函数用于创建一个 RedisLock 实例,需要传入 Redis 客户端、锁的 key、锁的值、锁的过期时间。Lock 方法用于尝试获取锁,如果获取成功,返回 true,否则返回 false。Unlock 方法用于释放锁。
以下是一个简单的使用示例:
```go
package main
import (
"fmt"
"time"
"github.com/go-redis/redis/v7"
"github.com/yourusername/redislock"
)
func main() {
redisClient := redis.NewClient(&redis.Options{
Addr: "localhost:6379",
DB: 0,
})
lock := redislock.NewRedisLock(redisClient, "my-lock", "my-value", 10*time.Second)
success, err := lock.Lock()
if err != nil {
fmt.Println("failed to acquire lock:", err)
return
}
if !success {
fmt.Println("lock is already held by another process")
return
}
defer lock.Unlock()
// Do some work
}
```
在上面的示例中,我们创建了一个 Redis 客户端,并且创建了一个 RedisLock 实例。然后,我们调用 Lock 方法尝试获取锁,如果获取成功,就可以进行一些需要加锁的操作。最后,我们调用 Unlock 方法释放锁。
阅读全文