puppet的hash合併函數

puppet官方提供的create_resources函數的參數生成版,主要使用場景用以轉換對應傳遞的hash字段給erb模板使用.

#　相關代碼的github地址,傳送門如下:

代碼: hash_merge.rb

erb部分的使用,可參考筆者前陣子的rsync管理模塊的 類引用 和 erb迭代

函數文件: hash_merge.rb  函數註釋部分寫在代碼中了,因此直接上代碼如下:

module Puppet::Parser::Functions
  newfunction(:hash_merge, :type => :rvalue, :doc => <<-EOS

*Examples:*
  $ddd = {'t1' => 'aa', 't2' => 'ab'}
  $ttt = {
     'key_1' => { 't3' => 'a1'},
     'key_2' => { 't1' => 'zz', 't3' => 'a8' }
  }
  hash_merge($ttt, $ddd)

Would result in: "
  {
    'key_1' => { 't3' => 'a1', 't1' => 'aa', 't2' => 'ab' }, 
    'key_2' => { 't1' => 'zz', 't3' => 'a8', 't2' => 'ab' }
  }
"
    EOS
  ) do |arguments|
      
     # Technically we support two arguments but only first is mandatory ...
     raise(Puppet::ParseError, "hash_merge(): Wrong number of arguments " +
       "given (#{arguments.size} for 1)") if arguments.size < 1
    
     res = arguments[0]
    
     unless res.is_a?(Hash)
       raise Puppet::ParseError, "hash_merge(): The argument must be a hash"
     end

     if arguments.size == 2
       t_hash = arguments[1] || {}
       res.each_key do |k|
         t_hash.each_key do |t|
           res[k][t] = t_hash[t] unless res[k].key?(t) 
         end 
       end
     end
     
     return res
  end
end
# vim: set ts=2 sw=2 et :