Welcome to mirror list, hosted at ThFree Co, Russian Federation.

gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'app/services/users/update_service.rb')
-rw-r--r--app/services/users/update_service.rb29
1 files changed, 26 insertions, 3 deletions
diff --git a/app/services/users/update_service.rb b/app/services/users/update_service.rb
index 96018db5974..36c41c03303 100644
--- a/app/services/users/update_service.rb
+++ b/app/services/users/update_service.rb
@@ -6,6 +6,7 @@ module Users
attr_reader :user, :identity_params
ATTRS_REQUIRING_PASSWORD_CHECK = %w[email].freeze
+ BATCH_SIZE = 100
def initialize(current_user, params = {})
@current_user = current_user
@@ -34,7 +35,7 @@ module Users
reset_unconfirmed_email
if @user.save(validate: validate) && update_status
- notify_success(user_exists)
+ after_update(user_exists)
else
messages = @user.errors.full_messages + Array(@user.status&.errors&.full_messages)
error(messages.uniq.join('. '))
@@ -80,8 +81,6 @@ module Users
def notify_success(user_exists)
notify_new_user(@user, nil) unless user_exists
-
- success
end
def discard_read_only_attributes
@@ -118,6 +117,30 @@ module Users
def provider_params
identity_params.slice(*provider_attributes)
end
+
+ def after_update(user_exists)
+ notify_success(user_exists)
+ remove_followers_and_followee! if ::Feature.enabled?(:disable_follow_users, user)
+
+ success
+ end
+
+ def remove_followers_and_followee!
+ return false unless user.user_preference.enabled_following_previously_changed?(from: true, to: false)
+
+ # rubocop: disable CodeReuse/ActiveRecord
+ loop do
+ inner_query = Users::UserFollowUser
+ .where(follower_id: user.id).or(Users::UserFollowUser.where(followee_id: user.id))
+ .select(:follower_id, :followee_id)
+ .limit(BATCH_SIZE)
+
+ deleted_records = Users::UserFollowUser.where('(follower_id, followee_id) IN (?)', inner_query).delete_all
+
+ break if deleted_records == 0
+ end
+ # rubocop: enable CodeReuse/ActiveRecord
+ end
end
end