okcomputer 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ce5b21b1d706641cf9b709167a93eddf24701338
-  data.tar.gz: 045d6b22a4107d0b643318e3a721cbf1af256709
+  metadata.gz: 9c8082e5decec046c189e6ccd432ffcafafdd863
+  data.tar.gz: e943f74ef3075248aae292a370ffdcb3ed433dc4
 SHA512:
-  metadata.gz: 79ed0bb362e58c18473688f2e0ffddd42a93de86654231964ecd4df798677af4c66c18ccf1895a5e69bcd5377c370033ede13b91c1cc5c867f0b42d04c8ab412
-  data.tar.gz: 0ecb90b754c9d4c2f0d4e5ef68e1ca3814ff9d1569fac57ea582c54ff4d2396f828672796505363244ae0f2343ec2828a865c3f903b4f555976c2b0846b089be
+  metadata.gz: bbfa6ee07eba2834f5fb2066ba75f3bf371ff622cbfc669f72ae52438ec341edad9153a90a40ce1aa9ecd4403c7b7f0d69b79cb36afcaec61b21452291ccf7a6
+  data.tar.gz: 5586d90b4e6798e9ad61a68d159766a519f61352ac6b907ed8d82fddcdf0db776785862858d8940146c0f1100b2d272092a1acb72d1504091efeb71ad7a342dc

data/lib/ok_computer/built_in_checks/cache_check.rb

@@ -1,7 +1,12 @@
 module OkComputer
+  # Verifies that the Rails cache is set up and can speak with Memcached
+  # running on the given host (defaults to local).
   class CacheCheck < Check
+    attr_accessor :host
 
-    ConnectionFailed = Class.new(StandardError)
+    def initialize(host=Socket.gethostname)
+      self.host = host
+    end
 
     # Public: Check whether the cache is active
     def check
@@ -14,9 +19,9 @@ module OkComputer
     # Public: Outputs stats string for cache
     def stats
       stats    = Rails.cache.stats
-      host     = stats.select{|k,v| k =~ Regexp.new(Socket.gethostname) }.values[0]
-      mem_used = to_megabytes host['bytes']
-      mem_max  = to_megabytes host['limit_maxbytes']
+      values     = stats.select{|k,v| k =~ Regexp.new(host) }.values[0]
+      mem_used = to_megabytes values['bytes']
+      mem_max  = to_megabytes values['limit_maxbytes']
       return "#{mem_used} / #{mem_max} MB, #{stats.count - 1} peers"
     rescue => e
       raise ConnectionFailed, e
@@ -28,5 +33,7 @@ module OkComputer
     def to_megabytes(bytes)
       bytes.to_i / (1024 * 1024)
     end
+
+    ConnectionFailed = Class.new(StandardError)
   end
 end

data/lib/ok_computer/built_in_checks/generic_cache_check.rb

@@ -0,0 +1,33 @@
+require "securerandom"
+
+module OkComputer
+  # Verifies that Rails can write to and read from its cache.
+  class GenericCacheCheck < Check
+    # Public: Check whether cache can be written to and read from
+    def check
+      test_value.tap do |value|
+        Rails.cache.write(cache_key, value)
+        if value == Rails.cache.read(cache_key)
+          mark_message "Able to read and write"
+        else
+          mark_failure
+          mark_message "Value read from the cache does not match the value written"
+        end
+      end
+    rescue => error
+      mark_failure
+      mark_message "Connection failure: #{error}"
+    end
+
+    private
+
+    # Private: Generate a unique value each time we check
+    def test_value
+      SecureRandom.hex
+    end
+
+    def cache_key
+      "ock-generic-cache-check-#{Socket.gethostname}"
+    end
+  end
+end

data/lib/ok_computer/built_in_checks/mongoid_replica_set_check.rb

