summaryrefslogtreecommitdiff
path: root/app/services/todos/destroy/destroyed_issuable_service.rb
blob: db12965224b39cba4da1dd4bc77bd5782efb9867 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
# frozen_string_literal: true

module Todos
  module Destroy
    class DestroyedIssuableService
      BATCH_SIZE = 100

      def initialize(target_id, target_type)
        @target_id = target_id
        @target_type = 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 = ActiveRecord::Base.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