Add functionality for retrieving multi row query responses (#29)

* Added functionality for retrieving multi row query responses.

* Updated changelog and fixed syntax to pass rubocop requirements.

* Added logic flow to handle a special case. Fixed the return loop to properly return the query rows.

* Fixed an offense from Rubocop, spacing was off.

* Updated multirow option to be more representative of its functionality.

Author: Micasou

CHANGELOG.md

@@ -4,6 +4,9 @@ This project adheres to [Semantic Versioning](http://semver.org/).
 This CHANGELOG follows the format listed at [Keep A Changelog](http://keepachangelog.com/)
 
 ## [Unreleased]
+### Updated
+- metrics-postgres-query.rb: Added option to return multi row queries.
+
 ### Fixed
 - check-postgres-alive.rb: Fix connections using a custom port (#25)
 - check-postgres-connections.rb: Fix connections using a custom port (#25)

bin/metrics-postgres-query.rb

@@ -78,6 +78,13 @@ class MetricsPostgresQuery < Sensu::Plugin::Metric::CLI::Graphite
          long: '--scheme SCHEME',
          default: 'postgres'
 
+  option :multirow,
+         description: 'Determines if we return first row or all rows',
+         short: '-m',
+         long: '--multirow',
+         boolean: true,
+         default: false
+
   option :timeout,
          description: 'Connection timeout (seconds)',
          short: '-T TIMEOUT',
@@ -100,10 +107,16 @@ def run
     value = if config[:count_tuples]
               res.ntuples
             else
-              res.first.values.first
+              res.values
             end
 
-    output config[:scheme], value
+    if config[:multirow] && !config[:count_tuples]
+      value.each do |row|
+        output "#{config[:scheme]}.#{row[0]}", row[1]
+      end
+    else
+      output config[:scheme], value
+    end
     ok
   end
 end