query_tracker 0.1.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: f36bdfeaac971e0d3feb504cc6b99bd1ced63405f081bee5f15a104a7edc21d7
+  data.tar.gz: 519e7814892c3ba01e71cacc2c03591fd2657a85725582090ed90b71aa133734
+SHA512:
+  metadata.gz: 56026f5ad302e525ba269ee5ff6bb88287a46b6777c14ac526df5de6903864301ada972fb03602b52ae2b704ef376a3bb06f083c294860c9d6d510bc7ba5f20c
+  data.tar.gz: a1c346e6b192ad8f53dadadd7edc3473fa96f83ac4c673d64dcb617c7ffcba1c476daead1ecf9995420218231df1ef5d4ad97d3b3bd6057615b0b3d6452c43cc

data/README.md

@@ -0,0 +1,103 @@
+# QueryTracker
+
+`QueryTracker` is a Ruby gem designed to help you track and analyze SQL queries executed by your ActiveRecord models. By subscribing to ActiveSupport notifications, it provides detailed insights into the queries being run, including the tables involved and the locations in your code where the queries are generated.
+There are three things this gem allows you to do
+
+1. You can compare two codes to view the difference in SQL counts on locations with a graph or a table.
+2. You can view an overview of the SQL that a code does in a graph, a table or in the console.
+3. (In progress) You can see an overview of the current request on a controller action
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'query_tracker'
+```
+
+And then execute:
+
+```bash
+    bundle install
+```
+
+Or install it yourself as:
+
+```bash
+    gem install query_tracker
+```
+
+## Usage
+
+There are three ways of using this gem:
+
+1. With a block of code
+
+```ruby
+require 'query_tracker'
+QueryTracker.start_with_block(printer: :html) do
+    # your code goes here
+end
+```
+
+this will open up a html table with the SQL stats of your code
+
+2. Starting recording manually
+
+```ruby
+require 'query_tracker'
+
+QueryTracker.start_recording
+# your code goes here
+QueryTracker.end_recording(printer: :html)
+```
+
+3. Comparing two blocks of code (only available for html printer)
+
+```ruby
+require 'query_tracker'
+QueryTracker.compare do |bench|
+    bench.code('script1') do
+    end
+    bench.code('script2') do
+    end
+    bench.compare!
+end
+```
+
+this will open up a graph comparing the quantity of SQL of the two codes
+
+4. (In progress) Enabling a middleware to see an overview of the current request SQL's in Rack Application.
+    On `config/development.rb` or the initializer of the application.
+
+```ruby
+  config.after_initialize do
+    QueryTracker.configure do |configuration|
+      configuration.enable_middleware = true
+      configuration.ignore_table_count = 1
+      configuration.max_locations_per_table = 4
+    end
+  end
+```
+
+### Printing options
+
+There are two ways of displaying the collected queries data, `:console` and `:html`, to select one pass the printer argument to `start_with_block` o `end_recording` methods.
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at <https://github.com/[USERNAME]/query_tracker>. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/[USERNAME]/query_tracker/blob/master/CODE_OF_CONDUCT.md).
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+## Code of Conduct
+
+Everyone interacting in the QueryTracker project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/[USERNAME]/query_tracker/blob/master/CODE_OF_CONDUCT.md).

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/query_tracker/compare/comparator.rb

@@ -0,0 +1,33 @@
+module QueryTracker
+  module Compare
+    class Comparator
+      attr_accessor :results, :scripts_loaded
+
+      def initialize
+        @results = {}
+        @scripts_loaded = 0
+      end
+
+      def code(name)
+        @scripts_loaded += 1
+
+        QueryTracker.start_with_block(printer: :none) do
+          yield
+          @results[name] = QueryTracker.tracker.query_tracker.dup
+        end
+      end
+
+      def compare!
+        raise 'Exactly two code blocks are required' if @scripts_loaded != 2
+
+        QueryTracker::Printer::HtmlCompare.new(data_1: results.slice(results.keys[0]),
+                                               data_2: results.slice(results.keys[1])).print
+      end
+    end
+
+    def self.compare
+      comparator = Comparator.new
+      yield(comparator)
+    end
+  end
+end

data/lib/query_tracker/configuration.rb

@@ -0,0 +1,12 @@
+require 'active_support'
+require 'active_support/core_ext/module/attribute_accessors'
+
+module QueryTracker
+  module Configuration
+    mattr_accessor :ignore_table_count, :max_locations_per_table, :enable_middleware
+
+    self.ignore_table_count = ENV['QUERY_COUNTER_IGNORE_TABLE_COUNT'] || 10
+    self.max_locations_per_table = ENV['MAX_LOCATIONS_PER_TABLE'] || 3
+    self.enable_middleware = false
+  end
+end

data/lib/query_tracker/middleware.rb

@@ -0,0 +1,37 @@
+require 'json'
+require 'nokogiri'
+
+module QueryTracker
+  class Middleware
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      return @app.call(env) unless Configuration.enable_middleware
+
+      status, headers, response, query_count_data = nil
+      QueryTracker.start_recording
+      status, headers, response = @app.call(env)
+      query_count_data = QueryTracker.tracker.query_tracker.clone
+      QueryTracker.end_recording(printer: :none)
+      if headers['Content-Type']&.include?('text/html') && response.respond_to?(:body) && response.body['<body']
+        response_body = inject_query_counter_js(response.body, query_count_data)
+        headers['Content-Length'] = response_body.bytesize.to_s
+        response = [response_body]
+      end
+      [status, headers, response]
+    end
+
+    def inject_query_counter_js(response_body, query_count_data)
+      injected_html = ::QueryTracker::Printer::Html.new(data: query_count_data).inject_in_html
+      # Parse the response body with Nokogiri
+      doc = Nokogiri::HTML(response_body)
+
+      # Inject the query counter JS before the closing </body> tag
+      doc.at('body').add_child(injected_html)
+
+      doc.to_html
+    end
+  end
+end

data/lib/query_tracker/printer/base.rb

@@ -0,0 +1,42 @@
+module QueryTracker
+  module Printer
+    class Base
+      def filter_data data
+        data = data.select { |_, v| v[:count] >= Configuration.ignore_table_count }
+        data = data.sort_by { |_, v| -v[:count] }.each do |_category, info|
+          info[:location] = info[:location].sort_by do |_, detail|
+            -detail[:count]
+          end.first(Configuration.max_locations_per_table).to_h
+        end
+        data.to_h
+      end
+
+      def sort_data data
+        data = data.sort_by { |_, v| -v[:count] }.each do |_category, info|
+          info[:location] = info[:location].sort_by do |_, detail|
+            -detail[:count]
+          end.to_h
+        end
+        data.to_h
+      end
+
+      def open_file html_dest
+        if ENV['WSL_DISTRIBUTION']
+          Launchy.open("file://wsl%24/#{ENV['WSL_DISTRIBUTION']}#{html_dest}")
+        else
+          Launchy.open(html_dest)
+        end
+      end
+
+      # this js is not used as erb, it could be changed
+      def js_content
+        File.read(@js_path)
+      end
+
+      # this js is not used as erb, it could be changed
+      def css_content
+        File.read(@css_path)
+      end
+    end
+  end
+end

data/lib/query_tracker/printer/console.rb

@@ -0,0 +1,30 @@
+require 'colorize'
+
+module QueryTracker
+  module Printer
+    class Console < Base
+      def initialize data:
+        super()
+        @data = data
+      end
+
+      def print
+        data = filter_data(@data)
+        puts '[QueryTracker] Query count per table:'.colorize(:blue)
+        puts "Total query count: #{data.values.sum { |v| v[:count] }}\n\n"
+        puts "All tables with less than #{Configuration.ignore_table_count} queries are ignored. \n\n"
+        puts "For each table, the top #{Configuration.max_locations_per_table} locations with the most queries will be shown.\n\n"
+        data.each do |category, info|
+          puts "Table #{category.colorize(:cyan)}"
+          puts "  Total query count: #{info[:count].to_s.colorize(:blue)}"
+          puts '  Locations where the table was called:'
+          info[:location].each do |loc, details|
+            puts "    - File location: #{loc}"
+            puts "        Query count: #{details[:count].to_s.colorize(:blue)}"
+          end
+          puts
+        end
+      end
+    end
+  end
+end

data/lib/query_tracker/printer/html.rb

@@ -0,0 +1,68 @@
+require 'erb'
+require 'tempfile'
+require 'launchy'
+require 'pry-byebug'
+require 'json'
+
+module QueryTracker
+  module Printer
+    class Html < Base
+      attr_accessor :data
+
+      def initialize data:
+        @template_path = File.join(__dir__, 'templates', 'template.html.erb')
+        @base_query_counter_path = File.join(__dir__, 'templates', 'template_base_query_counter.html.erb')
+        @inject_template_path = File.join(__dir__, 'templates', 'template_for_inject.html.erb')
+        @css_path = File.join(__dir__, 'templates', 'style.css')
+        # este no falla, quizá pueda poner lo mismo aquí con chart.js y así no hacer lo del nonce y csp
+        @js_path = File.join(__dir__, 'templates', 'bar_chart.js')
+        @data = data
+      end
+
+      def chart_data
+        @chart_data ||= generate_chart_data(data)
+      end
+
+      # maybe this should not be filtered
+      def total_query_count
+        @total_query_count ||= data.values.sum { |v| v[:count] }
+      end
+
+      def inject_in_html
+        ERB.new(File.read(@inject_template_path)).result(binding)
+      end
+
+      def render_query_counter_base_div
+        ERB.new(File.read(@base_query_counter_path)).result(binding)
+      end
+
+      def print
+        html_dest = generate_html(binding)
+        open_file(html_dest)
+      end
+
+      private
+
+      def generate_chart_data(data)
+        chart_data = { labels: [], data: [], locations: {} }
+        data.each do |table, info|
+          chart_data[:labels] << table
+          chart_data[:data] << info[:count]
+          chart_data[:locations][table] = info[:location].map do |loc, detail|
+            { location: loc, count: detail[:count] }
+          end
+        end
+        chart_data
+      end
+
+      def generate_html binding
+        template = ERB.new(File.read(@template_path))
+        html_content = template.result(binding)
+        temp_dir = Dir.mktmpdir
+        html_dest = File.join(temp_dir, 'query_counter_report.html')
+        File.write(html_dest, html_content)
+        html_dest
+      end
+    end
+  end
+end

data/lib/query_tracker/printer/html_compare.rb

@@ -0,0 +1,64 @@
+require 'erb'
+require 'tempfile'
+require 'launchy'
+require 'pry-byebug'
+require 'json'
+
+module QueryTracker
+  module Printer
+    class HtmlCompare < Base
+      def initialize data_1:, data_2:
+        super()
+        @template_path = File.join(__dir__, 'templates', 'comparing.html.erb')
+        @css_path = File.join(__dir__, 'templates', 'style.css')
+        @js_path = File.join(__dir__, 'templates', 'bar_chart.js')
+        @script_1_name = data_1.keys.first
+        @script_2_name = data_2.keys.first
+        @data_1 = data_1[@script_1_name]
+        @data_2 = data_2[@script_2_name]
+      end
+
+      def print
+        # used by binding on erb templates
+        data_1 = sort_data(@data_1)
+        data_2 = sort_data(@data_2)
+        tables = data_1.keys | data_2.keys
+        total_query_count_1 = data_1.values.sum { |v| v[:count] }
+        total_query_count_2 = data_2.values.sum { |v| v[:count] }
+        chart_data = generate_chart_data_compare(data_1, data_2)
+        # end
+        html_dest = generate_html(binding)
+        open_file(html_dest)
+      end
+
+      private
+
+      def generate_chart_data_compare(data_1, data_2)
+        labels = (data_1.keys | data_2.keys).sort
+        chart_data = { labels: [], data_1: {}, data_2: {}, locations: {} }
+        chart_data[:data_1][:name] = @script_1_name
+        chart_data[:data_2][:name] = @script_2_name
+        chart_data[:data_1][:data] = []
+        chart_data[:data_2][:data] = []
+        labels.each do |label|
+          chart_data[:labels] << label
+          chart_data[:data_1][:data] << (data_1[label] ? data_1[label][:count] : 0)
+          chart_data[:data_2][:data] << (data_2[label] ? data_2[label][:count] : 0)
+          chart_data[:locations][label] =
+            (data_1[label] ? data_1[label][:location] : {}).merge(data_2[label] ? data_2[label][:location] : {})
+        end
+        chart_data
+      end
+
+      def generate_html binding
+        template = ERB.new(File.read(@template_path))
+        html_content = template.result(binding)
+
+        temp_dir = Dir.mktmpdir
+        html_dest = File.join(temp_dir, 'query_counter_report.html')
+        File.write(html_dest, html_content)
+        html_dest
+      end
+    end
+  end
+end

data/lib/query_tracker/printer/templates/bar_chart.js

@@ -0,0 +1,153 @@
+
+function toggleView() {
+  const chart = document.getElementById('queryCountChart');
+  const chartContainer = chart.parentElement;
+  const table = document.getElementById('queryTable');
+  const button = document.getElementById('toggleButton');
+  if (chartContainer.style.display === 'none') {
+    chartContainer.style.display = 'block';
+    table.style.display = 'none';
+    button.textContent = 'Show Table View';
+  } else {
+    chartContainer.style.display = 'none';
+    table.style.display = 'table';
+    button.textContent = 'Show chartContainer View';
+  };
+};
+
+function toggleColumnContent() {
+  const columnHeader = document.getElementById('columnHeader');
+  const toggleContents = document.querySelectorAll('.toggle-content');
+  const showSQL = columnHeader.textContent === 'File Path';
+
+  columnHeader.textContent = showSQL ? 'SQL' : 'File Path';
+
+  toggleContents.forEach(content => {
+    const filepath = content.querySelector('.filepath');
+    const sql = content.querySelector('.sql');
+
+    if (showSQL) {
+      filepath.style.display = 'none';
+      sql.style.display = 'block';
+    } else {
+      filepath.style.display = 'block';
+      sql.style.display = 'none';
+    }
+  });
+}
+
+function initializeChart(chartData) {
+  console.log(chartData);
+  const ctx = document.getElementById('queryCountChart').getContext('2d');
+  const queryCountChart = new Chart(ctx, {
+    type: 'bar',
+    data: {
+      labels: chartData.labels,
+      datasets: [
+        {
+          label: 'Dataset 1',
+          data: chartData.data,
+          backgroundColor: 'rgba(75, 192, 192, 0.2)',
+          borderColor: 'rgba(75, 192, 192, 1)',
+          borderWidth: 1
+        }
+      ]
+    },
+    options: {
+      scales: {
+        y: {
+          type: 'logarithmic',
+          beginAtZero: true,
+          title: {
+            display: true,
+            text: 'Logarithmic Scale'
+          },
+          ticks: {
+            callback: function(value, index, values) {
+              if (value === 0 || value === 10 || value === 100 || value === 1000 || value === 10000) {
+                return value.toString();
+              }
+              return null;
+            }
+          }
+        }
+      },
+      plugins: {
+        tooltip: {
+          callbacks: {
+            label: function(tooltipItem) {
+              const index = tooltipItem.dataIndex;
+              const table = chartData.labels[index];
+              // const locations = chartData.locations[table];
+              // Aquí se podría hacer el stackgraph
+              return [
+                `Total Queries: ${tooltipItem.raw}`,
+              ];
+            }
+          }
+        }
+      }
+    }
+  });
+  document.getElementById('queryCountChart').parentElement.style.display = 'none';
+}
+
+function initializeChartCompare(chartData) {
+  const ctx = document.getElementById('queryCountChart').getContext('2d');
+  const queryCountChart = new Chart(ctx, {
+    type: 'bar',
+    data: {
+      labels: chartData.labels,
+      datasets: [
+        {
+          label: chartData.data_1.name,
+          data: chartData.data_1.data,
+          backgroundColor: 'rgba(75, 192, 192, 0.2)',
+          borderColor: 'rgba(75, 192, 192, 1)',
+          borderWidth: 1
+        },
+        {
+          label: chartData.data_2.name,
+          data: chartData.data_2.data,
+          backgroundColor: 'rgba(255, 99, 132, 0.2)',
+          borderColor: 'rgba(255, 99, 132, 1)',
+          borderWidth: 1
+        }
+      ]
+    },
+    options: {
+      scales: {
+        y: {
+          type: 'logarithmic',
+          beginAtZero: true,
+          title: {
+            display: true,
+            text: 'Logarithmic Scale'
+          },
+          ticks: {
+            callback: function(value, index, values) {
+              if (value === 0 || value === 10 || value === 100 || value === 1000 || value === 10000) {
+                return value.toString();
+              }
+              return null;
+            }
+          }
+        }
+      },
+      plugins: {
+        tooltip: {
+          callbacks: {
+            label: function(tooltipItem) {
+              const index = tooltipItem.dataIndex;
+              const table = chartData.labels[index];
+              return [
+                `Total Queries: ${tooltipItem.raw}`,
+              ];
+            }
+          }
+        }
+      }
+    }
+  });
+  document.getElementById('queryCountChart').parentElement.style.display = 'none';
+}

data/lib/query_tracker/printer/templates/comparing.html.erb

@@ -0,0 +1,66 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>Query Counter Report</title>
+</head>
+<body>
+  <script>
+    <%= js_content %>
+  </script>
+  <style>
+    <%= css_content %>
+  </style>
+  <div id='query_counter_report_gem'>
+    <h2>Query Counter Report</h2>
+    <p>Total query count on <%= @script_1_name %>: <%= total_query_count_1 %></p>
+    <p>Total query count on <%= @script_2_name %>: <%= total_query_count_2 %></p>
+    <button id="toggleButton" onclick="toggleView()">Show Chart View</button>
+    <button id="toggleColumnButton" onclick="toggleColumnContent()">Show SQL</button>
+
+    <div class="center-container">
+      <canvas id="queryCountChart"></canvas>
+    </div>
+  <table id="queryTable">
+      <tr>
+        <th>Table</th>
+        <th>Total Query Count <%= @script_1_name %></th>
+        <th>Total Query Count <%= @script_2_name %></th>
+        <th id="columnHeader">File Path</th>
+        <th>Method</th>
+        <th>Location Count <%= @script_1_name %></th>
+        <th>Location Count <%= @script_2_name %></th>
+      </tr>
+      <% tables.each do |table_name| %>
+        <% locations = (data_1.dig(table_name, :location)&.keys || []) | (data_2.dig(table_name, :location)&.keys || [])%>
+        <tr>
+          <td rowspan="<%= locations.count + 1 %>"><%= table_name %></td>
+          <td rowspan="<%= locations.count + 1 %>"><%= data_1.dig(table_name, :count) || 0%></td>
+          <td rowspan="<%= locations.count + 1 %>"><%= data_2.dig(table_name, :count) || 0 %></td>
+        </tr>
+        <% locations.each do |loc| %>
+          <% details_1 = data_1.dig(table_name, :location, loc) || {}%>
+          <% details_2 = data_2.dig(table_name, :location, loc) || {}%>
+          <% present_detail = details_1 || details_2 %>
+          <tr class="sub-row">
+            <% match = loc&.match(/^(?<file_path>.*):in.*`(?<method>.*)'$/) %>
+            <% file_path = match ? match[:file_path] : loc %>
+            <% method = match ? match[:method] : nil %>
+            <td class="toggle-content">
+              <span class="filepath"><%= file_path %></span>
+              <span class="sql" style="display: none;"><%= present_detail[:sql]&.gsub('"', '') %></span>
+            </td>
+            <td class="method-column"><%= method %></td>
+            <td><%= details_1[:count] || 0 %></td>
+            <td><%= details_2[:count] || 0 %></td>
+          </tr>
+        <% end %>
+      <% end %>
+    </table>
+    <script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
+    <script>
+      const chartData = <%= chart_data.to_json %>;
+      initializeChartCompare(chartData);
+    </script>
+  </div>
+</body>
+</html>

