redis: ZMSCORE

Author: Roman Janusz

commons-redis/src/main/scala/com/avsystem/commons/redis/RedisCommand.scala

@@ -189,6 +189,9 @@ abstract class RedisScanCommand[T](decoder: ReplyDecoder[Seq[T]])
 abstract class RedisSeqCommand[T](elementDecoder: ReplyDecoder[T])
   extends AbstractRedisCommand[Seq[T]](multiBulkAsSeq(elementDecoder))
 
+abstract class RedisSeqOptCommand[T](elementDecoder: ReplyDecoder[T])
+  extends AbstractRedisCommand[Seq[Opt[T]]](multiBulkAsSeq(nullBulkOr(elementDecoder)))
+
 abstract class RedisOptSeqCommand[T](elementDecoder: ReplyDecoder[T])
   extends AbstractRedisCommand[Opt[Seq[T]]](nullMultiBulkOr(multiBulkAsSeq(elementDecoder)))
 

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

@@ -127,6 +127,14 @@ trait SortedSetsApi extends ApiSubset {
   def zlexcount(key: Key, min: LexLimit[Value] = LexLimit.MinusInf, max: LexLimit[Value] = LexLimit.PlusInf): Result[Long] =
     execute(new Zlexcount(key, min, max))
 
+  /** Executes [[http://redis.io/commands/zmscore ZMSCORE]] */
+  def zmscore(key: Key, members: Value*): Result[Seq[Opt[Double]]] =
+    execute(new Zmscore(key, members))
+
+  /** Executes [[http://redis.io/commands/zmscore ZMSCORE]] */
+  def zmscore(key: Key, members: Iterable[Value]): Result[Seq[Opt[Double]]] =
+    execute(new Zmscore(key, members))
+
   /** Executes [[http://redis.io/commands/zpopmax ZPOPMAX]] */
   def zpopmax(key: Key): Result[Opt[(Value, Double)]] =
     execute(new Zpopmax(key, Opt.Empty).map(_.headOpt))
@@ -366,6 +374,11 @@ trait SortedSetsApi extends ApiSubset {
     val encoded: Encoded = encoder("ZPOPMIN").key(key).optAdd(count).result
   }
 
+  private final class Zmscore(key: Key, members: Iterable[Value])
+    extends RedisSeqOptCommand[Double](bulkAsDouble) with NodeCommand {
+    val encoded: Encoded = encoder("ZMSCORE").key(key).datas(members).result
+  }
+
   private final class Zpopmax(key: Key, count: Opt[Long])
     extends AbstractValuesWithScoresCommand with NodeCommand {
     val encoded: Encoded = encoder("ZPOPMAX").key(key).optAdd(count).result

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

@@ -115,6 +115,11 @@ trait SortedSetsApiSuite extends CommandsSuite {
     zlexcount("key", LexLimit.incl("a"), LexLimit.excl("c")).assertEquals(2)
   }
 
+  apiTest("ZMSCORE") {
+    setup(zadd("key", "a" -> 1.0, "b" -> 2.0, "c" -> 3.0, "d" -> 4.0))
+    zmscore("key", "a", "b", "e").assertEquals(Seq(Opt(1.0), Opt(2.0), Opt.Empty))
+  }
+
   apiTest("ZPOPMAX") {
     setup(zadd("key", "lol" -> 1.0, "fuu" -> 2.0, "bar" -> 3.0, "fag" -> 4.0))
     zpopmax("???").assertEquals(Opt.Empty)