Contents

require "rubygems"
require "bundler/setup"
Bundler.setup

require "active_support"
require "active_support/core_ext"
require "active_model"
require "elasticsearch"

module Elasticity
  class Config
    attr_writer :logger, :client, :settings, :namespace, :pretty_json

    def logger
      return @logger if defined?(@logger)
      @logger = Logger.new(STDOUT)
    end

    def client
      return @client if defined?(@client)
      @client = Elasticsearch::Client.new
    end

    def settings
      return @settings if defined?(@settings)
      @settings = {}
    end

    def namespace
      @namespace
    end

    def pretty_json
      @pretty_json || false
    end
  end

  def self.configure
    @config = Config.new
    yield(@config)
  end

  def self.config
    return @config if defined?(@config)
    @config = Config.new
  end
end

ActiveSupport::Notifications.subscribe(/^elasticity\./) do |name, start, finish, id, payload|
  time = (finish - start)*1000

  if logger = Elasticity.config.logger
    logger.debug "#{name} #{"%.2f" % time}ms #{MultiJson.dump(payload[:args], pretty: Elasticity.config.pretty_json)}"

    exception, message = payload[:exception]
    if exception
      logger.error "#{name} #{exception}: #{message}"
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
es-elasticity-0.2.7	lib/elasticity_base.rb
es-elasticity-0.2.6	lib/elasticity_base.rb
es-elasticity-0.2.5	lib/elasticity_base.rb
es-elasticity-0.2.4	lib/elasticity_base.rb
es-elasticity-0.2.3	lib/elasticity_base.rb
es-elasticity-0.2.2	lib/elasticity_base.rb
es-elasticity-0.2.1	lib/elasticity_base.rb