peek-active_record 0.1.0 → 0.2.0

data/lib/peek/extensions/active_record.rb

@@ -1,11 +1,12 @@
 require "active_record"
+require "atomic"
 
 class ActiveRecord::Base
   class << self
     attr_accessor :obj_count, :obj_types, :obj_types_enabled
   end
 
-  self.obj_count = 0
+  self.obj_count = Atomic.new(0)
   self.obj_types = Hash.new(0)
   self.obj_types_enabled = false
 
@@ -33,10 +34,14 @@ class ActiveRecord::Base
 
   protected
   def _update_object_stats
-    ActiveRecord::Base.obj_count += 1
+    ActiveRecord::Base.obj_count.update { |val| val + 1 }
 
     if ActiveRecord::Base.obj_types_enabled
-      ActiveRecord::Base.obj_types[ self.class.name ] += 1
+      if ActiveRecord::Base.obj_types.has_key? self.class.name
+        ActiveRecord::Base.obj_types[self.class.name].update { |val| val + 1 }
+      else
+        ActiveRecord::Base.obj_types[self.class.name] = Atomic.new(1)
+      end
     end
   end
 end

data/lib/peek/views/active_record.rb

@@ -1,13 +1,15 @@
+require "atomic"
+
 module Peek
   module Views
     class ActiveRecord < View
       def result
-        { :objects => ::ActiveRecord::Base.obj_count }
+        { :objects => ::ActiveRecord::Base.obj_count.value }
       end
 
       def context
         Hash.new.tap do |ctx|
-          ctx[:object_count] = ::ActiveRecord::Base.obj_count
+          ctx[:object_count] = ::ActiveRecord::Base.obj_count.value
           ctx[:object_types] = ::ActiveRecord::Base.obj_types
         end
       end
@@ -16,7 +18,7 @@ module Peek
 
       def setup_subscribers
         before_request do
-          ::ActiveRecord::Base.tap { |ar| ar.obj_count = 0 }
+          ::ActiveRecord::Base.tap { |ar| ar.obj_count = Atomic.new(0) }
         end
       end
     end

data/peek-active_record.gemspec

@@ -2,7 +2,7 @@
 #
 Gem::Specification.new do |spec|
   spec.name          = "peek-active_record"
-  spec.version       = "0.1.0"
+  spec.version       = "0.2.0"
   spec.authors       = ["Will Farrington"]
   spec.email         = ["wfarr@github.com"]
   spec.description   = %q{Peek into your ActiveRecord stats.}
@@ -16,6 +16,8 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_dependency "activerecord", "~> 3.2.0"
+  spec.add_dependency "atomic",       ">= 1.0.0"
+  spec.add_dependency "peek"
 
   spec.add_development_dependency "bundler", "~> 1.3"
   spec.add_development_dependency "rake"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: peek-active_record
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -27,6 +27,38 @@ dependencies:
     - - ~>
       - !ruby/object:Gem::Version
         version: 3.2.0
+- !ruby/object:Gem::Dependency
+  name: atomic
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: peek
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement