使用redis做分布式锁

思路步骤

1. 获取锁, 使用 SETNX 命令设置一个key.
2. 如果没获取到，从新拿锁，返回步骤1 ; 从新拿锁可以设置等待时间;也可以记录拿锁次数为了做“避免死循环”.
3. 如果获取到，使用 EXPIRE 给锁加存活时间；
4. 接步骤3，执行业务，执行完释放锁, 使用 DEL 命令

代码实现

package main

import (
	"fmt"
	"github.com/garyburd/redigo/redis"
	gt "github.com/mangenotwork/gathertool"
	"log"
	"time"
)

var red = gt.NewRedis("192.168.0.197", "6379", "love0021$%", 14)

func main(){
	_ = red.RedisPool() // 初始化redis连接池

	for i:=0; i<20; i++ {
		go Doing(i)
	}

	select {}

}

func Doing(i int){
	log.Println(fmt.Sprintf("[%d] 开始Doing... ", i))
	key := "lock"
	rd := red.GetConn()
	lockTime := 0 // 获取锁次数
Lo:
	if lockTime > 999 { //避免死循环
		log.Println("避免死循环")
		return
	}
	// 获取锁
	is, err := redis.Bool(rd.Do("SETNX", redis.Args{}.Add(key).AddFlat(1)...))
	if err != nil {
		panic(err)
	}
	//没有拿到锁
	if !is {
		//log.Println("没拿到锁")
		lockTime++
		time.Sleep(1*time.Second)
		goto Lo
	}

	// 给锁添加时间
	_,err = rd.Do("EXPIRE", redis.Args{}.Add(key).Add(5)...)
	if err != nil {
		log.Println("给锁添加时间 err = ", err)
	}

	// 执行
	log.Println(fmt.Sprintf("[%d]执行", i))
	time.Sleep(1*time.Second)

	// 释放锁
	log.Println(fmt.Sprintf("[%d]释放锁", i))
	_,err = rd.Do("DEL", key)
	if err != nil {
		log.Println("释放锁 err = ", err)
	}
}

运行结果

一些问题探讨

如果突然服务器宕机，那么必然造成锁无法释放，即造成死锁？
解决方案：设置超时时间。

加锁和设置超时时间中间引起服务器宕机，则一样会导致死锁。
解决方案：原子性操作，即同时加锁和设置超时时间；

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-14Tju5bW-1670470670608)(http://mange1.oss-cn-beijing.aliyuncs.com/test/a8ea5c06b619cb643a891497b15aee86.png#pic_center)]

总结

设计实现分布式锁需要满足一下条件：

1. 互斥性；在任意时刻，只有一个客户端能持有锁。
2. 不能发生死锁；即使存在一个线程持有锁的期间崩溃而没有主动解锁，也能保证后续其他线程能加锁。
3. 加锁和解锁必须是同一个线程。