ActiveRecord find_each combined with limit and order

后端 未结 13 2086
温柔的废话
温柔的废话 2020-12-02 11:41

I\'m trying to run a query of about 50,000 records using ActiveRecord\'s find_each method, but it seems to be ignoring my other parameters like so:



        
13条回答
  •  失恋的感觉
    2020-12-02 12:45

    The documentation says that find_each and find_in_batches don't retain sort order and limit because:

    • Sorting ASC on the PK is used to make the batch ordering work.
    • Limit is used to control the batch sizes.

    You could write your own version of this function like @rorra did. But you can get into trouble when mutating the objects. If for example you sort by created_at and save the object it might come up again in one of the next batches. Similarly you might skip objects because the order of results has changed when executing the query to get the next batch. Only use that solution with read only objects.

    Now my primary concern was that I didn't want to load 30000+ objects into memory at once. My concern was not the execution time of the query itself. Therefore I used a solution that executes the original query but only caches the ID's. It then divides the array of ID's into chunks and queries/creates the objects per chunk. This way you can safely mutate the objects because the sort order is kept in memory.

    Here is a minimal example similar to what I did:

    batch_size = 512
    ids = Thing.order('created_at DESC').pluck(:id) # Replace .order(:created_at) with your own scope
    ids.each_slice(batch_size) do |chunk|
        Thing.find(chunk, :order => "field(id, #{chunk.join(',')})").each do |thing|
          # Do things with thing
        end
    end
    

    The trade-offs to this solution are:

    • The complete query is executed to get the ID's
    • An array of all the ID's is kept in memory
    • Uses the MySQL specific FIELD() function

    Hope this helps!

提交回复
热议问题