okcomputer 1.1.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9c8082e5decec046c189e6ccd432ffcafafdd863
-  data.tar.gz: e943f74ef3075248aae292a370ffdcb3ed433dc4
+  metadata.gz: 24238bddd672bb4db2ca97ec0abd4460faff2da6
+  data.tar.gz: fce0dba5d41e276557f3fbdbbc6824d91571d109
 SHA512:
-  metadata.gz: bbfa6ee07eba2834f5fb2066ba75f3bf371ff622cbfc669f72ae52438ec341edad9153a90a40ce1aa9ecd4403c7b7f0d69b79cb36afcaec61b21452291ccf7a6
-  data.tar.gz: 5586d90b4e6798e9ad61a68d159766a519f61352ac6b907ed8d82fddcdf0db776785862858d8940146c0f1100b2d272092a1acb72d1504091efeb71ad7a342dc
+  metadata.gz: 74b1aca6e40aecbea93a345fd8ebe66adceee698266edc9f547f8526457fb63aa66445ede8df6d745925aad9e87b0bfb0f3fbaf80bb31b85186907e4da481a30
+  data.tar.gz: eee8e54d589b97a9fd6773a8d08ffbaf4111972b4b6bf437161654b7b211ca9c360d9d2f4cc6d42fee108738f0c9f9a162f0b503a6a6ca49a0aa98186d1344e1

data/lib/ok_computer/built_in_checks/app_version_check.rb

@@ -0,0 +1,39 @@
+module OkComputer
+  # Display app version SHA
+  #
+  # * If `ENV["SHA"]` is set, uses that value.
+  # * Otherwise, checks for Capistrano's REVISION file in the app root.
+  # * Failing these, the check fails
+  class AppVersionCheck < Check
+    # Public: Return the application version
+    def check
+      mark_message "Version: #{version}"
+    rescue UnknownRevision
+      mark_failure
+      mark_message "Unable to determine version"
+    end
+
+    # Public: The application version
+    #
+    # Returns a String
+    def version
+      version_from_env || version_from_file || raise(UnknownRevision)
+    end
+
+    private
+
+    # Private: Version stored in environment variable
+    def version_from_env
+      ENV["SHA"]
+    end
+
+    # Private: Version stored in Capistrano revision file
+    def version_from_file
+      if File.exist?(Rails.root.join("REVISION"))
+        File.read(Rails.root.join("REVISION")).chomp
+      end
+    end
+
+    UnknownRevision = Class.new(StandardError)
+  end
+end

data/lib/ok_computer/built_in_checks/elasticsearch_check.rb

@@ -0,0 +1,50 @@
+module OkComputer
+  # This class performs a health check on an elasticsearch cluster using the
+  # {cluster health API}[http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/cluster-health.html].
+  # 
+  # It reports the cluster's name, number of nodes, and status (green, yellow,
+  # or red). A cluster status of red is reported as a failure, since this means
+  # one or more primary shards are unavailable. Note that the app may still
+  # be able to perform some queries on the available indices/shards.
+  class ElasticsearchCheck < Check
+    attr_accessor :host
+    attr_accessor :request_timeout
+
+    # Public: Initialize a new elasticsearch check.
+    #
+    # host - The hostname of elasticsearch
+    # request_timeout - How long to wait to connect before timing out. Defaults to 5 seconds.
+    def initialize(host, request_timeout = 5)
+      self.host = host
+      self.request_timeout = request_timeout.to_i
+    end
+
+    # Public: Return the status of the elasticsearch cluster
+    def check
+      cluster_health = self.cluster_health
+
+      if cluster_health[:status] == 'red'
+        mark_failure
+      end
+
+      mark_message "Connected to elasticseach cluster '#{cluster_health[:cluster_name]}', #{cluster_health[:number_of_nodes]} nodes, status '#{cluster_health[:status]}'"
+    rescue ConnectionFailed => e
+      mark_failure
+      mark_message "Error: '#{e}'"
+    end
+
+    # Returns a hash from elasticsearch's cluster health API
+    def cluster_health
+      response = timeout(request_timeout) { health_url.read(read_timeout: request_timeout) }
+      JSON.parse(response, symbolize_names: true)
+    rescue => e
+      raise ConnectionFailed, e
+    end
+
+    def health_url
+      @health_url ||= URI.join(host, '_cluster/health')
+    end
+
+    ConnectionFailed = Class.new(StandardError)
+  end
+end

data/lib/ok_computer/version.rb

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

data/lib/okcomputer.rb

@@ -7,10 +7,12 @@ 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/app_version_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/elasticsearch_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"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: okcomputer
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors:
 - Patrick Byrne
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-21 00:00:00.000000000 Z
+date: 2015-01-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sqlite3
@@ -68,9 +68,11 @@ files:
 - app/controllers/ok_computer/ok_computer_controller.rb
 - config/routes.rb
 - lib/ok_computer/built_in_checks/active_record_check.rb
+- lib/ok_computer/built_in_checks/app_version_check.rb
 - 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/elasticsearch_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