fin blog embed

fin_active_record_extensions.rb

    Contract Integer, Func[Any => Any] => Any
    public def log_query_count(num_trace_lines=1)
      # Initialize our counter
      ActiveRecordQueryCounter.instance.reset

      # Define a callback to be called each time we encounter a query
      # event. The callback will print out the trace
      callback = lambda do |_name, _start, _finish, _id, payload|
        if payload[:name] != 'SCHEMA'
          Rails.logger.info(tag: 'log_query_count', message: "Query # #{ActiveRecordQueryCounter.instance.query_count}: #{payload[:name]}")
          ActiveRecordQueryCounter.instance.trace(num_trace_lines).each do |line|
            Rails.logger.info(tag: 'log_query_count', message: line)
          end
        end
      end
      
      Rails.logger.info(tag: 'log_query_count', message: "Starting to count queries, #{ActiveRecordQueryCounter.instance.query_count} so far")
      
      # Subscribe the block to query events, and hook up the events to
      # the callback we defined above
      result = ActiveSupport::Notifications.subscribed(callback, 'sql.active_record') { yield }

      Rails.logger.info(tag: 'log_query_count', message: "Total query count: #{ActiveRecordQueryCounter.instance.query_count}")

      # Remember to return the result from our block
      result
    end