Set flag expiration on attribute specification (#52)

Author: dhh

lib/kredis/attributes.rb

@@ -22,8 +22,8 @@ def kredis_datetime(name, key: nil, config: :shared, after_change: nil, expires_
       kredis_connection_with __method__, name, key, config: config, after_change: after_change, expires_in: expires_in
     end
 
-    def kredis_flag(name, key: nil, config: :shared, after_change: nil)
-      kredis_connection_with __method__, name, key, config: config, after_change: after_change
+    def kredis_flag(name, key: nil, config: :shared, after_change: nil, expires_in: nil)
+      kredis_connection_with __method__, name, key, config: config, after_change: after_change, expires_in: expires_in
 
       define_method("#{name}?") do
         send(name).marked?

lib/kredis/types.rb

@@ -47,8 +47,8 @@ def cycle(key, values:, expires_in: nil, config: :shared, after_change: nil)
     type_from(Cycle, config, key, after_change: after_change, values: values, expires_in: expires_in)
   end
 
-  def flag(key, config: :shared, after_change: nil)
-    type_from(Flag, config, key, after_change: after_change)
+  def flag(key, config: :shared, after_change: nil, expires_in: nil)
+    type_from(Flag, config, key, after_change: after_change, expires_in: expires_in)
   end
 
   def enum(key, values:, default:, config: :shared, after_change: nil)

lib/kredis/types/flag.rb

@@ -1,8 +1,10 @@
 class Kredis::Types::Flag < Kredis::Types::Proxying
   proxying :set, :exists?, :del
 
+  attr_accessor :expires_in
+
   def mark(expires_in: nil)
-    set 1, ex: expires_in
+    set 1, ex: expires_in || self.expires_in
   end
 
   def marked?

test/attributes_test.rb

@@ -11,6 +11,7 @@ class Person
   kredis_list :names_with_custom_key, key: ->(p) { "person:#{p.id}:names_customized" }
   kredis_unique_list :skills, limit: 2
   kredis_flag :special
+  kredis_flag :temporary_special, expires_in: 1.second
   kredis_string :address
   kredis_integer :age
   kredis_decimal :salary
@@ -236,4 +237,11 @@ def suddenly_implemented_person.id; 8; end
       sleep 1.1.seconds
     end
   end
+
+  test "expiring flag" do
+    @person.temporary_special.mark
+    assert_changes "@person.temporary_special.marked?", from: true, to: false do
+      sleep 1.1.seconds
+    end
+  end
 end