rails_mini_profiler 0.1.0

data/app/controllers/rails_mini_profiler/application_controller.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module RailsMiniProfiler
+  class ApplicationController < ::ApplicationController
+    rescue_from ActiveRecord::RecordNotFound, with: :not_found
+
+    before_action :check_current_user
+
+    protected
+
+    def present(model, presenter_class = nil, **kwargs)
+      klass = presenter_class || "RailsMiniProfiler::#{model.class.to_s.demodulize}Presenter".constantize
+      presenter = klass.new(model, view_context, **kwargs)
+      yield(presenter) if block_given?
+      presenter
+    end
+
+    private
+
+    def not_found(error)
+      redirect_back(fallback_location: profiled_requests_path, alert: error.to_s)
+    end
+
+    def check_current_user
+      redirect_back(fallback_location: root_path) unless User.get(request.env).present?
+    end
+  end
+end

data/app/controllers/rails_mini_profiler/flamegraphs_controller.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+require_dependency 'rails_mini_profiler/application_controller'
+
+module RailsMiniProfiler
+  class FlamegraphsController < ApplicationController
+    layout 'rails_mini_profiler/flamegraph'
+
+    before_action :set_flamegraph, only: %i[show]
+
+    def show; end
+
+    private
+
+    def set_flamegraph
+      @flamegraph = Flamegraph.find_by!(rmp_profiled_request_id: params[:id]).json_data
+    end
+
+    def configuration
+      @configuration ||= RailsMiniProfiler.configuration
+    end
+  end
+end

data/app/controllers/rails_mini_profiler/profiled_requests_controller.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require_dependency 'rails_mini_profiler/application_controller'
+
+module RailsMiniProfiler
+  class ProfiledRequestsController < ApplicationController
+    before_action :set_profiled_request, only: %i[show destroy]
+
+    def index
+      @profiled_requests = ProfiledRequest
+                             .includes(:flamegraph)
+                             .where(user_id: user_id).order(id: :desc)
+      @profiled_requests = @profiled_requests.where('request_path LIKE ?', "%#{params[:path]}%") if params[:path]
+      @profiled_requests = @profiled_requests.map { |request| present(request) }
+    end
+
+    def show
+      @traces = @profiled_request.traces
+      @traces = @traces.where('payload LIKE ?', "%#{params[:search]}%") if params[:search]
+      @traces = @traces
+                  .order(:start)
+                  .map { |trace| present(trace, profiled_request: @profiled_request) }
+      @profiled_request = present(@profiled_request)
+    end
+
+    def destroy
+      ProfiledRequest.where(user_id: user_id).destroy(@profiled_request.id)
+      redirect_to profiled_requests_url, notice: 'Profiled request was successfully destroyed.'
+    end
+
+    def destroy_all
+      ProfiledRequest.transaction do
+        requests_table_name = RailsMiniProfiler.storage_configuration.profiled_requests_table.to_sym
+        Flamegraph.joins(:profiled_request).where(requests_table_name => { user_id: user_id }).delete_all
+        Trace.joins(:profiled_request).where(requests_table_name => { user_id: user_id }).delete_all
+        ProfiledRequest.where(requests_table_name => { user_id: user_id }).delete_all
+      end
+      redirect_to profiled_requests_url, notice: 'Profiled Requests cleared'
+    end
+
+    private
+
+    def user_id
+      @user_id ||= User.get(request.env)
+    end
+
+    def set_profiled_request
+      @profiled_request = ProfiledRequest.where(user_id: user_id).find(params[:id])
+    end
+
+    def configuration
+      @configuration ||= RailsMiniProfiler.configuration
+    end
+  end
+end

data/app/helpers/rails_mini_profiler/application_helper.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module RailsMiniProfiler
+  module ApplicationHelper
+    def present(model, presenter_class = nil, **kwargs)
+      klass = presenter_class || "#{model.class}Presenter".constantize
+      presenter = klass.new(model, self, **kwargs)
+      yield(presenter) if block_given?
+      presenter
+    end
+  end
+end

