Remove usages of hasExpired

Author: katcharov

driver-core/src/main/com/mongodb/internal/TimeoutContext.java

@@ -144,6 +144,10 @@ public boolean hasExpired() {
         return Timeout.nullAsInfinite(timeout).call(NANOSECONDS, () -> false, (ns) -> false, () -> true);
     }
 
+    public void onExpired(final Runnable onExpired) {
+        Timeout.nullAsInfinite(timeout).onExpired(onExpired);
+    }
+
     /**
      * Sets the recent min round trip time
      * @param minRoundTripTimeMS the min round trip time
@@ -370,6 +374,7 @@ public Timeout startWaitQueueTimeout(final StartTime checkoutStart) {
         return checkoutStart.timeoutAfterOrInfiniteIfNegative(ms, MILLISECONDS);
     }
 
+    // TODO (CSOT) method not used in production;
     @Nullable
     public Timeout getTimeout() {
         return timeout;

driver-core/src/main/com/mongodb/internal/connection/InternalStreamConnection.java

@@ -771,24 +771,22 @@ private void updateSessionContext(final SessionContext sessionContext, final Res
     }
 
     private void throwTranslatedWriteException(final Throwable e, final OperationContext operationContext) {
-        throw translateWriteException(e, operationContext);
-    }
-
-    private MongoException translateWriteException(final Throwable e, final OperationContext operationContext) {
-        if (e instanceof MongoSocketWriteTimeoutException && operationContext.getTimeoutContext().hasExpired()) {
-            return createMongoTimeoutException(e);
+        if (e instanceof MongoSocketWriteTimeoutException) {
+            operationContext.getTimeoutContext().onExpired(() -> {
+                throw createMongoTimeoutException(e);
+            });
         }
 
         if (e instanceof MongoException) {
-            return (MongoException) e;
+            throw (MongoException) e;
         }
         Optional<MongoInterruptedException> interruptedException = translateInterruptedException(e, "Interrupted while sending message");
         if (interruptedException.isPresent()) {
-            return interruptedException.get();
+            throw interruptedException.get();
         } else if (e instanceof IOException) {
-            return new MongoSocketWriteException("Exception sending message", getServerAddress(), e);
+            throw new MongoSocketWriteException("Exception sending message", getServerAddress(), e);
         } else {
-            return new MongoInternalException("Unexpected exception", e);
+            throw new MongoInternalException("Unexpected exception", e);
         }
     }
 

driver-core/src/main/com/mongodb/internal/connection/SocketStream.java

@@ -163,9 +163,9 @@ public ByteBuf getBuffer(final int size) {
     public void write(final List<ByteBuf> buffers, final OperationContext operationContext) throws IOException {
         for (final ByteBuf cur : buffers) {
             outputStream.write(cur.array(), 0, cur.limit());
-            if (operationContext.getTimeoutContext().hasExpired()) {
+            operationContext.getTimeoutContext().onExpired(() -> {
                 throwMongoTimeoutException("Socket write exceeded the timeout limit.");
-            }
+            });
         }
     }