Contents

require "yaml"

module Librato
  module Rails
    # Adds yaml-based config and extra properties to the configuration
    # class provided by librato-rack
    #
    # https://github.com/librato/librato-rack/blob/master/lib/librato/rack/configuration.rb
    #
    class Configuration < Rack::Configuration
      CONFIG_SETTABLE = %w{user token flush_interval log_level prefix source source_pids proxy suites}

      DEFAULT_SUITES = [:rails_action, :rails_cache, :rails_controller, :rails_mail, :rails_method, :rails_render, :rails_sql, :rails_status, :rails_job]

      attr_accessor :config_by, :config_file

      # options:
      # * :config_file - alternate config file location
      #
      def initialize(options={})
        self.config_file = options[:config_file] || 'config/librato.yml'
        super()
        self.log_prefix = '[librato-rails] '
      end

      # detect and load configuration from config file or env vars
      def load_configuration
        if self.config_file && File.exists?(self.config_file)
          configure_with_config_file
        else
          self.config_by = :environment
          super
        end

        # respect autorun and log_level env vars regardless of config method
        self.autorun = detect_autorun
        self.log_level = :info if log_level.blank?
        self.suites = '' if suites.nil?
        self.log_level = ENV['LIBRATO_LOG_LEVEL'] if ENV['LIBRATO_LOG_LEVEL']
      end

      private

      def configure_with_config_file
        self.config_by = :config_file
        env_specific = YAML.load(ERB.new(File.read(config_file)).result)[::Rails.env]
        if env_specific
          settable = CONFIG_SETTABLE & env_specific.keys
          settable.each { |key| self.send("#{key}=", env_specific[key]) }
        end
      end

      def default_suites
        super + DEFAULT_SUITES
      end

    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
librato-rails-1.4.2	lib/librato/rails/configuration.rb
librato-rails-1.4.1	lib/librato/rails/configuration.rb
librato-rails-1.4.0	lib/librato/rails/configuration.rb
librato-rails-1.4.0.beta	lib/librato/rails/configuration.rb
librato-rails-1.3.0	lib/librato/rails/configuration.rb