mirror of
https://github.com/discourse/discourse.git
synced 2024-12-15 10:24:27 +08:00
710eafdd35
We pick the first topic with 30 responses as our bench topic. Previously we simply picked the last topic, but hand no guarantee on ordering. This also attempts to correct previous runs of the bench.
128 lines
2.9 KiB
Ruby
128 lines
2.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# can be used to generate a mock db for profiling purposes
|
|
|
|
# we want our script to generate a consistent output, to do so
|
|
# we monkey patch array sample so it always uses the same rng
|
|
class Array
|
|
RNG = Random.new(1098109928029800)
|
|
|
|
def sample
|
|
self[RNG.rand(size)]
|
|
end
|
|
end
|
|
|
|
# based on https://gist.github.com/zaius/2643079
|
|
def unbundled_require(gem)
|
|
if defined?(::Bundler)
|
|
spec_path = Dir.glob("#{Gem.dir}/specifications/#{gem}-*.gemspec").last
|
|
if spec_path.nil?
|
|
raise LoadError
|
|
end
|
|
|
|
spec = Gem::Specification.load spec_path
|
|
spec.activate
|
|
end
|
|
|
|
begin
|
|
require gem
|
|
end
|
|
end
|
|
|
|
def sentence
|
|
@gabbler ||= Gabbler.new.tap do |gabbler|
|
|
story = File.read(File.dirname(__FILE__) + "/alice.txt")
|
|
gabbler.learn(story)
|
|
end
|
|
|
|
sentence = +""
|
|
until sentence.length > 800 do
|
|
sentence << @gabbler.sentence
|
|
sentence << "\n"
|
|
end
|
|
sentence
|
|
end
|
|
|
|
def create_admin(seq)
|
|
User.new.tap { |admin|
|
|
admin.email = "admin@localhost#{seq}.fake"
|
|
admin.username = "admin#{seq}"
|
|
admin.password = "password12345abc"
|
|
admin.save!
|
|
admin.grant_admin!
|
|
admin.change_trust_level!(TrustLevel[4])
|
|
admin.activate
|
|
}
|
|
end
|
|
|
|
require File.expand_path(File.dirname(__FILE__) + "/../config/environment")
|
|
|
|
Jobs.run_immediately!
|
|
|
|
unless Rails.env == "profile"
|
|
puts "This script should only be used in the profile environment"
|
|
exit
|
|
end
|
|
|
|
def ensure_perf_test_topic_has_right_title!
|
|
title = "I am a topic used for perf tests"
|
|
# in case we have an old run and picked the wrong topic
|
|
Topic.where(title: title).update_all(title: "Test topic #{SecureRandom.hex}")
|
|
t = Topic.where(archetype: :regular, posts_count: 30).order(id: :desc).first
|
|
t.title = title
|
|
t.save!
|
|
end
|
|
|
|
# by default, Discourse has a "system" and `discobot` account
|
|
if User.count > 2
|
|
puts "Only run this script against an empty DB"
|
|
|
|
ensure_perf_test_topic_has_right_title!
|
|
exit
|
|
end
|
|
|
|
require 'optparse'
|
|
begin
|
|
unbundled_require 'gabbler'
|
|
rescue LoadError
|
|
puts "installing gabbler gem"
|
|
puts `gem install gabbler`
|
|
unbundled_require 'gabbler'
|
|
end
|
|
|
|
puts "Creating 100 users"
|
|
users = 100.times.map do |i|
|
|
putc "."
|
|
create_admin(i)
|
|
end
|
|
|
|
puts
|
|
puts "Creating 10 categories"
|
|
categories = 10.times.map do |i|
|
|
putc "."
|
|
Category.create(name: "category#{i}", text_color: "ffffff", color: "000000", user: users.first)
|
|
end
|
|
|
|
puts
|
|
puts "Creating 100 topics"
|
|
|
|
topic_ids = 100.times.map do
|
|
post = PostCreator.create(users.sample, raw: sentence, title: sentence[0..50].strip, category: categories.sample.id, skip_validations: true)
|
|
|
|
putc "."
|
|
post.topic_id
|
|
end
|
|
|
|
puts
|
|
puts "creating 2000 replies"
|
|
2000.times do
|
|
putc "."
|
|
PostCreator.create(users.sample, raw: sentence, topic_id: topic_ids.sample, skip_validations: true)
|
|
end
|
|
|
|
# no sidekiq so update some stuff
|
|
Category.update_stats
|
|
Jobs::PeriodicalUpdates.new.execute(nil)
|
|
|
|
ensure_perf_test_topic_has_right_title!
|