Welcome to mirror list, hosted at ThFree Co, Russian Federation.

gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'lib/gitlab/seeder.rb')
-rw-r--r--lib/gitlab/seeder.rb23
1 files changed, 20 insertions, 3 deletions
diff --git a/lib/gitlab/seeder.rb b/lib/gitlab/seeder.rb
index e2df60c46f1..ec514adafc8 100644
--- a/lib/gitlab/seeder.rb
+++ b/lib/gitlab/seeder.rb
@@ -4,12 +4,24 @@ module Gitlab
class Seeder
extend ActionView::Helpers::NumberHelper
- MASS_INSERT_PROJECT_START = 'mass_insert_project_'
- MASS_INSERT_USER_START = 'mass_insert_user_'
+ MASS_INSERT_PREFIX = 'mass_insert'
+ MASS_INSERT_PROJECT_START = "#{MASS_INSERT_PREFIX}_project_"
+ MASS_INSERT_GROUP_START = "#{MASS_INSERT_PREFIX}_group_"
+ MASS_INSERT_USER_START = "#{MASS_INSERT_PREFIX}_user_"
REPORTED_USER_START = 'reported_user_'
- ESTIMATED_INSERT_PER_MINUTE = 2_000_000
+ ESTIMATED_INSERT_PER_MINUTE = 250_000
MASS_INSERT_ENV = 'MASS_INSERT'
+ module NamespaceSeed
+ extend ActiveSupport::Concern
+
+ included do
+ scope :not_mass_generated, -> do
+ where.not("path LIKE '#{MASS_INSERT_GROUP_START}%'")
+ end
+ end
+ end
+
module ProjectSeed
extend ActiveSupport::Concern
@@ -30,6 +42,10 @@ module Gitlab
end
end
+ def self.log_message(message)
+ puts "#{Time.current}: #{message}"
+ end
+
def self.with_mass_insert(size, model)
humanized_model_name = model.is_a?(String) ? model : model.model_name.human.pluralize(size)
@@ -63,6 +79,7 @@ module Gitlab
def self.quiet
# Additional seed logic for models.
+ Namespace.include(NamespaceSeed)
Project.include(ProjectSeed)
User.include(UserSeed)