crowbar/barclamp-ganglia

#
# Copyright 2011-2013, Dell
# Copyright 2013-2014, SUSE LINUX Products GmbH
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

class GangliaService < ServiceObject
  
  def initialize(thelogger)
    @bc_name = "ganglia"
    @logger = thelogger
  end

  class << self
    def role_constraints
      {
        "ganglia-server" => {
          "unique" => true,
          "count" => 1,
          "admin" => true
        },
        "ganglia-client" => {
          "unique" => true,
          "count" => -1
        }
      }
    end
  end
  
  def create_proposal
    @logger.debug("Ganglia create_proposal: entering")
    base = super
    @logger.debug("Ganglia create_proposal: exiting")
    base
  end
  
  def transition(inst, name, state)
    @logger.debug("Ganglia transition: make sure that network role is on all nodes: #{name} for #{state}")
    
    #
    # If we are discovering the node, make sure that we add the ganglia client or server to the node
    #
    if state == "discovered"
      @logger.debug("Ganglia transition: discovered state for #{name} for #{state}")
      db = Proposal.where(barclamp: "ganglia", name: inst).first
      role = RoleObject.find_role_by_name "ganglia-config-#{inst}"
      
      if role.override_attributes["ganglia"]["elements"]["ganglia-server"].nil? or
        role.override_attributes["ganglia"]["elements"]["ganglia-server"].empty?
        @logger.debug("Ganglia transition: make sure that ganglia-server role is on first: #{name} for #{state}")
        result = add_role_to_instance_and_node("ganglia", inst, name, db, role, "ganglia-server")
      else
        node = NodeObject.find_node_by_name name
        unless node.role? "ganglia-server"
          @logger.debug("Ganglia transition: make sure that ganglia-client role is on all nodes but first: #{name} for #{state}")
          result = add_role_to_instance_and_node("ganglia", inst, name, db, role, "ganglia-client")
        end
      end
      
      # Set up the client url
      if result 
        role = RoleObject.find_role_by_name "ganglia-config-#{inst}"
        
        # Get the server IP address
        server_ip = nil
        [ "ganglia-server" ].each do |element|
          tnodes = role.override_attributes["ganglia"]["elements"][element]
          next if tnodes.nil? or tnodes.empty?
          tnodes.each do |n|
            next if n.nil?
            node = NodeObject.find_node_by_name(n)
            pub = node.get_network_by_type("public")
            if pub and pub["address"] and pub["address"] != ""
              server_ip = pub["address"]
            else
              server_ip = node.get_network_by_type("admin")["address"]
            end
          end
        end
        
        # The Ganglia interface expects IP addresses to be passed in the link references.
        # The background collection processes explicitly use the node IP addresses to name
        # the RRD image caching directories which are used for Ganglia graphs.
        # See /var/lib/ganglia/rrds/Crowbar PoC/* on the Ganalia server node for
        # an example. PW - 05/02/2012 
        unless server_ip.nil?
          node = NodeObject.find_node_by_name(name)
          node.crowbar["crowbar"] = {} if node.crowbar["crowbar"].nil?
          node.crowbar["crowbar"]["links"] = {} if node.crowbar["crowbar"]["links"].nil?
          node.crowbar["crowbar"]["links"]["Ganglia"] = "http://#{server_ip}/ganglia/?c=Crowbar PoC&h=#{node.name}&m=load_one&r=hour&s=descending&hc=4&mc=2"
          node.save
        end 
      end
      
      @logger.debug("Ganglia transition: leaving from discovered state for #{name} for #{state}")
      a = [200, NodeObject.find_node_by_name(name).to_hash ] if result
      a = [400, "Failed to add role to node"] unless result
      return a
    end
    
    @logger.debug("Ganglia transition: leaving for #{name} for #{state}")
    [200, NodeObject.find_node_by_name(name).to_hash ]
  end
  
end