Class: SolidApm::Middleware

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

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(app) ⇒ Middleware

Returns a new instance of Middleware.



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

def initialize(app)
  @app = app
end

Class Method Details

.call ⇒ Object 



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
 
# File 'lib/solid_apm/middleware.rb', line 23

def self.call
  transaction = SpanSubscriber::Base.transaction
  SpanSubscriber::Base.transaction = nil

  if transaction.nil? ||
      transaction_filtered?(transaction.name) ||
      !Sampler.should_sample?

    SpanSubscriber::Base.spans = nil
    return
  end

  with_silence_logger do
    ApplicationRecord.transaction do
      transaction.save!

      SpanSubscriber::Base.spans.each do |span|
        span[:transaction_id] = transaction.id
      end
      SolidApm::Span.insert_all SpanSubscriber::Base.spans
    end
  end
  SpanSubscriber::Base.spans = nil
end

.init_transaction ⇒ Object

Initialize a new transaction and reset spans



71
72
73
74
75
76
77
78
79
80
 
# File 'lib/solid_apm/middleware.rb', line 71

def self.init_transaction
  now = Time.zone.now
  SpanSubscriber::Base.transaction = Transaction.new(
    uuid: SecureRandom.uuid,
    unix_minute: (now.to_f / 60).to_i,
    type: 'request'
  )
  SpanSubscriber::Base.spans = []
  SpanSubscriber::Base.transaction
end

.transaction_filtered?(transaction_name) ⇒ Boolean

Returns:

  • (Boolean) 


48
49
50
51
52
53
54
55
56
57
58
59
 
# File 'lib/solid_apm/middleware.rb', line 48

def self.transaction_filtered?(transaction_name)
  SolidApm.transaction_filters.any? do |filter|
    case filter
    when String
      transaction_name == filter
    when Regexp
      filter.match?(transaction_name)
    else
      false
    end
  end
end

.with_silence_logger ⇒ Object 



61
62
63
64
65
66
67
 
# File 'lib/solid_apm/middleware.rb', line 61

def self.with_silence_logger
  if SolidApm.silence_active_record_logger && ActiveRecord::Base.logger
    ActiveRecord::Base.logger.silence { yield }
  else
    yield
  end
end

Instance Method Details

#call(env) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
19
20
21
 
# File 'lib/solid_apm/middleware.rb', line 9

def call(env)
  self.class.init_transaction
  status, headers, body = @app.call(env)

    env['rack.after_reply'] ||= []
    env['rack.after_reply'] << ->() do
      self.class.call
    rescue StandardError => e
      Rails.logger.error e
      Rails.logger.error e.backtrace&.join("\n")
    end
  [status, headers, body]
end