Contents

# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
# frozen_string_literal: true

module Contrast
  module Utils
    # A module that detects whether any job servers attached to
    # the application are running
    module JobServersRunning
      include Contrast::Components::Interface
      access_component :app_context, :logging

      class << self
        def job_servers_running?
          sidekiq_running? || rake_running?
        end

        private

        def sidekiq_running?
          return unless defined?(Sidekiq) && Sidekiq.cs__respond_to?(:server?) && Sidekiq.server?

          logger.trace('Detected the spawn of a Sidekiq process')
          true
        end

        def rake_running?
          unless defined?(Rake) &&
                Rake.cs__respond_to?(:application) &&
                Rake.application.cs__respond_to?(:top_level_tasks)

            return
          end

          disabled_rake_tasks = APP_CONTEXT.disabled_agent_rake_tasks
          has_disabled_task = Rake.application.top_level_tasks.any? do |top_level_task|
            disabled_rake_tasks.include?(top_level_task)
          end
          return false unless has_disabled_task

          logger.trace('Detected startup within Rake task')
          true
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
contrast-agent-4.8.0	lib/contrast/utils/job_servers_running.rb
contrast-agent-4.7.0	lib/contrast/utils/job_servers_running.rb