Prefer returning chainable ActiveRecord objects

matt swanson - Mar 18 '21 - - Dev Community

One of the best parts about ActiveRecord is the chainable query interface:

Post.includes(:comments)
  .where(published: true)
  .where(author: Current.user)
  .order(:name)
Enter fullscreen mode Exit fullscreen mode

To take advantage of this strength and give you flexibility in your code, always try to return chainable objects when querying data.

Usage

It’s common to extract complex queries as your application grows.

class SpecialOffer

  def self.find_eligible_products(store, shopper)
    return [] if store.restricted?

    store.products
      .where('price >= ?', 100)
      .select{ |p| shopper.can_order?(p) }
  end
end

@products = SpecialOffer.find_eligible_products(store, shopper)
#=> [#<Product:0x00007fb1719b7ec0>, #<Product:0x00007fb174744de8>, ...]
Enter fullscreen mode Exit fullscreen mode

While this code may work, what happens if you need to order the @products in a certain way? Or add additional logic? Or lazy-load some associations?

In this case, the return type of our SpecialOffer method are arrays. We would have to switch to using Ruby array methods like sort and select and maybe accidentally introduce an N+1 bug if we need more data.

Let’s refactor this code to make it return chainable objects.

class SpecialOffer

  def self.find_eligible_products(store, shopper)
    return Product.none if store.restricted?

    product_ids = store.products
      .where('price >= ?', 100)
      .select{ |p| shopper.can_order?(p) }
      .map(&:id)

    Product.where(id: product_ids)
  end
end

@products = SpecialOffer.find_eligible_products(store, shopper)
#=> Product::ActiveRecord_Relation
Enter fullscreen mode Exit fullscreen mode

First, we make use of the none query method: this returns an empty (but still chainable!) result. You can call ActiveRecord methods like order, includes, or where on this empty relation and it will simply return no results.

Second, instead of returning the result of our complex product query directly, we collect up the right products and then return “fresh” results for just those ids. While this does incur an additional database query, we can also manipulate the results as needed.

If we want to sort the results or load an association, we can do it in the database and not be worried about any existing conditions that were run as part of the computations.

@products = SpecialOffer.find_eligible_products(store, shopper)
  .includes(:variants)
  .order(:price)

@products = SpecialOffer.find_eligible_products(store, shopper)
  .joins(:sales)
  .where("sales.count > 15")
  .order(:sku)
Enter fullscreen mode Exit fullscreen mode

I’ve found this pattern to be extremely helpful for pulling out complex queries, while still maintaining flexibility to massage the data into the correct shape.

Additional Resources

Rails API: ActiveRecord::QueryMethods#none

Rails Docs: Active Record Query Interface


. . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . .