CMDB开发

在 /etc/puppet/modules 目录下创建如下文件结构： 

modules
└── cmdb
    ├── lib
    │   └── puppet
    │       └── reports
    │           └── cmdb.rb
    └── manifests
        └── init.pp

################ cmdb.rb ################
# cmdb.rb
require ‘puppet‘
require ‘fileutils‘
require ‘puppet/util‘
  
SEPARATOR = [Regexp.escape(File::SEPARATOR.to_s), Regexp.escape(File::ALT_SEPARATOR.to_s)].join
  
Puppet::Reports.register_report(:cmdb) do
  desc "Store server info
    These files collect quickly -- one every half hour -- so it is a good idea
    to perform some maintenance on them if you use this report (it‘s the only
    default report)."
  
  def process
    certname = self.name
    now = Time.now.gmtime
    File.open("/tmp/cmdb.json",‘a‘) do |f|
      f.write(certname)
      f.write(‘ | ‘)
      f.write(now)
      f.write("\r\n")
    end
  
  end
end


################ 配置 ################
/etc/puppet/puppet.conf
[main]
reports = cmdb
#report = true #默认
#pluginsync ＝ true #默认 

$LOAD_PATH.unshift(File.dirname(__FILE__)) unless $LOAD_PATH.include?(File.dirname(__FILE__))
require "rubygems"
require ‘pp‘
require ‘json‘
require ‘utils‘

def dmi_get_ram(cmd)

    ram_slot = []

    key_map = {
        ‘Size‘ => ‘capacity‘,
        ‘Serial Number‘ => ‘sn‘,
        ‘Type‘ => ‘model‘,
        ‘Manufacturer‘ => ‘manufactory‘,
        ‘Locator‘ => ‘slot‘,
    }

    output = Utils.facter_exec(cmd)
    devices = output.split(‘Memory Device‘)

    devices.each do |d|
      next if d.strip.empty?
      segment = {}
      d.strip.split("\n\t").each do |line|
        key, value = line.strip.split(":")
        if key_map.has_key?(key.strip)
          if key.strip == ‘Size‘
            segment[key_map[‘Size‘]] = value.chomp("MB").strip.to_i / 1024.0 # unit GB
          else
            segment[key_map[key.strip]] =  value ? value.strip : ‘‘
          end
        end
      end

      ram_slot.push(segment) unless segment.empty?
    end

    return ram_slot

end

Facter.add("ram") do
  confine :kernel => "Linux"
  setcode do

    ram_slot = []
    cmd = "dmidecode -q -t 17 2>/dev/null"
    ram_slot = dmi_get_ram(cmd)

    JSON.dump(ram_slot)

  end
end


Facter.add("ram") do
  confine :kernel => ‘windows‘
  setcode do

    ram_slot = []

    if Facter.value(:manufacturer)  =~ /.*HP.*/i
      cli = ‘C:\cmdb_report\dmidecode.exe‘
      cmd = "#{cli} -q -t 17"
      ram_slot = dmi_get_ram(cmd) if File.exist?(cli)

    else

      require ‘facter/util/wmi‘
      Facter::Util::WMI.execquery("select * from Win32_PhysicalMemory").each do | item |

        if item.DeviceLocator
          slot = item.DeviceLocator.strip
        else
          slot = ‘‘
        end

        if item.PartNumber
          model = item.PartNumber.strip
        else
          model = ‘‘
        end

        if item.SerialNumber
          sn = item.SerialNumber.strip
        else
          sn = ‘‘
        end

        if item.Manufacturer
          manufactory = item.Manufacturer.strip
        else
          manufactory = ‘‘
        end

        ram_slot.push({
         ‘capacity‘ => item.Capacity.to_i / (1024**3), # unit GB
         ‘slot‘ => slot,
         ‘model‘ => model,
         ‘sn‘ => sn,
         ‘manufactory‘ => manufactory,
       })

      end
    end

    JSON.dump(ram_slot)

  end
end