golang实现Redis分布式锁

Redis的分布式锁是通过利用Redis的单线程特性以及原子操作来实现的
Redis的SET命令具有原子性,这意味着只有一个客户端能够成功地设置该键,其他客户端将无法获得锁。如果SET命令成功,表示该客户端成功获得了锁。
Redis锁示例代码

package redis

import (
	"fmt"
	"github.com/go-redis/redis"
	"log"
	"strconv"
	"sync"
	"testing"
	"time"
)

var client *redis.Client
var mutex sync.Mutex

// 初始化连接
func initClient() (err error) {
	client = redis.NewClient(&redis.Options{
		Addr:     "localhost:6379",
		Password: "", // 没有密码,默认值
		DB:       2,  // 默认DB 0
	})
	_, err = client.Ping().Result()
	return err
}

// 加锁
func Lock(key string) (bool, error) {
	// 10秒之后,自动解锁,防止死锁
	delayUnLockTime := 10 * time.Second
	mutex.Lock()
	defer mutex.Unlock()
	bool, err := client.SetNX(key, `{"lock":1}`, delayUnLockTime).Result()
	if err != nil {
		return bool, err
	}
	return bool, err
}

// 解锁
func UnLock(key string, i int) (int64, error) {
	nums, err := client.Del(key).Result()
	if err != nil {
		log.Println(err.Error())
		return 0, err
	}
	fmt.Println(strconv.Itoa(i) + key + "解锁成功,删除key的成功:" + strconv.Itoa(int(nums)))
	return nums, nil
}
func LockTest(key string, i int) {
	result, err := Lock(key)
	if err != nil {
		fmt.Println(err)
		return
	}
	if result {
		fmt.Println(strconv.Itoa(i) + key + "锁定成功")
		return
	}
	fmt.Println(strconv.Itoa(i) + key + "锁定失败")
}

func TestRedis(t *testing.T) {
	err := initClient()
	if err != nil {
		fmt.Println(err)
		return
	}
	LockTest("lock_key", 1)
	go LockTest("lock_key", 2)
	go LockTest("lock_key", 3)
	time.Sleep(1 * time.Second)
	go UnLock("lock_key", 1)
	go UnLock("lock_key", 2)
	go UnLock("lock_key", 3)
	time.Sleep(10 * time.Second)
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值