refactor: 将redis各个业务分离
This commit is contained in:
59
backend/internal/middleware/redis/zset.go
Normal file
59
backend/internal/middleware/redis/zset.go
Normal file
@@ -0,0 +1,59 @@
|
||||
package redis
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
redis "github.com/redis/go-redis/v9"
|
||||
)
|
||||
|
||||
func (c *Client) ZincrBy(ctx context.Context, key string, member string, score float64) error {
|
||||
if c == nil || c.rdb == nil {
|
||||
return nil
|
||||
}
|
||||
return c.rdb.ZIncrBy(ctx, key, score, member).Err()
|
||||
}
|
||||
|
||||
func (c *Client) Expire(ctx context.Context, key string, ttl time.Duration) error {
|
||||
if c == nil || c.rdb == nil {
|
||||
return nil
|
||||
}
|
||||
return c.rdb.Expire(ctx, key, ttl).Err()
|
||||
}
|
||||
|
||||
func (c *Client) ZUnionStore(ctx context.Context, dst string, keys []string, aggregate string) error {
|
||||
if c == nil || c.rdb == nil {
|
||||
return nil
|
||||
}
|
||||
return c.rdb.ZUnionStore(ctx, dst, &redis.ZStore{
|
||||
Keys: keys,
|
||||
Aggregate: aggregate,
|
||||
}).Err()
|
||||
}
|
||||
|
||||
func (c *Client) Exists(ctx context.Context, key string) (bool, error) {
|
||||
if c == nil || c.rdb == nil {
|
||||
return false, nil
|
||||
}
|
||||
n, err := c.rdb.Exists(ctx, key).Result()
|
||||
return n > 0, err
|
||||
}
|
||||
|
||||
func (c *Client) ZRevRange(ctx context.Context, key string, start, stop int64) ([]string, error) {
|
||||
if c == nil || c.rdb == nil {
|
||||
return nil, nil
|
||||
}
|
||||
return c.rdb.ZRevRange(ctx, key, start, stop).Result()
|
||||
}
|
||||
|
||||
func (c *Client) ZRevRangeByScore(ctx context.Context, key string, max, min string, offset, count int64) ([]string, error) {
|
||||
if c == nil || c.rdb == nil {
|
||||
return nil, nil
|
||||
}
|
||||
return c.rdb.ZRevRangeByScore(ctx, key, &redis.ZRangeBy{
|
||||
Max: max,
|
||||
Min: min,
|
||||
Offset: offset,
|
||||
Count: count,
|
||||
}).Result()
|
||||
}
|
||||
Reference in New Issue
Block a user