class ElasticAPM::Middleware

def call(env)
  begin
    transaction = ElasticAPM.transaction 'Rack', 'request',
      context: ElasticAPM.build_context(env)
    resp = @app.call env
    status, headers, = resp
    if transaction
      result = "HTTP #{status.to_s[0]}xx"
      transaction.submit(result, status: status, headers: headers)
    end
  rescue InternalError
    raise # Don't report ElasticAPM errors
  rescue ::Exception => e
    ElasticAPM.report(e, handled: false)
    transaction.submit('HTTP 5xx', status: 500) if transaction
    raise
  ensure
    transaction.release if transaction
  end
  resp
end

def initialize(app)
  @app = app
end