Change delivery failure tracking to work with hostnames instead of URLs (#13437)

This commit is contained in:
Eugen Rochko
2020-04-15 20:33:24 +02:00
committed by GitHub
parent 5524258da9
commit 5edff32733
15 changed files with 103 additions and 44 deletions

View File

@ -3,47 +3,53 @@
class DeliveryFailureTracker
FAILURE_DAYS_THRESHOLD = 7
def initialize(inbox_url)
@inbox_url = inbox_url
def initialize(url_or_host)
@host = url_or_host.start_with?('https://') || url_or_host.start_with?('http://') ? Addressable::URI.parse(url_or_host).normalized_host : url_or_host
end
def track_failure!
Redis.current.sadd(exhausted_deliveries_key, today)
Redis.current.sadd('unavailable_inboxes', @inbox_url) if reached_failure_threshold?
UnavailableDomain.create(domain: @host) if reached_failure_threshold?
end
def track_success!
Redis.current.del(exhausted_deliveries_key)
Redis.current.srem('unavailable_inboxes', @inbox_url)
UnavailableDomain.find_by(domain: @host)&.destroy
end
def days
Redis.current.scard(exhausted_deliveries_key) || 0
end
class << self
def filter(arr)
arr.reject(&method(:unavailable?))
end
def available?
!UnavailableDomain.where(domain: @host).exists?
end
def unavailable?(url)
Redis.current.sismember('unavailable_inboxes', url)
alias reset! track_success!
class << self
def without_unavailable(urls)
unavailable_domains_map = Rails.cache.fetch('unavailable_domains') { UnavailableDomain.pluck(:domain).each_with_object({}) { |domain, hash| hash[domain] = true } }
urls.reject do |url|
host = Addressable::URI.parse(url).normalized_host
unavailable_domains_map[host]
end
end
def available?(url)
!unavailable?(url)
new(url).available?
end
def track_inverse_success!(from_account)
new(from_account.inbox_url).track_success! if from_account.inbox_url.present?
new(from_account.shared_inbox_url).track_success! if from_account.shared_inbox_url.present?
def reset!(url)
new(url).reset!
end
end
private
def exhausted_deliveries_key
"exhausted_deliveries:#{@inbox_url}"
"exhausted_deliveries:#{@host}"
end
def today