golang redis连接池式分布式锁小小的封装

摘要：把redis加锁方法单独抽离出去放入一个公共方法，暂未做数据类型断言

helper redisLock 文件

package helper

import (
"fmt"
"github.com/garyburd/redigo/redis"
"github.com/prometheus/common/log"
)

type Lock struct {
resource string
token    int64
conn     redis.Pool
timeout  int
}

func (lock *Lock) tryLock() (ok bool, err error) {
_, err = redis.String(lock.conn.Get().Do("SET", lock.key(), lock.token, "EX", int(lock.timeout), "NX"))
if err == redis.ErrNil {
// The lock was not successful, it already exists.
return false, nil
}
if err != nil {
return false, err
}
return true, nil
}

func (lock *Lock) Unlock() (err error) {
_, err = lock.conn.Get().Do("del", lock.key())
return
}

func (lock *Lock) Close() (err error) {
if lock != nil {
lock.conn.Close()
}
return
}

func (lock *Lock) key() string {
return fmt.Sprintf("redislock:%s", lock.resource)
}

func (lock *Lock) AddTimeout(ex_time int64) (ok bool, err error) {
ttl_time, err := redis.Int64(lock.conn.Get().Do("TTL", lock.key()))
fmt.Println(ttl_time)
if err != nil {
log.Fatal("redis get failed:", err)
}
if ttl_time > 0 {
fmt.Println(11)
_, err := redis.String(lock.conn.Get().Do("SET", lock.key(), lock.token, "EX", int(ttl_time+ex_time)))
if err == redis.ErrNil {
return false, nil
}
if err != nil {
return false, err
}
}
return false, nil
}
func (lock *Lock) Get(key string) int64 {
sum, err := redis.Int64(lock.conn.Get().Do("TTL", lock.resource))
if err != nil {
log.Fatal("redis get failed:", err)
}
return sum
}

func TryLockWithTimeout(conn redis.Pool, resource string, token int64, timeout int) (lock *Lock, ok bool, err error) {
lock = &Lock{resource, token, conn, timeout}

ok, err = lock.tryLock()

if !ok || err != nil {
lock = nil
}

return
}
func TryLock(conn redis.Pool, key string, value int64, DefaulTimeout int) (lock *Lock, ok bool, err error) {
return TryLockWithTimeout(conn, key, value, DefaulTimeout)
}

调用（有一些无用import我懒得删了，复制粘贴自行删除）

package main

import (
"apitest/helper"
"context"
"flag"
"fmt"
"github.com/garyburd/redigo/redis"
"github.com/panjf2000/ants"
"github.com/rpcx-ecosystem/rpcx-examples3"
"github.com/smallnest/rpcx/client"
"log"
"net/http"
"time"
)

var redisPool = &redis.Pool{
MaxIdle:     30,
MaxActive:   1000,
IdleTimeout: 30 * time.Second,
Dial: func() (conn redis.Conn, err error) {
//RedisHost := "127.0.0.1:6379" //连接地址
//RedisPassword := ""           //连接密码
return redis.Dial("tcp", "127.0.0.1:32769")
},
}

func main() {
//redisPool.Close()
rs := redisPool
//defer rs.Close()
res, ok, err := helper.TryLock(*rs, "key", 0, 12)
//res, ok, err := TryLock(*rs, "key", 0, 12)
if err != nil {
panic(err)
}
if ok {
fmt.Println("加锁成功")
res.Unlock()
} else {
fmt.Println("加锁失败")
}
res.Close()

}