Rails: How find array of ids with .where but sorted?

user_7602290 0
Viewed 725

I'm attempting to lookup a list of stores using .where but I'm also trying to keep them sorted as the same array of ids.

i.e

  ids = ["4", "15", "10", "20", "1"]
  stores = Store.published.where(id: ids)

It looks like stores is being returned in ascending order of the id so like [{id: 20}, {id: 15} {id: 10}, {id: 4}, {id: 1}]. I'd like to keep the returned stores ordered in the same way that ids is ordered. Also note the ids in each store are ints whereas the ids in the ids array are strings.

ruby-on-rails ruby
3 Answers

Because the numbers are the IDs of the records you can simply use:

ids = ["4", "15", "10", "20", "1"]
stores = Store.find(ids)

This works because find accepts a list of IDs too and when a list is provided then find will return the records in the same order as in the list.

And in Ruby on Rails 7.0 you will be able to use where in combination with in_order_of which has the benefit of not raising an error if a record is not found and that you can order by other columns and their values too:

ids = ["4", "15", "10", "20", "1"]
stores = Store.where(id: ids).in_order_of(:id, ids)
user_24833130

When doing a where(id: ids) query the database is simply running a query limiting the results to those ID's. The database won't return the results based on the ordering of the ids you give it, but you can explicitly order database results by id (or any other attribute) with where(id: ids).order('id desc') to return in decreasing order by id. Another option you can use ActiveRecord#find with multiple ids and it will return the results in the order you request them:

Store.published.find(ids)

However where will silently ignore any id's that are not found, whereas find will fail if any id passed in is missing.

user_30806100

Figured it out. I wanted to stick with using .where() so it wouldn't throw an error if the id didn't exist. This solution worked for me...

sorted_stores = stores.sort_by { |store| ids.index(store[:id]) }
user_76022900
Related