module TingYun::Instrumentation::MiddlewareTracing

Constants

CONTENT_TYPE

the trailing unless is for the benefit for Ruby 1.8.7 and can be removed when it is deprecated.

TXN_STARTED_KEY

Public Instance Methods

_nr_has_middleware_tracing() click to toggle source
# File lib/ting_yun/instrumentation/middleware_tracing.rb, line 15
def _nr_has_middleware_tracing
  true
end
build_transaction_options(env, first_middleware) click to toggle source
# File lib/ting_yun/instrumentation/middleware_tracing.rb, line 19
def build_transaction_options(env, first_middleware)
  opts = transaction_options
  opts = merge_first_middleware_options(opts, env) if first_middleware
  opts
end
call(env) click to toggle source
# File lib/ting_yun/instrumentation/middleware_tracing.rb, line 52
def call(env)
  return target.call(env) if sinatra_static?(env)

  first_middleware = note_transaction_started(env)
  state = TingYun::Agent::TransactionState.tl_get
  begin

    if first_middleware
      events.notify(:cross_app_before_call, env)
    end
    TingYun::Agent::Transaction.start(state, category, build_transaction_options(env, first_middleware))

    result = (target == self) ? traced_call(env) : target.call(env)

    if first_middleware
      capture_http_response_code(state, result)
      capture_response_content_type(state, result)
      events.notify(:cross_app_after_call, result)
    end

    result
  rescue Exception => e
    TingYun::Agent.notice_error(e,:type=>:error)
    raise e
  ensure
    TingYun::Agent::Transaction.stop(state)
  end
end
capture_http_response_code(state, result) click to toggle source
# File lib/ting_yun/instrumentation/middleware_tracing.rb, line 32
def capture_http_response_code(state, result)
  if result.is_a?(Array) && state.current_transaction
    state.current_transaction.attributes.add_agent_attribute(:httpStatus, result[0].to_s)
  end
end
capture_response_content_type(state, result) click to toggle source
# File lib/ting_yun/instrumentation/middleware_tracing.rb, line 41
def capture_response_content_type(state, result)
  if result.is_a?(Array) && state.current_transaction
    _, headers, _ = result
    state.current_transaction.attributes.add_agent_attribute(:contentType, headers[CONTENT_TYPE].to_s)
  end
end
events() click to toggle source
# File lib/ting_yun/instrumentation/middleware_tracing.rb, line 86
def events
  ::TingYun::Agent.instance.events
end
merge_first_middleware_options(opts, env) click to toggle source
# File lib/ting_yun/instrumentation/middleware_tracing.rb, line 25
def merge_first_middleware_options(opts, env)
  opts.merge(
      :request          => ::Rack::Request.new(env),
      :apdex_start_time => TingYun::Instrumentation::Support::QueueTime.parse_frontend_timestamp(env)
  )
end
note_transaction_started(env) click to toggle source
# File lib/ting_yun/instrumentation/middleware_tracing.rb, line 82
def note_transaction_started(env)
  env[TXN_STARTED_KEY] = true unless env[TXN_STARTED_KEY]
end
sinatra_static?(env) click to toggle source
# File lib/ting_yun/instrumentation/middleware_tracing.rb, line 48
def sinatra_static?(env)
  defined?(::Sinatra) && defined?(::Sinatra::Base) && env['REQUEST_URI'] && env['REQUEST_URI'] =~ /\.(css|js|html|png|jpg|jpeg|gif|bmp)\Z/i
end