#!/usr/bin/env ruby -KU

TIMES = (ENV['N'] || 10000).to_i

require 'rubygems'
gem 'addressable',  '~>2.0'
gem 'faker',        '~>0.3.1'
gem 'rbench',       '~>0.2.3'
require 'addressable/uri'
require 'faker'
require 'rbench'

__DIR__ = File.dirname(__FILE__)
$:.unshift "#{__DIR__}/../lib"
require 'active_record'

conn = { :adapter => 'mysql',
  :database => 'activerecord_unittest',
  :username => 'rails', :password => '',
  :encoding => 'utf8' }

conn[:socket] = Pathname.glob(%w[
  /opt/local/var/run/mysql5/mysqld.sock
  /tmp/mysqld.sock
  /tmp/mysql.sock
  /var/mysql/mysql.sock
  /var/run/mysqld/mysqld.sock
]).find { |path| path.socket? }

ActiveRecord::Base.establish_connection(conn)

class User < ActiveRecord::Base
  connection.create_table :users, :force => true do |t|
    t.string :name, :email
    t.timestamps
  end

  has_many :exhibits
end

class Exhibit < ActiveRecord::Base
  connection.create_table :exhibits, :force => true do |t|
    t.belongs_to :user
    t.string :name
    t.text :notes
    t.timestamps
  end

  belongs_to :user

  def look; attributes end
  def feel; look; user.name end

  def self.look(exhibits) exhibits.each { |e| e.look } end
  def self.feel(exhibits) exhibits.each { |e| e.feel } end
end

sqlfile = "#{__DIR__}/performance.sql"

if File.exists?(sqlfile)
  mysql_bin = %w[mysql mysql5].select { |bin| `which #{bin}`.length > 0 }
  `#{mysql_bin} -u #{conn[:username]} #{"-p#{conn[:password]}" unless conn[:password].blank?} #{conn[:database]} < #{sqlfile}`
else
  puts 'Generating data...'

  # pre-compute the insert statements and fake data compilation,
  # so the benchmarks below show the actual runtime for the execute
  # method, minus the setup steps

  # Using the same paragraph for all exhibits because it is very slow
  # to generate unique paragraphs for all exhibits.
  notes = Faker::Lorem.paragraphs.join($/)
  today = Date.today

  puts 'Inserting 10,000 users and exhibits...'
  10_000.times do
    user = User.create(
      :created_at => today,
      :name       => Faker::Name.name,
      :email      => Faker::Internet.email
    )

    Exhibit.create(
      :created_at => today,
      :name       => Faker::Company.name,
      :user       => user,
      :notes      => notes
    )
  end

  mysqldump_bin = %w[mysqldump mysqldump5].select { |bin| `which #{bin}`.length > 0 }
  `#{mysqldump_bin} -u #{conn[:username]} #{"-p#{conn[:password]}" unless conn[:password].blank?} #{conn[:database]} exhibits users > #{sqlfile}`
end


def find_model
  ar_obj = Exhibit.find(1)

  ar_obj.id
end

def instantiate_model
  Exhibit.new
end

def setting_attributes
  attrs = { :name => 'sam' }
  Exhibit.new(attrs)
end

def model_first
  Exhibit.first.look
end

def model_all_limit_100
  Exhibit.look Exhibit.all(:limit => 100)
end

def model_all_limit_100_with_relationship
  Exhibit.feel Exhibit.all(:limit => 100, :include => :user)
end

def model_all_limit_10000
  Exhibit.look Exhibit.all(:limit => 10000)
end

def model_create
  Exhibit.create({
    :name       => Faker::Company.name,
    :notes      => Faker::Lorem.paragraphs.join($/),
    :created_at => Date.today
  })
end

def model_attributes
  attrs_first  = { :name => 'sam' }
  attrs_second = { :name => 'tom' }
  exhibit = Exhibit.new(attrs_first); exhibit.attributes = attrs_second
end

def model_update
  Exhibit.first.update_attributes(:name => 'bob')
end

def model_destroy
  Exhibit.first.destroy
end


require "ruby-prof"
RubyProf.start

find_model
instantiate_model
setting_attributes
model_first
model_all_limit_100
model_all_limit_100_with_relationship
model_all_limit_10000
model_create
model_attributes
model_update
model_destroy

result = RubyProf.stop
printer = RubyProf::CallStackPrinter.new(result)
printer.print(File.open("output.html", "w"))

ActiveRecord::Migration.drop_table "exhibits"
ActiveRecord::Migration.drop_table "users"