add ES 2.0 support

sensu-plugins · Dec 30, 2015 · 622d9b9 · 622d9b9
1 parent 5e1f48d
commit 622d9b9
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/bin/metrics-es-node.rb b/bin/metrics-es-node.rb
@@ -86,7 +86,9 @@ def get_es_resource(resource)
   end
 
   def run
-    if Gem::Version.new(acquire_es_version) >= Gem::Version.new('1.0.0')
+    es_version = Gem::Version.new(acquire_es_version)
+
+    if es_version >= Gem::Version.new('1.0.0')
       ln = get_es_resource('/_nodes/_local')
       stats = get_es_resource('/_nodes/_local/stats')
     else
@@ -98,7 +100,11 @@ def run
     node = stats['nodes'].values.first
     node['jvm']['mem']['heap_max_in_bytes'] = ln['nodes'].values.first['jvm']['mem']['heap_max_in_bytes']
     metrics = {}
-    metrics['os.load_average'] = node['os']['load_average'][0]
+    if es_version >= Gem::Version.new('2.0.0')
+      metrics['os.load_average'] = node['os']['load_average']
+    else
+      metrics['os.load_average'] = node['os']['load_average'][0]
+    end
     metrics['os.mem.free_in_bytes'] = node['os']['mem']['free_in_bytes']
     metrics['process.mem.resident_in_bytes'] = node['process']['mem']['resident_in_bytes']
     metrics['jvm.mem.heap_used_in_bytes'] = node['jvm']['mem']['heap_used_in_bytes']