Browse Source

Close connection when succeeded posting (#5390)

* Close connection when succeeded posting

* Update webmock
master
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