Add latest changes from gitlab-org/gitlab@16-0-stable-eev16.0.0-rc42

1 files changed, 61 insertions, 0 deletions

diff --git a/lib/gitlab/resource_events/assignment_event_recorder.rb b/lib/gitlab/resource_events/assignment_event_recorder.rb
new file mode 100644
index 00000000000..94bd05a17ba
--- /dev/null
+++ b/lib/gitlab/resource_events/assignment_event_recorder.rb
@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module ResourceEvents
+    class AssignmentEventRecorder
+      BATCH_SIZE = 100
+
+      def initialize(parent:, old_assignees:)
+        @parent = parent
+        @old_assignees = old_assignees
+      end
+
+      def record
+        return if Feature.disabled?(:record_issue_and_mr_assignee_events, parent.project)
+
+        case parent
+        when Issue
+          record_for_parent(
+            ::ResourceEvents::IssueAssignmentEvent,
+            :issue_id,
+            parent,
+            old_assignees
+          )
+        when MergeRequest
+          record_for_parent(
+            ::ResourceEvents::MergeRequestAssignmentEvent,
+            :merge_request_id,
+            parent,
+            old_assignees
+          )
+        end
+      end
+
+      private
+
+      attr_reader :parent, :old_assignees
+
+      def record_for_parent(resource_klass, foreign_key, parent, old_assignees)
+        removed_events = (old_assignees - parent.assignees).map do |unassigned_user|
+          {
+            foreign_key => parent.id,
+            user_id: unassigned_user.id,
+            action: :remove
+          }
+        end.to_set
+
+        added_events = (parent.assignees.to_a - old_assignees).map do |added_user|
+          {
+            foreign_key => parent.id,
+            user_id: added_user.id,
+            action: :add
+          }
+        end.to_set
+
+        (removed_events + added_events).each_slice(BATCH_SIZE) do |events|
+          resource_klass.insert_all(events)
+        end
+      end
+    end
+  end
+end