forked from mirror/redis
add HRANDFIELD and ZRANDMEMBER commands
Signed-off-by: monkey <golang@88.com>
This commit is contained in:
monkey
parent
cb1be727e3
commit
e7dbdda439
113
commands.go
113
commands.go
|
|
@ -9,11 +9,18 @@ import (
|
|||
"github.com/go-redis/redis/v8/internal"
|
||||
)
|
||||
|
||||
const (
|
||||
// KeepTTL is an option for Set command to keep key's existing TTL.
|
||||
// For example:
|
||||
//
|
||||
// rdb.Set(ctx, key, value, redis.KeepTTL)
|
||||
const KeepTTL = -1
|
||||
KeepTTL = -1
|
||||
|
||||
// Persist is remove the time to live associated with the key.
|
||||
// For example:
|
||||
// rdb.GetEX(ctx, key, redis.Persist)
|
||||
Persist = -2
|
||||
)
|
||||
|
||||
func usePrecise(dur time.Duration) bool {
|
||||
return dur < time.Second || dur%time.Second != 0
|
||||
|
|
@ -117,7 +124,7 @@ type Cmdable interface {
|
|||
Get(ctx context.Context, key string) *StringCmd
|
||||
GetRange(ctx context.Context, key string, start, end int64) *StringCmd
|
||||
GetSet(ctx context.Context, key string, value interface{}) *StringCmd
|
||||
GetEX(ctx context.Context, key string, ttl *SetTTL) *StringCmd
|
||||
GetEX(ctx context.Context, key string, expiration time.Duration) *StringCmd
|
||||
GetDel(ctx context.Context, key string) *StringCmd
|
||||
Incr(ctx context.Context, key string) *IntCmd
|
||||
IncrBy(ctx context.Context, key string, value int64) *IntCmd
|
||||
|
|
@ -162,6 +169,7 @@ type Cmdable interface {
|
|||
HMSet(ctx context.Context, key string, values ...interface{}) *BoolCmd
|
||||
HSetNX(ctx context.Context, key, field string, value interface{}) *BoolCmd
|
||||
HVals(ctx context.Context, key string) *StringSliceCmd
|
||||
HRandField(ctx context.Context, key string, count int, withValues bool) *StringSliceCmd
|
||||
|
||||
BLPop(ctx context.Context, timeout time.Duration, keys ...string) *StringSliceCmd
|
||||
BRPop(ctx context.Context, timeout time.Duration, keys ...string) *StringSliceCmd
|
||||
|
|
@ -265,6 +273,7 @@ type Cmdable interface {
|
|||
ZRevRank(ctx context.Context, key, member string) *IntCmd
|
||||
ZScore(ctx context.Context, key, member string) *FloatCmd
|
||||
ZUnionStore(ctx context.Context, dest string, store *ZStore) *IntCmd
|
||||
ZRandMember(ctx context.Context, key string, count int, withScores bool) *StringSliceCmd
|
||||
|
||||
PFAdd(ctx context.Context, key string, els ...interface{}) *IntCmd
|
||||
PFCount(ctx context.Context, keys ...string) *IntCmd
|
||||
|
|
@ -363,58 +372,6 @@ type statefulCmdable func(ctx context.Context, cmd Cmder) error
|
|||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
type ttlAttr int
|
||||
|
||||
const (
|
||||
TExpire ttlAttr = 1 << iota
|
||||
TExpireAT
|
||||
TKeepTTL
|
||||
TPersist
|
||||
)
|
||||
|
||||
// TTL related parameters, not all commands support all ttl attributes.
|
||||
// priority: Expire > ExpireAt > KeepTTL > Persist
|
||||
type SetTTL struct {
|
||||
// set the specified expire time.
|
||||
// Expire > time.Second AND Expire % time.Second == 0: set key EX Expire/time.Second
|
||||
// Expire < time.Second OR Expire % time.Second != 0: set key PX Expire/time.Millisecond
|
||||
Expire time.Duration
|
||||
|
||||
// set the specified Unix time at which the key will expire.
|
||||
// Example: set key EXAT ExpireAt.Unix()
|
||||
// Don't consider milliseconds for now(PXAT)
|
||||
ExpireAt time.Time
|
||||
|
||||
// Retain the time to live associated with the key.
|
||||
KeepTTL bool
|
||||
|
||||
// Remove the time to live associated with the key, Change to never expire
|
||||
Persist bool
|
||||
}
|
||||
|
||||
func appendTTL(ctx context.Context, args []interface{}, t *SetTTL, attr ttlAttr) []interface{} {
|
||||
if t == nil {
|
||||
return args
|
||||
}
|
||||
|
||||
switch {
|
||||
case attr&TExpire == 1 && t.Expire > 0:
|
||||
if usePrecise(t.Expire) {
|
||||
args = append(args, "px", formatMs(ctx, t.Expire))
|
||||
} else {
|
||||
args = append(args, "ex", formatSec(ctx, t.Expire))
|
||||
}
|
||||
case attr&TExpireAT == 1 && !t.ExpireAt.IsZero():
|
||||
args = append(args, "exat", t.ExpireAt.Unix())
|
||||
case attr&TKeepTTL == 1 && t.KeepTTL:
|
||||
args = append(args, "keepttl")
|
||||
case attr&TPersist == 1 && t.Persist:
|
||||
args = append(args, "persist")
|
||||
}
|
||||
|
||||
return args
|
||||
}
|
||||
|
||||
func (c statefulCmdable) Auth(ctx context.Context, password string) *StatusCmd {
|
||||
cmd := NewStatusCmd(ctx, "auth", password)
|
||||
_ = c(ctx, cmd)
|
||||
|
|
@ -764,17 +721,29 @@ func (c cmdable) GetSet(ctx context.Context, key string, value interface{}) *Str
|
|||
return cmd
|
||||
}
|
||||
|
||||
// redis-server version >= 6.2.0
|
||||
func (c cmdable) GetEX(ctx context.Context, key string, ttl *SetTTL) *StringCmd {
|
||||
args := make([]interface{}, 2, 4)
|
||||
// redis-server version >= 6.2.0.
|
||||
//
|
||||
// A value of zero means that the expiration time will not be changed.
|
||||
// Persist(-2) Remove the time to live associated with the key.
|
||||
func (c cmdable) GetEX(ctx context.Context, key string, expiration time.Duration) *StringCmd {
|
||||
args := make([]interface{}, 0, 4)
|
||||
args = append(args, "getex", key)
|
||||
args = appendTTL(ctx, args, ttl, TExpire|TExpireAT|TPersist)
|
||||
if expiration > 0 {
|
||||
if usePrecise(expiration) {
|
||||
args = append(args, "px", formatMs(ctx, expiration))
|
||||
} else {
|
||||
args = append(args, "ex", formatSec(ctx, expiration))
|
||||
}
|
||||
} else if expiration == Persist {
|
||||
args = append(args, "persist")
|
||||
}
|
||||
|
||||
cmd := NewStringCmd(ctx, args...)
|
||||
_ = c(ctx, cmd)
|
||||
return cmd
|
||||
}
|
||||
|
||||
// redis-server version >= 6.2.0
|
||||
// redis-server version >= 6.2.0.
|
||||
func (c cmdable) GetDel(ctx context.Context, key string) *StringCmd {
|
||||
cmd := NewStringCmd(ctx, "getdel", key)
|
||||
_ = c(ctx, cmd)
|
||||
|
|
@ -1253,6 +1222,20 @@ func (c cmdable) HVals(ctx context.Context, key string) *StringSliceCmd {
|
|||
return cmd
|
||||
}
|
||||
|
||||
func (c cmdable) HRandField(ctx context.Context, key string, count int, withValues bool) *StringSliceCmd {
|
||||
args := make([]interface{}, 0, 4)
|
||||
|
||||
// Although count=0 is meaningless, redis accepts count=0.
|
||||
args = append(args, "hrandfield", key, count)
|
||||
if withValues {
|
||||
args = append(args, "withvalues")
|
||||
}
|
||||
|
||||
cmd := NewStringSliceCmd(ctx, args...)
|
||||
_ = c(ctx, cmd)
|
||||
return cmd
|
||||
}
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
func (c cmdable) BLPop(ctx context.Context, timeout time.Duration, keys ...string) *StringSliceCmd {
|
||||
|
|
@ -2327,6 +2310,18 @@ func (c cmdable) ZUnionStore(ctx context.Context, dest string, store *ZStore) *I
|
|||
return cmd
|
||||
}
|
||||
|
||||
func (c cmdable) ZRandMember(ctx context.Context, key string, count int, withScores bool) *StringSliceCmd {
|
||||
args := make([]interface{}, 0, 4)
|
||||
args = append(args, "zrandmember", key, count)
|
||||
if withScores {
|
||||
args = append(args, "withscores")
|
||||
}
|
||||
|
||||
cmd := NewStringSliceCmd(ctx, args...)
|
||||
_ = c(ctx, cmd)
|
||||
return cmd
|
||||
}
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
func (c cmdable) PFAdd(ctx context.Context, key string, els ...interface{}) *IntCmd {
|
||||
|
|
|
|||
|
|
@ -1092,7 +1092,7 @@ var _ = Describe("Commands", func() {
|
|||
Expect(ttl.Err()).NotTo(HaveOccurred())
|
||||
Expect(ttl.Val()).To(BeNumerically("~", 100*time.Second, 3*time.Second))
|
||||
|
||||
getEX := client.GetEX(ctx, "key", &redis.SetTTL{Expire: 200 * time.Second})
|
||||
getEX := client.GetEX(ctx, "key", 200*time.Second)
|
||||
Expect(getEX.Err()).NotTo(HaveOccurred())
|
||||
Expect(getEX.Val()).To(Equal("value"))
|
||||
|
||||
|
|
@ -1832,6 +1832,22 @@ var _ = Describe("Commands", func() {
|
|||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(slice).To(Equal([]string{"hello1", "hello2"}))
|
||||
})
|
||||
|
||||
It("should HRandField", func() {
|
||||
err := client.HSet(ctx, "hash", "key1", "hello1").Err()
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
err = client.HSet(ctx, "hash", "key2", "hello2").Err()
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
|
||||
v := client.HRandField(ctx, "hash", 1, false)
|
||||
Expect(v.Err()).NotTo(HaveOccurred())
|
||||
Expect(v.Result()).To(Or(Equal([]string{"key1"}), Equal([]string{"key2"})))
|
||||
|
||||
var slice []string
|
||||
err = client.HRandField(ctx, "hash", 1, true).ScanSlice(&slice)
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(slice).To(Or(Equal([]string{"key1", "hello1"}), Equal([]string{"key2", "hello2"})))
|
||||
})
|
||||
})
|
||||
|
||||
Describe("hyperloglog", func() {
|
||||
|
|
@ -3870,6 +3886,22 @@ var _ = Describe("Commands", func() {
|
|||
Member: "two",
|
||||
}}))
|
||||
})
|
||||
|
||||
It("should ZRandMember", func() {
|
||||
err := client.ZAdd(ctx, "zset", &redis.Z{Score: 1, Member: "one"}).Err()
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
err = client.ZAdd(ctx, "zset", &redis.Z{Score: 2, Member: "two"}).Err()
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
|
||||
v := client.ZRandMember(ctx, "zset", 1, false)
|
||||
Expect(v.Err()).NotTo(HaveOccurred())
|
||||
Expect(v.Val()).To(Or(Equal([]string{"one"}), Equal([]string{"two"})))
|
||||
|
||||
var slice []string
|
||||
err = client.ZRandMember(ctx, "zset", 1, true).ScanSlice(&slice)
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(slice).To(Or(Equal([]string{"one", "1"}), Equal([]string{"two", "2"})))
|
||||
})
|
||||
})
|
||||
|
||||
Describe("streams", func() {
|
||||
|
|
|
|||
Loading…
Reference in New Issue