Store sample of sql when ActiveRecord::QueryCanceled occured

application_job.rb

class ApplicationJob < ActiveJob::Base
  include ActiveSupport::Rescuable
  # PG::QueryCanceled will be wrapped with ActiveRecord::QueryCanceled errors and it's a direct
  # descendant of ActiveRecord::StatementInvalid error.
  rescue_from ActiveRecord::StatementInvalid do |exception|
    # It's hard to know, on which queries we're seeing statement timeouts -- so we are making this explicit in AppSignal error.
    Appsignal::Transaction.current.set_sample_data("custom_data", sql_statement: exception.sql)

    raise exception
  end
end