lib/id/timestamps.rb in id-0.0.12 vs lib/id/timestamps.rb in id-0.1

@@ -1,20 +1,20 @@
-module Id
-  module Timestamps
-    def self.included(base)
-      base.field :created_at
-      base.field :updated_at
-    end
+module Id::Timestamps
+  def self.included(base)
+    base.field :created_at
+    base.field :updated_at
+  end
 
-    def initialize(data = {})
-      super data.merge(:created_at => data.fetch('created_at', Time.now))
-    end
+  def initialize(_data = {})
+    now = Time.now
+    super ({ created_at: now, updated_at: now }).merge(_data)
+  end
 
-    def set(values)
-      self.class.new(super.data.merge(:updated_at => Time.now))
-    end
+  def set(update)
+    super update.merge(updated_at: Time.now)
+  end
 
-    def unset(*keys)
-      self.class.new(super.data.merge(:updated_at => Time.now))
-    end
+  def unset(update)
+    super.set({})
   end
+
 end