diff options
author | Jeff McCune <jeff@puppetlabs.com> | 2012-08-14 08:48:25 -0700 |
---|---|---|
committer | Jeff McCune <jeff@puppetlabs.com> | 2012-08-14 08:48:25 -0700 |
commit | 29aec0df1eabf8e903f45086f20e884fc3f928ed (patch) | |
tree | 41e6902f9292518f59ebedbe148442dcfab507d9 /lib/puppet/parser/functions/to_bytes.rb | |
parent | 313bc438abb32707ea8e1244b73e2ac8cda59764 (diff) | |
parent | f30885118843f2eef15c048fe2cb504d7eaf8f77 (diff) |
Merge branch '2.4.x'
* 2.4.x:
(#15872) Add to_bytes function
Diffstat (limited to 'lib/puppet/parser/functions/to_bytes.rb')
-rw-r--r-- | lib/puppet/parser/functions/to_bytes.rb | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/lib/puppet/parser/functions/to_bytes.rb b/lib/puppet/parser/functions/to_bytes.rb new file mode 100644 index 0000000..8ff73d1 --- /dev/null +++ b/lib/puppet/parser/functions/to_bytes.rb @@ -0,0 +1,28 @@ +module Puppet::Parser::Functions + newfunction(:to_bytes, :type => :rvalue, :doc => <<-EOS + Converts the argument into bytes, for example 4 kB becomes 4096. + Takes a single string value as an argument. + EOS + ) do |arguments| + + raise(Puppet::ParseError, "to_bytes(): Wrong number of arguments " + + "given (#{arguments.size} for 1)") if arguments.size != 1 + + arg = arguments[0] + + return arg if arg.is_a? Numeric + + value,prefix = */([0-9.e+-]*)\s*([^bB]?)/.match(arg)[1,2] + + value = value.to_f + case prefix + when '' then return value.to_i + when 'k' then return (value*(1<<10)).to_i + when 'M' then return (value*(1<<20)).to_i + when 'G' then return (value*(1<<30)).to_i + when 'T' then return (value*(1<<40)).to_i + when 'E' then return (value*(1<<50)).to_i + else raise Puppet::ParseError, "to_bytes(): Unknown prefix #{prefix}" + end + end +end |