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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
 
# 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

      # Handle command execution
      if self.resource[:query]
        @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
      elsif self.resource[:path]
        if self.class == Puppet::Type::Wait_for::State
          status = case self.should
                   when :absent
                     not File.exist?(self.resource[:path])
                   when :directory
                     File.directory?(self.resource[:path])
                   when :file
                     File.file?(self.resource[:path])
                   when :present
                     File.exist?(self.resource[:path])
                   end
        end
      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, "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"
    elsif self.class == Puppet::Type::Wait_for::State
      case self.should
      when :absent
        self.fail Puppet::Error, "Filesystem element still present after max_retries"
      when :directory
        self.fail Puppet::Error, "Filesystem element isn't a directory or doesn't exist after max_retries"
      when :file
        self.fail Puppet::Error, "Filesystem element isn't a file or doesn't exist after max_retries"
      when :present
        self.fail Puppet::Error, "Filesystem element doesn't exist after max_retries"
      end
    end
  end
end