Skip to content

Instantly share code, notes, and snippets.

@rakibulislam
Forked from mtrolle/memory-test.rb
Created December 3, 2018 02:51
Show Gist options
  • Save rakibulislam/448327b79319363d91f7fb5b46e17716 to your computer and use it in GitHub Desktop.
Save rakibulislam/448327b79319363d91f7fb5b46e17716 to your computer and use it in GitHub Desktop.
High memory consumption, question asked on StackOverflow: https://stackoverflow.com/questions/53169751/ruby-memory-usage-gone-wild
require 'bundler/inline'
require 'json'
gemfile do
source 'https://rubygems.org'
gem 'rest-client'
gem 'nokogiri'
gem 'memory_profiler'
end
xml_base_url = 'http://noneed.dk/test.xml?date='
auth_end_point = 'http://noneed.dk/session.php'
auth_payload = {"user" => "hi@noneed.dk", "password" => "secret"}
image_search_base_url = 'http://noneed.dk/images.php?id='
image_base_url = 'http://noneed.dk/'
logger = Logger.new(STDOUT)
download_size = 0
time_start = Time.now
requests = 0
report = MemoryProfiler.report do
# Authenticates
response = RestClient::Request.execute({
method: :post,
url: auth_end_point,
payload: auth_payload,
})
cookie_jar = response.cookie_jar
logger.debug "Got a cookiejar - #{cookie_jar.cookies}"
download_size += response.body.bytesize
requests += 1
# Run X days of data
1.times do |n|
date = (Time.now + n * 86400).strftime("%F")
logger.info "Downloads data for #{date}"
response = RestClient.get(xml_base_url + date)
download_size += response.body.bytesize
requests += 1
logger.debug "Download completed with code #{response.code} with a total size of #{response.body.bytesize/1024}Kb."
# Parse downloaded XML and process data items
xml = Nokogiri::XML(response.body)
items = xml.xpath("//noneed//item")
logger.debug "-- found #{items.count} records in the file"
items.each do |item|
id = item.at("id").content
unless id.nil?
logger.info "Find images for (id:#{id})"
request = {
method: :post,
url: image_search_base_url + id,
cookies: cookie_jar
}
images = RestClient::Request.execute(request)
download_size += images.body.bytesize
requests += 1
logger.debug "Download completed with code #{images.code} with a total size of #{images.body.bytesize}."
image_json = JSON.parse(images.body)
image_json['ids'].each do |img_id|
logger.debug "-- downloading image #{img_id}"
request = {
method: :post,
url: image_base_url + img_id.to_s,
cookies: cookie_jar
}
image = RestClient::Request.execute(request)
download_size += image.body.bytesize
requests += 1
logger.debug "---- download status: #{image.code} with a total size of #{image.body.bytesize/1024.0/1024.0}Mb."
end
end
end
end
logger.info "Total download size: #{(download_size / 1024.0 / 1024.0).round(2)}Mb in #{Time.now - time_start} seconds through #{requests} requests."
end
report.pretty_print
logger.info "Completed!"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment