47 lines
1.1 KiB
Ruby
47 lines
1.1 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
# Centralized class for repository size related calculations.
|
||
|
|
class RepositorySizeChecker
|
||
|
|
attr_reader :limit
|
||
|
|
|
||
|
|
# @param current_size_proc [Proc] returns repository size in bytes
|
||
|
|
def initialize(current_size_proc:, limit:, enabled: true)
|
||
|
|
@current_size_proc = current_size_proc
|
||
|
|
@limit = limit
|
||
|
|
@enabled = enabled && limit != 0
|
||
|
|
end
|
||
|
|
|
||
|
|
# @return [Integer] bytes
|
||
|
|
def current_size
|
||
|
|
@current_size ||= @current_size_proc.call
|
||
|
|
end
|
||
|
|
|
||
|
|
def enabled?
|
||
|
|
@enabled
|
||
|
|
end
|
||
|
|
|
||
|
|
def above_size_limit?
|
||
|
|
return false unless enabled?
|
||
|
|
|
||
|
|
current_size > limit
|
||
|
|
end
|
||
|
|
|
||
|
|
# @param change_size [int] in bytes
|
||
|
|
def changes_will_exceed_size_limit?(change_size)
|
||
|
|
return false unless enabled?
|
||
|
|
|
||
|
|
change_size > limit || exceeded_size(change_size) > 0
|
||
|
|
end
|
||
|
|
|
||
|
|
# @param change_size [int] in bytes
|
||
|
|
def exceeded_size(change_size = 0)
|
||
|
|
current_size + change_size - limit
|
||
|
|
end
|
||
|
|
|
||
|
|
def error_message
|
||
|
|
@error_message_object ||= Gitlab::RepositorySizeErrorMessage.new(self)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|