data/app/helpers/rails_mini_profiler/profiled_requests_helper.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module RailsMiniProfiler
+  module ProfiledRequestsHelper
+    include ApplicationHelper
+
+    def formatted_duration(duration)
+      duration = (duration.to_f / 100)
+      duration < 1 ? duration : duration.round
+    end
+
+    def formatted_allocations(allocations)
+      number_to_human(allocations, units: { unit: '', thousand: 'k', million: 'M', billion: 'B', trillion: 'T' })
+    end
+  end
+end

data/app/models/rails_mini_profiler/application_record.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module RailsMiniProfiler
+  class ApplicationRecord < ActiveRecord::Base
+    if RailsMiniProfiler.storage_configuration.database.present?
+      establish_connection(RailsMiniProfiler.storage_configuration.database.present?)
+    end
+
+    self.abstract_class = true
+
+    def self.record_timestamps
+      # Some applications may disable timestamp setting, but in the context of the engine we always want to record
+      # timestamps, as engine functionality relies on it.
+      true
+    end
+  end
+end

data/app/models/rails_mini_profiler/controller_trace.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+# == Schema Information
+#
+# Table name: rmp_traces
+#
+#  id                      :integer          not null, primary key
+#  rmp_profiled_request_id :bigint           not null
+#  name                    :string
+#  start                   :integer
+#  finish                  :integer
+#  duration                :integer
+#  allocations             :integer
+#  payload                 :json
+#  backtrace               :json
+#  created_at              :datetime         not null
+#  updated_at              :datetime         not null
+#
+module RailsMiniProfiler
+  class ControllerTrace < Trace
+    store :payload, accessors: %i[view_runtime db_runtime]
+
+    class << self
+      def find_sti_class(_)
+        super(name)
+      end
+
+      def sti_name
+        'process_action.action_controller'
+      end
+    end
+  end
+end

data/app/models/rails_mini_profiler/flamegraph.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+# == Schema Information
+#
+# Table name: rmp_flamegraphs
+#
+#  id                      :integer          not null, primary key
+#  rmp_profiled_request_id :bigint           not null
+#  data                    :binary
+#  created_at              :datetime         not null
+#  updated_at              :datetime         not null
+#
+module RailsMiniProfiler
+  class Flamegraph < RailsMiniProfiler::ApplicationRecord
+    self.table_name = RailsMiniProfiler.storage_configuration.flamegraphs_table
+
+    belongs_to :profiled_request,
+               class_name: 'RailsMiniProfiler::ProfiledRequest',
+               foreign_key: :rmp_profiled_request_id
+
+    before_save :compress
+
+    def json_data
+      @json_data = ActiveSupport::Gzip.decompress(data)
+    end
+
+    private
+
+    def compress
+      self.data = ActiveSupport::Gzip.compress(data)
+    end
+  end
+end

data/app/models/rails_mini_profiler/instantiation_trace.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+# == Schema Information
+#
+# Table name: rmp_traces
+#
+#  id                      :integer          not null, primary key
+#  rmp_profiled_request_id :bigint           not null
+#  name                    :string
+#  start                   :integer
+#  finish                  :integer
+#  duration                :integer
+#  allocations             :integer
+#  payload                 :json
+#  backtrace               :json
+#  created_at              :datetime         not null
+#  updated_at              :datetime         not null
+#
+module RailsMiniProfiler
+  class InstantiationTrace < Trace
+    store :payload, accessors: %i[record_count class_name]
+
+    class << self
+      def find_sti_class(_)
+        super(name)
+      end
+
+      def sti_name
+        'instantiation.active_record'
+      end
+    end
+  end
+end

