1
2
3
4
5
6
7
8
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
require 'fileutils'
module StackProf
class Middleware
def initialize(app, options = {})
@app = app
@options = options
@num_reqs = options[:save_every] || nil
Middleware.mode = options[:mode] || :cpu
Middleware.interval = options[:interval] || 1000
Middleware.raw = options[:raw] || false
Middleware.enabled = options[:enabled]
options[:path] = 'tmp/' if options[:path].to_s.empty?
Middleware.path = options[:path]
Middleware.metadata = options[:metadata] || {}
at_exit{ Middleware.save } if options[:save_at_exit]
end
def call(env)
enabled = Middleware.enabled?(env)
StackProf.start(
mode: Middleware.mode,
interval: Middleware.interval,
raw: Middleware.raw,
metadata: Middleware.metadata,
) if enabled
@app.call(env)
ensure
if enabled
StackProf.stop
if @num_reqs && (@num_reqs-=1) == 0
@num_reqs = @options[:save_every]
Middleware.save
end
end
end
class << self
attr_accessor :enabled, :mode, :interval, :raw, :path, :metadata
def enabled?(env)
if enabled.respond_to?(:call)
enabled.call(env)
else
enabled
end
end
def save
if results = StackProf.results
path = Middleware.path
is_directory = path != path.chomp('/')
if is_directory
filename = "stackprof-#{results[:mode]}-#{Process.pid}-#{Time.now.to_i}.dump"
else
filename = File.basename(path)
path = File.dirname(path)
end
FileUtils.mkdir_p(path)
File.open(File.join(path, filename), 'wb') do |f|
f.write Marshal.dump(results)
end
filename
end
end
end
end
end
|
