Add a hash of hash response to Heartcheck

Add a hash of hash response to Heartcheck

lib/heartcheck.rb

@@ -13,9 +13,13 @@ class << self
     # @attr [Array<Checks>] the checks to use when checking
     attr_accessor :checks
 
-    # @attr [Heartcheck::Executors::Base] the checks executor backend
+    # @attr [Heartcheck::Executors::Base] or [Heartcheck::Executors::HashResponse]
+    # the checks executor backend
     attr_accessor :executor
 
+    # @attr [Boolean] the option for using a hash response or an array response
+    attr_reader :hash_response
+
     # @attr_writer [Object] change the default logger
     attr_writer :logger
 
@@ -38,7 +42,8 @@ def logger
     #   end
     #
     # @return [void]
-    def setup
+    def setup(options = {})
+      @hash_response = options.fetch(:hash_response, false)
       yield(self)
     end
 
@@ -97,9 +102,11 @@ def info_checks
 
     # an executor class that respond to dispatch(checkers)
     #
-    # @return [Heartcheck::Executors::Base]
+    # @return [Heartcheck::Executors::Base] or [Heartcheck::Executors::HashResponse]
     def executor
-      @executor ||= Heartcheck::Executors::Base.new
+      @executor ||= hash_response ?
+        Heartcheck::Executors::HashResponse.new :
+        Heartcheck::Executors::Base.new
     end
 
     # change current executor to a threaded implementation

lib/heartcheck/executors/hash_response.rb

@@ -0,0 +1,20 @@
+module Heartcheck
+    module Executors
+      class HashResponse
+        def dispatch(checkers)
+            checkers.map do |checker|
+              track_and_check(checker)
+            end.reduce({}, :merge)
+        end
+
+        def track_and_check(checker)
+          started = Time.now
+          checker.check.tap do |checked|
+            value = checked.values.first.merge(time:((Time.now - started) * 1_000.0))
+            checked.merge!(Hash[checked.keys.first => value])
+            Logger.info MultiJson.dump(checked)
+          end
+        end
+    end
+  end
+end

spec/lib/heartcheck/executors/hash_response_spec.rb

@@ -0,0 +1,39 @@
+require 'heartcheck/executors/hash_response'
+
+module Heartcheck
+    module Executors
+      describe HashResponse do
+        subject { described_class.new }
+
+        describe '#dispatch' do
+          let(:registered) { 3 }
+
+          let(:checkers) do
+            registered.times.collect do |current|
+              double.tap do |checker|
+                expect(checker).to receive(:check).and_return({ current: { status: 'ok' } })
+              end
+            end
+          end
+
+          it 'should register a log entry for each checker' do
+            expect(Logger).to receive(:info).exactly(registered).times
+            subject.dispatch(checkers)
+          end
+
+          it "should have a :time key in the checker response" do
+            subject.dispatch(checkers).each do |current|
+              expect(current.last).to include(:time)
+            end
+          end
+
+          it "should have a float value (time key)" do
+            subject.dispatch(checkers).each do |current|
+              expect(current.last[:time]).to be_a(Float)
+            end
+          end
+        end
+      end
+    end
+  end
+

spec/lib/heartcheck_spec.rb

@@ -1,3 +1,4 @@
+require 'heartcheck/executors/hash_response'
 describe Heartcheck do
   let(:essential)  { Heartcheck::Checks::Base.new }
   let(:functional) { Heartcheck::Checks::Base.new.tap { |c| c.functional = true } }
@@ -22,21 +23,31 @@
   end
 
   describe '#executor' do
+    context 'with hash_response' do
+      before(:each) { described_class.instance_variable_set :@executor, nil }
+      it 'returns a Heartcheck::Executors::HashResponse' do
+        described_class.setup hash_response: true do |monitor|
+          expect(monitor.executor).to be_a(Heartcheck::Executors::HashResponse)
+        end
+      end
+    end
+
     context 'with default' do
+      before(:each) { described_class.instance_variable_set :@executor, nil }
       it 'returns a Heartcheck::Executors::Base' do
         described_class.setup do |monitor|
           expect(monitor.executor).to be_a(Heartcheck::Executors::Base)
         end
       end
     end
 
-    context 'with custom' do
+    context 'with custom' do   
       it 'returns a custom custom executor' do
         described_class.executor = 'lala'
         expect(described_class.executor).to be_a(String)
       end
     end
-
+    
     context 'with threaded' do
       it 'returns a threaded  executor' do
         described_class.use_threaded_executor!