data/lib/query_tracker/printer/templates/style.css

@@ -0,0 +1,41 @@
+#query_counter_report_gem {
+  font-family: Arial, sans-serif;
+  table {
+    width: 100%;
+    border-collapse: collapse;
+    margin-top: 20px;
+
+    height: 100%;
+    overflow:auto;
+    display: block; /* Ensure the table behaves as a block element */
+    box-sizing: border-box;
+  }
+  th, td {
+    border: 1px solid #ddd;
+    padding: 8px;
+  }
+  th {
+    background-color: #f2f2f2;
+  }
+  tr:nth-child(even) {
+    background-color: #f9f9f9;
+  }
+  .sub-row {
+    background-color: #f9f9f9;
+  }
+  .center-container {
+    text-align: center;
+    width: 100%;
+    height: 500px;
+  }
+
+  #queryCountChart {
+    display: none;
+    margin: 0 auto;
+    width: 1200px;
+    height: 500px;
+  }
+  #queryTable {
+    display: table;
+  }
+}

data/lib/query_tracker/printer/templates/template.html.erb

@@ -0,0 +1,9 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>Query Counter Report</title>
+</head>
+<body>
+  <%= render_query_counter_base_div %>
+</body>
+</html>

data/lib/query_tracker/printer/templates/template_base_query_counter.html.erb

