Search in sources :

Example 1 with AsyncCallableWithConnectionAndSource

use of com.mongodb.internal.operation.OperationHelper.AsyncCallableWithConnectionAndSource in project mongo-java-driver by mongodb.

the class FindOperation method executeAsync.

@Override
public void executeAsync(final AsyncReadBinding binding, final SingleResultCallback<AsyncBatchCursor<T>> callback) {
    RetryState retryState = initialRetryState(retryReads);
    binding.retain();
    AsyncCallbackSupplier<AsyncBatchCursor<T>> asyncRead = CommandOperationHelper.<AsyncBatchCursor<T>>decorateReadWithRetries(retryState, funcCallback -> {
        logRetryExecute(retryState);
        withAsyncSourceAndConnection(binding::getReadConnectionSource, false, funcCallback, (source, connection, releasingCallback) -> {
            if (retryState.breakAndCompleteIfRetryAnd(() -> !canRetryRead(source.getServerDescription(), connection.getDescription(), binding.getSessionContext()), releasingCallback)) {
                return;
            }
            if (serverIsAtLeastVersionThreeDotTwo(connection.getDescription())) {
                final SingleResultCallback<AsyncBatchCursor<T>> wrappedCallback = exceptionTransformingCallback(releasingCallback);
                createReadCommandAndExecuteAsync(retryState, binding, source, namespace.getDatabaseName(), getCommandCreator(binding.getSessionContext()), CommandResultDocumentCodec.create(decoder, FIRST_BATCH), asyncTransformer(), connection, wrappedCallback);
            } else {
                retryState.markAsLastAttempt();
                validateFindOptions(source, connection, binding.getSessionContext().getReadConcern(), collation, allowDiskUse, new AsyncCallableWithConnectionAndSource() {

                    @Override
                    public void call(final AsyncConnectionSource source, final AsyncConnection connection, final Throwable t) {
                        if (t != null) {
                            releasingCallback.onResult(null, t);
                        } else {
                            connection.queryAsync(namespace, asDocument(connection.getDescription(), binding.getReadPreference()), projection, skip, limit, batchSize, isSecondaryOk() || binding.getReadPreference().isSecondaryOk(), isTailableCursor(), isAwaitData(), isNoCursorTimeout(), isPartial(), isOplogReplay(), decoder, binding.getRequestContext(), new SingleResultCallback<QueryResult<T>>() {

                                @Override
                                public void onResult(final QueryResult<T> result, final Throwable t) {
                                    if (t != null) {
                                        releasingCallback.onResult(null, t);
                                    } else {
                                        releasingCallback.onResult(new AsyncQueryBatchCursor<T>(result, limit, batchSize, getMaxTimeForCursor(), decoder, source, connection), null);
                                    }
                                }
                            });
                        }
                    }
                });
            }
        });
    }).whenComplete(binding::release);
    asyncRead.get(errorHandlingCallback(callback, LOGGER));
}
Also used : AsyncReadBinding(com.mongodb.internal.binding.AsyncReadBinding) ServerDescription(com.mongodb.connection.ServerDescription) CommandOperationHelper.initialRetryState(com.mongodb.internal.operation.CommandOperationHelper.initialRetryState) BsonValue(org.bson.BsonValue) DocumentHelper.putIfNotNull(com.mongodb.internal.operation.DocumentHelper.putIfNotNull) ConnectionDescription(com.mongodb.connection.ConnectionDescription) ExplainHelper.asExplainCommand(com.mongodb.internal.operation.ExplainHelper.asExplainCommand) DocumentHelper.putIfNotNullOrEmpty(com.mongodb.internal.operation.DocumentHelper.putIfNotNullOrEmpty) MIN_WIRE_VERSION(com.mongodb.internal.operation.ServerVersionHelper.MIN_WIRE_VERSION) CursorType(com.mongodb.CursorType) AsyncCallableWithConnectionAndSource(com.mongodb.internal.operation.OperationHelper.AsyncCallableWithConnectionAndSource) OperationHelper.validateFindOptions(com.mongodb.internal.operation.OperationHelper.validateFindOptions) OperationHelper.canRetryRead(com.mongodb.internal.operation.OperationHelper.canRetryRead) QueryResult(com.mongodb.internal.connection.QueryResult) OperationHelper.withSourceAndConnection(com.mongodb.internal.operation.OperationHelper.withSourceAndConnection) ConnectionSource(com.mongodb.internal.binding.ConnectionSource) ReadBinding(com.mongodb.internal.binding.ReadBinding) AsyncCallbackSupplier(com.mongodb.internal.async.function.AsyncCallbackSupplier) OperationReadConcernHelper.appendReadConcernToCommand(com.mongodb.internal.operation.OperationReadConcernHelper.appendReadConcernToCommand) NoOpSessionContext(com.mongodb.internal.connection.NoOpSessionContext) SingleResultCallback(com.mongodb.internal.async.SingleResultCallback) Decoder(org.bson.codecs.Decoder) RetryState(com.mongodb.internal.async.function.RetryState) CommandReadTransformerAsync(com.mongodb.internal.operation.CommandOperationHelper.CommandReadTransformerAsync) Assertions.notNull(com.mongodb.assertions.Assertions.notNull) ReadPreference.primary(com.mongodb.ReadPreference.primary) ExplainVerbosity(com.mongodb.ExplainVerbosity) ReadPreference(com.mongodb.ReadPreference) CommandOperationHelper.logRetryExecute(com.mongodb.internal.operation.CommandOperationHelper.logRetryExecute) OperationHelper.cursorDocumentToQueryResult(com.mongodb.internal.operation.OperationHelper.cursorDocumentToQueryResult) CommandCreator(com.mongodb.internal.operation.CommandOperationHelper.CommandCreator) BsonBoolean(org.bson.BsonBoolean) MongoCommandException(com.mongodb.MongoCommandException) ErrorHandlingResultCallback.errorHandlingCallback(com.mongodb.internal.async.ErrorHandlingResultCallback.errorHandlingCallback) CommandReadTransformer(com.mongodb.internal.operation.CommandOperationHelper.CommandReadTransformer) BsonString(org.bson.BsonString) CommandOperationHelper.createReadCommandAndExecuteAsync(com.mongodb.internal.operation.CommandOperationHelper.createReadCommandAndExecuteAsync) MongoQueryException(com.mongodb.MongoQueryException) Supplier(java.util.function.Supplier) BsonDocument(org.bson.BsonDocument) LOGGER(com.mongodb.internal.operation.OperationHelper.LOGGER) Connection(com.mongodb.internal.connection.Connection) SHARD_ROUTER(com.mongodb.connection.ServerType.SHARD_ROUTER) BsonInt64(org.bson.BsonInt64) ServerVersionHelper.serverIsAtLeastVersionThreeDotTwo(com.mongodb.internal.operation.ServerVersionHelper.serverIsAtLeastVersionThreeDotTwo) AsyncConnectionSource(com.mongodb.internal.binding.AsyncConnectionSource) SessionContext(com.mongodb.internal.session.SessionContext) Collation(com.mongodb.client.model.Collation) MongoNamespace(com.mongodb.MongoNamespace) CommandOperationHelper.createReadCommandAndExecute(com.mongodb.internal.operation.CommandOperationHelper.createReadCommandAndExecute) CommandOperationHelper.decorateReadWithRetries(com.mongodb.internal.operation.CommandOperationHelper.decorateReadWithRetries) TimeUnit(java.util.concurrent.TimeUnit) AsyncBatchCursor(com.mongodb.internal.async.AsyncBatchCursor) Assertions.isTrueArgument(com.mongodb.assertions.Assertions.isTrueArgument) OperationHelper.withAsyncSourceAndConnection(com.mongodb.internal.operation.OperationHelper.withAsyncSourceAndConnection) Nullable(com.mongodb.lang.Nullable) BsonInt32(org.bson.BsonInt32) AsyncConnection(com.mongodb.internal.connection.AsyncConnection) AsyncBatchCursor(com.mongodb.internal.async.AsyncBatchCursor) SingleResultCallback(com.mongodb.internal.async.SingleResultCallback) QueryResult(com.mongodb.internal.connection.QueryResult) OperationHelper.cursorDocumentToQueryResult(com.mongodb.internal.operation.OperationHelper.cursorDocumentToQueryResult) AsyncConnectionSource(com.mongodb.internal.binding.AsyncConnectionSource) AsyncConnection(com.mongodb.internal.connection.AsyncConnection) AsyncCallableWithConnectionAndSource(com.mongodb.internal.operation.OperationHelper.AsyncCallableWithConnectionAndSource) CommandOperationHelper.initialRetryState(com.mongodb.internal.operation.CommandOperationHelper.initialRetryState) RetryState(com.mongodb.internal.async.function.RetryState)

