Run callback after records load with ActiveRecord::Association

Hi all,

I was wondering if there is a way to run a hook after records load with ActiveRecord::Relation.

I want to write a gem for including complex data in a rails query, for example

User.includes(:sales).special_includes(:profits)

class User < ActiveRecord::Base

special_includes :profits do |result_set|

sales = Sale.select(“user_id, SUM(profit)”).group(:user_id).where(user_id: result_set.map(&:user_id)).to_h

result_set.each do |result|

result.profits = sales[result.user_id]

end

end

end

I want to do this to improve performance on complex pages with too much legacy code to completely refactor