一文看懂Go+Redis实现分布式锁

2024-01-27 21:07:01

分布式锁简介

分布式锁是一种在分布式系统中协调对共享资源的访问的机制。它确保只有一个进程或线程能够同时访问共享资源。分布式锁通常用于防止多个进程或线程同时修改共享数据，从而导致数据不一致或损坏。

Redis实现分布式锁

Redis是一个键值存储数据库，它支持多种数据类型，包括字符串、散列、列表、集合和有序集合。Redis还支持事务和发布/订阅功能。这些特性使得Redis成为实现分布式锁的理想选择。

最简单的Go+Redis分布式锁实现

以下是最简单的Go+Redis分布式锁实现：

// 获取锁
func lock(key string, expiration time.Duration) (string, error) {
  // 尝试获取锁
  value, err := client.SetNX(key, "locked", expiration).Result()
  if err != nil {
    return "", err
  }

  // 如果获取成功，返回锁值
  if value {
    return "locked", nil
  }

  // 如果获取失败，返回nil
  return nil, nil
}

// 释放锁
func unlock(key string, value string) error {
  // 使用LUA脚本原子性地比较并删除锁
  script := redis.NewScript(`
    if redis.call("get", KEYS[1]) == ARGV[1] then
      return redis.call("del", KEYS[1])
    else
      return 0
    end
  `)
  _, err := client.Eval(script, []string{key}, []interface{}{value}).Result()
  return err
}

这个实现使用Redis的SetNX命令来尝试获取锁。如果获取成功，则返回锁值；否则，返回nil。释放锁时，使用LUA脚本原子性地比较并删除锁。这样可以确保只有持有锁的进程或线程才能释放锁。

更复杂的Go+Redis分布式锁实现

上面的实现非常简单，但它也有局限性。例如，它没有考虑锁的过期时间。如果持有锁的进程或线程崩溃，锁将永远不会被释放。

为了解决这个问题，我们可以使用更复杂的实现。例如，我们可以使用Redis的SET命令来设置锁，并指定一个过期时间。这样，如果持有锁的进程或线程崩溃，锁将在过期时间后自动释放。

以下是一个更复杂的Go+Redis分布式锁实现：

// 获取锁
func lock(key string, expiration time.Duration) (string, error) {
  // 使用SET命令设置锁，并指定过期时间
  value, err := client.Set(key, "locked", expiration).Result()
  if err != nil {
    return "", err
  }

  // 返回锁值
  return value, nil
}

// 释放锁
func unlock(key string, value string) error {
  // 使用LUA脚本原子性地比较并删除锁
  script := redis.NewScript(`
    if redis.call("get", KEYS[1]) == ARGV[1] then
      return redis.call("del", KEYS[1])
    else
      return 0
    end
  `)
  _, err := client.Eval(script, []string{key}, []interface{}{value}).Result()
  return err
}