Sha256: 03e085567700280258d86a4b7450ebe9ba8febded27fe0ac57b1669e0a0da4d8

Contents?: true

Size: 1.48 KB

Versions: 5

Compression:

Stored size: 1.48 KB

Contents

# frozen_string_literal: true

module Datadog
  module Profiling
    module Tasks
      # Wraps command with Datadog profiling
      class Exec
        attr_reader :args

        def initialize(args)
          @args = args
        end

        def run
          set_rubyopt!
          exec_with_error_handling(args)
        end

        def rubyopts
          [
            '-rdatadog/profiling/preload'
          ]
        end

        private

        def set_rubyopt!
          existing_rubyopt = ENV['RUBYOPT']

          ENV['RUBYOPT'] = existing_rubyopt ? "#{existing_rubyopt} #{rubyopts.join(' ')}" : rubyopts.join(' ')
        end

        # If there's an error here, rather than throwing a cryptic stack trace, let's instead have clearer messages, and
        # follow the same status codes as the shell uses
        # See also:
        # * https://www.gnu.org/software/bash/manual/html_node/Exit-Status.html
        # * https://github.com/rubygems/rubygems/blob/dd93966cac224532035deda533cba2685dfa30cc/bundler/lib/bundler/cli/exec.rb#L45
        def exec_with_error_handling(args)
          Kernel.exec(*args)
        rescue Errno::ENOENT => e
          Kernel.warn "ddprofrb exec failed: #{e.class.name} #{e.message} (command was '#{args.join(' ')}')"
          Kernel.exit 127
        rescue Errno::EACCES, Errno::ENOEXEC => e
          Kernel.warn "ddprofrb exec failed: #{e.class.name} #{e.message} (command was '#{args.join(' ')}')"
          Kernel.exit 126
        end
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version Path
datadog-2.2.0 lib/datadog/profiling/tasks/exec.rb
datadog-2.1.0 lib/datadog/profiling/tasks/exec.rb
datadog-2.0.0 lib/datadog/profiling/tasks/exec.rb
datadog-2.0.0.beta2 lib/datadog/profiling/tasks/exec.rb
datadog-2.0.0.beta1 lib/datadog/profiling/tasks/exec.rb