Redmine

require 'benchmark'

users_count = (ENV['USERS_COUNT'] || '10000').to_i

iterations = (ENV['ITER'] || '10').to_i

mentions_count = (ENV['MENTIONS_COUNT'] || '100').to_i

Setting.text_formatting = 'common_mark'

Setting.cache_formatted_text = 0

puts "ruby=#{RUBY_VERSION} db=#{ActiveRecord::Base.connection.adapter_name} users=#{users_count} mentions_per_text=#{mentions_count} iter=#{iterations}"

class SimpleHelper

  include ApplicationHelper

  include ActionView::Helpers::TagHelper

  include ActionView::Helpers::AssetTagHelper

  include ActionView::Helpers::UrlHelper

  include Rails.application.routes.url_helpers

  def controller

    nil

  end

  def current_menu_item

    nil

  end

end

helper = SimpleHelper.new

ActiveRecord::Base.transaction do

  puts "Inserting users..."

  total_inserted = 0

  users_count.times.each_slice(10000) do |slice|

    users_attrs = slice.map do |i|

      {

        login: "bench_user_#{i}",

        firstname: "Bench",

        lastname: "User #{i}",

        status: Principal::STATUS_ACTIVE,

        type: 'User',

        language: 'en',

        created_on: Time.now,

        updated_on: Time.now

      }

    end

    User.insert_all(users_attrs)

    total_inserted += users_attrs.size

    print "."

  end

  puts "\nUsers inserted: #{total_inserted}"

  sample_users = User.where("login LIKE 'bench_user_%'").limit(mentions_count).to_a

  if sample_users.length < mentions_count

    puts "Warning: Only #{sample_users.length} users generated"

  end

  project = Project.find(1)

  role = Role.givable.first

  puts "Assigning sample users to project..."

  # Only add the sampled users to the project for speed

  member_attrs = sample_users.map do |u|

    {

      user_id: u.id,

      project_id: project.id,

      created_on: Time.now

    }

  end

  Member.insert_all(member_attrs) if member_attrs.any?

  # Also need member_roles

  members = Member.where(user_id: sample_users.map(&:id)).pluck(:id)

  member_role_attrs = members.map do |member_id|

    {

      member_id: member_id,

      role_id: role.id

    }

  end

  MemberRole.insert_all(member_role_attrs) if member_role_attrs.any?

  text_login = sample_users.map { |u| "@#{u.login}" }.join(" ")

  text_user_link = sample_users.map { |u| "user##{u.id}" }.join(" ")

  # Worst case: mentions that look valid but don't exist in the DB,

  # forcing a sequential scan across all rows

  text_invalid_login = (1..mentions_count).map { |i| "@missing_user_#{i}" }.join(" ")

  prefix = "Here are some mentions: "

  text_login = prefix + text_login

  text_user_link = prefix + text_user_link

  text_invalid_login = prefix + text_invalid_login

  # Warm up

  helper.textilizable(text_login, project: project)

  helper.textilizable(text_user_link, project: project)

  helper.textilizable(text_invalid_login, project: project)

  puts "\nBenchmarking @login..."

  bm_login = Benchmark.realtime do

    ApplicationRecord.uncached do

      iterations.times { helper.textilizable(text_login, project: project) }

    end

  end

  puts "login_ms=#{(bm_login*1000).round(2)} avg_login=#{(bm_login*1000/iterations).round(2)}"

  puts "\nBenchmarking user#id..."

  bm_user_link = Benchmark.realtime do

    ApplicationRecord.uncached do

      iterations.times { helper.textilizable(text_user_link, project: project) }

    end

  end

  puts "user_link_ms=#{(bm_user_link*1000).round(2)} avg_user_link=#{(bm_user_link*1000/iterations).round(2)}"

  puts "\nBenchmarking @login (worst-case non-existent, forces full scan)..."

  bm_invalid_login = Benchmark.realtime do

    ApplicationRecord.uncached do

      iterations.times { helper.textilizable(text_invalid_login, project: project) }

    end

  end

  puts "invalid_login_ms=#{(bm_invalid_login*1000).round(2)} avg_invalid_login=#{(bm_invalid_login*1000/iterations).round(2)}"

  raise ActiveRecord::Rollback

end