Last change
on this file since 30960 was 30960, checked in by davidb, 7 years ago |
Switch to using Puppet to provision machine. Strongly based on files developed for spark-hdfs cluster
|
File size:
1.1 KB
|
Line | |
---|
1 | module Puppet::Parser::Functions
|
---|
2 | newfunction(:to_bytes, :type => :rvalue, :doc => <<-EOS
|
---|
3 | Converts the argument into bytes, for example 4 kB becomes 4096.
|
---|
4 | Takes a single string value as an argument.
|
---|
5 | These conversions reflect a layperson's understanding of
|
---|
6 | 1 MB = 1024 KB, when in fact 1 MB = 1000 KB, and 1 MiB = 1024 KiB.
|
---|
7 | EOS
|
---|
8 | ) do |arguments|
|
---|
9 |
|
---|
10 | raise(Puppet::ParseError, "to_bytes(): Wrong number of arguments " +
|
---|
11 | "given (#{arguments.size} for 1)") if arguments.size != 1
|
---|
12 |
|
---|
13 | arg = arguments[0]
|
---|
14 |
|
---|
15 | return arg if arg.is_a? Numeric
|
---|
16 |
|
---|
17 | value,prefix = */([0-9.e+-]*)\s*([^bB]?)/.match(arg)[1,2]
|
---|
18 |
|
---|
19 | value = value.to_f
|
---|
20 | case prefix
|
---|
21 | when '' then return value.to_i
|
---|
22 | when 'k' then return (value*(1<<10)).to_i
|
---|
23 | when 'M' then return (value*(1<<20)).to_i
|
---|
24 | when 'G' then return (value*(1<<30)).to_i
|
---|
25 | when 'T' then return (value*(1<<40)).to_i
|
---|
26 | when 'P' then return (value*(1<<50)).to_i
|
---|
27 | when 'E' then return (value*(1<<60)).to_i
|
---|
28 | else raise Puppet::ParseError, "to_bytes(): Unknown prefix #{prefix}"
|
---|
29 | end
|
---|
30 | end
|
---|
31 | end
|
---|
Note:
See
TracBrowser
for help on using the repository browser.