通过Puppet角色提取配置文件，推送到另一台服务器上

云中漫步

#!/usr/bin/ruby　　
# encoding: utf-8
　　
require 'md5'
　　
require 'fileutils'
　　

　　
class Conf
　　
  def initialize(groups)
　　
    @role = ['ha','ssl']
　　
    @conf_dir = '/modules'
　　
    @tmp_dir = '/w5_sync'
　　
    @groups = groups
　　
    @time = Time.now.strftime('%F %X')
　　
  end
　　

　　
  def ha_file?
　　
     File.exists?("#{@conf_dir}/ha/files/#{@groups}/haproxy.cfg")
　　
  end
　　

　　
  def ssl_file?(ssl_conf)
　　
    File.exists?("#{@conf_dir}/ssl/files/#{@groups}/#{ssl_conf}")
　　
  end
　　

　　
  def nodes
　　
    `grep -rw #{@groups} /var/yaml/node/|awk -F'[ /]' '{print $7}'|sed 's/.yaml://'`.split("\n").sort.uniq
　　
  end
　　

　　
  def node_file?(node,file)
　　
    File.exists?("#{@tmp_dir}/#{node}/#{file}")
　　
  end
　　

　　
  def check_conf(file,role)
　　
    if role == 'ha'
　　
      MD5.hexdigest(File.readlines("#{@conf_dir}/ha/files/#{@groups}/#{file}").to_s)
　　
    elsif role == 'ssl'
　　
      MD5.hexdigest(File.readlines("#{@conf_dir}/ssl/files/#{@groups}/#{file}").to_s)
　　
    end
　　
  end
　　

　　
  def check_sync(nodes,file)
　　
    `md5sum "#{@tmp_dir}/#{nodes}/#{file}"|awk '{print $1}'`.chomp
　　
  end
　　

　　
  def ha_start
　　
    puts "#{@time} 开始执行ha角色,#{@groups}"
　　
    if self.ha_file?
　　
      self.nodes.each do |n|
　　
        if ! File.directory?("#{@tmp_dir}/#{n}")
　　
          FileUtils.mkdir("#{@tmp_dir}/#{n}") unless File.directory?("#{@tmp_dir}/#{n}")
　　
        end
　　

　　
        #sync haproxy.cfg
　　
        if self.node_file?(n,'haproxy_cfg')
　　
          if self.check_conf('haproxy.cfg','ha') != self.check_sync(n,'haproxy_cfg')
　　
            puts "发现#{@tmp_dir}/#{n}/haproxy_cfg 和 #{@groups}/haproxy.cfg MD5不一致, 通过#{@groups}/haproxy.cfg 进行覆盖"
　　
            FileUtils.cp("#{@conf_dir}/ha/files/#{@groups}/haproxy.cfg","#{@tmp_dir}/#{n}/haproxy_cfg")
　　
          end
　　
        else
　　
          puts "发现#{@tmp_dir}/#{n}/haproxy_cfg 不存在, 从#{@groups}/haproxy.cfg 进行拷贝"
　　
          FileUtils.cp("#{@conf_dir}/ha/files/#{@groups}/haproxy.cfg","#{@tmp_dir}/#{n}/haproxy_cfg")
　　
        end
　　

　　
        #sync backend.d
　　
        unless system("rsync --delete -raz #{@conf_dir}/ha/files/#{@groups}/backend.d #{@tmp_dir}/#{n}")
　　
          puts "rsync --delete -raz #{@conf_dir}/ha/files/#{@groups}/backend.d #{@tmp_dir}/#{n} 执行失败"
　　
        end
　　
      end
　　
    else
　　
      self.nodes.each do |n|
　　
        FileUtils.mkdir("#{@tmp_dir}/#{n}") unless File.directory?("#{@tmp_dir}/#{n}")
　　
        if self.node_file?(n,'haproxy_cfg') && ! self.ha_file?
　　
          puts "发现#{@groups}/haproxy.cfg 不存在, 从#{n}移除文件 #{@tmp_dir}/#{n}/haproxy_cfg"
　　
          FileUtils.remove_file("#{@tmp_dir}/#{n}/haproxy_cfg")
　　
        end
　　
      end
　　
    end
　　
  end
　　

　　
  def ssl_start
　　
    puts "#{@time} 开始执行ssl角色,#{@groups}"
　　
    if self.ssl_file?('nginx-dx.conf')
　　
      self.nodes.each do |n|
　　
        FileUtils.mkdir("#{@tmp_dir}/#{n}") unless File.directory?("#{@tmp_dir}/#{n}")
　　
        if self.node_file?(n,'nginx_cfg')
　　
          if self.check_conf('nginx-dx.conf','ssl') != self.check_sync(n,'nginx_cfg')
　　
            puts "发现#{@tmp_dir}/#{n}/nginx_cfg 和 #{@groups}/nginx-dx.conf MD5不一致, 通过#{@groups}/nginx-dx.conf 进行覆盖"
　　
            FileUtils.cp("#{@conf_dir}/ssl/files/#{@groups}/nginx-dx.conf","#{@tmp_dir}/#{n}/nginx_cfg")
　　
          end
　　
        else
　　
          puts "发现#{@tmp_dir}/#{n}/nginx_cfg 不存在, 从#{@groups}/nginx-dx.conf 进行拷贝"
　　
          FileUtils.cp("#{@conf_dir}/ssl/files/#{@groups}/nginx-dx.conf","#{@tmp_dir}/#{n}/nginx_cfg")
　　
        end
　　
      end
　　
    else
　　
      if self.ssl_file?('nginx.conf')
　　
        self.nodes.each do |n|
　　
          FileUtils.mkdir("#{@tmp_dir}/#{n}") unless File.directory?("#{@tmp_dir}/#{n}")
　　
          if self.node_file?(n,'nginx_cfg')
　　
            if self.check_conf('nginx.conf','ssl') != self.check_sync(n,'nginx_cfg')
　　
              puts "发现#{@tmp_dir}/#{n}/nginx_cfg 和 #{@groups}/nginx.conf MD5不一致, 通过#{@groups}/nginx.conf 进行覆盖"
　　
              FileUtils.cp("#{@conf_dir}/ssl/files/#{@groups}/nginx.conf","#{@tmp_dir}/#{n}/nginx_cfg")
　　
            end
　　
          else
　　
            puts "发现#{@tmp_dir}/#{n}/nginx_cfg 不存在, 从#{@groups}/nginx.conf 进行拷贝"
　　
            FileUtils.cp("#{@conf_dir}/ssl/files/#{@groups}/nginx.conf","#{@tmp_dir}/#{n}/nginx_cfg")
　　
          end
　　
        end
　　
      else
　　
        self.nodes.each do |n|
　　
          FileUtils.mkdir("#{@tmp_dir}/#{n}") unless File.directory?("#{@tmp_dir}/#{n}")
　　
          if self.node_file?(n,'nginx_cfg')
　　
            puts "发现#{@groups}/nginx.conf或nginx-dx.conf 不存在, 从#{n}移除文件 #{@tmp_dir}/#{n}/nginx_cfg"
　　
            FileUtils.remove_file("#{@tmp_dir}/#{n}/nginx_cfg")
　　
          end
　　
        end
　　
      end
　　
    end
　　
  end
　　
end
　　

　　
run = []
　　
Dir.glob("/modules/ha/files/*").reject{|i| File.basename(i) == 'common'}.each do |groups|
　　
  run << Thread.new { Conf.new(File.basename(groups)).ha_start }
　　
end
　　

　　
Dir.glob("/modules/ssl/files/*").reject{|i| File.basename(i) == 'common'}.each do |groups|
　　
  run << Thread.new { Conf.new(File.basename(groups)).ssl_start }
　　
end
　　
run.each { |t| t.join }
　　

　　
tmp_cfg='/tmp/w5_sync'
　　
system("/usr/bin/rsync --delete --exclude='getdate' --exclude='inventorynumber' --exclude='rtable' -avz #{tmp_cfg}/* 0.0.0.0::sync")