131 lines
4.3 KiB
Ruby
131 lines
4.3 KiB
Ruby
# File-writing code is from the store report processor in puppet's master branch.
|
|
# The rest of the code is from the puppetdb report processor in puppetdb's 2.3.x branch.
|
|
require 'puppet'
|
|
|
|
Puppet::Reports.register_report(:puppetdb_file) do
|
|
desc <<-DESC
|
|
Save report information to a file for sending to PuppetDB via the REST API
|
|
later. Reports are serialized to JSON format and may then submitted to puppetdb.
|
|
DESC
|
|
|
|
# Process the report by formatting it into a PuppetDB 'store report'
|
|
# written to disk to be submitted to PuppetDB later.
|
|
#
|
|
# @return [void]
|
|
def process
|
|
dir = File.join(Puppet[:reportdir], host)
|
|
if ! Puppet::FileSystem.exist?(dir)
|
|
FileUtils.mkdir_p(dir)
|
|
FileUtils.chmod_R(0750, dir)
|
|
end
|
|
now = Time.now.gmtime
|
|
name = %w{year month day hour min}.collect do |method|
|
|
"%02d" % now.send(method).to_s
|
|
end.join("") + "_puppetdb.json"
|
|
file = File.join(dir, name)
|
|
begin
|
|
Puppet::Util.replace_file(file, 0640) do |fh|
|
|
fh.print({ "command" => "store report", "version" => 3, "payload" => report_to_hash }.to_json)
|
|
end
|
|
rescue => detail
|
|
Puppet.log_exception(detail, "Could not write report for #{host} at #{file}: #{detail}")
|
|
end
|
|
|
|
nil
|
|
end
|
|
|
|
# Convert `self` (an instance of `Puppet::Transaction::Report`) to a hash
|
|
# suitable for sending over the wire to PuppetDB
|
|
#
|
|
# @return Hash[<String, Object>]
|
|
# @api private
|
|
def report_to_hash
|
|
if environment.nil?
|
|
raise Puppet::Error, "Environment is nil, unable to submit report. This may be due a bug with Puppet. Ensure you are running the latest revision, see PUP-2508 for more details."
|
|
end
|
|
|
|
{
|
|
"certname" => host,
|
|
"puppet-version" => puppet_version,
|
|
"report-format" => report_format,
|
|
"configuration-version" => configuration_version.to_s,
|
|
"start-time" => time.iso8601(9),
|
|
"end-time" => (time + run_duration).iso8601(9),
|
|
"resource-events" => build_events_list,
|
|
"environment" => environment,
|
|
"transaction-uuid" => transaction_uuid,
|
|
"status" => status,
|
|
}
|
|
end
|
|
|
|
# Build a resource-events array from both evaluated and skipped resources.
|
|
#
|
|
# @return Array[Hash]
|
|
# @api private
|
|
def build_events_list
|
|
resource_statuses.inject([]) do |events, status_entry|
|
|
_, status = *status_entry
|
|
if ! (status.events.empty?)
|
|
events.concat(status.events.map { |event| event_to_hash(status, event) })
|
|
elsif status.skipped
|
|
events.concat([fabricate_event(status, "skipped")])
|
|
end
|
|
events
|
|
end
|
|
end
|
|
|
|
# Calculate run duration.
|
|
#
|
|
# @return Number
|
|
# @api private
|
|
def run_duration
|
|
if metrics["time"] and metrics["time"]["total"]
|
|
metrics["time"]["total"]
|
|
else
|
|
0
|
|
end
|
|
end
|
|
|
|
# Convert an instance of `Puppet::Transaction::Event` to a hash
|
|
# suitable for sending over the wire to PuppetDB
|
|
#
|
|
# @return Hash[<String, Object>]
|
|
# @api private
|
|
def event_to_hash(resource_status, event)
|
|
{
|
|
"status" => event.status,
|
|
"timestamp" => event.time.iso8601(9),
|
|
"resource-type" => resource_status.resource_type,
|
|
"resource-title" => resource_status.title.to_s,
|
|
"property" => event.property,
|
|
"new-value" => event.desired_value,
|
|
"old-value" => event.previous_value,
|
|
"message" => event.message,
|
|
"file" => resource_status.file,
|
|
"line" => resource_status.line,
|
|
"containment-path" => resource_status.containment_path,
|
|
}
|
|
end
|
|
|
|
# Given an instance of `Puppet::Resource::Status` and a status
|
|
# string, this method fabricates a PuppetDB event object with the
|
|
# provided `"status"`.
|
|
#
|
|
# @api private
|
|
def fabricate_event(resource_status, event_status)
|
|
{
|
|
"status" => event_status,
|
|
"timestamp" => resource_status.time.iso8601(9),
|
|
"resource-type" => resource_status.resource_type,
|
|
"resource-title" => resource_status.title.to_s,
|
|
"property" => nil,
|
|
"new-value" => nil,
|
|
"old-value" => nil,
|
|
"message" => nil,
|
|
"file" => resource_status.file,
|
|
"line" => resource_status.line,
|
|
"containment-path" => resource_status.containment_path,
|
|
}
|
|
end
|
|
end
|