oesgalha/xylem

require 'bundler/inline'

BENCH_GEM = ENV['BENCH_GEM']

unless ['xylem', 'acts_as_tree', 'awesome_nested_set', 'ancestry'].include?(BENCH_GEM)
  fail 'Please provide a environment variable BENCH_GEM with one o the following values: [xylem, acts_as_tree, awesome_nested_set, ancestry]'
end

gemfile(true) do
  source 'https://rubygems.org'
  gem 'activerecord', require: 'active_record'
  gem 'sqlite3'
  gem 'benchmark-ips'

  case BENCH_GEM
  when 'xylem'
    gem 'xylem', path: '..'
  when 'acts_as_tree'
    gem 'acts_as_tree'
  when 'awesome_nested_set'
    gem 'activesupport', require: 'active_support/core_ext/module/delegation'
    gem 'awesome_nested_set'
  when 'ancestry'
    gem 'ancestry'
  end
end

ActiveRecord::Base.establish_connection(adapter: 'sqlite3', database: ':memory:')

ActiveRecord::Base.send(:include, ActsAsTree) if BENCH_GEM == 'acts_as_tree'

class Comment < ActiveRecord::Base
  acts_as_tree        if ['xylem', 'acts_as_tree'].include?(BENCH_GEM)
  acts_as_nested_set  if BENCH_GEM == 'awesome_nested_set'
  has_ancestry        if BENCH_GEM == 'ancestry'

  connection.create_table table_name, force: true do |t|
    t.string :payload
    t.integer :parent_id, index: true   if BENCH_GEM != 'ancestry'
    t.integer :lft, index: true         if BENCH_GEM == 'awesome_nested_set'
    t.integer :rgt, index: true         if BENCH_GEM == 'awesome_nested_set'
    t.string :ancestry, index: true     if BENCH_GEM == 'ancestry'
  end
end

def recursive_child(par, depth)
  unless depth == 0
    5.times do
      recursive_child(Comment.create!(payload: SecureRandom.hex(128), parent: par), depth - 1)
    end
  end
end

puts 'INSERTING TEST DATA...'

recursive_child(nil, 5)

@comment1 = Comment.roots.first
@comment11 = @comment1.children.sample
@comment111 = @comment11.children.sample
@comment1111 = @comment111.children.sample
@comment11111 = @comment1111.children.sample

Benchmark.ips do |x|
  x.report('insertion') do |times|
    times.times { Comment.create!(parent: @comment111) }
  end
  x.report('parent') do |times|
    times.times { @comment11111.reload.parent }
  end
  x.report('children') do |times|
    times.times { @comment111.reload.children.to_a }
  end
  x.report('roots') do |times|
    times.times { Comment.roots.to_a }
  end
  x.report('ancestors') do |times|
    times.times { @comment11111.reload.ancestors.to_a }
  end
  x.report('descendants') do |times|
    times.times { @comment1.reload.descendants.to_a }
  end
  x.json!("#{BENCH_GEM}.json")
end