ruby-on-rails - 带有锁定的 Rails find_in_batches

Question

我需要批量处理大量记录。并且每个批处理都应该在它自己的事务中处理。有没有办法把每一个batch包裹在transaction中，同时锁定batch的所有记录？

Model.scheduled_now.lock.find_in_batches do |batch|
  model_ids = batch.map(&:id)
  Model.update_all({status: 'delivering'}, {"id IN (?)" , model_ids})

  # creates and updates other DB records
  # and triggers background job
  perform_delivery_actions(batch)
end

此示例中的 SELECT FOR UPDATE 是否在每批之后提交事务？或者我需要在每个批处理中手动放置内部事务 block 和锁定记录(这意味着多一个查询)？

我不想放置外部事务 block 的原因是我想分别提交每个批处理，而不是一次提交整个事务。

Answer 1

我最终实现了自己的 find_in_batches_with_lock:

def find_in_batches_with_lock(scope, user, batch_size: 1000)
  last_processed_id = 0
  records = []
  begin
    user.transaction do
      records = scope.where("id > ?", last_processed_id)
                     .order(:id).limit(batch_size).lock.all
      next if records.empty?
      yield records
      last_processed_id = records.last.id
    end
  end while !records.empty?
end