/
graphing_stress_test.rb
executable file
·80 lines (71 loc) · 2.03 KB
/
graphing_stress_test.rb
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
#!/usr/bin/env ruby
$:.unshift File.expand_path("../../lib", __FILE__)
require "all_my_circuits"
require_relative "./graphing_server"
require "thread"
require "net/http"
require "uri"
require "timeout"
require "securerandom"
WORKERS = Integer(ENV["WORKERS"]) rescue 25
def setup
@responses_queue = Queue.new
@breaker = AllMyCircuits::Breaker.new(
name: "test service circuit breaker",
sleep_seconds: 5,
strategy: AllMyCircuits::Strategies::PercentageOverWindowStrategy.new(
requests_window: 40,
failure_rate_percent_threshold: 25
)
)
@datapoints_queue = Queue.new
@commands_queue = Queue.new
end
def run
setup
workers = run_workers
graphing_server = run_graphing_server
workers.each(&:join)
graphing_server.join
end
def run_workers
WORKERS.times.map do
Thread.new(@responses_queue, @breaker) do |responses, breaker|
loop do
begin
t1 = Time.now
@breaker.run do
uri = URI("http://localhost:8081")
http = Net::HTTP.new(uri.host, uri.port)
http.open_timeout = 2
http.read_timeout = 2
http.start do |h|
request = Net::HTTP::Get.new(uri)
response = h.request(request)
response.value
log "success"
responses.push(status: :succeeded, started: t1, finished: Time.now)
end
end
rescue AllMyCircuits::BreakerOpen
log "breaker open"
responses.push(status: :skipped, started: t1, finished: Time.now)
rescue
log "failure #{$!.inspect}: #{$!.backtrace.first(2).join(", ")}"
responses.push(status: :failed, started: t1, finished: Time.now)
end
end
end
end
end
def run_graphing_server
Thread.new(@responses_queue) do |responses_queue|
GraphingServer.start(responses_queue)
end
end
def log(msg)
@mtx ||= Mutex.new
timestamp = "%10.6f" % Time.now.to_f
@mtx.synchronize { puts "[#{Thread.current.object_id}] #{timestamp}: #{msg}" if ENV["DEBUG"] }
end
run