diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2020-10-21 07:08:36 +0000 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2020-10-21 07:08:36 +0000 |
commit | 48aff82709769b098321c738f3444b9bdaa694c6 (patch) | |
tree | e00c7c43e2d9b603a5a6af576b1685e400410dee /app/services/web_hooks/destroy_service.rb | |
parent | 879f5329ee916a948223f8f43d77fba4da6cd028 (diff) | |
download | gitlab-ce-48aff82709769b098321c738f3444b9bdaa694c6.tar.gz |
Add latest changes from gitlab-org/gitlab@13-5-stable-eev13.5.0-rc42
Diffstat (limited to 'app/services/web_hooks/destroy_service.rb')
-rw-r--r-- | app/services/web_hooks/destroy_service.rb | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/app/services/web_hooks/destroy_service.rb b/app/services/web_hooks/destroy_service.rb new file mode 100644 index 00000000000..58117985b56 --- /dev/null +++ b/app/services/web_hooks/destroy_service.rb @@ -0,0 +1,78 @@ +# frozen_string_literal: true + +module WebHooks + class DestroyService + include BaseServiceUtility + + BATCH_SIZE = 1000 + LOG_COUNT_THRESHOLD = 10000 + + DestroyError = Class.new(StandardError) + + attr_accessor :current_user, :web_hook + + def initialize(current_user) + @current_user = current_user + end + + def execute(web_hook) + @web_hook = web_hook + + async = false + # For a better user experience, it's better if the Web hook is + # destroyed right away without waiting for Sidekiq. However, if + # there are a lot of Web hook logs, we will need more time to + # clean them up, so schedule a Sidekiq job to do this. + if needs_async_destroy? + Gitlab::AppLogger.info("User #{current_user&.id} scheduled a deletion of hook ID #{web_hook.id}") + async_destroy(web_hook) + async = true + else + sync_destroy(web_hook) + end + + success({ async: async }) + end + + def sync_destroy(web_hook) + @web_hook = web_hook + + delete_web_hook_logs + result = web_hook.destroy + + if result + success({ async: false }) + else + error("Unable to destroy #{web_hook.model_name.human}") + end + end + + private + + def async_destroy(web_hook) + WebHooks::DestroyWorker.perform_async(current_user.id, web_hook.id) + end + + # rubocop: disable CodeReuse/ActiveRecord + def needs_async_destroy? + web_hook.web_hook_logs.limit(LOG_COUNT_THRESHOLD).count == LOG_COUNT_THRESHOLD + end + # rubocop: enable CodeReuse/ActiveRecord + + def delete_web_hook_logs + loop do + count = delete_web_hook_logs_in_batches + break if count < BATCH_SIZE + end + end + + # rubocop: disable CodeReuse/ActiveRecord + def delete_web_hook_logs_in_batches + # We can't use EachBatch because that does an ORDER BY id, which can + # easily time out. We don't actually care about ordering when + # we are deleting these rows. + web_hook.web_hook_logs.limit(BATCH_SIZE).delete_all + end + # rubocop: enable CodeReuse/ActiveRecord + end +end |