Add latest changes from gitlab-org/gitlab@master

1 files changed, 28 insertions, 0 deletions

diff --git a/lib/gitlab/background_migration/populate_canonical_emails.rb b/lib/gitlab/background_migration/populate_canonical_emails.rb
new file mode 100644
index 00000000000..052e75c5655
--- /dev/null
+++ b/lib/gitlab/background_migration/populate_canonical_emails.rb
@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module BackgroundMigration
+    # Class to populate new rows of UserCanonicalEmail based on existing email addresses
+    class PopulateCanonicalEmails
+      def perform(start_id, stop_id)
+        ActiveRecord::Base.connection.execute <<~SQL
+          INSERT INTO
+            user_canonical_emails (
+                user_id,
+                canonical_email,
+                created_at,
+                updated_at
+            )
+            SELECT users.id AS user_id,
+                   concat(translate(split_part(split_part(users.email, '@', 1), '+', 1), '.', ''), '@gmail.com') AS canonical_email,
+                   NOW() AS created_at,
+                   NOW() AS updated_at
+             FROM users
+            WHERE users.email ILIKE '%@gmail.com'
+              AND users.id BETWEEN #{start_id} AND #{stop_id}
+            ON CONFLICT DO NOTHING;
+        SQL
+      end
+    end
+  end
+end