Contents

# frozen_string_literal: true

module Orchestration
  module Services
    module HealthcheckBase
      attr_reader :configuration

      def self.included(base)
        base.extend(ClassMethods)
      end

      module ClassMethods
        def start(env = nil, terminal = nil, options = {})
          load_dependencies
          exit_on_error = options.fetch(:exit_on_error, true)
          options.delete(:exit_on_error)
          env ||= Environment.new
          terminal ||= Terminal.new
          name = options.delete(:service_name)
          check = ServiceCheck.new(new(env, name), terminal, options)

          exit 1 if !check.run && exit_on_error
        end

        def dependencies(*args)
          @dependencies = args
        end

        private

        def load_dependencies
          return if @dependencies.nil?

          @dependencies.map { |dependency| require dependency }
        end
      end

      def initialize(env, service_name = nil)
        @configuration = configuration_class.new(env, service_name)
      end

      def service_name
        @configuration.service_name
      end

      private

      def configuration_class
        # Find the relevant `Configuration` class for whatever `Healthcheck`
        # class we happen to be included in.
        self.class.parent.const_get(:Configuration)
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
orchestration-0.2.6	lib/orchestration/services/healthcheck_base.rb
orchestration-0.2.5	lib/orchestration/services/healthcheck_base.rb
orchestration-0.2.4	lib/orchestration/services/healthcheck_base.rb