@@ -0,0 +1,72 @@
+module OkComputer
+  # This class provides a check for a mongodb replica set via the Mongoid ORM.
+  #
+  # The check first refreshes the cluster status, which communicates with all
+  # the nodes, discovers any new ones, and figures out which node is the
+  # primary and which ones are the secondaries. Nodes that are recovering or
+  # unavailable are automatically removed from rotation. It's okay to do this
+  # fairly frequently.
+  #
+  # The second part of the check attempts to contact the primary node (to
+  # ensure writes are accepted) and a secondary node (to ensure reads
+  # can be distributed).
+  #
+  # This calls the
+  # {replSetGetStatus}[http://docs.mongodb.org/manual/reference/command/replSetGetStatus/]
+  # command on the admin database of each node. This provides further
+  # information as well as the replica set's name. This could potentially be
+  # parsed for more actionable information.
+  class MongoidReplicaSetCheck < OkComputer::Check
+    attr_accessor :session
+
+    # Public: Initialize a check for a Mongoid replica set
+    #
+    # session - The name of the Mongoid session to use. Defaults to the
+    #   default session.
+    def initialize(session = :default)
+      self.session = Mongoid::Sessions.with_name(session)
+    end
+
+    # Public: Return the status of the mongodb replica set
+    def check
+      refresh
+      primary_status = self.primary_status
+      secondary_status = self.secondary_status
+      mark_message "Connected to #{session.cluster.nodes.count} nodes in mongodb replica set '#{primary_status['set']}'"
+    rescue ConnectionFailed => e
+      mark_failure
+      mark_message "Error: '#{e}'"
+    end
+
+    # Public: Refresh the cluster status
+    def refresh
+      session.cluster.refresh
+    rescue => e
+      raise ConnectionFailed, e
+    end
+
+    # Public: The status for the session's mongodb replica set primary
+    #
+    # Returns a hash with the status of the primary
+    def primary_status
+      session.cluster.with_primary do |primary|
+        primary.command(:admin, replSetGetStatus: 1)
+      end
+    rescue => e
+      raise ConnectionFailed, e
+    end
+
+    # Public: The status for the session's mongodb replica set secondary
+    #
+    # Returns a hash with the status of the secondary
+    def secondary_status
+      session.cluster.with_secondary do |secondary|
+        secondary.command(:admin, replSetGetStatus: 1)
+      end
+    rescue => e
+      raise ConnectionFailed, e
+    end
+
+    ConnectionFailed = Class.new(StandardError)
+  end
+end

data/lib/ok_computer/version.rb

@@ -1,3 +1,3 @@
 module OkComputer
-  VERSION = "1.0.0"
+  VERSION = "1.1.0"
 end

data/lib/okcomputer.rb

@@ -7,15 +7,17 @@ require "ok_computer/registry"
 # and the built-in checks
 require "ok_computer/built_in_checks/size_threshold_check"
 require "ok_computer/built_in_checks/active_record_check"
+require "ok_computer/built_in_checks/cache_check"
 require "ok_computer/built_in_checks/default_check"
+require "ok_computer/built_in_checks/delayed_job_backed_up_check"
+require "ok_computer/built_in_checks/generic_cache_check"
 require "ok_computer/built_in_checks/mongoid_check"
+require "ok_computer/built_in_checks/mongoid_replica_set_check"
 require "ok_computer/built_in_checks/resque_backed_up_check"
-require "ok_computer/built_in_checks/resque_failure_threshold_check"
 require "ok_computer/built_in_checks/resque_down_check"
-require "ok_computer/built_in_checks/delayed_job_backed_up_check"
-require "ok_computer/built_in_checks/sidekiq_latency_check"
+require "ok_computer/built_in_checks/resque_failure_threshold_check"
 require "ok_computer/built_in_checks/ruby_version_check"
-require "ok_computer/built_in_checks/cache_check"
+require "ok_computer/built_in_checks/sidekiq_latency_check"
 
 OkComputer::Registry.register "default", OkComputer::DefaultCheck.new
 OkComputer::Registry.register "database", OkComputer::ActiveRecordCheck.new

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: okcomputer
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Patrick Byrne
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-20 00:00:00.000000000 Z
+date: 2015-01-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sqlite3
@@ -71,7 +71,9 @@ files:
 - lib/ok_computer/built_in_checks/cache_check.rb
 - lib/ok_computer/built_in_checks/default_check.rb
 - lib/ok_computer/built_in_checks/delayed_job_backed_up_check.rb
+- lib/ok_computer/built_in_checks/generic_cache_check.rb
 - lib/ok_computer/built_in_checks/mongoid_check.rb
+- lib/ok_computer/built_in_checks/mongoid_replica_set_check.rb
 - lib/ok_computer/built_in_checks/resque_backed_up_check.rb
 - lib/ok_computer/built_in_checks/resque_down_check.rb
 - lib/ok_computer/built_in_checks/resque_failure_threshold_check.rb
@@ -89,11 +91,7 @@ homepage: https://github.com/sportngin/okcomputer
 licenses:
 - MIT
 metadata: {}
-post_install_message: |
-  OkComputer Namespace Change (v0.7.0)
-  ------------------------------------
-  OkComputer has changed its namespace from `OKComputer` (uppercase K) to `OkComputer` (lowercase k)
-  as of version 0.7.0. Please update your configuration accordingly.
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib