Sha256: 945452aa7c568ec3fff1560ac2e1b717e69560de9b12fd7868df7b63036a0b8e
Contents?: true
Size: 1.77 KB
Versions: 8
Compression:
Stored size: 1.77 KB
Contents
# encoding: utf-8
# This file is distributed under New Relic's license terms.
# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.
module FlakyProxy
class Rule
class ActionBuilder
attr_reader :actions
def initialize
@actions = []
end
def pass
@actions << [:pass]
end
def respond(response_spec)
@actions << [:respond, response_spec]
end
def delay(amount)
@actions << [:delay, amount]
end
def close
@actions << [:close]
end
end
def initialize(url_regex=nil, builder=nil, &blk)
@url_regex = url_regex
@action = blk
@builder = builder || ActionBuilder.new
@default_action = [:pass]
end
def match?(request)
@url_regex.nil? || @url_regex.match(request.request_url)
end
def next_action
@builder.actions.shift || @default_action
end
def relay(request, connection)
request.relay_to(connection.server_socket)
response = Response.read_from(connection.server_socket)
response.relay_to(connection.client_socket)
end
def evaluate(request, connection)
@builder.instance_exec(request, connection, &@action) if @action
action, *params = next_action
FlakyProxy.logger.info(" [#{action.upcase}] #{request.request_method} #{request.request_url}")
case action
when :pass
relay(request, connection)
when :respond
response_spec = params.first
response = Response.build(response_spec)
response.relay_to(connection.client_socket)
when :delay
amount = params.first
sleep(amount)
relay(request, connection)
when :close
connection.client_socket.close
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems