rorvswild 1.5.1 → 1.5.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9273a6bd33ffb223f1d894975c3d25ad0e4f291295e3267fb6d995ad367b777a
-  data.tar.gz: 3e32db66b9ee0984352b5e559efa94b3b6a2f35b9c43af0631a1c95a3ecdb6ba
+  metadata.gz: a9e6a2963e38a065dfaba2af93a5d05275a18d6a57c9e3b12b80b027f8f6c605
+  data.tar.gz: 953c30a7d822f9ea69b2c2ef16d96f7a16ed5765da2fa8d013982a553c1a06f8
 SHA512:
-  metadata.gz: 53433ea46f20730ee3813ecac8437761fb700645e2c9e4ced1a69c67dafe22e1f6ec3905fa7d320798b984fdaecd1c4c252c6c2373a465f4788a0a9e905edebe
-  data.tar.gz: 8b27c4b3eb3e4f37a5e2c4dbacce88c52ff022ecd7cacf328fe39f7466bef8aaa9d35739c369481368f5ac147b62d73ab41a2d6b0a4ea979720ef0a8ab94215a
+  metadata.gz: bad6d0f028f564619f7283f582fbf5f008f8cf5fcd5b38ab8036fb570438ef1e4e75d2bffeeff93f562f541dc4bdf7cbdf3b38435fc44d354c82f9fd01b7cc5b
+  data.tar.gz: 31a9d64bbc3a6468022705025bab670c2dd3d156a9e35ba9241450f2d98a45c4f30448f25ceab7d7b983a21d251907dc3c4aeef6900084559a47b1c27841748b

data/lib/rorvswild/agent.rb

@@ -48,11 +48,11 @@ module RorVsWild
     end
 
     def measure_block(name, kind = "code".freeze, &block)
-      data[:name] ? measure_section(name, kind: kind, &block) : measure_job(name, &block)
+      current_data ? measure_section(name, kind: kind, &block) : measure_job(name, &block)
     end
 
     def measure_section(name, kind: "code", appendable_command: false, &block)
-      return block.call unless data[:name]
+      return block.call unless current_data
       begin
         RorVsWild::Section.start do |section|
           section.appendable_command = appendable_command
@@ -66,29 +66,27 @@ module RorVsWild
     end
 
     def measure_job(name, parameters: nil, &block)
-      return measure_section(name, &block) if data[:name] # For recursive jobs
+      return measure_section(name, &block) if current_data # For recursive jobs
       return block.call if ignored_job?(name)
-      initialize_data(name)
+      initialize_data[:name] = name
       begin
         block.call
       rescue Exception => ex
         push_exception(ex, parameters: parameters)
         raise
       ensure
-        data[:runtime] = RorVsWild.clock_milliseconds - data[:started_at]
+        current_data[:runtime] = RorVsWild.clock_milliseconds - current_data[:started_at]
         post_job
       end
     end
 
-    def start_request(payload)
-      return if data[:name]
-      initialize_data(payload[:name])
-      data[:path] = payload[:path]
+    def start_request
+      current_data || initialize_data
     end
 
     def stop_request
-      return unless data[:name]
-      data[:runtime] = RorVsWild.clock_milliseconds - data[:started_at]
+      return unless current_data
+      current_data[:runtime] = RorVsWild.clock_milliseconds - current_data[:started_at]
       post_request
     end
 
@@ -107,21 +105,21 @@ module RorVsWild
 
     def push_exception(exception, options = nil)
       return if ignored_exception?(exception)
-      data[:error] = exception_to_hash(exception)
-      data[:error].merge!(options) if options
-      data[:error]
+      current_data[:error] = exception_to_hash(exception)
+      current_data[:error].merge!(options) if options
+      current_data[:error]
     end
 
-    def data
-      Thread.current[:rorvswild_data] ||= {}
+    def current_data
+      Thread.current[:rorvswild_data]
     end
 
     def add_section(section)
-      return unless data[:sections]
-      if sibling = data[:sections].find { |s| s.sibling?(section) }
+      return unless current_data[:sections]
+      if sibling = current_data[:sections].find { |s| s.sibling?(section) }
         sibling.merge(section)
       else
-        data[:sections] << section
+        current_data[:sections] << section
       end
     end
 
@@ -139,11 +137,8 @@ module RorVsWild
 
     private
 
-    def initialize_data(name)
-      data[:name] = name
-      data[:sections] = []
-      data[:section_stack] = []
-      data[:started_at] = RorVsWild.clock_milliseconds
+    def initialize_data
+      Thread.current[:rorvswild_data] = {sections: [], section_stack: [], started_at: RorVsWild.clock_milliseconds}
     end
 
     def cleanup_data
@@ -153,7 +148,7 @@ module RorVsWild
     end
 
     def post_request
