sensu-plugins-azurerm 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f64abc5f248cc77f5fac058d2a0ae1c1bbf5d8cd
-  data.tar.gz: 002b3616bf8fe69d3eb0ba3342b3b0f56a678c5f
+  metadata.gz: eb55a621b355424cc965fb60bfbd7a8eb18efb34
+  data.tar.gz: 85f6319fd7de0a5bce89a71a5d5ffd4bf1f9f2b6
 SHA512:
-  metadata.gz: 58bf2d82e569a834da181226e69364e42fe0f514d6a92b5f21ed7437b3f247ec5878f30863825dbd22d34ae82511b770ec7623844eadbccfc4fe4016b058410e
-  data.tar.gz: 9900e040280568af5adbf9f27f4d097b1ff8b4ee14f6edf81fa64760d721b5ff47ef8b2d1118a0bacbe6c6c0b6b220f12534778cf657d7b07ce7a48edae588fb
+  metadata.gz: 901c1808a79b961d78247e4c00189bad6d787b111fc11721702b2bcd201ed7c2ad40f1beed1febef744e0f8e7a292a4ade8c927464f1ac6f9e69c8587617fa4c
+  data.tar.gz: f82fc17d7f9ab3328b9f5716b059ce544fcf80685d165968f111f0e0bcd7acbb5945baeb98229ccc76bce84948a365f1397696cf6a41ffd345477ba6aa96819f

data/CHANGELOG.md

