query_counter 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: dde4c69db1dfdb26db42ab2dd2d549120db69565
+  data.tar.gz: 46d0d9e9ba5593004d2fb9fe034a8b26a417fc28
+SHA512:
+  metadata.gz: 25870e36da7884d02c2b5bcf63011bf88233cf39979684a3684431019b302898e2b6fe90af6fbf9ff13cfc4e14ebc71bce10bcb6046a006e9fe45a6fdc00eb7e
+  data.tar.gz: 476ad9e7953ce4e9012988e026b4fc2b3dbc608cf494a4e7be12d8aa923da95d012806a04814c97f268585848a97ed9e3866f8c30289d6c17cedd989992074da

data/lib/query_counter/collector.rb

@@ -0,0 +1,54 @@
+module QueryCounter
+  class Collector
+    attr_reader :stats,
+                :notification_events
+
+    def initialize
+      reset
+    end
+
+    def record(resource, duration, by=1)
+      @stats[resource] ||= ::QueryCounter::Stat.new
+      @stats[resource].increment(duration, by)
+    end
+
+    # used for debugging resource events
+    def record_event(resource, event)
+      record(resource, event.duration)
+
+      @notification_events ||= {}
+      @notification_events[resource] ||= []
+      @notification_events[resource] << event
+    end
+
+    def reset
+      if instance_variable_defined?(:@notification_events)
+        remove_instance_variable(:@notification_events)
+      end
+
+      # just away to return the current stats and reset them back to an empty hash
+      @stats.tap { @stats = {} }
+    end
+
+    def count(resource)
+      (@stats[resource] && @stats[resource].count) || 0
+    end
+
+    def events(resource)
+      (@notification_events && @notification_events[resource]) || []
+    end
+
+    def add(collector)
+      if collector.notification_events
+        collector.notification_events.each do |resource, list|
+          list.each { |event| record_event(resource, event) }
+        end
+      else
+        collector.stats.each do |resource, stat|
+          record(record, stat.time, stat.count)
+        end
+      end
+      self
+    end
+  end
+end

data/lib/query_counter/global.rb

@@ -0,0 +1,20 @@
+require 'singleton'
+
+# Purpose is to collect stats for the life time of the process
+module QueryCounter
+  class Global
+    include Singleton
+
+    def initialize
+      @collector = ::QueryCounter::Collector.new
+    end
+
+    def record(resource, duration, by=1)
+      @collector.record(resource, duration, by)
+    end
+
+    def reset
+      @collector.reset
+    end
+  end
+end

data/lib/query_counter/stat.rb

@@ -0,0 +1,16 @@
+module QueryCounter
+  class Stat
+    attr_reader :count,
+                :time
+
+    def initialize
+      @count = 0
+      @time = 0.0
+    end
+
+    def increment(duration, by=1)
+      @count += by
+      @time += duration
+    end
+  end
+end

data/lib/query_counter.rb

@@ -0,0 +1,61 @@
+module QueryCounter
+  autoload :Collector, 'query_counter/collector'
+  autoload :Global, 'query_counter/global'
+  autoload :Stat, 'query_counter/stat'
+
+  def self.global_collector
+    ::QueryCounter::Global.instance
+  end
+
+  def self.current_collector
+    Thread.current[:query_counter_collector] ||= ::QueryCounter::Collector.new
+  end
+
+  def self.temporary_collector
+    Thread.current[:temporary_query_counter_collector]
+  end
+
+  def self.record(resource, duration, by=1)
+    global_collector.record(resource, duration, by)
+    current_collector.record(resource, duration, by)
+  end
+
+  def self.record_event(resource, event)
+    temporary_collector && temporary_collector.record_event(resource, event)
+    record(resource, event.duration)
+  end
+
+  def self.reset
+    current_collector.reset
+  end
+
+  def self.around
+    new_collector = ::QueryCounter::Collector.new
+    old_collector, Thread.current[:temporary_query_counter_collector] = Thread.current[:temporary_query_counter_collector], new_collector
+
+    yield
+
+    if old_collector
+      Thread.current[:temporary_query_counter_collector] = old_collector.add(new_collector)
+    else
+      Thread.current[:temporary_query_counter_collector] = nil
+    end
+
+    new_collector
+  end
+
+  def self.count(resource)
+    current_collector.count(resource)
+  end
+
+  def self.global_count(resource)
+    current_collector.count(resource)
+  end
+
+  def self.auto_subscribe!(resource, event_name)
+    require 'active_support/notifications'
+    ActiveSupport::Notifications.subscribe(event_name) do |*args|
+      ::QueryCounter.record_event(resource, ActiveSupport::Notifications::Event.new(*args))
+    end
+  end
+end

metadata

@@ -0,0 +1,46 @@
+--- !ruby/object:Gem::Specification
+name: query_counter
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Doug Youch
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-12-18 00:00:00.000000000 Z
+dependencies: []
+description: Used for monitoring number of external calls
+email: dougyouch@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/query_counter.rb
+- lib/query_counter/collector.rb
+- lib/query_counter/global.rb
+- lib/query_counter/stat.rb
+homepage: https://github.com/evertrue/query_counter
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: Count requests to external systems
+test_files: []