79 lines
2 KiB
Ruby
79 lines
2 KiB
Ruby
|
# 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
|