package/delayqueue
1
0
Fork 0
mirror of https://github.com/HDT3213/delayqueue.git synced 2025-10-05 15:16:57 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

try to intercept messages
Some checks failed
Go / build (push) Has been cancelled
Details

Browse Source
This commit is contained in:
finley
2025-01-22 22:11:50 +08:00
parent b78a27f399
commit 71751b9b34
6 changed files with 275 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
wrapper.go
View File

@@ -44,6 +44,15 @@ func (r *redisV9Wrapper) Set(key string, value string, expiration time.Duration)
return wrapErr(r.inner.Set(ctx, key, value, expiration).Err())
}
func (r *redisV9Wrapper) LRem(key string, count int64, value string) (int64, error) {
ctx := context.Background()
count, err := r.inner.LRem(ctx, key, count, value).Result()
if err != nil {
return 0, wrapErr(err)
}
return count, nil
}
func (r *redisV9Wrapper) Get(key string) (string, error) {
ctx := context.Background()
ret, err := r.inner.Get(ctx, key).Result()
@@ -92,13 +101,17 @@ func (r *redisV9Wrapper) ZAdd(key string, values map[string]float64) error {
return wrapErr(r.inner.ZAdd(ctx, key, zs...).Err())
}
func (r *redisV9Wrapper) ZRem(key string, members []string) error {
func (r *redisV9Wrapper) ZRem(key string, members []string) (int64, error) {
ctx := context.Background()
members2 := make([]interface{}, len(members))
for i, v := range members {
members2[i] = v
}
return wrapErr(r.inner.ZRem(ctx, key, members2...).Err())
removed, err := r.inner.ZRem(ctx, key, members2...).Result()
if err != nil {
return 0, wrapErr(err)
}
return removed, nil
}
func (r *redisV9Wrapper) ZCard(key string) (int64, error) {
@@ -106,6 +119,15 @@ func (r *redisV9Wrapper) ZCard(key string) (int64, error) {
return r.inner.ZCard(ctx, key).Result()
}
func (r *redisV9Wrapper) ZScore(key string, member string) (float64, error) {
ctx := context.Background()
v, err := r.inner.ZScore(ctx, key, member).Result()
if err != nil {
return 0, wrapErr(err)
}
return v, nil
}
func (r *redisV9Wrapper) LLen(key string) (int64, error) {
ctx := context.Background()
return r.inner.LLen(ctx, key).Result()
@@ -207,13 +229,17 @@ func (r *redisClusterWrapper) ZAdd(key string, values map[string]float64) error
return wrapErr(r.inner.ZAdd(ctx, key, zs...).Err())
}
func (r *redisClusterWrapper) ZRem(key string, members []string) error {
func (r *redisClusterWrapper) ZRem(key string, members []string) (int64, error) {
ctx := context.Background()
members2 := make([]interface{}, len(members))
for i, v := range members {
members2[i] = v
}
return wrapErr(r.inner.ZRem(ctx, key, members2...).Err())
removed, err := r.inner.ZRem(ctx, key, members2...).Result()
if err != nil {
return 0, wrapErr(err)
}
return removed, nil
}
func (r *redisClusterWrapper) ZCard(key string) (int64, error) {
@@ -221,11 +247,29 @@ func (r *redisClusterWrapper) ZCard(key string) (int64, error) {
return r.inner.ZCard(ctx, key).Result()
}
func (r *redisClusterWrapper) ZScore(key string, member string) (float64, error) {
ctx := context.Background()
v, err := r.inner.ZScore(ctx, key, member).Result()
if err != nil {
return 0, wrapErr(err)
}
return v, nil
}
func (r *redisClusterWrapper) LLen(key string) (int64, error) {
ctx := context.Background()
return r.inner.LLen(ctx, key).Result()
}
func (r *redisClusterWrapper) LRem(key string, count int64, value string) (int64, error) {
ctx := context.Background()
count, err := r.inner.LRem(ctx, key, count, value).Result()
if err != nil {
return 0, wrapErr(err)
}
return count, nil
}
func (r *redisClusterWrapper) Publish(channel string, payload string) error {
ctx := context.Background()
return r.inner.Publish(ctx, channel, payload).Err()