service_handler.rb

# Copyright (c) 2009-2012 VMware, Inc.

module Collector
  class ServiceHandler < Handler
    def initialize(tsdb_connection, job, index, now)
      super(tsdb_connection, job, index, now)
    end

    def send_metric(name, value, tags = {})
      default_tags = {:service_type => service_type,
                      :component => component}
      tags = tags.merge(default_tags)
      super(name, value, tags)
    end

    # Process healthy instances percent for each service, default is 0 if
    # no instance provisioned.
    #
    def process_healthy_instances_metric(varz)
      healthy_instances = 0
      if varz["instances"]
        total_instances = varz["instances"].length
        healthy_instances = varz["instances"].values.count("ok")
        if (total_instances != 0)
          healthy_instances = format("%.2f",
                  healthy_instances.to_f / total_instances.to_f * 100)
        end
      end
      send_metric("services.healthy_instances", healthy_instances)
    end

    # Sum up all nodes' available_capacity value for each service, report
    # low_water & high_water value at the same time.
    #
    def process_plan_score_metric(varz)
      return unless varz.include?("plans")
      varz["plans"].each do |plan|
        send_metric("services.plans.high_water", plan["high_water"] || 0, :plan => plan["plan"])
        send_metric("services.plans.low_water", plan["low_water"] || 0, :plan => plan["plan"])
        send_metric("services.plans.score", plan["score"] || 0, :plan => plan["plan"])
        send_metric("services.plans.allow_over_provisioning",
                    plan["allow_over_provisioning"] ? 1 : 0, :plan => plan["plan"])
      end if varz["plans"]
    end

    # Get online nodes varz for each service gateway, report the total
    # number of online nodes
    #
    def process_online_nodes(varz)
      return unless varz.include?("nodes")
      send_metric("services.online_nodes", varz["nodes"].length)
    end

    def service_type    # "mysql", "postgresql", "mongodb" ...
      "unknown"
    end

    def component       # "node", "gateway"
      "unknown"
    end
  end
end