@@ -3,6 +3,14 @@ This project adheres to [Semantic Versioning](http://semver.org/).
 
 This CHANGELOG follows the format listed at [Keep A Changelog](http://keepachangelog.com/)
 
+## 0.0.3 - 2016-11-24
+### Added
+- Virtual Network Gateways:
+  - Added functionality to check Failover connections
+  - Made it possible to specify a custom Scheme for Metrics
+- Service Bus:
+  - Added a Metric to expose the the Message Count for a Service Bus Topic
+
 ## 0.0.2 - 2016-11-23
 ### Added
 - Virtual Network Gateways:

data/README.md

@@ -11,14 +11,17 @@
  - check-azurerm-core-usage.rb
  - check-azurerm-virtual-machines-usage.rb
  - check-azurerm-virtual-network-gateway-connected.rb
+ - check-azurerm-virtual-network-gateway-failover-connected.rb
+ - metric-azurerm-service-bus-subscription-message-count.rb
  - metric-azurerm-virtual-network-gateway-usage.rb
 
-
 ## Files
 
 * /bin/check-azurerm-core-usage.rb
 * /bin/check-azurerm-virtual-machines-usage.rb
 * /bin/check-azurerm-virtual-network-gateway-connected.rb
+* /bin/check-azurerm-virtual-network-gateway-failover-connected.rb
+* /bin/metric-azurerm-service-bus-subscription-message-count.rb
 * /bin/metric-azurerm-virtual-network-gateway-usage.rb
 
 
@@ -83,6 +86,63 @@
                              -name "gatewayname"
 ```
 
+**check-azurerm-virtual-network-gateway-failover-connected.rb**
+
+```
+./check-azurerm-virtual-network-gateway-failover-connected.rb
+                           -r "resourcegroup"
+                           -p "primaryname"
+                           -s "secondaryname"
+
+./check-azurerm-virtual-network-gateway-failover-connected.rb
+                           -t "00000000-0000-0000-0000-000000000000"
+                           -c "00000000-0000-0000-0000-000000000000"
+                           -S "00000000-0000-0000-0000-000000000000"
+                           -s "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
+                           -r "resourcegroup"
+                           -p "primaryname"
+                           -s "secondaryname"
+
+check-azurerm-virtual-network-gateway-failover-connected.rb
+                          -tenant "00000000-0000-0000-0000-000000000000"
+                          -client "00000000-0000-0000-0000-000000000000"
+                          -clientSecret "00000000-0000-0000-0000-000000000000"
+                          -subscription_id "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
+                          -resourceGroup "resourcegroup"
+                          -primaryName "gatewayname"
+                          -secondaryName "gatewayname"
+```
+
+**metric-azurerm-service-bus-subscription-message-count.rb**
+```
+./metric-azurerm-service-bus-subscription-message-count.rb
+                           --resourceGroup "resourcegroup"
+                           --namespace "namespace"
+                           --topic "topic"
+                           --subscriptionName "subscriptionName"
+
+./metric-azurerm-service-bus-subscription-message-count.rb
+                           -t "00000000-0000-0000-0000-000000000000"
+                           -c "00000000-0000-0000-0000-000000000000"
+                           -S "00000000-0000-0000-0000-000000000000"
+                           -s "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
+                           --resourceGroup "resourcegroup"
+                           --namespace "namespace"
+                           --topic "topic"
+                           --subscriptionName "subscriptionName"
+
+./metric-azurerm-service-bus-subscription-message-count.rb
+                           --tenant "00000000-0000-0000-0000-000000000000"
+                           --client "00000000-0000-0000-0000-000000000000"
+                           --clientSecret "00000000-0000-0000-0000-000000000000"
+                           --subscription_id "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
+                           --resourceGroup "resourcegroup"
+                           --namespaceName "namespace"
+                           --topicName "topic"
+                           --subscriptionName "subscriptionName"
+                           --customScheme "foo"
+```
+
 **metric-azurerm-virtual-network-gateway-usagerb**
 ```
 ./metric-azurerm-virtual-network-gateway-usage.rb -r "resourcegroup" -n "gatewayname"
@@ -95,12 +155,13 @@
                           -r "resourcegroup" -n "gatewayname"
 
 ./metric-azurerm-virtual-network-gateway-usage.rb
-                          -tenant "00000000-0000-0000-0000-000000000000"
-                          -client "00000000-0000-0000-0000-000000000000"
-                          -clientSecret "00000000-0000-0000-0000-000000000000"
-                          -subscription_id "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
-                          -resourceGroup "resourcegroup"
-                          -name "gatewayname"
+                          --tenant "00000000-0000-0000-0000-000000000000"
+                          --client "00000000-0000-0000-0000-000000000000"
+                          --clientSecret "00000000-0000-0000-0000-000000000000"
+                          --subscription_id "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
+                          --resourceGroup "resourcegroup"
+                          --name "gatewayname"
+                          --customScheme "foo"
 ```
 
 ## Installation

data/bin/check-azurerm-virtual-machines-usage.rb

@@ -116,9 +116,9 @@ class CheckAzureRMVMUsage < Sensu::Plugin::Check::CLI
       ok message
     end
 
-  #rescue => e
-  #  puts "Error: exception: #{e}"
-  #  critical
+  rescue => e
+    puts "Error: exception: #{e}"
+    critical
   end
 
 end

data/bin/check-azurerm-virtual-network-gateway-failure-connected.rb

@@ -0,0 +1,136 @@
+#! /usr/bin/env ruby
+#
+# check-azurerm-virtual-network-gateway-failover-connected
+#
+# DESCRIPTION:
+#   This plugin checks that at least one of the specified Virtual Network Gateways is connected.
+#   Firing a Critical alert if both are not in the Connected state.
+#
+# OUTPUT:
+#   plain-text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: azure_mgmt_network
+#   gem: sensu-plugin
+#
+# USAGE:
+#   ./check-azurerm-virtual-network-gateway-failover-connected.rb
+#                             -r "resourcegroup"
+#                             -p "primaryname"
+#                             -s "secondaryname"
+#
+#   ./check-azurerm-virtual-network-gateway-failover-connected.rb
+#                             -t "00000000-0000-0000-0000-000000000000"
+#                             -c "00000000-0000-0000-0000-000000000000"
+#                             -S "00000000-0000-0000-0000-000000000000"
+#                             -s "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
+#                             -r "resourcegroup"
+#                             -p "primaryname"
+#                             -s "secondaryname"
+#
+#   ./check-azurerm-virtual-network-gateway-failover-connected.rb
+#                             -tenant "00000000-0000-0000-0000-000000000000"
+#                             -client "00000000-0000-0000-0000-000000000000"
+#                             -clientSecret "00000000-0000-0000-0000-000000000000"
+#                             -subscription_id "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
+#                             -resourceGroup "resourcegroup"
+#                             -primaryName "gatewayname"
+#                             -secondaryName "gatewayname"
+#
+# NOTES:
+#
+# LICENSE:
+#   Tom Harvey
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+#
+
+require 'sensu-plugin/check/cli'
+require 'sensu-plugins-azurerm'
+require 'azure_mgmt_network'
+
+class CheckAzureRMVirtualNetworkGatewayConnected < Sensu::Plugin::Check::CLI
+  include SensuPluginsAzureRM
+
+  option :tenant_id,
+       description: 'ARM Tenant ID. Either set ENV[\'ARM_TENANT_ID\'] or provide it as an option',
+       short: '-t ID',
+       long: '--tenant ID',
+       default: ENV['ARM_TENANT_ID'] # TODO: can we pull these out from the Check too?
+
+  option :client_id,
+       description: 'ARM Client ID. Either set ENV[\'ARM_CLIENT_ID\'] or provide it as an option',
+       short: '-c ID',
+       long: '--client ID',
+       default: ENV['ARM_CLIENT_ID']
+
+  option :client_secret,
+       description: 'ARM Client Secret. Either set ENV[\'ARM_CLIENT_SECRET\'] or provide it as an option',
+       short: '-s SECRET',
+       long: '--clientSecret SECRET',
+       default: ENV['ARM_CLIENT_SECRET']
+
+  option :subscription_id,
+       description: 'ARM Subscription ID',
+       short: '-S ID',
+       long: '--subscription ID',
+       default: ENV['ARM_SUBSCRIPTION_ID']
+
+  option :resource_group_name,
+       description: 'Azure Resource Group Name',
+       short: '-r RESOURCEGROUP',
+       long: '--resourceGroup RESOURCEGROUP'
+
+  option :primary_name,
+       description: 'Azure Virtual Network Connection Gateway Primary Name',
+       short: '-p NAME',
+       long: '--primaryName NAME'
+
+  option :secondary_name,
+       description: 'Azure Virtual Network Connection Gateway Secondary Name',
+       short: '-s NAME',
+       long: '--secondaryName NAME'
+
+  def run
+    tenantId = config[:tenant_id]
+    clientId = config[:client_id]
+    clientSecret = config[:client_secret]
+    subscriptionId = config[:subscription_id]
+
+    resource_group_name = config[:resource_group_name]
+
+    primaryName = config[:primary_name]
+    secondaryName = config[:secondary_name]
+
+    usage = NetworkUsage.new()
+
+    networkClient = usage.buildVirtualNetworkClient(tenantId, clientId, clientSecret, subscriptionId)
+
+    primaryResult = networkClient.get(resource_group_name, primaryName)
+    primary_connection_state = primaryResult.connection_status
+    primary_inbound = primaryResult.ingress_bytes_transferred
+    primary_outbound = primaryResult.egress_bytes_transferred
+
+    secondaryResult = networkClient.get(resource_group_name, secondaryName)
+    secondary_connection_state = secondaryResult.connection_status
+    secondary_inbound = secondaryResult.ingress_bytes_transferred
+    secondary_outbound = secondaryResult.egress_bytes_transferred
+    message = "Primary: State is '#{primary_connection_state}'. Usage is #{primary_inbound} in / #{primary_outbound} out.\n"
+    message += "Secondary: State is '#{secondary_connection_state}'. Usage is #{secondary_inbound} in / #{secondary_outbound} out."
+
+    if primaryResult.connection_status.casecmp("connected") == 0 ||
+       secondaryResult.connection_status.casecmp("connected") == 0
+      ok message
+    else
+      critical message
+    end
+
+  rescue => e
+    puts "Error: exception: #{e}"
+    critical
+  end
+
+end

data/bin/metric-azurerm-service-bus-subscription-message-count.rb

@@ -0,0 +1,137 @@
+#! /usr/bin/env ruby
+#
+# metric-azurerm-service-bus-subscription-message-count
+#
+# DESCRIPTION:
+#   This plugin exposes the Service Bus Subscription Message Counts as a Metric
+#
+# OUTPUT:
+#   plain-text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: azure_mgmt_service_bus
+#   gem: sensu-plugin
+#
+# USAGE:
+#
+#   ./metric-azurerm-service-bus-subscription-message-count.rb
+#                             --resourceGroup "resourcegroup"
+#                             --namespace "namespace"
+#                             --topic "topic"
+#                             --subscriptionName "subscriptionName"
+#
+#   ./metric-azurerm-service-bus-subscription-message-count.rb
+#                             -t "00000000-0000-0000-0000-000000000000"
+#                             -c "00000000-0000-0000-0000-000000000000"
+#                             -S "00000000-0000-0000-0000-000000000000"
+#                             -s "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
+#                             --resourceGroup "resourcegroup"
+#                             --namespace "namespace"
+#                             --topic "topic"
+#                             --subscriptionName "subscriptionName"
+#
+#   ./metric-azurerm-service-bus-subscription-message-count.rb
+#                             --tenant "00000000-0000-0000-0000-000000000000"
+#                             --client "00000000-0000-0000-0000-000000000000"
+#                             --clientSecret "00000000-0000-0000-0000-000000000000"
+#                             --subscription_id "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
+#                             --resourceGroup "resourcegroup"
+#                             --namespaceName "namespace"
+#                             --topicName "topic"
+#                             --subscriptionName "subscriptionName"
+#                             --customScheme "foo"
+#
+# NOTES:
+#
+# LICENSE:
+#   Tom Harvey
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+#
+
+require 'sensu-plugin/metric/cli'
+require 'sensu-plugins-azurerm'
+require 'azure_mgmt_service_bus'
+
+class MetricAzureRMServiceBusSubscriptionMessageCount < Sensu::Plugin::Metric::CLI::Statsd
+  include SensuPluginsAzureRM
+
+  option :tenant_id,
+       description: 'ARM Tenant ID. Either set ENV[\'ARM_TENANT_ID\'] or provide it as an option',
+       short: '-t ID',
+       long: '--tenant ID',
+       default: ENV['ARM_TENANT_ID'] # TODO: can we pull these out from the Check too?
+
+  option :client_id,
+       description: 'ARM Client ID. Either set ENV[\'ARM_CLIENT_ID\'] or provide it as an option',
+       short: '-c ID',
+       long: '--client ID',
+       default: ENV['ARM_CLIENT_ID']
+
+  option :client_secret,
+       description: 'ARM Client Secret. Either set ENV[\'ARM_CLIENT_SECRET\'] or provide it as an option',
+       short: '-s SECRET',
+       long: '--clientSecret SECRET',
+       default: ENV['ARM_CLIENT_SECRET']
+
+  option :subscription_id,
+       description: 'ARM Subscription ID',
+       short: '-S ID',
+       long: '--subscription ID',
+       default: ENV['ARM_SUBSCRIPTION_ID']
+
+  option :resource_group_name,
+       description: 'Azure Resource Group Name',
+       long: '--resourceGroup RESOURCEGROUP'
+
+  option :namespace_name,
+       description: 'Azure Service Bus Namespace Name',
+       long: '--namespaceName NAMESPACE'
+
+  option :topic_name,
+       description: 'Azure Service Bus Topic Name',
+       long: '--topicName TOPIC'
+
+  option :subscription_name,
+       description: 'Azure Service Bus Topic Name',
+       long: '--subscriptionName TOPIC'
+
+ option :customScheme,
+        description: 'Metric naming scheme, text to prepend to .$parent.$child',
+        long: '--customScheme SCHEME',
+        default: "azurerm.servicebus"
+
+  def run
+    tenantId = config[:tenant_id]
+    clientId = config[:client_id]
+    clientSecret = config[:client_secret]
+    subscriptionId = config[:subscription_id]
+
+    resource_group_name = config[:resource_group_name]
+    namespace_name = config[:namespace_name]
+    topic_name = config[:topic_name]
+    subscription_name = config[:subscription_name]
+
+    usage = ServiceBusUsage.new()
+
+    serviceBusClient = usage.buildServiceBusClient(tenantId, clientId, clientSecret, subscriptionId)
+    result = serviceBusClient.get(resource_group_name, namespace_name, topic_name, subscription_name)
+
+    count = result.message_count
+
+    timestamp = Time.now.utc.to_i
+    scheme = config[:customScheme]
+    name = [scheme, resource_group_name, namespace_name, topic_name, subscription_name].join('.').tr(' ', '_').tr('{}', '').tr('[]', '')
+    metricName = [name, "message_count"].join('.')
+
+    output metricName, count, timestamp
+    ok
+  rescue => e
+    puts "Error: exception: #{e}"
+    critical
+  end
+
+end

data/bin/metric-azurerm-virtual-network-gateway-usage.rb

@@ -26,12 +26,13 @@
 #                             -r "resourcegroup" -n "gatewayname"
 #
 #   ./metric-azurerm-virtual-network-gateway-usage.rb
-#                             -tenant "00000000-0000-0000-0000-000000000000"
-#                             -client "00000000-0000-0000-0000-000000000000"
-#                             -clientSecret "00000000-0000-0000-0000-000000000000"
-#                             -subscription_id "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
-#                             -resourceGroup "resourcegroup"
-#                             -name "gatewayname"
+#                             --tenant "00000000-0000-0000-0000-000000000000"
+#                             --client "00000000-0000-0000-0000-000000000000"
+#                             --clientSecret "00000000-0000-0000-0000-000000000000"
+#                             --subscription_id "1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ12345678901234"
+#                             --resourceGroup "resourcegroup"
+#                             --name "gatewayname"
+#                             --customScheme "foo"
 #
 # NOTES:
 #
@@ -41,11 +42,11 @@
 #   for details.
 #
 
-require 'sensu-plugin/check/cli'
+require 'sensu-plugin/metric/cli'
 require 'sensu-plugins-azurerm'
 require 'azure_mgmt_network'
 
-class MetricAzureRMVirtualNetworkGatewayUsage < Sensu::Plugin::Check::CLI
+class MetricAzureRMVirtualNetworkGatewayUsage < Sensu::Plugin::Metric::CLI::Statsd
   include SensuPluginsAzureRM
 
   option :tenant_id,
@@ -82,6 +83,11 @@ class MetricAzureRMVirtualNetworkGatewayUsage < Sensu::Plugin::Check::CLI
        short: '-n NAME',
        long: '--name NAME'
 
+ option :customScheme,
+        description: 'Metric naming scheme, text to prepend to .$parent.$child',
+        long: '--customScheme SCHEME',
+        default: "azurerm.virtualnetworkgateway"
+
   def run
     tenantId = config[:tenant_id]
     clientId = config[:client_id]
@@ -100,14 +106,14 @@ class MetricAzureRMVirtualNetworkGatewayUsage < Sensu::Plugin::Check::CLI
     outbound = result.egress_bytes_transferred
 
     timestamp = Time.now.utc.to_i
-    scheme = config[:scheme]
+    scheme = config[:customScheme]
     name = [scheme, resource_group_name, name].join('.').tr(' ', '_').tr('{}', '').tr('[]', '')
     inboundName = [name, "inbound"].join('.')
     outboundName = [name, "outbound"].join('.')
 
     output inboundName, inbound, timestamp
     output outboundName, outbound, timestamp
-
+    ok
   rescue => e
     puts "Error: exception: #{e}"
     critical

data/lib/sensu-plugins-azurerm.rb

@@ -1,3 +1,4 @@
 require 'sensu-plugins-azurerm/compute_usage'
 require 'sensu-plugins-azurerm/network_usage'
+require 'sensu-plugins-azurerm/servicebus_usage'
 require 'sensu-plugins-azurerm/version'

data/lib/sensu-plugins-azurerm/servicebus_usage.rb

@@ -0,0 +1,16 @@
+module SensuPluginsAzureRM
+
+  class ServiceBusUsage
+
+    def buildServiceBusClient(tenant_id, client_id, secret, subscription_id)
+      token_provider = MsRestAzure::ApplicationTokenProvider.new(tenant_id, client_id, secret)
+      credentials = MsRest::TokenCredentials.new(token_provider)
+      client = Azure::ARM::ServiceBus::ServiceBusManagementClient.new(credentials)
+      client.subscription_id = subscription_id
+
+      usageClient = Azure::ARM::ServiceBus::Subscriptions.new(client)
+      usageClient
+    end
+
+  end
+end

data/lib/sensu-plugins-azurerm/version.rb

@@ -2,7 +2,7 @@ module SensuPluginsAzureRM
   module Version
     MAJOR = 0
     MINOR = 0
-    PATCH = 2
+    PATCH = 3
     VER_STRING = [MAJOR, MINOR, PATCH].compact.join('.')
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sensu-plugins-azurerm
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Tom Harvey
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-11-23 00:00:00.000000000 Z
+date: 2016-11-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: azure_mgmt_compute
@@ -38,6 +38,20 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 0.8.0
+- !ruby/object:Gem::Dependency
+  name: azure_mgmt_service_bus
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.8.0
 - !ruby/object:Gem::Dependency
   name: ms_rest_azure
   requirement: !ruby/object:Gem::Requirement
@@ -198,6 +212,8 @@ executables:
 - check-azurerm-core-usage.rb
 - check-azurerm-virtual-machines-usage.rb
 - check-azurerm-virtual-network-gateway-connected.rb
+- check-azurerm-virtual-network-gateway-failure-connected.rb
+- metric-azurerm-service-bus-subscription-message-count.rb
 - metric-azurerm-virtual-network-gateway-usage.rb
 extensions: []
 extra_rdoc_files: []
@@ -205,9 +221,12 @@ files:
 - bin/check-azurerm-core-usage.rb
 - bin/check-azurerm-virtual-machines-usage.rb
 - bin/check-azurerm-virtual-network-gateway-connected.rb
+- bin/check-azurerm-virtual-network-gateway-failure-connected.rb
+- bin/metric-azurerm-service-bus-subscription-message-count.rb
 - bin/metric-azurerm-virtual-network-gateway-usage.rb
 - lib/sensu-plugins-azurerm/compute_usage.rb
 - lib/sensu-plugins-azurerm/network_usage.rb
+- lib/sensu-plugins-azurerm/servicebus_usage.rb
 - lib/sensu-plugins-azurerm/version.rb
 - lib/sensu-plugins-azurerm.rb
 - LICENSE