Merge pull request #433 from redis/fix-zpopmax-count

fix: rueidiscompat's ZPopMax and ZPopMin with count 1
redis · Dec 23, 2023 · 3ec15df · 3ec15df
2 parents 0249ec8 + a81d7e4
commit 3ec15df
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/rueidiscompat/adapter.go b/rueidiscompat/adapter.go
@@ -2028,9 +2028,7 @@ func (c *Compat) ZPopMax(ctx context.Context, key string, count ...int64) *ZSlic
 		resp = c.client.Do(ctx, c.client.B().Zpopmax().Key(key).Build())
 	case 1:
 		resp = c.client.Do(ctx, c.client.B().Zpopmax().Key(key).Count(count[0]).Build())
-		if count[0] > 1 {
-			return newZSliceCmd(resp)
-		}
+		return newZSliceCmd(resp)
 	default:
 		panic("too many arguments")
 	}
@@ -2044,9 +2042,7 @@ func (c *Compat) ZPopMin(ctx context.Context, key string, count ...int64) *ZSlic
 		resp = c.client.Do(ctx, c.client.B().Zpopmin().Key(key).Build())
 	case 1:
 		resp = c.client.Do(ctx, c.client.B().Zpopmin().Key(key).Count(count[0]).Build())
-		if count[0] > 1 {
-			return newZSliceCmd(resp)
-		}
+		return newZSliceCmd(resp)
 	default:
 		panic("too many arguments")
 	}