|
|
package dorm
|
|
|
|
|
|
import (
|
|
|
"context"
|
|
|
"github.com/redis/go-redis/v9"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
// Set 设置一个key的值
|
|
|
func (r *RedisClient) Set(ctx context.Context, key string, value interface{}, expiration time.Duration) *redis.StatusCmd {
|
|
|
return r.Db.Set(ctx, key, value, expiration)
|
|
|
}
|
|
|
|
|
|
// Get 查询key的值
|
|
|
func (r *RedisClient) Get(ctx context.Context, key string) *redis.StringCmd {
|
|
|
return r.Db.Get(ctx, key)
|
|
|
}
|
|
|
|
|
|
// GetSet 设置一个key的值，并返回这个key的旧值
|
|
|
func (r *RedisClient) GetSet(ctx context.Context, key string, value interface{}) *redis.StringCmd {
|
|
|
return r.Db.GetSet(ctx, key, value)
|
|
|
}
|
|
|
|
|
|
// SetNX 如果key不存在，则设置这个key的值
|
|
|
func (r *RedisClient) SetNX(ctx context.Context, key string, value interface{}, expiration time.Duration) *redis.BoolCmd {
|
|
|
return r.Db.SetNX(ctx, key, value, expiration)
|
|
|
}
|
|
|
|
|
|
// MGet 批量查询key的值
|
|
|
func (r *RedisClient) MGet(ctx context.Context, keys ...string) *redis.SliceCmd {
|
|
|
return r.Db.MGet(ctx, keys...)
|
|
|
}
|
|
|
|
|
|
// MSet 批量设置key的值
|
|
|
// MSet(map[string]interface{}{"key1": "value1", "key2": "value2"})
|
|
|
func (r *RedisClient) MSet(ctx context.Context, values map[string]interface{}) *redis.StatusCmd {
|
|
|
return r.Db.MSet(ctx, values)
|
|
|
}
|
|
|
|
|
|
// Incr 针对一个key的数值进行递增操作
|
|
|
func (r *RedisClient) Incr(ctx context.Context, key string) *redis.IntCmd {
|
|
|
return r.Db.Incr(ctx, key)
|
|
|
}
|
|
|
|
|
|
// IncrBy 针对一个key的数值进行递增操作，指定每次递增多少
|
|
|
func (r *RedisClient) IncrBy(ctx context.Context, key string, value int64) *redis.IntCmd {
|
|
|
return r.Db.IncrBy(ctx, key, value)
|
|
|
}
|
|
|
|
|
|
// Decr 针对一个key的数值进行递减操作
|
|
|
func (r *RedisClient) Decr(ctx context.Context, key string) *redis.IntCmd {
|
|
|
return r.Db.Decr(ctx, key)
|
|
|
}
|
|
|
|
|
|
// DecrBy 针对一个key的数值进行递减操作，指定每次递减多少
|
|
|
func (r *RedisClient) DecrBy(ctx context.Context, key string, value int64) *redis.IntCmd {
|
|
|
return r.Db.DecrBy(ctx, key, value)
|
|
|
}
|
|
|
|
|
|
// Del 删除key操作，支持批量删除
|
|
|
func (r *RedisClient) Del(ctx context.Context, keys ...string) *redis.IntCmd {
|
|
|
return r.Db.Del(ctx, keys...)
|
|
|
}
|
|
|
|
|
|
// Keys 按前缀获取所有key名
|
|
|
func (r *RedisClient) Keys(ctx context.Context, prefix string) []string {
|
|
|
values, _ := r.Db.Keys(ctx, prefix).Result()
|
|
|
keys := make([]string, 0, len(values))
|
|
|
if len(values) <= 0 {
|
|
|
return keys
|
|
|
}
|
|
|
for _, value := range values {
|
|
|
keys = append(keys, value)
|
|
|
}
|
|
|
return keys
|
|
|
}
|
|
|
|
|
|
// KeysValue 按前缀获取所有key值
|
|
|
func (r *RedisClient) KeysValue(ctx context.Context, prefix string) *redis.SliceCmd {
|
|
|
values, _ := r.Db.Keys(ctx, prefix).Result()
|
|
|
if len(values) <= 0 {
|
|
|
return &redis.SliceCmd{}
|
|
|
}
|
|
|
keys := make([]string, 0, len(values))
|
|
|
for _, value := range values {
|
|
|
keys = append(keys, value)
|
|
|
}
|
|
|
return r.MGet(ctx, keys...)
|
|
|
}
|
