Clean up TLS Socket Factory in key management service

JAVA-3734

Author: jyemin

driver-core/src/main/com/mongodb/internal/async/client/Crypt.java

@@ -18,7 +18,6 @@
 
 import com.mongodb.MongoClientException;
 import com.mongodb.MongoInternalException;
-import com.mongodb.internal.async.SingleResultCallback;
 import com.mongodb.client.model.vault.DataKeyOptions;
 import com.mongodb.client.model.vault.EncryptOptions;
 import com.mongodb.crypt.capi.MongoCrypt;
@@ -29,6 +28,7 @@
 import com.mongodb.crypt.capi.MongoKeyDecryptor;
 import com.mongodb.diagnostics.logging.Logger;
 import com.mongodb.diagnostics.logging.Loggers;
+import com.mongodb.internal.async.SingleResultCallback;
 import com.mongodb.lang.Nullable;
 import org.bson.BsonBinary;
 import org.bson.BsonDocument;
@@ -242,6 +242,7 @@ public void close() {
             commandMarker.close();
         }
         keyRetriever.close();
+        keyManagementService.close();
     }
 
     private void executeStateMachine(final MongoCryptContext cryptContext, final String databaseName,

driver-core/src/main/com/mongodb/internal/async/client/KeyManagementService.java

@@ -18,14 +18,14 @@
 
 import com.mongodb.MongoSocketException;
 import com.mongodb.ServerAddress;
-import com.mongodb.internal.async.SingleResultCallback;
 import com.mongodb.connection.AsyncCompletionHandler;
 import com.mongodb.connection.SocketSettings;
 import com.mongodb.connection.SslSettings;
 import com.mongodb.connection.Stream;
 import com.mongodb.connection.StreamFactory;
 import com.mongodb.connection.TlsChannelStreamFactoryFactory;
 import com.mongodb.crypt.capi.MongoKeyDecryptor;
+import com.mongodb.internal.async.SingleResultCallback;
 import com.mongodb.internal.connection.AsynchronousChannelStream;
 import org.bson.ByteBuf;
 import org.bson.ByteBufNIO;
@@ -40,17 +40,24 @@
 
 class KeyManagementService {
     private final int defaultPort;
+    private final TlsChannelStreamFactoryFactory tlsChannelStreamFactoryFactory;
     private final StreamFactory streamFactory;
 
     KeyManagementService(final SSLContext sslContext, final int defaultPort, final int timeoutMillis) {
         this.defaultPort = defaultPort;
-        this.streamFactory = new TlsChannelStreamFactoryFactory().create(SocketSettings.builder()
+        this.tlsChannelStreamFactoryFactory = new TlsChannelStreamFactoryFactory();
+        this.streamFactory = tlsChannelStreamFactoryFactory.create(SocketSettings.builder()
                         .connectTimeout(timeoutMillis, TimeUnit.MILLISECONDS)
                         .readTimeout(timeoutMillis, TimeUnit.MILLISECONDS)
                         .build(),
                 SslSettings.builder().enabled(true).context(sslContext).build());
     }
 
+
+    public void close() {
+        tlsChannelStreamFactoryFactory.close();
+    }
+
     void decryptKey(final MongoKeyDecryptor keyDecryptor, final SingleResultCallback<Void> callback) {
         streamOpen(keyDecryptor, callback);
     }