Sha256: bb560a72dcb9dca637f3cad742adf4c63fc3a68d0f246dad926ecf841061f502

Contents?: true

Size: 1.6 KB

Versions: 6

Compression:

Stored size: 1.6 KB

Contents

begin
  require "oj"
rescue LoadError
  require "json"
end

module LogjamAgent
  class Request
    attr_reader :caller_id, :fields

    def initialize(app, env, logger, initial_fields)
      @logger = logger
      @app = app
      @env = env
      @forwarder = Forwarders.get(app, env)
      @lines = []
      @id = UUID4R::uuid(1).gsub('-','')
      @fields = initial_fields.merge(:request_id => @id, :host => LogjamAgent.hostname, :process_id => Process.pid, :lines => @lines)
      @mutex = Mutex.new
    end

    def id
      "#{@app}-#{@env}-#{@id}"
    end

    def caller_id
      @fields[:caller_id]
    end

    def add_line(severity, timestamp, message)
      @mutex.synchronize do
        @lines << [severity, format_time(timestamp), message.strip]
      end
    end

    def add_exception(exception)
      @mutex.synchronize do
        ((@fields[:exceptions] ||= []) << exception).uniq!
      end
    end

    def forward
      engine = @fields.delete(:engine)
      @forwarder.forward(encode_payload, engine)
    rescue Exception => e
      handle_forwarding_error(e)
    end

    private

    if defined?(Oj)
      def encode_payload
        Oj.dump(@fields, :mode => :compat)
      rescue
        @fields.to_son
      end
    else
      def encode_payload
        @fields.to_json
      end
    end

    def format_time(t)
      # iso time with microseconds
      t.strftime("%Y-%m-%dT%H:%M:%S.#{"%06d" % t.usec}")
    end

    def handle_forwarding_error(exception)
      @logger.error exception.to_s
      LogjamAgent.error_handler.call(exception)
    rescue Exception
      # swallow all exceptions
    end

  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version Path
logjam_agent-0.6.3 lib/logjam_agent/request.rb
logjam_agent-0.6.2 lib/logjam_agent/request.rb
logjam_agent-0.6.1 lib/logjam_agent/request.rb
logjam_agent-0.6.0 lib/logjam_agent/request.rb
logjam_agent-0.5.9 lib/logjam_agent/request.rb
logjam_agent-0.5.8 lib/logjam_agent/request.rb