Browse Source

Close connection when succeeded posting (#5390)

* Close connection when succeeded posting

* Update webmock
pull/4/head
abcang 6 years ago
committed by Eugen Rochko
parent
commit
2eab41cd1a
6 changed files with 13 additions and 7 deletions
  1. +2
    -2
      Gemfile.lock
  2. +3
    -1
      app/services/send_interaction_service.rb
  3. +2
    -1
      app/services/subscribe_service.rb
  4. +2
    -1
      app/services/unsubscribe_service.rb
  5. +2
    -1
      app/workers/activitypub/delivery_worker.rb
  6. +2
    -1
      app/workers/pubsubhubbub/delivery_worker.rb

+ 2
- 2
Gemfile.lock View File

@ -193,7 +193,7 @@ GEM
railties (>= 4.0.1)
hamster (3.0.0)
concurrent-ruby (~> 1.0)
hashdiff (0.3.6)
hashdiff (0.3.7)
highline (1.7.8)
hiredis (0.6.1)
hkdf (0.3.0)
@ -513,7 +513,7 @@ GEM
uniform_notifier (1.10.0)
warden (1.2.7)
rack (>= 1.0)
webmock (3.0.1)
webmock (3.1.0)
addressable (>= 2.3.6)
crack (>= 0.3.2)
hashdiff

+ 3
- 1
app/services/send_interaction_service.rb View File

@ -12,9 +12,11 @@ class SendInteractionService < BaseService
return if !target_account.ostatus? || block_notification?
delivery = build_request.perform.flush
delivery = build_request.perform
raise Mastodon::UnexpectedResponseError, delivery unless delivery.code > 199 && delivery.code < 300
delivery.connection&.close
end
private

+ 2
- 1
app/services/subscribe_service.rb View File

@ -6,7 +6,7 @@ class SubscribeService < BaseService
@account = account
@account.secret = SecureRandom.hex
@response = build_request.perform.flush
@response = build_request.perform
if response_failed_permanently?
# We're not allowed to subscribe. Fail and move on.
@ -20,6 +20,7 @@ class SubscribeService < BaseService
# We need to retry at a later time. Fail loudly!
raise Mastodon::UnexpectedResponseError, @response
end
@response.connection&.close
end
private

+ 2
- 1
app/services/unsubscribe_service.rb View File

@ -7,9 +7,10 @@ class UnsubscribeService < BaseService
@account = account
begin
@response = build_request.perform.flush
@response = build_request.perform
Rails.logger.debug "PuSH unsubscribe for #{@account.acct} failed: #{@response.status}" unless @response.status.success?
@response.connection&.close
rescue HTTP::Error, OpenSSL::SSL::SSLError => e
Rails.logger.debug "PuSH unsubscribe for #{@account.acct} failed: #{e}"
end

+ 2
- 1
app/workers/activitypub/delivery_worker.rb View File

@ -16,6 +16,7 @@ class ActivityPub::DeliveryWorker
raise Mastodon::UnexpectedResponseError, @response unless response_successful?
@response.connection&.close
failure_tracker.track_success!
rescue => e
failure_tracker.track_failure!
@ -31,7 +32,7 @@ class ActivityPub::DeliveryWorker
end
def perform_request
@response = build_request.perform.flush
@response = build_request.perform
end
def response_successful?

+ 2
- 1
app/workers/pubsubhubbub/delivery_worker.rb View File

@ -27,6 +27,7 @@ class Pubsubhubbub::DeliveryWorker
raise Mastodon::UnexpectedResponseError, payload_delivery unless response_successful?
payload_delivery.connection&.close
subscription.touch(:last_successful_delivery_at)
end
@ -37,7 +38,7 @@ class Pubsubhubbub::DeliveryWorker
def callback_post_payload
request = Request.new(:post, subscription.callback_url, body: payload)
request.add_headers(headers)
request.perform.flush
request.perform
end
def blocked_domain?

Loading…
Cancel
Save