komachi_heartbeat 2.1.4 → 2.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 15dbd7348a5b785253e7a137596aab5c2c256dea
-  data.tar.gz: 2a0f337583c5a758aba47aa2fd45247403354b88
+  metadata.gz: e1f36fb59434d775367692fde238f5634bcd371e
+  data.tar.gz: 7c8165c671542407deef0aeaa0929bb37f503afe
 SHA512:
-  metadata.gz: 597fbaf17f26ec3e5fd2743b6a9804cad1ebe205b402dbec54add581349e49cf9d0b0b6311479055424ea3fe809d54f2a847031faa5f14ceb4b11601d95dd352
-  data.tar.gz: ef2a2c3a60518d307afb0b384613504fe9c8c5eb1607cccd8505b35136160d211fd05af06a5a13ccf6531df5c08d5e85533da440b6509ab43c55b77c26d1ed51
+  metadata.gz: 7fd337e5867f1cef835d9b85374be7e8049e13227722b1b9c46902a6daa8eaafefcf2bd0638262608f06286b8b95c9c394a6e71c279e7eac65daa64b36047af1
+  data.tar.gz: 5917e40b2904dca5211ab7bd80d7c07319a38a05f44ff0ce0915f2a43f3b4a499c70c75373c8042828a66fa8af8f98b318b976a346faa1b52465ba8ed5e26331

data/app/controllers/komachi_heartbeat/stats_controller.rb

@@ -0,0 +1,38 @@
+module KomachiHeartbeat
+  class StatsController < ApplicationController
+    def worker
+      return head :not_found unless worker_stats_enabled?
+      render json: worker_stats
+    end
+
+    private
+
+    def worker_stats
+      if defined?(Sidekiq)
+        sidekiq_stats
+      elsif defined?(Resque)
+        resque_stats
+      end
+    end
+
+    def sidekiq_stats
+      stats = Sidekiq::Stats.new
+      { enqueued: stats.enqueued, processed: stats.processed }
+    end
+
+    def resque_stats
+      unless Resque.respond_to?(:pending_queues)
+        Resque.class_eval {
+          def pending_queues
+            queues.inject(0) { |m,k| m + size(k) }
+          end
+        }
+      end
+      { enqueued: Resque.pending_queues, processed: Resque::Stat[:processed]}
+    end
+
+    def worker_stats_enabled?
+      KomachiHeartbeat.config.worker_stats_enabled
+    end
+  end
+end

data/config/routes.rb

@@ -1,4 +1,6 @@
 KomachiHeartbeat::Engine.routes.draw do
   get "heartbeat", :to => "heartbeat#index"
   get "version", :to => "heartbeat#version"
+
+  get "stats/worker", :to => "stats#worker"
 end

data/lib/komachi_heartbeat/config.rb

@@ -1,7 +1,7 @@
 module KomachiHeartbeat
   class Config
     include ActiveSupport::Configurable
-    config_accessor :application_version, :application_name, :db_check_enabled, :redis_check_enabled, :memcached_check_enabled, :database_class_names, :memcached_server_names, :redis_servers
+    config_accessor :application_version, :application_name, :db_check_enabled, :redis_check_enabled, :memcached_check_enabled, :database_class_names, :memcached_server_names, :redis_servers, :worker_stats_enabled
 
     configure do |config|
       config.application_name = "Unknown"
@@ -14,6 +14,8 @@ module KomachiHeartbeat
 
       config.memcached_check_enabled = false
       config.memcached_server_names = ["localhost"]
+
+      config.worker_stats_enabled = false
     end
   end
 end

data/lib/komachi_heartbeat/version.rb

@@ -1,3 +1,3 @@
 module KomachiHeartbeat
-  VERSION = "2.1.4"
+  VERSION = "2.1.5"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: komachi_heartbeat
 version: !ruby/object:Gem::Version
-  version: 2.1.4
+  version: 2.1.5
 platform: ruby
 authors:
 - Takumi MIURA
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-03-07 00:00:00.000000000 Z
+date: 2014-04-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -164,6 +164,7 @@ files:
 - app/assets/stylesheets/komachi_heartbeat/application.css
 - app/controllers/komachi_heartbeat/application_controller.rb
 - app/controllers/komachi_heartbeat/heartbeat_controller.rb
+- app/controllers/komachi_heartbeat/stats_controller.rb
 - app/helpers/komachi_heartbeat/application_helper.rb
 - app/helpers/komachi_heartbeat/heartbeat_helper.rb
 - app/views/layouts/komachi_heartbeat/application.html.erb