-      queue.push_request(cleanup_data)
+      (data = cleanup_data) && data[:name] && queue.push_request(data)
     end
 
     def post_job

data/lib/rorvswild/locator.rb

@@ -34,15 +34,15 @@ module RorVsWild
     end
 
     def relative_path(path)
-      path.index(current_path) == 0 ? path.sub(current_path, "".freeze) : path
+      path.start_with?(current_path) ? path.sub(current_path, "".freeze) : path
     end
 
     def relevant_path?(path)
-      path.index(current_path) == 0 && !irrelevant_path?(path)
+      path.start_with?(current_path) && !irrelevant_path?(path)
     end
 
     def irrelevant_path?(path)
-      irrelevant_paths.any? { |irrelevant_path| path.index(irrelevant_path) }
+      path.start_with?(*irrelevant_paths)
     end
 
     def irrelevant_paths
@@ -52,7 +52,7 @@ module RorVsWild
     private
 
     def initialize_irrelevant_paths
-      array = ["RUBYLIB", "GEM_HOME", "GEM_PATH", "BUNDLER_ORIG_PATH", "BUNDLER_ORIG_GEM_PATH"].flat_map do |name|
+      array = ["RUBYLIB", "GEM_HOME", "GEM_PATH", "BUNDLER_ORIG_GEM_PATH"].flat_map do |name|
         ENV[name].split(":".freeze) if ENV[name]
       end
       array += [heroku_ruby_lib_path] if File.exists?(heroku_ruby_lib_path)

data/lib/rorvswild/plugin/action_controller.rb

@@ -5,11 +5,28 @@ module RorVsWild
         return if @installed
         return unless defined?(::ActionController::Base)
         ActiveSupport::Notifications.subscribe("process_action.action_controller", new)
-        ::ActionController::Base.around_action(&method(:around_action))
         ::ActionController::Base.rescue_from(StandardError) { |ex| RorVsWild::Plugin::ActionController.after_exception(ex, self) }
         @installed = true
       end
 
+      def start(name, id, payload)
+        controller_action = "#{payload[:controller]}##{payload[:action]}"
+        if !RorVsWild.agent.ignored_request?(controller_action)
+          section = RorVsWild::Section.start
+          RorVsWild.agent.current_data[:name] = controller_action
+          controller = payload[:headers]["action_controller.instance".freeze]
+          method_name = controller.method_for_action(payload[:action])
+          section.file, section.line = controller.method(method_name).source_location
+          section.file = RorVsWild.agent.locator.relative_path(section.file)
+          section.command = "#{controller.class}##{method_name}"
+          section.kind = "code".freeze
+        end
+      end
+
+      def finish(name, id, payload)
+        RorVsWild::Section.stop
+      end
+
       def self.after_exception(exception, controller)
         if hash = RorVsWild.agent.push_exception(exception)
           hash[:session] = controller.session.to_hash
@@ -19,32 +36,6 @@ module RorVsWild
         raise exception
       end
 
-      def self.around_action(controller, block)
-        begin
-          RorVsWild::Section.start do |section|
-            method_name = controller.method_for_action(controller.action_name)
-            section.file, section.line = controller.method(method_name).source_location
-            section.file = RorVsWild.agent.locator.relative_path(section.file)
-            section.command = "#{controller.class}##{method_name}"
-            section.kind = "code".freeze
-          end
-          block.call
-        ensure
-          RorVsWild::Section.stop
-        end
-      end
-
-      # Payload: controller, action, params, format, method, path
-      def start(name, id, payload)
-        if !RorVsWild.agent.ignored_request?(name = "#{payload[:controller]}##{payload[:action]}")
-          RorVsWild.agent.start_request(name: name, path: payload[:path])
-        end
-      end
-
-      def finish(name, id, payload)
-        RorVsWild.agent.stop_request
-      end
-
       def self.extract_http_headers(headers)
         headers.reduce({}) do |hash, (name, value)|
           if name.index("HTTP_".freeze) == 0 && name != "HTTP_COOKIE".freeze

data/lib/rorvswild/plugin/action_dispatch.rb

@@ -0,0 +1,27 @@
+module RorVsWild
+  module Plugin
+    class ActionDispatch
+      def self.setup
+        return if @installed
+        return unless defined?(::ActiveSupport::Notifications)
+        ActiveSupport::Notifications.subscribe("request.action_dispatch", new)
+        @installed = true
+      end
+
+      def start(name, id, payload)
+        RorVsWild.agent.start_request
+        RorVsWild.agent.current_data[:path] = payload[:request].original_fullpath
+        @action_dispath_location ||= ::ActionDispatch::Executor.instance_method(:call).source_location
+        section = RorVsWild::Section.start
+        section.file, section.line = @action_dispath_location
+        section.command = "ActionDispatch::Executor.#call".freeze
+        section.kind = "code".freeze
+      end
+
+      def finish(name, id, payload)
+        RorVsWild::Section.stop
+        RorVsWild.agent.stop_request
+      end
+    end
+  end
+end

