Merge pull request #53 from q7164518/master

新增ZRANDMEMBER方法
2881099 · Mar 22, 2021 · 9b29e4a · 9b29e4a
2 parents 159aef4 + f38dd4e
commit 9b29e4a
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 0 deletions.
diff --git a/src/FreeRedis/FreeRedis.xml b/src/FreeRedis/FreeRedis.xml
diff --git a/src/FreeRedis/RedisClient/SortedSets.cs b/src/FreeRedis/RedisClient/SortedSets.cs
@@ -268,6 +268,29 @@ long ZStore(bool inter, string destination, string[] keys, int[] weights, ZAggre
             .InputIf(weights?.Any() == true, weights)
             .InputIf(aggregate != null, "AGGREGATE", aggregate ?? ZAggregate.max), rt => rt
             .ThrowOrValue<long>());
+
+        /// <summary>
+        /// 随机返回N个元素
+        /// <para>Redis 6.2.0+以上才支持该命令</para>
+        /// </summary>
+        /// <param name="key">Key</param>
+        /// <param name="count">返回的个数</param>
+        /// <param name="repetition">是否允许有重复元素返回</param>
+        /// <returns></returns>
+        public string[] ZRandMember(string key, int count, bool repetition) => Call("ZRANDMEMBER".InputKey(key, repetition ? -count : count), rt => rt.ThrowOrValue<string[]>());
+
+        /// <summary>
+        /// 随机返回N个元素, 包含分数
+        /// <para>Redis 6.2.0+以上才支持该命令</para>
+        /// </summary>
+        /// <param name="key">Key</param>
+        /// <param name="count">返回的个数</param>
+        /// <param name="repetition">是否允许有重复元素返回</param>
+        /// <returns></returns>
+        public ZMember[] ZRandMemberWithScores(string key, int count, bool repetition) => Call("ZRANDMEMBER"
+            .InputKey(key, repetition ? -count : count)
+            .Input("WITHSCORES"), rt => rt
+            .ThrowOrValue((a, _) => a == null || a.Length == 0 ? new ZMember[0] : a.MapToHash<decimal>(rt.Encoding).Select(b => new ZMember(b.Key, b.Value)).ToArray()));
     }
 
     public class ZMember