Custom find with date range to handle table partition

active_record.rb

module ActiveRecord
  class Base
    class << self
      #
      # SELECT  "payments".* 
      # FROM "payments"  
      # WHERE "payments"."id" = <id> 
      #   AND ("payments"."created_at" BETWEEN '<start month>' AND '<end month>')  
      # ORDER BY "payments"."id" ASC LIMIT 1
      #
      def find_between id, date
        date = date.to_datetime || Time.zone.now
        beginning_month = date.beginning_of_month
        end_month = date.end_of_month

        where(id: id)
          .where(created_at: beginning_month..end_month)
          .first
      end
    end
  end
end