data/lib/rorvswild/plugin/active_job.rb

@@ -1,6 +1,6 @@
 module RorVsWild
   module Plugin
-    module ActiveJob
+    class ActiveJob
       def self.setup
         return if @installed
         return unless defined?(::ActiveJob::Base)
@@ -9,7 +9,17 @@ module RorVsWild
       end
 
       def self.around_perform(job, block)
-        RorVsWild.agent.measure_job(job.class.name, parameters: job.arguments, &block)
+        RorVsWild.agent.measure_job(job.class.name, parameters: job.arguments) do
+          begin
+            section = RorVsWild::Section.start
+            section.command = "#{job.class}#perform"
+            section.file, section.line = job.method(:perform).source_location
+            section.file = RorVsWild.agent.locator.relative_path(section.file)
+            block.call
+          ensure
+            RorVsWild::Section.stop
+          end
+        end
       end
     end
   end

data/lib/rorvswild/queue.rb

@@ -58,7 +58,6 @@ module RorVsWild
     end
 
     def flush
-      RorVsWild.logger.info("RorVsWild::Queue#flush".freeze)
       data = pull_jobs and client.post("/jobs", jobs: data)
       data = pull_requests and client.post("/requests", requests: data)
     end

data/lib/rorvswild/section.rb

@@ -6,12 +6,12 @@ module RorVsWild
     def self.start(&block)
       section = Section.new
       block.call(section) if block_given?
-      stack.push(section)
+      stack && stack.push(section)
       section
     end
 
     def self.stop(&block)
-      section = stack.pop
+      return unless stack && section = stack.pop
       block.call(section) if block_given?
       section.total_runtime = RorVsWild.clock_milliseconds - section.started_at
       current.children_runtime += section.total_runtime if current
@@ -19,17 +19,18 @@ module RorVsWild
     end
 
     def self.stack
-      RorVsWild.agent.data[:section_stack] ||= []
+      (data = RorVsWild.agent.current_data) && data[:section_stack]
     end
 
     def self.current
-      stack.last
+      (sections = stack) && sections.last
     end
 
     def initialize
       @calls = 1
       @total_runtime = 0
       @children_runtime = 0
+      @kind = "code".freeze
       @started_at = RorVsWild.clock_milliseconds
       location = RorVsWild.agent.locator.find_most_relevant_location(caller_locations)
       @file = RorVsWild.agent.locator.relative_path(location.path)

data/lib/rorvswild/version.rb

@@ -1,3 +1,3 @@
 module RorVsWild
-  VERSION = "1.5.1".freeze
+  VERSION = "1.5.2".freeze
 end

data/lib/rorvswild.rb

@@ -40,7 +40,7 @@ module RorVsWild
   end
 
   def self.initialize_logger(destination = nil)
-    if destination.is_a?(Logger)
+    if destination.respond_to?(:info) && destination.respond_to?(:warn) && destination.respond_to?(:error)
       destination
     elsif destination
       Logger.new(destination)
@@ -54,6 +54,16 @@ module RorVsWild
   def self.clock_milliseconds
     Process.clock_gettime(Process::CLOCK_MONOTONIC, :millisecond)
   end
+
+  def self.check
+    api_key = RorVsWild.agent.config[:api_key]
+    return puts "You API key is missing and has to be defined in config/rorvswild.yml." if !api_key || api_key.empty?
+    puts case response = agent.client.post("/jobs", jobs: [{sections: [], name: "RorVsWild.check", runtime: 0}])
+    when Net::HTTPOK then "Connection to RorVsWild works fine !"
+    when Net::HTTPUnauthorized then "Wrong API key"
+    else puts "Something went wrong: #{response.inspect}"
+    end
+  end
 end
 
 if defined?(Rails)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rorvswild
 version: !ruby/object:Gem::Version
-  version: 1.5.1
+  version: 1.5.2
 platform: ruby
 authors:
 - Alexis Bernard
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-03-04 00:00:00.000000000 Z
+date: 2020-04-25 00:00:00.000000000 Z
 dependencies: []
 description: Performances and quality insights for rails developers.
 email:
@@ -39,6 +39,7 @@ files:
 - lib/rorvswild/local/stylesheet/vendor/prism.css
 - lib/rorvswild/locator.rb
 - lib/rorvswild/plugin/action_controller.rb
+- lib/rorvswild/plugin/action_dispatch.rb
 - lib/rorvswild/plugin/action_mailer.rb
 - lib/rorvswild/plugin/action_view.rb
 - lib/rorvswild/plugin/active_job.rb