2018-07-17 16:50:37 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2017-05-04 10:13:33 +00:00
|
|
|
module Projects
|
2020-02-07 12:09:13 +00:00
|
|
|
class PropagateInstanceLevelService
|
2017-05-04 10:13:33 +00:00
|
|
|
BATCH_SIZE = 100
|
|
|
|
|
2017-05-04 16:11:28 +00:00
|
|
|
def self.propagate(*args)
|
|
|
|
new(*args).propagate
|
2017-05-04 10:13:33 +00:00
|
|
|
end
|
|
|
|
|
2020-02-07 12:09:13 +00:00
|
|
|
def initialize(instance_level_service)
|
|
|
|
@instance_level_service = instance_level_service
|
2017-05-04 10:13:33 +00:00
|
|
|
end
|
|
|
|
|
2017-05-04 16:11:28 +00:00
|
|
|
def propagate
|
2020-02-07 12:09:13 +00:00
|
|
|
return unless @instance_level_service.active?
|
2017-05-04 10:13:33 +00:00
|
|
|
|
2020-02-07 12:09:13 +00:00
|
|
|
Rails.logger.info("Propagating services for instance_level_service #{@instance_level_service.id}") # rubocop:disable Gitlab/RailsLogger
|
2017-05-04 10:13:33 +00:00
|
|
|
|
2020-02-07 12:09:13 +00:00
|
|
|
propagate_projects_with_instance_level_service
|
2017-05-04 10:13:33 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
2020-02-07 12:09:13 +00:00
|
|
|
def propagate_projects_with_instance_level_service
|
2017-05-04 10:13:33 +00:00
|
|
|
loop do
|
2019-06-25 21:47:17 +00:00
|
|
|
batch = Project.uncached { project_ids_batch }
|
2017-05-04 10:13:33 +00:00
|
|
|
|
2020-02-07 12:09:13 +00:00
|
|
|
bulk_create_from_instance_level_service(batch) unless batch.empty?
|
2017-05-04 10:13:33 +00:00
|
|
|
|
2017-05-04 16:11:28 +00:00
|
|
|
break if batch.size < BATCH_SIZE
|
2017-05-04 10:13:33 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2020-02-07 12:09:13 +00:00
|
|
|
def bulk_create_from_instance_level_service(batch)
|
2017-05-05 12:43:22 +00:00
|
|
|
service_list = batch.map do |project_id|
|
2017-05-05 16:57:52 +00:00
|
|
|
service_hash.values << project_id
|
2017-05-05 07:40:44 +00:00
|
|
|
end
|
|
|
|
|
2017-05-05 14:16:02 +00:00
|
|
|
Project.transaction do
|
2017-05-05 16:57:52 +00:00
|
|
|
bulk_insert_services(service_hash.keys << 'project_id', service_list)
|
2017-05-05 14:16:02 +00:00
|
|
|
run_callbacks(batch)
|
|
|
|
end
|
2017-05-04 10:13:33 +00:00
|
|
|
end
|
|
|
|
|
2017-05-05 08:43:56 +00:00
|
|
|
def project_ids_batch
|
2017-05-05 08:51:25 +00:00
|
|
|
Project.connection.select_values(
|
2017-05-05 07:40:44 +00:00
|
|
|
<<-SQL
|
|
|
|
SELECT id
|
|
|
|
FROM projects
|
|
|
|
WHERE NOT EXISTS (
|
|
|
|
SELECT true
|
|
|
|
FROM services
|
|
|
|
WHERE services.project_id = projects.id
|
2020-02-07 12:09:13 +00:00
|
|
|
AND services.type = '#{@instance_level_service.type}'
|
2017-05-05 07:40:44 +00:00
|
|
|
)
|
2017-05-05 16:01:33 +00:00
|
|
|
AND projects.pending_delete = false
|
|
|
|
AND projects.archived = false
|
2017-05-05 08:43:56 +00:00
|
|
|
LIMIT #{BATCH_SIZE}
|
2017-05-05 07:40:44 +00:00
|
|
|
SQL
|
2017-05-05 08:51:25 +00:00
|
|
|
)
|
2017-05-05 07:40:44 +00:00
|
|
|
end
|
|
|
|
|
2017-05-05 16:01:33 +00:00
|
|
|
def bulk_insert_services(columns, values_array)
|
|
|
|
ActiveRecord::Base.connection.execute(
|
|
|
|
<<-SQL.strip_heredoc
|
|
|
|
INSERT INTO services (#{columns.join(', ')})
|
|
|
|
VALUES #{values_array.map { |tuple| "(#{tuple.join(', ')})" }.join(', ')}
|
|
|
|
SQL
|
|
|
|
)
|
|
|
|
end
|
|
|
|
|
2017-05-05 07:40:44 +00:00
|
|
|
def service_hash
|
2017-05-05 12:43:22 +00:00
|
|
|
@service_hash ||=
|
|
|
|
begin
|
2020-02-07 12:09:13 +00:00
|
|
|
instance_hash = @instance_level_service.as_json(methods: :type).except('id', 'instance', 'project_id')
|
2017-05-05 12:43:22 +00:00
|
|
|
|
2020-02-07 12:09:13 +00:00
|
|
|
instance_hash.each_with_object({}) do |(key, value), service_hash|
|
2017-05-05 12:43:22 +00:00
|
|
|
value = value.is_a?(Hash) ? value.to_json : value
|
|
|
|
|
2017-05-05 16:57:52 +00:00
|
|
|
service_hash[ActiveRecord::Base.connection.quote_column_name(key)] =
|
2019-01-15 21:05:36 +00:00
|
|
|
ActiveRecord::Base.connection.quote(value)
|
2017-05-05 12:43:22 +00:00
|
|
|
end
|
|
|
|
end
|
2017-05-04 10:13:33 +00:00
|
|
|
end
|
2017-05-05 14:16:02 +00:00
|
|
|
|
2018-08-27 15:31:01 +00:00
|
|
|
# rubocop: disable CodeReuse/ActiveRecord
|
2017-05-05 14:16:02 +00:00
|
|
|
def run_callbacks(batch)
|
|
|
|
if active_external_issue_tracker?
|
|
|
|
Project.where(id: batch).update_all(has_external_issue_tracker: true)
|
|
|
|
end
|
|
|
|
|
|
|
|
if active_external_wiki?
|
|
|
|
Project.where(id: batch).update_all(has_external_wiki: true)
|
|
|
|
end
|
|
|
|
end
|
2018-08-27 15:31:01 +00:00
|
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
2017-05-05 14:16:02 +00:00
|
|
|
|
|
|
|
def active_external_issue_tracker?
|
2020-02-07 12:09:13 +00:00
|
|
|
@instance_level_service.issue_tracker? && !@instance_level_service.default
|
2017-05-05 14:16:02 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def active_external_wiki?
|
2020-02-07 12:09:13 +00:00
|
|
|
@instance_level_service.type == 'ExternalWikiService'
|
2017-05-05 14:16:02 +00:00
|
|
|
end
|
2017-05-04 10:13:33 +00:00
|
|
|
end
|
|
|
|
end
|