Support expires_in for list

lib/kredis/types.rb

@@ -61,8 +61,8 @@ def hash(key, typed: :string, default: nil, config: :shared, after_change: nil)
     type_from(Hash, config, key, after_change: after_change, default: default, typed: typed)
   end
 
-  def list(key, default: nil, typed: :string, config: :shared, after_change: nil)
-    type_from(List, config, key, after_change: after_change, default: default, typed: typed)
+  def list(key, default: nil, typed: :string, config: :shared, after_change: nil, expires_in: nil)
+    type_from(List, config, key, after_change: after_change, default: default, typed: typed, expires_in: expires_in)
   end
 
   def unique_list(key, default: nil, typed: :string, limit: nil, config: :shared, after_change: nil)

lib/kredis/types/list.rb

@@ -3,9 +3,9 @@
 class Kredis::Types::List < Kredis::Types::Proxying
   prepend Kredis::DefaultValues
 
-  proxying :lrange, :lrem, :lpush, :ltrim, :rpush, :exists?, :del
+  proxying :lrange, :lrem, :lpush, :ltrim, :rpush, :exists?, :del, :expire, :ttl
 
-  attr_accessor :typed
+  attr_accessor :typed, :expires_in
 
   def elements
     strings_to_types(lrange(0, -1) || [], typed)
@@ -17,11 +17,20 @@ def remove(*elements)
   end
 
   def prepend(*elements)
-    lpush types_to_strings(elements, typed) if elements.flatten.any?
+    return if elements.flatten.empty?
+
+    with_expiration do
+      lpush types_to_strings(elements, typed)
+    end
   end
 
   def append(*elements)
-    rpush types_to_strings(elements, typed) if elements.flatten.any?
+    return if elements.flatten.empty?
+
+
+    with_expiration do
+      rpush types_to_strings(elements, typed)
+    end
   end
   alias << append
 
@@ -37,4 +46,12 @@ def last(n = nil)
     def set_default
       append default
     end
+
+    def with_expiration(&block)
+      result = block.call
+      if expires_in && ttl < 0
+        expire expires_in.to_i
+      end
+      result
+    end
 end

test/types/list_test.rb

@@ -76,6 +76,33 @@ class ListTest < ActiveSupport::TestCase
     assert_equal %w[ 2 3 ], @list.elements
   end
 
+  test "append with expiring list" do
+    @list = Kredis.list "mylist", expires_in: 1.second
+    @list.append(%w[1 2])
+
+    sleep 0.2.seconds
+    @list.append(3)
+
+    sleep 0.3.seconds
+    assert_equal %w[ 1 2 3 ], @list.elements
+
+    sleep 0.6.seconds
+    assert_equal [], @list.elements
+  end
+
+  test "prepend with expiring list" do
+    @list = Kredis.list "mylist", expires_in: 1.second
+    @list.prepend(%w[1 2])
+
+    sleep 0.2.seconds
+    @list.prepend(3)
+
+    sleep 0.3.seconds
+    assert_equal %w[ 3 2 1 ], @list.elements
+
+    sleep 0.6.seconds
+    assert_equal [], @list.elements
+  end
 
   test "default" do
     @list = Kredis.list "mylist", default: %w[ 1 2 3 ]