Module: Mixins::InstanceMethods

Defined in:
lib/puppet/type/wait_for.rb

Instance Method Summary collapse

Instance Method Details

#retrieve ⇒ Object

Defining a retrieve method stops Puppet looking for the getter method in the provider.



25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/puppet/type/wait_for.rb', line 25

def retrieve

  # Daniel Pittman's comment in the Exec source code:
  #
  # "We need to return :notrun to trigger evaluation; when that isn't
  # true, we *LIE* about what happened and return a "success" for the
  # value, which causes us to be treated as in_sync?, which means we
  # don't actually execute anything.  I think. --daniel 2011-03-10"
  #
  if @resource.check_all_attributes
    return :notrun
  else
    return self.should
  end
end

#sync ⇒ Object

Defining a sync stops Puppet looking for a setter method in the provider. Basically, all the provider logic is implemented here. Except the run method, although I think even that could really be moved to here.



46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
 
# File 'lib/puppet/type/wait_for.rb', line 46

def sync
  if self.resource[:seconds]
    seconds = resource[:seconds]
    info "Waiting for #{seconds} seconds..."
    sleep seconds
    return seconds
  end

  tries = self.resource[:max_retries]
  polling_frequency = self.resource[:polling_frequency]

  status = false

  begin
    tries.times do |try|

      # Only add debug messages for tries > 1 to reduce log spam.
      #
      debug("Wait_for try #{try+1}/#{tries}") if tries > 1
      @output = provider.run(self.resource[:query])

      if self.class == Puppet::Type::Wait_for::Exit_code
        status = self.should.include?(@output.exitstatus.to_i)
      elsif self.class == Puppet::Type::Wait_for::Regex
        status = @output =~ /#{self.should}/
      end

      break if status

      if polling_frequency > 0 and tries > 1
        debug("Sleeping for #{polling_frequency} seconds between tries")
        sleep polling_frequency
      end
    end
  rescue Timeout::Error
    self.fail Puppet::Error, "Query exceeded timeout", $!
  end

  unless status
    if self.class == Puppet::Type::Wait_for::Exit_code
      self.fail Puppet::Error, "Exit status #{@output.exitstatus.to_i} after max_retries"
    elsif self.class == Puppet::Type::Wait_for::Regex
      self.fail Puppet::Error, "Did not match regex after max_retries"
    end
  end
end