discourse/script/profile_db_generator.rb

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

160 lines
3.2 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
# can be used to generate a mock db for profiling purposes
2013-08-15 02:59:38 -04:00
# 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(1_098_109_928_029_800)
def sample
self[RNG.rand(size)]
end
end
2013-08-05 00:29:04 -04:00
# based on https://gist.github.com/zaius/2643079
2013-08-05 00:26:11 -04:00
def unbundled_require(gem)
if defined?(::Bundler)
spec_path = Dir.glob("#{Gem.dir}/specifications/#{gem}-*.gemspec").last
raise LoadError if spec_path.nil?
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
sentence << @gabbler.sentence
sentence << "\n"
end
sentence
end
def create_user(seq, admin: false, username: nil)
User.new.tap do |user|
user.email = "user@localhost#{seq}.fake"
user.username = username || "user#{seq}"
user.password = "password12345abc"
user.save!
if admin
user.grant_admin!
user.change_trust_level!(TrustLevel[4])
end
user.activate
end
end
require File.expand_path(File.dirname(__FILE__) + "/../config/environment")
Jobs.run_immediately!
2014-01-08 23:56:03 -05:00
unless Rails.env == "profile"
puts "This script should only be used in the profile environment"
exit
end
2017-09-04 23:23:03 -04:00
# by default, Discourse has a "system" and `discobot` account
if User.count > 2
puts "Only run this script against an empty DB"
exit
end
require "optparse"
begin
unbundled_require "gabbler"
rescue LoadError
puts "installing gabbler gem"
puts `gem install gabbler`
unbundled_require "gabbler"
end
number_of_users = 100
puts "Creating #{number_of_users} users"
number_of_users.times.map do |i|
putc "."
create_user(i)
end
puts
puts "Creating 1 admin user"
admin_user = create_user(number_of_users + 1, admin: true, username: "admin1")
users = User.human_users.all
puts
puts "Creating 10 categories"
categories =
10.times.map do |i|
putc "."
Category.create(name: "category#{i}", text_color: "ffffff", color: "000000", user: admin_user)
end
puts
puts "Creating 100 topics"
topic_ids =
100.times.map do
post =
PostCreator.create(
admin_user,
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
2014-01-08 23:56:03 -05:00
puts
puts "creating perf test topic"
first_post =
PostCreator.create(
users.sample,
raw: sentence,
title: "I am a topic used for perf tests",
category: categories.sample.id,
skip_validations: true,
)
puts
puts "Creating 100 replies for perf test topic"
100.times do
putc "."
PostCreator.create(
users.sample,
raw: sentence,
topic_id: first_post.topic_id,
skip_validations: true,
)
end
2014-01-08 23:56:03 -05:00
# no sidekiq so update some stuff
Category.update_stats
Jobs::PeriodicalUpdates.new.execute(nil)