Support expiration in Set

Author: heka1024

lib/kredis/types.rb

@@ -69,8 +69,8 @@ def unique_list(key, default: nil, typed: :string, limit: nil, config: :shared,
     type_from(UniqueList, config, key, after_change: after_change, default: default, typed: typed, limit: limit)
   end
 
-  def set(key, default: nil, typed: :string, config: :shared, after_change: nil)
-    type_from(Set, config, key, after_change: after_change, default: default, typed: typed)
+  def set(key, default: nil, typed: :string, config: :shared, after_change: nil, expires_in: nil)
+    type_from(Set, config, key, after_change: after_change, default: default, typed: typed, expires_in: expires_in)
   end
 
   def ordered_set(key, default: nil, typed: :string, limit: nil, config: :shared, after_change: nil)

lib/kredis/types/set.rb

@@ -2,6 +2,7 @@
 
 class Kredis::Types::Set < Kredis::Types::Proxying
   prepend Kredis::DefaultValues
+  include Kredis::Expiration
 
   proxying :smembers, :sadd, :srem, :multi, :del, :sismember, :scard, :spop, :exists?, :srandmember, :smove
 
@@ -16,7 +17,11 @@ def members
   alias to_a members
 
   def add(*members)
-    sadd types_to_strings(members, typed) if members.flatten.any?
+    return unless members.flatten.any?
+
+    with_expiration do
+      sadd types_to_strings(members, typed)
+    end
   end
   alias << add
 

test/types/set_test.rb

@@ -2,6 +2,7 @@
 
 require "test_helper"
 require "active_support/core_ext/object/inclusion"
+require "active_support/core_ext/integer"
 
 class SetTest < ActiveSupport::TestCase
   setup { @set = Kredis.set "myset" }
@@ -157,6 +158,18 @@ class SetTest < ActiveSupport::TestCase
     assert_equal [ 1, 2, 3, 5, 6, 7 ], @set.members
   end
 
+  test "add with expiration" do
+    @set = Kredis.set "mylist", typed: :integer, expires_in: 1.second
+    @set.add(%w[ 1 2 3 ])
+
+    sleep 0.7.seconds
+    @set.add(%w[ 4 5 ])
+    assert_equal [1, 2, 3, 4, 5], @set.members
+
+    sleep 0.5.seconds
+    assert_equal [], @set.members
+  end
+
   test "remove with default" do
     @set = Kredis.set "mylist", default: -> () { %w[ 1 2 3 4 ] }
     @set.remove(%w[ 2 3 ])