Sha256: 0a2357f2ca4a796cf5553a9ed0fb89d3ea4c1005ee0f20e1da10db47fa7fdc13
Contents?: true
Size: 1.76 KB
Versions: 55
Compression:
Stored size: 1.76 KB
Contents
# encoding: utf-8
module Mongoid #:nodoc:
module Collections #:nodoc:
# Provides behaviour for retrying commands on connection failure.
module Retry
# Retries command on connection failures.
#
# This is useful when using replica sets. When a primary server wents
# down and a command is issued, the driver will raise a
# Mongo::ConnectionFailure. We wait a little bit, because nodes are
# electing themselves, and then retry the given command.
#
# By setting Mongoid.max_retries_on_connection_failure to a value of 0,
# no attempt will be made, immediately raising connection failure.
# Otherwise it will attempt to make the specified number of retries
# and then raising the exception to clients.
#
# @example Retry the command.
# retry_on_connection_failure do
# collection.send(name, *args)
# end
#
# @since 2.0.0
def retry_on_connection_failure
retries = 0
begin
yield
rescue Mongo::ConnectionFailure => ex
retries = increase_retry_attempts(retries, ex)
retry
rescue Mongo::OperationFailure => ex
if ex.message =~ /not master/
retries = increase_retry_attempts(retries, ex)
retry
else
raise ex
end
end
end
private
def increase_retry_attempts(retries, ex)
retries += 1
raise ex if retries > Mongoid.max_retries_on_connection_failure
Kernel.sleep(0.5)
log_retry retries, ex
retries
end
def log_retry(retry_number, ex)
Mongoid.logger.warn "A #{ex.class.name} was raised. Retry attempt ##{retry_number}." if Mongoid.logger
end
end
end
end
Version data entries
55 entries across 55 versions & 3 rubygems