diff --git a/lib/puppet/parser/functions/seeded_rand.rb b/lib/puppet/parser/functions/seeded_rand.rb index c51e248..848e092 100644 --- a/lib/puppet/parser/functions/seeded_rand.rb +++ b/lib/puppet/parser/functions/seeded_rand.rb @@ -1,32 +1,32 @@ # # seeded_rand.rb # Puppet::Parser::Functions.newfunction( :seeded_rand, :arity => 2, :type => :rvalue, :doc => <<-DOC @summary Generates a random whole number greater than or equal to 0 and less than MAX, using the value of SEED for repeatable randomness. @return random number greater than or equal to 0 and less than MAX @example **Usage:** seeded_rand(MAX, SEED). MAX must be a positive integer; SEED is any string. Generates a random whole number greater than or equal to 0 and less than MAX, using the value of SEED for repeatable randomness. If SEED starts with "$fqdn:", this is behaves the same as `fqdn_rand`. DOC ) do |args| require 'digest/md5' raise(ArgumentError, 'seeded_rand(): first argument must be a positive integer') unless function_is_integer([args[0]]) && args[0].to_i > 0 raise(ArgumentError, 'seeded_rand(): second argument must be a string') unless args[1].is_a? String max = args[0].to_i seed = Digest::MD5.hexdigest(args[1]).hex - Puppet::Util.deterministic_rand(seed, max) + Puppet::Util.deterministic_rand_int(seed, max) end