feat: add automatic timestamp handling

Author: sncalvo

lib/staging_table/bulk_inserter.rb

@@ -18,6 +18,8 @@ def insert(records)
         raise RecordError, "All records must be hashes. If passing ActiveRecord objects, use Session#insert which normalizes them automatically."
       end
 
+      records = apply_timestamps(records)
+
       columns = records.first.keys.map(&:to_s)
       quoted_columns = columns.map { |c| connection.quote_column_name(c) }.join(", ")
       quoted_table = connection.quote_table_name(model.table_name)
@@ -34,6 +36,32 @@ def insert(records)
 
     private
 
+    TIMESTAMP_COLUMNS = %w[created_at updated_at].freeze
+
+    def apply_timestamps(records)
+      return records if records.empty?
+
+      sample_record = records.first
+      model_columns = model.column_names
+
+      missing_timestamps = TIMESTAMP_COLUMNS.select do |col|
+        model_columns.include?(col) && !record_has_key?(sample_record, col)
+      end
+
+      return records if missing_timestamps.empty?
+
+      now = Time.current
+      records.map do |record|
+        record = record.dup
+        missing_timestamps.each { |col| record[col.to_sym] = now }
+        record
+      end
+    end
+
+    def record_has_key?(record, key)
+      record.key?(key.to_sym) || record.key?(key.to_s)
+    end
+
     def connection
       model.connection
     end

sig/staging_table/bulk_inserter.rbs

@@ -10,6 +10,10 @@ module StagingTable
 
     private
 
+    TIMESTAMP_COLUMNS: Array[String]
+
+    def apply_timestamps: (Array[Hash[String | Symbol, untyped]] records) -> Array[Hash[String | Symbol, untyped]]
+    def record_has_key?: (Hash[String | Symbol, untyped] record, String key) -> bool
     def connection: () -> ActiveRecord::ConnectionAdapters::AbstractAdapter
     def quote: (untyped value) -> String
   end

spec/staging_table/bulk_inserter_spec.rb

@@ -85,6 +85,47 @@
         user = staging_model.first
         expect(user.created_at).to be_within(1.second).of(now)
       end
+
+      it "automatically sets created_at and updated_at when not provided" do
+        records = [
+          {name: "John", email: "john@example.com"}
+        ]
+
+        freeze_time = Time.current
+        allow(Time).to receive(:current).and_return(freeze_time)
+
+        inserter.insert(records)
+
+        user = staging_model.first
+        expect(user.created_at).to be_within(1.second).of(freeze_time)
+        expect(user.updated_at).to be_within(1.second).of(freeze_time)
+      end
+
+      it "does not override provided timestamps" do
+        custom_time = 1.day.ago
+        records = [
+          {name: "John", email: "john@example.com", created_at: custom_time, updated_at: custom_time}
+        ]
+
+        inserter.insert(records)
+
+        user = staging_model.first
+        expect(user.created_at).to be_within(1.second).of(custom_time)
+        expect(user.updated_at).to be_within(1.second).of(custom_time)
+      end
+
+      it "handles string keys for timestamps" do
+        custom_time = 2.days.ago
+        records = [
+          {"name" => "John", "email" => "john@example.com", "created_at" => custom_time, "updated_at" => custom_time}
+        ]
+
+        inserter.insert(records)
+
+        user = staging_model.first
+        expect(user.created_at).to be_within(1.second).of(custom_time)
+        expect(user.updated_at).to be_within(1.second).of(custom_time)
+      end
     end
   end