data-transfer-test/models/rate.rb

55 lines
1.2 KiB
Ruby

require 'speedtest'
require 'sinatra/activerecord'
Time.zone = "America/Campo_Grande"
ActiveRecord.default_timezone = :local
class Rate < ActiveRecord::Base
def self.transfer(type_transfer)
self.find_rate.collect {
|p| [p.created_at.strftime('%d/%m %H:%M'), p.public_send(type_transfer)]
}
end
def self.extremes(type, value)
hash = self.group("date(created_at)").public_send(type, value)
hash.transform_keys!{ |k| k.to_date.strftime('%d-%m') }
hash.transform_values!{ |v| v.ceil(2) }
end
def self.find_rate
where(created_at: Date.today.all_day)
end
def self.last_time
self.transfer('download').last.first
end
def self.last_value(type)
self.transfer(type).last.last
end
def self.save
results = run_speedtest
if results
Rate.create({
download: results[:download],
upload: results[:upload],
})
end
end
def run_speedtest
output = `speedtest`
if $?.success?
download_speed = output.match(/Download:\s+(\d+\.\d+)\sMbps/)[1]
upload_speed = output.match(/Upload:\s+(\d+\.\d+)\sMbps/)[1]
{
download: download_speed.to_f,
upload: upload_speed.to_f
}
else
nil
end
end
end