sensu-plugins-azure-burstable-metrics 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: d5cdf2ac9d672268a82bbb734e5f397ceb03078f
+  data.tar.gz: 7a34963590928735f6c95cc11d27674e9ea35eb5
+SHA512:
+  metadata.gz: d74553d99db3544b7a273fa4cdf8d3811e7faef5d1996d9d629cc9f8965354d4c0f2ae717fb6cf3f3c1318401eac8c5a869c502566776cc1e9d00f7177d230e7
+  data.tar.gz: 4d570ab6fc5ca5e5b9547ba58a23dbe940d31384e59bb09a59d01aa17a808d7edb655b7a10e666d20ddb092f60fc4582bb372fb280e1faf5bd8c07913723af6e

data/README.md

@@ -0,0 +1,25 @@
+Sensu Plugins Azure Burstable Metrics
+=====================================
+
+This is a hacky Sensu plug-in to call out to the `az` to list all vms in a
+subscription and then gather metrics about burstable credits and output them in
+a Graphite way.
+
+The gem assumes `virualenv`, `python` and `pip` are installed, and will run
+`install.bash` as a pre-gem hook to install a virualenv and the required `az`
+command. Ick.
+
+Tested on Ruby 2.3.
+
+The unit test is a bit dubious, and it's almost impossible to run unit tests on
+Sensu plug-ins as they run as soon as scoped...
+The test must be run from the root directory also.
+
+`bundle exec rspec` should run the test, after bundle installing of course.
+
+Why call out to `az` instead of using the Ruby SDK?
+---------------------------------------------------
+There's very little documentation and this was the path of least resistance.
+
+I'm open to re-writes which use the Ruby Gems instead. You'll need the
+`azure_mgmt_compute` and `azure_mgmt_monitor` gems.

data/bin/sensu-plugins-azure-burstable-metrics.rb

@@ -0,0 +1,79 @@
+#! /usr/bin/env ruby
+
+require 'json'
+require 'sensu-plugin/metric/cli'
+require 'time'
+
+class BurstableMetrics < Sensu::Plugin::Metric::CLI::Graphite
+  option :schema,
+         description: 'Metric naming scheme, text to prepend to metric',
+         short: '-s SCHEME',
+         long: '--scheme SCHEME',
+         default: 'sensu'
+
+  option :spn_id,
+         description: 'The SPN ID with read access to the subscription',
+         short: '-i',
+         long: '--spn-id <SPN ID>',
+         default: nil
+
+  option :spn_secret,
+         description: 'The SPN secret for the SPN ID',
+         short: '-p',
+         long: '--spn-secret <SPN Secret>',
+         default: nil
+
+  option :tenant_id,
+         description: 'The Tenant ID for the SPN ID',
+         short: '-t',
+         long: '--tenant-id <Tenant ID>',
+         default: nil
+
+  @@time_format = '%Y-%m-%dT%H:%M:%SZ'
+
+  def initialize(argv = ARGV, metrics_client = 'az')
+    super argv
+
+    dir = __FILE__ === '(eval)' ? '.' : "#{__dir__}/../"
+    require "#{dir}/lib/provider/#{metrics_client}.rb"
+    metrics_upper = metrics_client.capitalize
+    @metrics_client = Object.const_get("#{metrics_upper}Client").new
+  end
+
+  def run
+    @metrics_client.set_credentials(
+      config[:spn_id],
+      config[:spn_secret],
+      config[:tenant_id],
+  )
+
+    machines = @metrics_client.vm_list.select! { |machine| machine['hardwareProfile']['vmSize'] =~ /.*B\d{1,2}s$/ }
+
+    time = Time.now.utc
+    start_time = (time - 120).strftime(@@time_format)
+    end_time = (time - 60).strftime(@@time_format)
+
+    machines.each do |machine|
+      process_machine(machine, time, start_time, end_time)
+    end
+
+    ok
+  end
+
+private
+
+  def process_machine(machine, time, start_time, end_time)
+    machine_id = machine['id']
+    machine_name = machine['name']
+
+    metrics = @metrics_client.machine_burstable_metrics(machine_id, start_time, end_time)
+
+    metrics['value'].each_with_index do |metric, index|
+      process_metric(machine_name, metric, index, time)
+    end
+  end
+
+  def process_metric(machine_name, metric, index, time)
+    output [config[:schema], machine_name, @metrics_client.graphite_burstable_metrics_list[index]].join('.'), metric['timeseries'][0]['data'][0]['average'], time.to_i
+  end
+end

data/install.bash

@@ -0,0 +1,24 @@
+#!/bin/bash
+
+ENV_DIR="/az-python"
+
+check_error()
+{
+	if [ $1 -ne 0 ]; then
+		echo 'Failed to install!' >&2
+		exit $1
+	fi
+}
+
+virtualenv $ENV_DIR
+check_error $?
+
+. $ENV_DIR/bin/activate
+check_error $?
+
+pip install --upgrade 'wheel==0.31.0'
+check_error $?
+pip install --upgrade 'setuptools==39.0.1'
+check_error $?
+pip install --upgrade 'azure-cli==2.0.31'
+check_error $?

