diff --git a/lib/puppet/transaction.rb b/lib/puppet/transaction.rb index e95b3fdd9..89b321f99 100644 --- a/lib/puppet/transaction.rb +++ b/lib/puppet/transaction.rb @@ -1,404 +1,409 @@ # the class that actually walks our object/state tree, collects the changes, # and performs them require 'puppet' require 'puppet/statechange' module Puppet class Transaction attr_accessor :component, :objects, :tags, :ignoreschedules, :ignoretags include Puppet::Util Puppet.config.setdefaults(:transaction, :tags => ["", "Tags to use to find objects. If this is set, then only objects tagged with the specified tags will be applied. Values must be comma-separated."] ) # Add some additional times for reporting def addtimes(hash) hash.each do |name, num| @timemetrics[name] = num end end # Apply all changes for a child, returning a list of the events # generated. def apply(child) # First make sure there are no failed dependencies child.eachdependency do |dep| skip = false if @failures[dep] > 0 child.notice "Dependency %s[%s] has %s failures" % [dep.class.name, dep.name, @failures[dep]] skip = true end if skip child.warning "Skipping because of failed dependencies" @objectmetrics[:skipped] += 1 return [] end end begin changes = child.evaluate rescue => detail if Puppet[:trace] puts detail.backtrace end child.err "Failed to retrieve current state: %s" % detail # Mark that it failed @failures[child] += 1 # And then return return [] end unless changes.is_a? Array changes = [changes] end if changes.length > 0 @objectmetrics[:out_of_sync] += 1 end childevents = changes.collect { |change| @changes << change @count += 1 change.transaction = self events = nil begin # use an array, so that changes can return more than one # event if they want events = [change.forward].flatten.reject { |e| e.nil? } rescue => detail if Puppet[:debug] puts detail.backtrace end change.state.err "change from %s to %s failed: %s" % [change.state.is_to_s, change.state.should_to_s, detail] @failures[child] += 1 next # FIXME this should support using onerror to determine # behaviour; or more likely, the client calling us # should do so end # Mark that our change happened, so it can be reversed # if we ever get to that point unless events.nil? or (events.is_a?(Array) and events.empty?) change.changed = true @objectmetrics[:applied] += 1 end events }.flatten.reject { |e| e.nil? } unless changes.empty? # Record when we last synced child.cache(:synced, Time.now) + + # Flush, if appropriate + if child.respond_to?(:flush) + child.flush + end end childevents end # Find all of the changed objects. def changed? @changes.find_all { |change| change.changed }.collect { |change| change.state.parent }.uniq end # Collect all of the targets for the list of events. This is an unintuitive # method because it recurses up through the source the event. def collecttargets(events) events.each do |event| source = event.source start = source while source Puppet::Event::Subscription.targets_of(event, source) do |sub| if target = sub.target start.info "Scheduling %s of %s[%s]" % [sub.callback, sub.target.class.name, sub.target.title] @targets[sub.target][event] = sub else raise Puppet::DevError, "Failed to find a target for %s" % sub.inspect end end source = source.parent end end end # This method does all the actual work of running a transaction. It # collects all of the changes, executes them, and responds to any # necessary events. def evaluate @count = 0 # Allow the tags to be overriden tags = self.tags || Puppet[:tags] if tags.nil? or tags == "" tags = nil else tags = [tags] unless tags.is_a? Array tags = tags.collect do |tag| tag.split(/\s*,\s*/) end.flatten end # Start logging. Puppet::Log.newdestination(@report) begin allevents = @objects.collect { |child| events = [] if (self.ignoretags or tags.nil? or child.tagged?(tags)) if self.ignoreschedules or child.scheduled? @objectmetrics[:scheduled] += 1 # Perform the actual changes seconds = thinmark do events = apply(child) end # Keep track of how long we spend in each type of object @timemetrics[child.class.name] += seconds else child.debug "Not scheduled" end else child.debug "Not tagged with %s" % tags.join(", ") end # Check to see if there are any events for this child if triggedevents = trigger(child) events += triggedevents end # Collect the targets of any subscriptions to those events collecttargets(events) # And return the events for collection events }.flatten.reject { |e| e.nil? } ensure # And then close the transaction log. Puppet::Log.close(@report) end Puppet.debug "Finishing transaction %s with %s changes" % [self.object_id, @count] allevents end # Determine whether a given object has failed. def failed?(obj) @failures[obj] > 0 end # this should only be called by a Puppet::Container object now # and it should only receive an array def initialize(objects) @objects = objects @objectmetrics = { :total => @objects.length, :out_of_sync => 0, # The number of objects that had changes :applied => 0, # The number of objects fixed :skipped => 0, # The number of objects skipped :restarted => 0, # The number of objects triggered :failed_restarts => 0, # The number of objects that fail a trigger :scheduled => 0 # The number of objects scheduled } # Metrics for distributing times across the different types. @timemetrics = Hash.new(0) # The number of objects that were triggered in this run @triggered = Hash.new { |hash, key| hash[key] = Hash.new(0) } # Targets of being triggered. @targets = Hash.new do |hash, key| hash[key] = {} end # The changes we're performing @changes = [] # The objects that have failed and the number of failures each. This # is used for skipping objects because of failed dependencies. @failures = Hash.new do |h, key| h[key] = 0 end @report = Report.new end # Generate a transaction report. def report @objectmetrics[:failed] = @failures.find_all do |name, num| num > 0 end.length # Get the total time spent @timemetrics[:total] = @timemetrics.inject(0) do |total, vals| total += vals[1] total end # Unfortunately, RRD does not deal well with changing lists of values, # so we have to pick a list of values and stick with it. In this case, # that means we record the total time, the config time, and that's about # it. We should probably send each type's time as a separate metric. @timemetrics.dup.each do |name, value| if Puppet::Type.type(name) @timemetrics.delete(name) end end # Add all of the metrics related to object count and status @report.newmetric(:objects, @objectmetrics) # Record the relative time spent in each object. @report.newmetric(:time, @timemetrics) # Then all of the change-related metrics @report.newmetric(:changes, :total => @changes.length ) @report.time = Time.now return @report end # Roll all completed changes back. def rollback @targets.clear @triggered.clear allevents = @changes.reverse.collect { |change| # skip changes that were never actually run unless change.changed Puppet.debug "%s was not changed" % change.to_s next end begin events = change.backward rescue => detail Puppet.err("%s rollback failed: %s" % [change,detail]) if Puppet[:debug] puts detail.backtrace end next # at this point, we would normally do error handling # but i haven't decided what to do for that yet # so just record that a sync failed for a given object #@@failures[change.state.parent] += 1 # this still could get hairy; what if file contents changed, # but a chmod failed? how would i handle that error? dern end collecttargets(events) if events # Now check to see if there are any events for this child. # Kind of hackish, since going backwards goes a change at a # time, not a child at a time. trigger(change.state.parent) # And return the events for collection events }.flatten.reject { |e| e.nil? } end # Trigger any subscriptions to a child. This does an upwardly recursive # search -- it triggers the passed object, but also the object's parent # and so on up the tree. def trigger(child) obj = child callbacks = Hash.new { |hash, key| hash[key] = [] } sources = Hash.new { |hash, key| hash[key] = [] } trigged = [] while obj if @targets.include?(obj) callbacks.clear sources.clear @targets[obj].each do |event, sub| # Collect all of the subs for each callback callbacks[sub.callback] << sub # And collect the sources for logging sources[event.source] << sub.callback end sources.each do |source, callbacklist| obj.debug "%s[%s] results in triggering %s" % [source.class.name, source.name, callbacklist.join(", ")] end callbacks.each do |callback, subs| message = "Triggering '%s' from %s dependencies" % [callback, subs.length] obj.notice message # At this point, just log failures, don't try to react # to them in any way. begin obj.send(callback) @objectmetrics[:restarted] += 1 rescue => detail obj.err "Failed to call %s on %s: %s" % [callback, obj, detail] @objectmetrics[:failed_restarts] += 1 if Puppet[:debug] puts detail.backtrace end end # And then add an event for it. trigged << Puppet::Event.new( :event => :triggered, :transaction => self, :source => obj, :message => message ) triggered(obj, callback) end end obj = obj.parent end if trigged.empty? return nil else return trigged end end def triggered(object, method) @triggered[object][method] += 1 end def triggered?(object, method) @triggered[object][method] end end end require 'puppet/transaction/report' # $Id$