51 lines
1.5 KiB
Ruby
51 lines
1.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Todos
|
|
module Destroy
|
|
class DestroyedIssuableService
|
|
BATCH_SIZE = 100
|
|
|
|
# Since we are moving towards work items, in some instances we create todos with
|
|
# `target_type: WorkItem` in other instances we still create todos with `target_type: Issue`
|
|
# So when an issue/work item is deleted, we just make sure to delete todos for both target types
|
|
BOUND_TARGET_TYPES = %w(Issue WorkItem).freeze
|
|
|
|
def initialize(target_id, target_type)
|
|
@target_id = target_id
|
|
@target_type = BOUND_TARGET_TYPES.include?(target_type) ? BOUND_TARGET_TYPES : target_type
|
|
end
|
|
|
|
def execute
|
|
inner_query = Todo.select(:id).for_target(target_id).for_type(target_type).limit(BATCH_SIZE)
|
|
|
|
delete_query = <<~SQL
|
|
DELETE FROM "#{Todo.table_name}"
|
|
WHERE id IN (#{inner_query.to_sql})
|
|
RETURNING user_id
|
|
SQL
|
|
|
|
loop do
|
|
result = Todo.connection.execute(delete_query)
|
|
|
|
break if result.cmd_tuples == 0
|
|
|
|
user_ids = result.map { |row| row['user_id'] }.uniq
|
|
|
|
invalidate_todos_cache_counts(user_ids)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
attr_reader :target_id, :target_type
|
|
|
|
def invalidate_todos_cache_counts(user_ids)
|
|
user_ids.each do |id|
|
|
# Only build a user instance since we only need its ID for
|
|
# `User#invalidate_todos_cache_counts` to work.
|
|
User.new(id: id).invalidate_todos_cache_counts
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|