data/lib/provider/az.bash

@@ -0,0 +1,20 @@
+#!/bin/bash
+
+ENV_DIR="/az-python"
+
+while getopts c:i:p:t: option
+do
+	case "${option}"
+	in
+		c) AZ_COMMAND=${OPTARG};;
+		i) SPN_ID=${OPTARG};;
+		p) SPN_SECRET=${OPTARG};;
+		t) TENANT_ID=${OPTARG};;
+	esac
+done
+
+. $ENV_DIR/bin/activate &> /dev/null
+
+az login --service-principle -u "$SPN_ID" -p "$SPN_SECRET" --tenant "$TENANT_ID" &> /dev/null
+
+eval az $AZ_COMMAND

data/lib/provider/az.rb

@@ -0,0 +1,19 @@
+require_relative 'base'
+require 'json'
+
+class AzClient < BaseClient
+  def vm_list
+    az('vm list')
+  end
+
+  def machine_burstable_metrics(machine_id, start_time, end_time)
+    az("monitor metrics list --resource #{machine_id} --metric '#{@@burstable_metrics_list}' --interval PT1M --start-time #{start_time} --end-time #{end_time}")
+  end
+
+private
+
+  def az(command)
+    az_output = %x(#{__dir__}/az.bash -c "#{command}" -i "#{@spn_id}" -p "#{@spn_secret}" -t "#{@tenant_id}")
+    JSON.parse(az_output)
+  end
+end

data/lib/provider/base.rb

@@ -0,0 +1,16 @@
+class BaseClient
+  @@burstable_metrics = ['CPU Credits Consumed', 'CPU Credits Remaining'].freeze
+  @@burstable_metrics_list = @@burstable_metrics.join(',').freeze
+  @@output_burstable_metrics = @@burstable_metrics.map { |metric| metric.downcase.gsub!(' ','_') }.freeze
+
+  def set_credentials(spn_id = nil, spn_secret = nil, tenant_id = nil)
+    @spn_id = spn_id
+    @spn_secret = spn_secret
+    @tenant_id = tenant_id
+  end
+
+  def graphite_burstable_metrics_list
+    @@output_burstable_metrics
+  end
+
+end

data/lib/provider/file.rb

@@ -0,0 +1,18 @@
+require_relative 'base'
+require 'json'
+
+class FileClient < BaseClient
+  def vm_list
+    file_get_contents('spec/vm_list.json')
+  end
+
+  def machine_burstable_metrics(machine_id, start_time, end_time)
+    file_get_contents('spec/machine_burstable_metrics.json')
+  end
+
+  private
+
+  def file_get_contents(file_name)
+    JSON.parse(File.open(file_name, 'r') { |file| file.read })
+  end
+end

data/lib/rubygems_plugin.rb

@@ -0,0 +1,5 @@
+Gem.pre_install do
+  %x(#{File.dirname(__FILE__) + '/../install.bash'})
+
+  $?.success?
+end

data/lib/sensu-plugins-azure-burstable-metrics/version.rb

@@ -0,0 +1,9 @@
+module SensuPluginsAzureBurstableMetrics
+  class Version
+    MAJOR = 0
+    MINOR = 0
+    PATCH = 3
+
+    VER_STRING = [MAJOR, MINOR, PATCH].compact.join('.')
+  end
+end

metadata

@@ -0,0 +1,102 @@
+--- !ruby/object:Gem::Specification
+name: sensu-plugins-azure-burstable-metrics
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- Jack
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-05-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.1.0
+- !ruby/object:Gem::Dependency
+  name: sensu-plugin
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: A hacky Sensu plug-in to provide burstable metrics by calling out to
+  `az`
+email: 
+executables:
+- sensu-plugins-azure-burstable-metrics.rb
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- bin/sensu-plugins-azure-burstable-metrics.rb
+- install.bash
+- lib/provider/az.bash
+- lib/provider/az.rb
+- lib/provider/base.rb
+- lib/provider/file.rb
+- lib/rubygems_plugin.rb
+- lib/sensu-plugins-azure-burstable-metrics/version.rb
+homepage: https://github.com/ElvenSpellmaker/sensu-plugins-azure-burstable-metrics
+licenses:
+- MIT
+metadata:
+  maintainer: "@ElvenSpellmaker"
+  development_status: active
+  production_status: unstable - testing recommended
+  release_draft: 'false'
+  release_prerelease: 'false'
+post_install_message: You can use the embedded Ruby by setting EMBEDDED_RUBY=true
+  in /etc/default/sensu
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.3'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.8
+signing_key: 
+specification_version: 4
+summary: Sensu metrics plugin for Azure Burstable Machines
+test_files: []