Replace BiFunction with BinaryOperator

JAVA-4254

Author: stIncMale

driver-core/src/main/com/mongodb/internal/operation/AsyncOperationHelper.java

@@ -44,7 +44,7 @@
 
 import java.util.Collections;
 import java.util.List;
-import java.util.function.BiFunction;
+import java.util.function.BinaryOperator;
 
 import static com.mongodb.assertions.Assertions.assertNotNull;
 import static com.mongodb.internal.async.ErrorHandlingResultCallback.errorHandlingCallback;
@@ -286,7 +286,7 @@ static <D, T> void createReadCommandAndExecuteAsync(
 
     static <R> AsyncCallbackSupplier<R> decorateReadWithRetriesAsync(final RetryState retryState, final OperationContext operationContext,
             final AsyncCallbackSupplier<R> asyncReadFunction) {
-        BiFunction<Throwable, Throwable, Throwable> onAttemptFailure =
+        BinaryOperator<Throwable> onAttemptFailure =
                 (@Nullable Throwable previouslyChosenException, Throwable mostRecentAttemptException) ->
                         CommandOperationHelper.onRetryableReadAttemptFailure(
                                 operationContext, previouslyChosenException, mostRecentAttemptException);
@@ -299,7 +299,7 @@ static <R> AsyncCallbackSupplier<R> decorateReadWithRetriesAsync(final RetryStat
 
     static <R> AsyncCallbackSupplier<R> decorateWriteWithRetriesAsync(final RetryState retryState, final OperationContext operationContext,
             final AsyncCallbackSupplier<R> asyncWriteFunction) {
-        BiFunction<Throwable, Throwable, Throwable> onAttemptFailure =
+        BinaryOperator<Throwable> onAttemptFailure =
                 (@Nullable Throwable previouslyChosenException, Throwable mostRecentAttemptException) ->
                         CommandOperationHelper.onRetryableWriteAttemptFailure(
                                 operationContext, previouslyChosenException, mostRecentAttemptException);

driver-core/src/main/com/mongodb/internal/operation/MixedBulkWriteOperation.java

@@ -51,7 +51,7 @@
 import java.util.List;
 import java.util.Optional;
 import java.util.Set;
-import java.util.function.BiFunction;
+import java.util.function.BinaryOperator;
 import java.util.function.Supplier;
 import java.util.stream.Collectors;
 
@@ -141,7 +141,7 @@ public Boolean getRetryWrites() {
 
     private <R> Supplier<R> decorateWriteWithRetries(final RetryState retryState, final OperationContext operationContext,
             final Supplier<R> writeFunction) {
-        BiFunction<Throwable, Throwable, Throwable> onAttemptFailure =
+        BinaryOperator<Throwable> onAttemptFailure =
                 (@Nullable Throwable previouslyChosenException, Throwable mostRecentAttemptException) ->
                         CommandOperationHelper.onRetryableWriteAttemptFailure(
                                 operationContext, previouslyChosenException, mostRecentAttemptException);
@@ -154,7 +154,7 @@ private <R> Supplier<R> decorateWriteWithRetries(final RetryState retryState, fi
 
     private <R> AsyncCallbackSupplier<R> decorateWriteWithRetries(final RetryState retryState, final OperationContext operationContext,
             final AsyncCallbackSupplier<R> writeFunction) {
-        BiFunction<Throwable, Throwable, Throwable> onAttemptFailure =
+        BinaryOperator<Throwable> onAttemptFailure =
                 (@Nullable Throwable previouslyChosenException, Throwable mostRecentAttemptException) ->
                         CommandOperationHelper.onRetryableWriteAttemptFailure(
                                 operationContext, previouslyChosenException, mostRecentAttemptException);

driver-core/src/main/com/mongodb/internal/operation/SyncOperationHelper.java

@@ -41,6 +41,7 @@
 import org.bson.codecs.Decoder;
 
 import java.util.function.BiFunction;
+import java.util.function.BinaryOperator;
 import java.util.function.Function;
 import java.util.function.Supplier;
 
@@ -274,7 +275,7 @@ static <D, T> T createReadCommandAndExecute(
 
     static <R> Supplier<R> decorateWriteWithRetries(final RetryState retryState,
             final OperationContext operationContext, final Supplier<R> writeFunction) {
-        BiFunction<Throwable, Throwable, Throwable> onAttemptFailure =
+        BinaryOperator<Throwable> onAttemptFailure =
                 (@Nullable Throwable previouslyChosenException, Throwable mostRecentAttemptException) ->
                         CommandOperationHelper.onRetryableWriteAttemptFailure(
                                 operationContext, previouslyChosenException, mostRecentAttemptException);
@@ -287,7 +288,7 @@ static <R> Supplier<R> decorateWriteWithRetries(final RetryState retryState,
 
     static <R> Supplier<R> decorateReadWithRetries(final RetryState retryState, final OperationContext operationContext,
             final Supplier<R> readFunction) {
-        BiFunction<Throwable, Throwable, Throwable> onAttemptFailure =
+        BinaryOperator<Throwable> onAttemptFailure =
                 (@Nullable Throwable previouslyChosenException, Throwable mostRecentAttemptException) ->
                         CommandOperationHelper.onRetryableReadAttemptFailure(
                                 operationContext, previouslyChosenException, mostRecentAttemptException);