Class: OpenTrace::Middleware

Inherits:
Object
  • Object
show all
Defined in:
lib/opentrace/middleware.rb

Instance Method Summary collapse

Constructor Details

#initialize(app) ⇒ Middleware

Returns a new instance of Middleware.



5
6
7
 
# File 'lib/opentrace/middleware.rb', line 5

def initialize(app)
  @app = app
end

Instance Method Details

#call(env) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/opentrace/middleware.rb', line 9

def call(env)
  request_id = env["action_dispatch.request_id"] || env["HTTP_X_REQUEST_ID"]
  OpenTrace.current_request_id = request_id
  Fiber[:opentrace_sql_count] = 0
  Fiber[:opentrace_sql_total_ms] = 0.0

  # Create RequestCollector for accumulate-and-summarize pattern
  if OpenTrace.enabled? && OpenTrace.config.request_summary
    require_relative "request_collector"
    collector = OpenTrace::RequestCollector.new(
      max_timeline: OpenTrace.config.timeline_max_events
    )
    Fiber[:opentrace_collector] = collector

    # Memory snapshot before request (opt-in)
    if OpenTrace.config.memory_tracking
      collector.memory_before = current_rss_mb
    end
  end

  @app.call(env)
ensure
  # Memory snapshot after request (opt-in)
  collector = Fiber[:opentrace_collector]
  if collector && OpenTrace.config.memory_tracking && collector.memory_before
    collector.memory_after = current_rss_mb
  end

  Fiber[:opentrace_collector] = nil
  Fiber[:opentrace_sql_count] = nil
  Fiber[:opentrace_sql_total_ms] = nil
  OpenTrace.current_request_id = nil
end