@@ -0,0 +1,51 @@
+<script>
+  <%= js_content %>
+</script>
+<style>
+  <%= css_content %>
+</style>
+<div id='query_counter_report_gem'>
+  <h2>Query Counter Report</h2>
+  <p>Total query count on process: <%= total_query_count %></p>
+  <p>The table will show only tables with more than <%= ::QueryTracker::Configuration.ignore_table_count %> queries</p>
+  <p>Only the top  <%= ::QueryTracker::Configuration.max_locations_per_table %> locations with the most occurrences will be shown.</p>
+  <button id="toggleButton" onclick="toggleView()">Show Chart View</button>
+  <button id="toggleColumnButton" onclick="toggleColumnContent()">Show SQL</button>
+
+  <div class="center-container">
+    <canvas id="queryCountChart"></canvas>
+  </div>
+<table id="queryTable">
+    <tr>
+      <th>Table</th>
+      <th>Total Query Count</th>
+      <th id="columnHeader">File Path</th>
+      <th>Method</th>
+      <th>Location Count</th>
+    </tr>
+    <% filter_data(data).each do |table, info| %>
+      <tr>
+        <td rowspan="<%= info[:location].size + 1 %>"><%= table %></td>
+        <td rowspan="<%= info[:location].size + 1 %>"><%= info[:count] %></td>
+      </tr>
+      <% info[:location].each do |loc, detail| %>
+        <tr class="sub-row">
+          <% match = loc&.match(/^(?<file_path>.*):in.*`(?<method>.*)'$/) %>
+          <% file_path = match ? match[:file_path] : loc %>
+          <% method = match ? match[:method] : nil %>
+          <td class="toggle-content">
+            <span class="filepath"><%= file_path %></span>
+            <span class="sql" style="display: none;"><%= detail[:sql].gsub('"', '') %></span>
+          </td>
+          <td class="method-column"><%= method %></td>
+          <td><%= detail[:count] %></td>
+        </tr>
+      <% end %>
+    <% end %>
+  </table>
+  <script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
+  <script>
+    const chartData = <%= chart_data.to_json %>;
+    initializeChart(chartData);
+  </script>
+</div>

data/lib/query_tracker/printer/templates/template_for_inject.html.erb

@@ -0,0 +1,64 @@
+<style>
+  #query-modal-button {
+    position: fixed;
+    top: 10px;
+    left: 10px;
+    z-index: 1000000;
+    background-color: #007bff;
+    color: white;
+    border: none;
+    padding: 10px;
+    cursor: pointer;
+  }
+
+  #query-modal-button:hover {
+    background-color: #0056b3;
+  }
+
+  #modal-close-button {
+    background-color: #d9534f;
+    color: white;
+    border: none;
+    padding: 5px;
+    cursor: pointer;
+  }
+
+  #modal-close-button:hover {
+    background-color: #c9302c;
+  }
+
+  #modal-content-query-count {
+    display: none;
+    position: fixed;
+    top: 10%;
+    left: 10%;
+    width: 80%;
+    height: 700px;
+    max-height: 80%;
+    background-color: white;
+    border: 1px solid #ccc;
+    padding: 10px;
+    box-shadow: 0px 0px 10px rgba(0, 0, 0, 0.1);
+    z-index: 10000000;
+    overflow: auto;
+  }
+
+  <%= css_content %>
+</style>
+<script>
+  function toggleQueryModal() {
+    var modal = document.getElementById('modal-content-query-count');
+    if (modal.style.display === 'none' || modal.style.display === '') {
+      modal.style.display = 'block';
+    } else {
+      modal.style.display = 'none';
+    }
+  }
+</script>
+<div id="query-modal-button">
+  <button id="query-modal-button" onclick="toggleQueryModal()">Toggle Query Counter</button>
+</div>
+<div id="modal-content-query-count">
+  <button id="modal-close-button" onclick="toggleQueryModal()">Close</button>
+  <%= render_query_counter_base_div %>
+</div>

data/lib/query_tracker/recording/base.rb

@@ -0,0 +1,28 @@
+module QueryTracker
+  module Recording
+    module Base
+      def start_with_block(printer: :console)
+        raise 'Block not given' unless block_given?
+
+        start_recording
+        yield
+        end_recording(printer:)
+      end
+
+      def start_recording
+        tracker.reset_query_count
+        tracker.subscribe
+      end
+
+      def end_recording(printer: :console)
+        tracker.unsubscribe
+        case printer
+        when :html
+          Printer::Html.new(data: tracker.query_tracker).print
+        when :console
+          Printer::Console.new(data: tracker.query_tracker).print
+        end
+      end
+    end
+  end
+end

data/lib/query_tracker/recording/tracker.rb

@@ -0,0 +1,40 @@
+module QueryTracker
+  class Tracker
+    REGEX_TABLE_SQL = /FROM\s+"(?<table>[^"]+)"/
+    attr_accessor :query_tracker, :subscription
+
+    def initialize
+      reset_query_count
+    end
+
+    # This assums that in the same location of the code it will always be the same sql query
+    def reset_query_count
+      @query_tracker = Hash.new do |hash, key|
+        hash[key] = { count: 0, location: Hash.new do |loc_hash, loc_key|
+                                            loc_hash[loc_key] = { count: 0, sql: nil }
+                                          end }
+      end
+    end
+
+    def subscribe
+      return unless subscription.nil?
+
+      @subscription = ActiveSupport::Notifications.subscribe('sql.active_record') do |_a, _b, _c, _d, payload|
+        caller_from_sql = caller
+        sql = payload[:sql]
+        match = sql.match(REGEX_TABLE_SQL)
+        if match.present? && match[:table]
+          actual_location = Rails.backtrace_cleaner.clean(caller_from_sql).first
+          query_tracker[match[:table]][:count] += 1
+          query_tracker[match[:table]][:location][actual_location][:count] += 1
+          query_tracker[match[:table]][:location][actual_location][:sql] = sql
+        end
+      end
+    end
+
+    def unsubscribe
+      ActiveSupport::Notifications.unsubscribe(@subscription)
+      @subscription = nil
+    end
+  end
+end

data/lib/query_tracker/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module QueryTracker
+  VERSION = '0.1.1'
+end

data/lib/query_tracker.rb

@@ -0,0 +1,36 @@
+require 'active_support/notifications'
+require_relative 'query_tracker/version'
+require_relative 'query_tracker/configuration'
+require_relative 'query_tracker/printer/base'
+require_relative 'query_tracker/printer/console'
+require_relative 'query_tracker/printer/html'
+require_relative 'query_tracker/recording/base'
+require_relative 'query_tracker/recording/tracker'
+require_relative 'query_tracker/compare/comparator'
+require_relative 'query_tracker/middleware'
+require_relative 'query_tracker/printer/html_compare'
+module QueryTracker
+  extend Recording::Base
+  if defined?(Rails::Railtie)
+    class QueryTrackerRailtie < Rails::Railtie
+      initializer 'query_tracker.configure_rails_initialization' do |app|
+        app.middleware.use Middleware
+      end
+    end
+  end
+
+  class << self
+    def configure
+      yield(Configuration)
+    end
+
+    def tracker
+      Thread.current[:query_counter_data] ||= Tracker.new
+    end
+
+    def compare
+      comparator = Compare::Comparator.new
+      yield(comparator)
+    end
+  end
+end

metadata

@@ -0,0 +1,258 @@
+--- !ruby/object:Gem::Specification
+name: query_tracker
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- Jose Lara
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2024-07-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: launchy
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.16.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.16.5
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: ruby-lsp
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: shoulda-context
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: shoulda-matchers
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.4'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.4'
+description: 
+email:
+- jvlara@uc.cl
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- bin/setup
+- lib/query_tracker.rb
+- lib/query_tracker/compare/comparator.rb
+- lib/query_tracker/configuration.rb
+- lib/query_tracker/middleware.rb
+- lib/query_tracker/printer/base.rb
+- lib/query_tracker/printer/console.rb
+- lib/query_tracker/printer/html.rb
+- lib/query_tracker/printer/html_compare.rb
+- lib/query_tracker/printer/templates/bar_chart.js
+- lib/query_tracker/printer/templates/comparing.html.erb
+- lib/query_tracker/printer/templates/style.css
+- lib/query_tracker/printer/templates/template.html.erb
+- lib/query_tracker/printer/templates/template_base_query_counter.html.erb
+- lib/query_tracker/printer/templates/template_for_inject.html.erb
+- lib/query_tracker/recording/base.rb
+- lib/query_tracker/recording/tracker.rb
+- lib/query_tracker/version.rb
+homepage: https://github.com/jvlara/query_tracker
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.2.2
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.10
+signing_key: 
+specification_version: 4
+summary: Display an overview of quantity of queries and their origin in Rails applications.
+test_files: []