golock/redis.go

package golock

import (
	"context"
	"errors"
	"go.dtapp.net/dorm"
	"time"
)

type LockRedis struct {
	client *dorm.RedisClient // 驱动
}

func NewLockRedis(client *dorm.RedisClient) *LockRedis {
	return &LockRedis{client: client}
}

// Lock 上锁
// key 锁名
// val 锁内容
// ttl 锁过期时间
func (r *LockRedis) Lock(key string, val string, ttl int64) (string, error) {
	if ttl <= 0 {
		return "", errors.New("长期请使用 LockForever 方法")
	}
	// 获取
	get, err := r.client.Get(context.Background(), key).Result()
	if err != nil {
		return "", errors.New("获取异常")
	}
	if get != "" {
		return "", errors.New("上锁失败，已存在")
	}
	// 设置
	err = r.client.Set(context.Background(), key, val, time.Duration(ttl)).Err()
	if err != nil {
		return "", errors.New("上锁失败")
	}
	return val, nil
}

// Unlock 解锁
// key 锁名
func (r *LockRedis) Unlock(key string) error {
	_, err := r.client.Del(context.Background(), key).Result()
	return err
}

// LockForever 永远上锁
// key 锁名
// val 锁内容
func (r *LockRedis) LockForever(key string, val string) (string, error) {
	// 获取
	get, err := r.client.Get(context.Background(), key).Result()
	if err != nil {
		return "", errors.New("获取异常")
	}
	if get != "" {
		return "", errors.New("上锁失败，已存在")
	}
	// 设置
	err = r.client.Set(context.Background(), key, val, 0).Err()
	if err != nil {
		return "", errors.New("上锁失败")
	}
	return val, nil
}