redis: ZRANDMEMBER

Author: Roman Janusz

commons-redis/src/main/scala/com/avsystem/commons/redis/commands/sortedSets.scala

@@ -151,6 +151,18 @@ trait SortedSetsApi extends ApiSubset {
   def zpopmin(key: Key, count: Long): Result[Seq[(Value, Double)]] =
     execute(new Zpopmin(key, Opt(count)))
 
+  /** Executes [[http://redis.io/commands/zrandmember ZRANDMEMBER]] */
+  def zrandmember(key: Key): Result[Opt[Value]] =
+    execute(new Zrandmember(key))
+
+  /** Executes [[http://redis.io/commands/zrandmember ZRANDMEMBER]] */
+  def zrandmember(key: Key, count: Int, distinct: Boolean = true): Result[Seq[Value]] =
+    execute(new ZrandmemberCount(key, if (distinct) count else -count))
+
+  /** Executes [[http://redis.io/commands/zrandmember ZRANDMEMBER]] */
+  def zrandmemberWithscores(key: Key, count: Int, distinct: Boolean = true): Result[Seq[(Value, Double)]] =
+    execute(new ZrandmemberWithscores(key, if (distinct) count else -count))
+
   /** Executes [[http://redis.io/commands/zrange ZRANGE]] */
   def zrange(key: Key, start: Long = 0, stop: Long = -1): Result[Seq[Value]] =
     execute(new Zrange(key, start, stop))
@@ -384,6 +396,21 @@ trait SortedSetsApi extends ApiSubset {
     val encoded: Encoded = encoder("ZPOPMAX").key(key).optAdd(count).result
   }
 
+  private final class Zrandmember(key: Key)
+    extends RedisOptDataCommand[Value] with NodeCommand {
+    val encoded: Encoded = encoder("ZRANDMEMBER").key(key).result
+  }
+
+  private final class ZrandmemberCount(key: Key, count: Int)
+    extends RedisDataSeqCommand[Value] with NodeCommand {
+    val encoded: Encoded = encoder("ZRANDMEMBER").key(key).add(count).result
+  }
+
+  private final class ZrandmemberWithscores(key: Key, count: Int)
+    extends AbstractValuesWithScoresCommand with NodeCommand {
+    val encoded: Encoded = encoder("ZRANDMEMBER").key(key).add(count).add("WITHSCORES").result
+  }
+
   private final class Zrange(key: Key, start: Long, stop: Long)
     extends AbstractZrange[Value]("ZRANGE", multiBulkAsSeqOf[Value])(key, start, stop, withscores = false)
 

commons-redis/src/test/scala/com/avsystem/commons/redis/commands/SortedSetsApiSuite.scala

@@ -136,6 +136,13 @@ trait SortedSetsApiSuite extends CommandsSuite {
     zpopmin("key", 2).assertEquals(Seq("fuu" -> 2.0, "bar" -> 3.0))
   }
 
+  apiTest("ZRANDMEMBER") {
+    setup(zadd("key", "lol" -> 1.0, "fuu" -> 2.0, "bar" -> 3.0, "fag" -> 4.0))
+    zrandmember("key").assert(_.isDefined)
+    zrandmember("key", count = 2).assert(_.size == 2)
+    zrandmemberWithscores("key", count = 2).assert(_.size == 2)
+  }
+
   apiTest("ZRANGE") {
     setup(zadd("key", "lol" -> 1.0, "fuu" -> 2.0))
     zrange("???").assertEquals(Seq.empty)