// DistributeLockRedis 基于redis的分布式可重入鎖，自動(dòng)續(xù)租
type DistributeLockRedis struct {
	key       string             // 鎖的key
	expire    int64              // 鎖超時(shí)時(shí)間
	status    bool               // 上鎖成功標(biāo)識(shí)
	cancelFun context.CancelFunc // 用于取消自動(dòng)續(xù)租攜程
	redis     redis.Client       // redis句柄
}

// 創(chuàng)建可
func NewDistributeLockRedis(key string, expire int64) *DistributeLockRedis {
	return &DistributeLockRedis{
		 key : key,
		 expire : expire,
	}
}

// TryLock 上鎖
func (dl *DistributeLockRedis) TryLock() (err error) {
	if err = dl.lock(); err != nil {
		return err
	}
	ctx, cancelFun := context.WithCancel(context.Background())
	dl.cancelFun = cancelFun
	dl.startWatchDog(ctx) // 創(chuàng)建守護(hù)協(xié)程，自動(dòng)對(duì)鎖進(jìn)行續(xù)期
	dl.status = true
	return nil
}

// competition 競(jìng)爭(zhēng)鎖
func (dl *DistributeLockRedis) lock() error {
	if res, err := redis.String(dl.redis.Do(context.Background(), "SET", dl.key, 1, "NX", "EX", dl.expire)); err != nil {
		return err
	} 
	return nil
}


// guard 創(chuàng)建守護(hù)協(xié)程，自動(dòng)續(xù)期
func (dl *DistributeLockRedis) startWatchDog(ctx context.Context) {
	safeGo(func() error {
		for {
			select {
			// Unlock通知結(jié)束
			case <-ctx.Done():
				return nil
			default:
				// 否則只要開(kāi)始了，就自動(dòng)重入（續(xù)租鎖）
				if dl.status {
					if res, err := redis.Int(dl.redis.Do(context.Background(), "EXPIRE", dl.key, dl.expire)); err != nil {
						return nil
					} 
					// 續(xù)租時(shí)間為 expire/2 秒
					time.Sleep(time.Duration(dl.expire/2) * time.Second)
				}
			}
		}
	})
}

// Unlock 釋放鎖
func (dl *DistributeLockRedis) Unlock() (err error) {
	// 這個(gè)重入鎖必須取消，放在第一個(gè)地方執(zhí)行
	if dl.cancelFun != nil {
		dl.cancelFun() // 釋放成功，取消重入鎖
	}
	var res int
	if dl.status {
		if res, err = redis.Int(dl.redis.Do(context.Background(), "Del", dl.key)); err != nil {
			return fmt.Errorf("釋放鎖失敗")
		}
		if res == 1 {
			dl.status = false
			return nil
		}
	}
	return fmt.Errorf("釋放鎖失敗")
}

ctx, cancelFun := context.WithCancel(context.Background())
dl.cancelFun = cancelFun
dl.startWatchDog(ctx) // 創(chuàng)建守護(hù)協(xié)程，自動(dòng)對(duì)鎖進(jìn)行續(xù)期
dl.status = true

func Foo() error {
  key := foo
  
  // 創(chuàng)建可重入的分布式鎖
	dl := NewDistributeLockRedis(key, 10)
	// 爭(zhēng)搶鎖
	err := dl.TryLock()
	if err != nil {
		// 沒(méi)有搶到鎖
		return err
	}
	
	// 搶到鎖的記得釋放鎖
	defer func() {
		dl.Unlock()
	}
	
	// 做真正的任務(wù)
	DoJob()
}