Sha256: 08739f28d4a50e1ec17c94cdf69d712749e3e0621e20a690c2f2277f93b80dec

Contents?: true

Size: 1.73 KB

Versions: 2

Compression:

Stored size: 1.73 KB

Contents

# frozen_string_literal: true

# Catches errors in a given block and returns them. This will return the
# output of the block if no errors are raised. Accepts an optional list of
# error kinds to catch.
#
# **NOTE:** Not available in apply block
Puppet::Functions.create_function(:catch_errors) do
  # @param error_types An array of error types to catch
  # @param block The block of steps to catch errors on
  # @return Undef If an error is raised in the block then the error will be
  # returned, otherwise the result will be returned
  # @example Catch errors for a block
  #   catch_errors() || {
  #     run_command("whoami", $nodes)
  #     run_command("adduser ryan", $nodes)
  #   }
  # @example Catch parse errors for a block of code
  #   catch_errors(['bolt/parse-error']) || {
  #    run_plan('canary', $nodes)
  #    run_plan('other_plan)
  #    apply($nodes) || {
  #      notify { "Hello": }
  #    }
  #   }
  dispatch :catch_errors do
    optional_param 'Array[String[1]]', :error_types
    block_param 'Callable[0, 0]', :block
  end

  def catch_errors(error_types = nil)
    unless Puppet[:tasks]
      raise Puppet::ParseErrorWithIssue
        .from_issue_and_stack(Bolt::PAL::Issues::PLAN_OPERATION_NOT_SUPPORTED_WHEN_COMPILING,
                              action: self.class.name)
    end

    executor = Puppet.lookup(:bolt_executor)
    executor.report_function_call(self.class.name)

    begin
      yield
    rescue Puppet::PreformattedError => e
      if e.cause.is_a?(Bolt::Error)
        if error_types.nil?
          return e.cause.to_puppet_error
        elsif error_types.include?(e.cause.to_h['kind'])
          return e.cause.to_puppet_error
        else
          raise e
        end
      else
        raise e
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version Path
bolt-1.24.0 bolt-modules/boltlib/lib/puppet/functions/catch_errors.rb
bolt-1.23.0 bolt-modules/boltlib/lib/puppet/functions/catch_errors.rb