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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
# frozen_string_literal: true
namespace :perf do
task :rails_load do
ENV["RAILS_ENV"] ||= "production"
ENV['RACK_ENV'] = ENV["RAILS_ENV"]
ENV["DISABLE_SPRING"] = "true"
ENV["SECRET_KEY_BASE"] ||= "foofoofoo"
ENV['LOG_LEVEL'] ||= "FATAL"
require 'rails'
puts "Booting: #{Rails.env}"
%W{ . lib test config }.each do |file|
$LOAD_PATH << File.expand_path(file)
end
require 'application'
Rails.env = ENV["RAILS_ENV"]
DERAILED_APP = Rails.application
if DERAILED_APP.respond_to?(:initialized?)
DERAILED_APP.initialize! unless DERAILED_APP.initialized?
else
DERAILED_APP.initialize! unless DERAILED_APP.instance_variable_get(:@initialized)
end
if !ENV["DERAILED_SKIP_ACTIVE_RECORD"] && defined? ActiveRecord
if defined? ActiveRecord::Tasks::DatabaseTasks
ActiveRecord::Tasks::DatabaseTasks.create_current
else # Rails 3.2
raise "No valid database for #{ENV['RAILS_ENV']}, please create one" unless ActiveRecord::Base.connection.active?.inspect
end
ActiveRecord::Migrator.migrations_paths = DERAILED_APP.paths['db/migrate'].to_a
ActiveRecord::Migration.verbose = true
if Rails.version.start_with? '6'
ActiveRecord::MigrationContext.new(ActiveRecord::Migrator.migrations_paths, ActiveRecord::SchemaMigration).migrate
elsif Rails.version.start_with? '5.2'
ActiveRecord::MigrationContext.new(ActiveRecord::Migrator.migrations_paths).migrate
else
ActiveRecord::Migrator.migrate(ActiveRecord::Migrator.migrations_paths, nil)
end
end
DERAILED_APP.config.consider_all_requests_local = true
end
task :rack_load do
puts "You're not using Rails"
puts "You need to tell derailed how to boot your app"
puts "In your perf.rake add:"
puts
puts "namespace :perf do"
puts " task :rack_load do"
puts " # DERAILED_APP = your code here"
puts " end"
puts "end"
end
task :setup do
if DerailedBenchmarks.gem_is_bundled?("railties")
Rake::Task["perf:rails_load"].invoke
else
Rake::Task["perf:rack_load"].invoke
end
WARM_COUNT = (ENV['WARM_COUNT'] || 0).to_i
TEST_COUNT = (ENV['TEST_COUNT'] || ENV['CNT'] || 1_000).to_i
PATH_TO_HIT = ENV["PATH_TO_HIT"] || ENV['ENDPOINT'] || "/"
puts "Endpoint: #{ PATH_TO_HIT.inspect }"
HTTP_HEADER_PREFIX = "HTTP_".freeze
RACK_HTTP_HEADERS = ENV.select { |key| key.start_with?(HTTP_HEADER_PREFIX) }
HTTP_HEADERS = RACK_HTTP_HEADERS.keys.inject({}) do |hash, rack_header_name|
# e.g. "HTTP_ACCEPT_CHARSET" -> "Accept-Charset"
header_name = rack_header_name[HTTP_HEADER_PREFIX.size..-1].split("_").map(&:downcase).map(&:capitalize).join("-")
hash[header_name] = RACK_HTTP_HEADERS[rack_header_name]
hash
end
puts "HTTP headers: #{HTTP_HEADERS}" unless HTTP_HEADERS.empty?
CURL_HTTP_HEADER_ARGS = HTTP_HEADERS.map { |http_header_name, value| "-H \"#{http_header_name}: #{value}\"" }.join(" ")
require 'rack/test'
require 'rack/file'
DERAILED_APP = DerailedBenchmarks.add_auth(Object.class_eval { remove_const(:DERAILED_APP) })
if server = ENV["USE_SERVER"]
@port = (3000..3900).to_a.sample
puts "Port: #{ @port.inspect }"
puts "Server: #{ server.inspect }"
thread = Thread.new do
Rack::Server.start(app: DERAILED_APP, :Port => @port, environment: "none", server: server)
end
sleep 1
def call_app(path = File.join("/", PATH_TO_HIT))
cmd = "curl #{CURL_HTTP_HEADER_ARGS} 'http://localhost:#{@port}#{path}' -s --fail 2>&1"
response = `#{cmd}`
unless $?.success?
STDERR.puts "Couldn't call app."
STDERR.puts "Bad request to #{cmd.inspect} \n\n***RESPONSE***:\n\n#{ response.inspect }"
FileUtils.mkdir_p("tmp")
File.open("tmp/fail.html", "w+") {|f| f.write response.body }
`open #{File.expand_path("tmp/fail.html")}` if ENV["DERAILED_DEBUG"]
exit(1)
end
end
else
@app = Rack::MockRequest.new(DERAILED_APP)
def call_app
response = @app.get(PATH_TO_HIT, RACK_HTTP_HEADERS)
if response.status != 200
STDERR.puts "Couldn't call app. Bad request to #{PATH_TO_HIT}! Resulted in #{response.status} status."
STDERR.puts "\n\n***RESPONSE BODY***\n\n"
STDERR.puts response.body
FileUtils.mkdir_p("tmp")
File.open("tmp/fail.html", "w+") {|f| f.write response.body }
`open #{File.expand_path("tmp/fail.html")}` if ENV["DERAILED_DEBUG"]
exit(1)
end
response
end
end
if WARM_COUNT > 0
puts "Warming up app: #{WARM_COUNT} times"
WARM_COUNT.times { call_app }
end
end
end
|
