busy-administrator 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 883afd8faafc52f568eea6a1393d2a945654bcf7
-  data.tar.gz: a08973fcbc8caddb7db15d787751ebdc5a151235
+  metadata.gz: 028975af3001dec517754873ef39df4df057b97f
+  data.tar.gz: 4b79d312d9a12f12a0bebb1f23c67a5d3917f4be
 SHA512:
-  metadata.gz: 5ad360f51265c9d1a901327d1a2577ea6ae44504f0a5647b7deeba2753112bbfd093723cbbe1f38ee45cca743cddf35df1ef7a1b7e56e756a278a67f51bd40f4
-  data.tar.gz: a4ea6628c47693c4b551a4e58cea763de0ba9a7ca1897237222e5e491440b7be64ddaa04249e5df9a12985a149de0e050e00bf1fff89ebb83f61fbbee3e35a75
+  metadata.gz: e4f27c38524312770087828fb57fdf4556f5254a547bf4e0c9d42e2ec43cef94cbd57c2d172b716087b4ba1205c97d75cedc5b4d99c2324ac8bdb3c00405cdd3
+  data.tar.gz: a75c84f74e463cc7631967d3a3f13474db460b4093e8eb3c0fa4683b78c9ca7b1de8bd8d7b4417b48d1fb307a82b1bc30692b5f58e6ad0c619ecf981fab11223

data/lib/busy-administrator.rb

@@ -6,6 +6,7 @@ require 'busy-administrator/memory_size'
 require 'busy-administrator/memory_utils'
 require 'busy-administrator/process_utils'
 require 'busy-administrator/version'
+require 'busy-administrator/memory_middleware'
 
 """
 require 'busy-administrator'

data/lib/busy-administrator/memory_middleware.rb

@@ -0,0 +1,49 @@
+module BusyAdministrator
+  class MemoryMiddleware
+    module Analyzer
+      def analyze(key:, value:)
+        if analyzer = request.env['busy-administrator-analyzer']
+          analyzer.include(key: key, value: value)
+        end
+      end
+    end
+
+    def initialize(app)
+      @app = app
+    end
+
+    def assets_request?(env)
+      env['PATH_INFO'].ends_with?('.js') || env['PATH_INFO'].ends_with?('.css')
+    end
+
+    def enabled?
+      ENV['BUSY_ADMINISTRATOR_PROFILE'] == "YES"
+    end
+
+    def disabled?
+      not enabled?
+    end
+
+    def gc_enabled?
+      ENV['BUSY_ADMINISTRATOR_GC_ENABLED'] == "YES"
+    end
+
+    def call(env)
+      response = nil
+
+      if assets_request?(env) || disabled?
+        response = @app.call(env)
+      else
+        results = BusyAdministrator::MemoryUtils.profile(gc_enabled: gc_enabled?) do |analyzer|
+          env['busy-administrator-analyzer'] = analyzer
+
+          response = @app.call(env)
+        end
+
+        BusyAdministrator::Display.debug(results)
+      end
+
+      response
+    end
+  end
+end

data/lib/busy-administrator/memory_utils.rb

@@ -82,7 +82,7 @@ module BusyAdministrator
           excluded_object_id_storage = get_created_object_ids
           log("[Before] Object Count: #{ excluded_object_id_storage.size }", verbose: verbose)          
         end
-        
+
         action("Get Memory Usage Before Execution") do
           memory_usage_before = ProcessUtils.get_memory_usage(:rss)
           output_container[:memory_usage][:before] = memory_usage_before
@@ -147,11 +147,12 @@ module BusyAdministrator
           general = {}
 
           all_object_ids_after_execution.each do |object_id|
-            target = ObjectSpace._id2ref(object_id)
-            target_class = target.class.name
+            target = ObjectSpace._id2ref(object_id) rescue nil
 
-            general[target_class.to_sym] ||= 0
-            general[target_class.to_sym] += ObjectSpace.memsize_of(target)
+            if target_class = target.class.name rescue nil
+              general[target_class.to_sym] ||= 0
+              general[target_class.to_sym] += ObjectSpace.memsize_of(target)
+            end
           end
 
           general.each do |key, value|

data/lib/busy-administrator/version.rb

@@ -1,3 +1,3 @@
 module BusyAdministrator
-  VERSION = '1.0.0'
+  VERSION = '1.1.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: busy-administrator
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Joshua Arvin Lat
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-09-25 00:00:00.000000000 Z
+date: 2016-10-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -48,6 +48,7 @@ files:
 - lib/busy-administrator.rb
 - lib/busy-administrator/display.rb
 - lib/busy-administrator/example_generator.rb
+- lib/busy-administrator/memory_middleware.rb
 - lib/busy-administrator/memory_size.rb
 - lib/busy-administrator/memory_utils.rb
 - lib/busy-administrator/process_utils.rb