philae 0.3.1 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c630b6201d786e2d1b4cc1943d11de2dcd129d9a353a4e727e056a6f13e2678b
-  data.tar.gz: c0708fa30f24db6a8224380b9cb328a35fa64d38ce0095bca2dfb455fd2f4740
+  metadata.gz: a72261c385f330a6ca14c7ddaccd3774aa4e0e0918eb754bf757ccf16627b0db
+  data.tar.gz: f38b28573ee5829bcd41c9a772c074c43dff746a44b18842aab6c4cd7059bc21
 SHA512:
-  metadata.gz: f3074348a2e7848a7d6810da5552813cf610730eba369523e292c0f34560f5c22a08ee0ea9d28f3dc3801a095e6deb24c16895ff481a3c6f65e20fb7059304e7
-  data.tar.gz: 7350d66cce4ec8e6cffd71a108e981854cd3fa329bb39ad259c307841b926e43767c40d6d886e2bdbdfb3cc90cbbf68ff3075b49aa382ca8de95a31f76fda8a9
+  metadata.gz: 270fd5405071f9f12369bafe35693eda220a29180107ae32c5e4aeb0ac16f8d34ad3e069c5f886695b84707512a84054fa51e94fcbfce1bcb904df85f9c78e3c
+  data.tar.gz: 80543e2c7e06be18b3569c50ba373fe5c071d290f1f7ad51ba65af41033c32b62f324b05cffdadd230f8f87ca342888e61218cb1df3d5dd58ff70377a5d66827

data/lib/philae/collection_probe.rb

@@ -0,0 +1,54 @@
+module Philae
+  class CollectionProbe
+    def initialize(*args)
+      raise ArgumentError, "can't be instantiate, use specialized class"
+    end
+
+    def probes
+      raise ArgumentError, 'specialized class should implement :probes'
+    end
+
+    def check
+      results = []
+      result_queue = Queue.new
+      probes.each do |probe|
+        Thread.new do
+          begin
+            result = probe.check
+          rescue StandardError => e
+            result = {
+              healthy: false,
+              error: "probe #{probe.name} check failed: #{e.class} - #{e.message}",
+            }
+          end
+          result[:probe] = probe
+          result_queue.push result
+        end
+      end
+
+      loop do
+        results.push result_queue.pop
+        break if results.count == probes.length
+      end
+
+      failed_results = results.select do |result|
+        result[:healthy] == false
+      end
+
+      error = failed_results.map do |result|
+        msg = "#{result[:probe].name}: #{result[:comment]} "
+        msg += result[:code] if result[:code]
+        msg += result[:error] if result[:error]
+        next msg
+      end.join(', ')
+
+      if failed_results.count == probes.count
+        return { healthy: false, comment: "All #{name} nodes are down", error: error }
+      elsif failed_results.count > 0
+        return { healthy: true, comment: "#{failed_results.count} #{name} nodes are down: #{error}" }
+      else
+        return { healthy: true, comment: '' }
+      end
+    end
+  end
+end

data/lib/philae/etcd_cluster_probe.rb

@@ -0,0 +1,30 @@
+require_relative 'collection_probe'
+require_relative 'etcd_probe'
+
+module Philae
+  class EtcdClusterProbe < CollectionProbe
+    attr_reader :name, :etcd_probes
+
+    # @param [Integer] read_timeout Timeout in second for the HTTP request
+    def initialize(name, endpoints, read_timeout: 1, cacert: nil, ssl_cert: nil, ssl_key: nil)
+      raise ArgumentError, 'empty endpoints' if endpoints.nil? || endpoints.empty?
+
+      @etcd_probes = endpoints.map do |endpoint|
+        endpoint_uri = URI(endpoint)
+        next EtcdProbe.new(
+          "#{name}-#{endpoint_uri.host}-#{endpoint_uri.port}",
+          endpoint_uri.host,
+          endpoint_uri.port,
+          read_timeout: read_timeout, cacert: cacert,
+          ssl_cert: ssl_cert, ssl_key: ssl_key,
+        )
+      end
+
+      @name = name
+    end
+
+    def probes
+      return @etcd_probes
+    end
+  end
+end

data/lib/philae/nsq_cluster_probe.rb

@@ -0,0 +1,46 @@
+require_relative 'collection_probe.rb'
+require_relative 'nsq_probe.rb'
+
+module Philae
+  class NSQClusterProbe < CollectionProbe
+    DEFAULT_HTTP_PORT = 4152
+
+    attr_reader :name, :nsq_probes
+
+    def self.new_from_env(name)
+      if ENV['NSQD_HOSTS'].nil? || ENV['NSQD_HTTP_PORT'].nil?
+        raise ArgumentError, 'no NSQD_HOSTS and NSQD_HTTP_PORT defined'
+      end
+
+      tls_context = nil
+      if ENV['NSQD_TLS'] == 'true'
+        tls_context = {
+          cert: ENV['NSQD_TLS_CERT'],
+          key: ENV['NSQD_TLS_KEY'],
+          ca: ENV['NSQD_TLS_CACERT'],
+        }
+      end
+
+      http_hosts = ENV['NSQD_HOSTS'].split(',').map do |host|
+        host.split(':')[0] + ":#{ENV['NSQD_HTTP_PORT']}"
+      end
+
+      new(name, http_hosts, tls_context)
+    end
+
+    def initialize(name, hosts, tls_context = nil)
+      raise ArgumentError, 'should have at least one host' if hosts.nil? || hosts.empty?
+
+      @name = name
+      @nsq_probes = hosts.map do |hostport|
+        host, port = hostport.split(':')
+        port = DEFAULT_HTTP_PORT if port.nil?
+        next Philae::NSQProbe.new("#{name}-#{host}-#{port}", host, port, tls_context)
+      end
+    end
+
+    def probes
+      @nsq_probes
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: philae
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.2
 platform: ruby
 authors:
 - leo@scalingo.com
@@ -89,10 +89,13 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/philae.rb
+- lib/philae/collection_probe.rb
 - lib/philae/docker_probe.rb
+- lib/philae/etcd_cluster_probe.rb
 - lib/philae/etcd_probe.rb
 - lib/philae/http_probe.rb
 - lib/philae/mongo_probe.rb
+- lib/philae/nsq_cluster_probe.rb
 - lib/philae/nsq_probe.rb
 - lib/philae/pgsql_probe.rb
 - lib/philae/philae_probe.rb