elastic_search_client.rb

module SearchApp
  class ElasticSearchClient
    def self.production_client
      Elasticsearch::Client.new
    end

    def self.debug_client
      @client = Elasticsearch::Client.new log: true

      @client.transport.logger = Logger.new($stdout)
      @client.transport.logger.progname = 'ElasticSearchClient'
      @client.transport.logger.formatter = proc do |serverity, time, progname, msg|
        puts ["[#{time.utc.iso8601}]", serverity, progname, format_msg(msg)].join(' ')
      end

      @client
    end

    private

    # requests start with "> "
    # responses start with "< "
    def self.format_msg(msg)
      if msg.start_with?('< ')
        '< ' + JSON.pretty_generate(JSON.parse(msg[2..-1]))
      elsif msg.start_with?('> ')
        '> ' + JSON.pretty_generate(JSON.parse(msg[2..-1]))
      else
        msg
      end
    end
  end
end