Add ZCOUNT.

ezmobius · Jul 13, 2010 · 1dcf6a8 · 1dcf6a8
1 parent 8fa2151
commit 1dcf6a8
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/lib/redis.rb b/lib/redis.rb
@@ -284,6 +284,10 @@ def zrangebyscore(key, min, max, options = {})
     @client.call(:zrangebyscore, key, min, max, *command.to_a)
   end
 
+  def zcount(key, start, stop)
+    @client.call(:zcount, key, start, stop)
+  end
+
   def zrevrange(key, start, stop, options = {})
     command = CommandOptions.new(options) do |c|
       c.bool :with_scores

diff --git a/test/redis_test.rb b/test/redis_test.rb
@@ -898,6 +898,14 @@ class RedisTest < Test::Unit::TestCase
       assert_equal ["s3", "3"], @r.zrangebyscore("foo", 2, 4, :limit => [1, 1], :with_scores => true)
     end
 
+    test "ZCOUNT" do
+      @r.zadd "foo", 1, "s1"
+      @r.zadd "foo", 2, "s2"
+      @r.zadd "foo", 3, "s3"
+
+      assert_equal 2, @r.zcount("foo", 2, 3)
+    end
+
     test "ZCARD" do
       assert_equal 0, @r.zcard("foo")