summaryrefslogtreecommitdiff
path: root/lib/puppet/parser/functions/round.rb
blob: 489c3014dc7c51d5e9d96c73cdd4cf45791ad390 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#
# round.rb
#

module Puppet::Parser::Functions
  newfunction(:round, :type => :rvalue, :doc => <<-EOS
  Rounds a number to the nearest integer

  *Examples:*

  round(2.9)

  returns: 3

  round(2.4)

  returns: 2

  EOS
  ) do |args|

    raise Puppet::ParseError, "round(): Wrong number of arguments given #{args.size} for 1" if args.size != 1
    raise Puppet::ParseError, "round(): Expected a Numeric, got #{args[0].class}" unless args[0].is_a? Numeric

    value = args[0]

    if value >= 0
      Integer(value + 0.5)
    else
      Integer(value - 0.5)
    end
  end
end