Skip to content
Fetching contributors…
Cannot retrieve contributors at this time
130 lines (121 sloc) 3.14 KB
require 'bm_common'
url = ARGV.shift or raise
proxy = ENV['http_proxy'] || ENV['HTTP_PROXY']
url = URI.parse(url)
proxy = URI.parse(proxy) if proxy
threads = 1
number = 10
download_size = 10 * 1024 * 1024
def do_threads(number)
threads = []
results = []
number.times do
threads << Thread.new {
results << yield
}
end
threads.map { |th| th.join }
results
end
Benchmark.bmbm do |bm|
if defined?(Curl)
bm.report('curb') do
do_threads(threads) {
(1..number).collect {
Curl::Easy.download(url.to_s, 'download_curb')
raise if download_size != File.lstat('download_curb').size
}
}
end
end
if defined?(RFuzz)
bm.report('RFuzz::HttpClient') do
do_threads(threads) {
if proxy
host, port = proxy.host, proxy.port
else
host, port = url.host, url.port
end
path = proxy ? url.to_s : url.path
c = RFuzz::HttpClient.new(host, port)
result = (1..number).collect {
File.open('download_rfuzz', 'wb') do |file|
file.write(c.get(path).http_body)
end
raise if download_size != File.lstat('download_rfuzz').size
}
c.reset
result
}
end
end
if defined?(Net::HTTP)
bm.report('Net::HTTP') do
do_threads(threads) {
if proxy
c = Net::HTTP::Proxy(proxy.host, proxy.port).new(url.host, url.port)
else
c = Net::HTTP.new(url.host, url.port)
end
c.start
result = (1..number).collect {
File.open('download_net_http', 'wb') do |file|
c.get(url.path) do |data|
file.write(data)
end
end
raise if download_size != File.lstat('download_net_http').size
}
c.finish
result
}
end
end
if defined?(HTTPClient)
bm.report('HTTPClient') do
c = HTTPClient.new(proxy)
do_threads(threads) {
(1..number).collect {
File.open('download_httpclient', 'wb') do |file|
c.get_content(url) do |data|
file.write(data)
end
end
raise if download_size != File.lstat('download_httpclient').size
}
}
c.reset_all
end
end
if defined?(OpenURI)
bm.report('open-uri') do
size = 16 * 1024
do_threads(threads) {
(1..number).collect {
File.open('download_open-uri', 'wb') do |file|
open(url, :proxy => proxy) { |f|
FileUtils.copy_stream(f, file)
}
end
raise if download_size != File.lstat('download_open-uri').size
}
}
end
end
if defined?(HTTParty)
class HTTPartyClient # need to create subclass for http_proxy
include HTTParty
end
bm.report('HTTParty') do
HTTPartyClient.http_proxy(proxy.host, proxy.port) if proxy
do_threads(threads) {
(1..number).collect {
File.open('download_httparty', 'wb') do |file|
file.write(HTTPartyClient.get(url.to_s))
end
raise if download_size != File.lstat('download_httparty').size
}
}
end
end
end
Jump to Line
Something went wrong with that request. Please try again.