go redis锁

摘要:
=Nil{fmt.Printf(“Redis锁定-解锁失败密钥:%s,令牌:%s,错误:%v r n”,lockKey,令牌,错误)returnerr}ifv,ok:=ret。(int64);确定&&v==1{fmt.Printf(“Redis锁定-解锁成功密钥:%s,令牌:%s r n”,lockKey,令牌)returnnil}fmt。Printf(“Redis锁定-解锁失败密钥:%s,令牌:%s,密钥过期r n”,lockKey,令牌)returnnil}

redis经常用作分布式锁,这里记录一个简单的锁代码如下:

package main

import (
    "crypto/rand"
    "encoding/base64"
    "fmt"
    "github.com/go-redis/redis"
    "io"
    "sync"
    "time"
)

func main() {
    addr := "192.168.100.30:6379"
    rdb := redis.NewUniversalClient(&redis.UniversalOptions{
        Addrs:    []string {addr},
        Password: "", // no password set
        DB:       0,  // use default DB
    })
    defer func() {
        _ = rdb.Close()
    }()

    key := "gavin-lock"
    uid, err := Lock(rdb, key, time.Second*1, time.Second*2)
    if err == nil {
        _ = Unlock(rdb, key, uid)
    }
}

var randMutex sync.Mutex
var randBytes []byte

// 获取随机字串
func randToken() string {
    randMutex.Lock()
    defer randMutex.Unlock()

    if len(randBytes) == 0 {
        randBytes = make([]byte, 16)
    }

    if _, err := io.ReadFull(rand.Reader, randBytes); err != nil {
        return ""
    }
    return base64.RawURLEncoding.EncodeToString(randBytes)
}

func Lock(redisClient redis.UniversalClient, lockKey string, acquireTimeOut, lockTimeOut time.Duration) (token string, err error) {
    token = randToken()
    var acquireDuration = acquireTimeOut

    for acquireDuration > 0 {
        ok, err := redisClient.SetNX(lockKey, token, lockTimeOut).Result()
        if err != nil {
            fmt.Printf("redis锁 - 加锁失败 key: %s, token: %s, ttl: %v, err: %v\r\n", lockKey, token, lockTimeOut, err)
            return "", err
        }

        if ok {
            fmt.Printf("redis锁 - 加锁成功 key: %s, token: %s, ttl: %v\r\n", lockKey, token, lockTimeOut)
            return token, nil
        }
        time.Sleep(time.Millisecond * 100) // next: sleep 500ms
        acquireDuration -= time.Millisecond * 100
    }
    fmt.Printf("redis锁 - 加锁超时 key: %s, token: %s, ttl: %v\r\n", lockKey, token, acquireTimeOut)
    return "", fmt.Errorf("获取redis锁超时: %v\r\n", acquireTimeOut)
}

func Unlock(redisClient redis.UniversalClient, lockKey, token string) error {
    const script = `if redis.call('GET', KEYS[1]) == ARGV[1] then return redis.call('DEL', KEYS[1]) else return 0 end`
    ret, err := redisClient.Eval(script, []string{lockKey}, token).Result()
    if err != nil {
        fmt.Printf("redis锁 - 解锁失败 key: %s, token: %s,  err: %v\r\n", lockKey, token, err)
        return err
    }

    if v, ok := ret.(int64); ok && v == 1 {
        fmt.Printf("redis锁 - 解锁成功 key: %s, token: %s\r\n", lockKey, token)
        return nil
    }

    fmt.Printf("redis锁 - 解锁失败 key: %s, token: %s, key已经过期\r\n", lockKey, token)
    return nil
}

免责声明:文章转载自《go redis锁》仅用于学习参考。如对内容有疑问,请及时联系本站处理。

上篇利用OpenXml生成Word2007文档iOS制作自己的Framework框架下篇

宿迁高防,2C2G15M,22元/月;香港BGP,2C5G5M,25元/月 雨云优惠码:MjYwNzM=

相关文章

将Kafka收到的数据传入到redis中

首先得配置GateWay中的config.properties 然后再看一下TBox中的properties main方法中  Test中的config.properties  Test中 先启动网管(GateWay),再启动终端(TBox),然后再运行KafkaTest 启动TBox   在启动Test 可以看到 接收到数据了  然...

线上redis热key问题

一个安静的晚上突然被群里报警打断,然后果断不情愿的爬起来找问题,通过监控发现一个服务的QPS暴增,接口响应延迟也上来了,达到s级别了,所以赶紧找问题。 因为应用很依赖redis,第一时间先看是不是它,果然还真是它,看下图: 确定了问题的方向就该找具体原因了,进一步查看Zabbix发现其中一台redis的CPU使用率明显升高,如下图: 连到机器上top发...

SpringCloud-OAuth2(二):实战篇

如果不了解Oauth2 是什么、工作流程的可以看我上一篇文章:SpringCloud-OAuth2(一):基础篇 这篇讲的内容是:Oauth2在SpringBoot/SpringCloud中的实战。 SpringBoot版本:2.2.5.Release SpringCloud版本:Hoxton.SR9 JDK版本:1.8 1:POM配置 <d...

RedisTemplate访问Redis数据结构(三)——Hash

HashOperations提供一系列方法操作hash。首先初始化spring工厂获得redisTemplate和opsForHash private RedisTemplate<String,Object> redisTemplate; private HashOperations<String,String,Objec...

SpringBoot + Redis + Shiro 实现权限管理(转)

概述 本文基于网上整理,为了实现将Shiro框架的session存储到redis里面,进而实现基于Niginx负载均衡,多站点部署; maven下shiro依赖 <!-- shiro --> <dependency> <groupId>org.apache.shiro</gr...

JS全局添加token

全局添加token var token = sessionStorage.getItem("UserTocken"); if(token){ $.ajaxSetup({ //发送请求前触发 beforeSend: function(xhr) { //可以设置自定义标头 xhr....