Add custom database role and poc to avoid comparing integer ids as strings

Author: etravassos

lib/job-iteration/active_record_cursor.rb

@@ -55,26 +55,37 @@ def position=(position)
     def update_from_record(record)
       self.position = @columns.map do |column|
         method = column.to_s.split(".").last
-
-        if ActiveRecord.version >= Gem::Version.new("7.1.0.alpha") && method == "id"
-          record.id_value
-        else
-          record.send(method.to_sym)
-        end
+        record.send(method.to_sym)
       end
     end
 
-    def next_batch(batch_size)
+    def next_batch(batch_size, database_role: nil)
       return if @reached_end
 
       relation = @base_relation.limit(batch_size)
 
       if (conditions = self.conditions).any?
-        relation = relation.where(*conditions)
+        column = @columns.first.split(".").first
+        model = column.gsub(/_/, " ").split.map(&:capitalize).join("::").singularize.constantize
+
+        attribute = @columns.first.split(".").second.to_sym
+
+        relation = relation.where(model.arel_table[attribute].gt(@position.first))
+        # model.id > id
+
+        # relation = relation.where(*conditions)
+        # model.id > 'id'
+
       end
 
       records = relation.uncached do
-        relation.to_a
+        if database_role.present?
+          ActiveRecord::Base.connected_to(role: database_role) do
+            relation.to_a
+          end
+        else
+          relation.to_a
+        end
       end
 
       update_from_record(records.last) unless records.empty?
@@ -93,6 +104,7 @@ def conditions
       else
         "#{column} >= ?"
       end
+
       while i > 0
         i -= 1
         column = @columns[i]

lib/job-iteration/active_record_enumerator.rb

@@ -7,9 +7,10 @@ module JobIteration
   class ActiveRecordEnumerator
     SQL_DATETIME_WITH_NSEC = "%Y-%m-%d %H:%M:%S.%N"
 
-    def initialize(relation, columns: nil, batch_size: 100, cursor: nil)
+    def initialize(relation, columns: nil, batch_size: 100, cursor: nil, database_role: nil)
       @relation = relation
       @batch_size = batch_size
+      @database_role = database_role
       @columns = if columns
         Array(columns)
       else
@@ -31,7 +32,7 @@ def records
     def batches
       cursor = finder_cursor
       Enumerator.new(method(:size)) do |yielder|
-        while (records = cursor.next_batch(@batch_size))
+        while (records = cursor.next_batch(@batch_size, database_role: @database_role))
           yielder.yield(records, cursor_value(records.last)) if records.any?
         end
       end