data/app/models/rails_mini_profiler/profiled_request.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+# == Schema Information
+#
+# Table name: rmp_profiled_requests
+#
+#  id               :integer          not null, primary key
+#  user_id          :string
+#  start            :integer
+#  finish           :integer
+#  duration         :integer
+#  allocations      :integer
+#  request_path     :string
+#  request_method   :string
+#  request_headers  :json
+#  request_body     :json
+#  response_status  :integer
+#  response_body    :json
+#  response_headers :json
+#  created_at       :datetime         not null
+#  updated_at       :datetime         not null
+#
+module RailsMiniProfiler
+  class ProfiledRequest < RailsMiniProfiler::ApplicationRecord
+    self.table_name = RailsMiniProfiler.storage_configuration.profiled_requests_table
+
+    has_one :flamegraph,
+            class_name: 'RailsMiniProfiler::Flamegraph',
+            foreign_key: :rmp_profiled_request_id,
+            dependent: :destroy
+
+    has_many :traces,
+             class_name: 'RailsMiniProfiler::Trace',
+             foreign_key: :rmp_profiled_request_id,
+             dependent: :destroy
+
+    def request=(request)
+      self.request_body = request.body
+      self.request_headers = request.headers
+      self.request_method = request.method
+      self.request_path = request.path
+      self.request_query_string = request.query_string
+    end
+
+    def response=(response)
+      self.response_body = response.body
+      self.response_media_type = response.media_type
+      self.response_headers = response.headers
+      self.response_status = response.status
+    end
+
+    def total_time=(total_time)
+      self.start = total_time.start
+      self.finish = total_time.finish
+      self.duration = total_time.duration
+      self.allocations = total_time.allocations
+    end
+  end
+end

data/app/models/rails_mini_profiler/render_partial_trace.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+# == Schema Information
+#
+# Table name: rmp_traces
+#
+#  id                      :integer          not null, primary key
+#  rmp_profiled_request_id :bigint           not null
+#  name                    :string
+#  start                   :integer
+#  finish                  :integer
+#  duration                :integer
+#  allocations             :integer
+#  payload                 :json
+#  backtrace               :json
+#  created_at              :datetime         not null
+#  updated_at              :datetime         not null
+#
+module RailsMiniProfiler
+  class RenderPartialTrace < Trace
+    store :payload, accessors: %i[identifier]
+
+    class << self
+      def find_sti_class(_)
+        super(name)
+      end
+
+      def sti_name
+        'render_partial.action_view'
+      end
+    end
+  end
+end

data/app/models/rails_mini_profiler/render_template_trace.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+# == Schema Information
+#
+# Table name: rmp_traces
+#
+#  id                      :integer          not null, primary key
+#  rmp_profiled_request_id :bigint           not null
+#  name                    :string
+#  start                   :integer
+#  finish                  :integer
+#  duration                :integer
+#  allocations             :integer
+#  payload                 :json
+#  backtrace               :json
+#  created_at              :datetime         not null
+#  updated_at              :datetime         not null
+#
+module RailsMiniProfiler
+  class RenderTemplateTrace < Trace
+    store :payload, accessors: %i[identifier]
+
+    class << self
+      def find_sti_class(_)
+        super(name)
+      end
+
+      def sti_name
+        'render_template.action_view'
+      end
+    end
+  end
+end

data/app/models/rails_mini_profiler/rmp_trace.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+# == Schema Information
+#
+# Table name: rmp_traces
+#
+#  id                      :integer          not null, primary key
+#  rmp_profiled_request_id :bigint           not null
+#  name                    :string
+#  start                   :integer
+#  finish                  :integer
+#  duration                :integer
+#  allocations             :integer
+#  payload                 :json
+#  backtrace               :json
+#  created_at              :datetime         not null
+#  updated_at              :datetime         not null
+#
+module RailsMiniProfiler
+  class RmpTrace < Trace
+    class << self
+      def find_sti_class(_)
+        super(name)
+      end
+
+      def sti_name
+        'rails_mini_profiler.total_time'
+      end
+    end
+  end
+end

data/app/models/rails_mini_profiler/sequel_trace.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+# == Schema Information
+#
+# Table name: rmp_traces
+#
+#  id                      :integer          not null, primary key
+#  rmp_profiled_request_id :bigint           not null
+#  name                    :string
+#  start                   :integer
+#  finish                  :integer
+#  duration                :integer
+#  allocations             :integer
+#  payload                 :json
+#  backtrace               :json
+#  created_at              :datetime         not null
+#  updated_at              :datetime         not null
+#
+module RailsMiniProfiler
+  class SequelTrace < Trace
+    store :payload, accessors: %i[name sql binds]
+
+    class << self
+      def find_sti_class(_)
+        super(name)
+      end
+
+      def sti_name
+        'sql.active_record'
+      end
+    end
+  end
+end