Aggregations

CursorType (com.mongodb.CursorType)1 ExplainVerbosity (com.mongodb.ExplainVerbosity)1 MongoCommandException (com.mongodb.MongoCommandException)1 MongoNamespace (com.mongodb.MongoNamespace)1 MongoQueryException (com.mongodb.MongoQueryException)1 ReadPreference (com.mongodb.ReadPreference)1 ReadPreference.primary (com.mongodb.ReadPreference.primary)1 Assertions.isTrueArgument (com.mongodb.assertions.Assertions.isTrueArgument)1 Assertions.notNull (com.mongodb.assertions.Assertions.notNull)1 Collation (com.mongodb.client.model.Collation)1 ConnectionDescription (com.mongodb.connection.ConnectionDescription)1 ServerDescription (com.mongodb.connection.ServerDescription)1 SHARD_ROUTER (com.mongodb.connection.ServerType.SHARD_ROUTER)1 AsyncBatchCursor (com.mongodb.internal.async.AsyncBatchCursor)1 ErrorHandlingResultCallback.errorHandlingCallback (com.mongodb.internal.async.ErrorHandlingResultCallback.errorHandlingCallback)1 SingleResultCallback (com.mongodb.internal.async.SingleResultCallback)1 AsyncCallbackSupplier (com.mongodb.internal.async.function.AsyncCallbackSupplier)1 RetryState (com.mongodb.internal.async.function.RetryState)1 AsyncConnectionSource (com.mongodb.internal.binding.AsyncConnectionSource)1 AsyncReadBinding (com.mongodb.internal.binding.AsyncReadBinding)1