def write_with_retry(session, write_concern, ending_transaction = false, &block)
      if ending_transaction && !session
        raise ArgumentError, 'Cannot end a transaction without a session'
      end

      unless ending_transaction || retry_write_allowed?(session, write_concern)
        return legacy_write_with_retry(nil, session, &block)
      end

      # If we are here, session is not nil. A session being nil would have
      # failed retry_write_allowed? check.

      server = cluster.next_primary

      unless ending_transaction || server.retry_writes?
        return legacy_write_with_retry(server, session, &block)
      end

      begin
        txn_num = session.in_transaction? ? session.txn_num : session.next_txn_num
        yield(server, txn_num, false)
      rescue Error::SocketError, Error::SocketTimeoutError => e
        if session.in_transaction? && !ending_transaction
          raise
        end
        retry_write(e, txn_num, &block)
      rescue Error::OperationFailure => e
        if (session.in_transaction? && !ending_transaction) || !e.write_retryable?
          raise
        end
        retry_write(e, txn_num, &block)
      end
    end
def write_with_retry(session, write_concern, ending_transaction = false, &block)
      if ending_transaction && !session
        raise ArgumentError, 'Cannot end a transaction without a session'
      end

      unless ending_transaction || retry_write_allowed?(session, write_concern)
        return legacy_write_with_retry(nil, session, &block)
      end

      # If we are here, session is not nil. A session being nil would have
      # failed retry_write_allowed? check.

      server = cluster.next_primary

      unless ending_transaction || server.retry_writes?
        return legacy_write_with_retry(server, session, &block)
      end

      begin
        txn_num = session.in_transaction? ? session.txn_num : session.next_txn_num
        yield(server, txn_num, false)
      rescue Error::SocketError, Error::SocketTimeoutError => e
        if session.in_transaction? && !ending_transaction
          raise
        end
        retry_write(e, txn_num, &block)
      rescue Error::OperationFailure => e
        if (session.in_transaction? && !ending_transaction) || !e.write_retryable?
          raise
        end
        retry_write(e, txn_num, &block)
      end
    end
Implements write retrying functionality by yielding to the passed
 block one or more times.

 If the session is provided (hence, the deployment supports sessions),
 and modern retry writes are enabled on the client, the modern retry
 logic is invoked. Otherwise the legacy retry logic is invoked.

 If ending_transaction parameter is true, indicating that a transaction
 is being committed or aborted, the operation is executed exactly once.
 Note that, since transactions require sessions, this method will raise
 ArgumentError if ending_transaction is true and session is nil.

 @api private

 @example Execute the write.
   write_with_retry do
     ...
   end

 @note This only retries operations on not master failures, since it is
   the only case we can be sure a partial write did not already occur.

 @param [ nil | Session ] session Optional session to use with the operation.
 @param [ nil | Hash | WriteConcern::Base ] write_concern The write concern.
 @param [ true | false ] ending_transaction True if the write operation is abortTransaction or
   commitTransaction, false otherwise.
 @param [ Proc ] block The block to execute.

 @yieldparam [ Server ] server The server to which the write should be sent.
 @yieldparam [ Integer ] txn_num Transaction number (NOT the ACID kind).

 @return [ Result ] The result of the operation.

 @since 2.1.0