com.mongodb.reactivestreams.client.internal.MongoOperationPublisher Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongodb-driver-reactivestreams Show documentation
Show all versions of mongodb-driver-reactivestreams Show documentation
A Reactive Streams implementation of the MongoDB Java driver
/*
* Copyright 2008-present MongoDB, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.mongodb.reactivestreams.client.internal;
import com.mongodb.AutoEncryptionSettings;
import com.mongodb.MongoBulkWriteException;
import com.mongodb.MongoClientException;
import com.mongodb.MongoException;
import com.mongodb.MongoNamespace;
import com.mongodb.MongoWriteConcernException;
import com.mongodb.MongoWriteException;
import com.mongodb.ReadConcern;
import com.mongodb.ReadPreference;
import com.mongodb.WriteConcern;
import com.mongodb.WriteConcernResult;
import com.mongodb.WriteError;
import com.mongodb.bulk.BulkWriteResult;
import com.mongodb.bulk.WriteConcernError;
import com.mongodb.client.model.BulkWriteOptions;
import com.mongodb.client.model.ClusteredIndexOptions;
import com.mongodb.client.model.CountOptions;
import com.mongodb.client.model.CreateCollectionOptions;
import com.mongodb.client.model.CreateIndexOptions;
import com.mongodb.client.model.CreateViewOptions;
import com.mongodb.client.model.DeleteOptions;
import com.mongodb.client.model.DropCollectionOptions;
import com.mongodb.client.model.DropIndexOptions;
import com.mongodb.client.model.EstimatedDocumentCountOptions;
import com.mongodb.client.model.FindOneAndDeleteOptions;
import com.mongodb.client.model.FindOneAndReplaceOptions;
import com.mongodb.client.model.FindOneAndUpdateOptions;
import com.mongodb.client.model.IndexModel;
import com.mongodb.client.model.IndexOptionDefaults;
import com.mongodb.client.model.IndexOptions;
import com.mongodb.client.model.InsertManyOptions;
import com.mongodb.client.model.InsertOneOptions;
import com.mongodb.client.model.RenameCollectionOptions;
import com.mongodb.client.model.ReplaceOptions;
import com.mongodb.client.model.UpdateOptions;
import com.mongodb.client.model.ValidationOptions;
import com.mongodb.client.model.WriteModel;
import com.mongodb.client.result.DeleteResult;
import com.mongodb.client.result.InsertManyResult;
import com.mongodb.client.result.InsertOneResult;
import com.mongodb.client.result.UpdateResult;
import com.mongodb.internal.async.SingleResultCallback;
import com.mongodb.internal.bulk.WriteRequest;
import com.mongodb.internal.operation.AsyncReadOperation;
import com.mongodb.internal.operation.AsyncWriteOperation;
import com.mongodb.internal.operation.CommandReadOperation;
import com.mongodb.internal.operation.CreateCollectionOperation;
import com.mongodb.internal.operation.CreateViewOperation;
import com.mongodb.internal.operation.DropDatabaseOperation;
import com.mongodb.internal.operation.IndexHelper;
import com.mongodb.internal.operation.Operations;
import com.mongodb.lang.Nullable;
import com.mongodb.reactivestreams.client.ClientSession;
import org.bson.BsonDocument;
import org.bson.BsonValue;
import org.bson.UuidRepresentation;
import org.bson.codecs.configuration.CodecRegistry;
import org.bson.conversions.Bson;
import org.reactivestreams.Publisher;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import reactor.core.publisher.MonoSink;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import java.util.function.Function;
import java.util.function.Supplier;
import static com.mongodb.assertions.Assertions.notNull;
import static java.util.Collections.singletonList;
import static java.util.stream.Collectors.toList;
import static org.bson.codecs.configuration.CodecRegistries.withUuidRepresentation;
public final class MongoOperationPublisher {
private final Operations operations;
private final UuidRepresentation uuidRepresentation;
private final AutoEncryptionSettings autoEncryptionSettings;
private final OperationExecutor executor;
MongoOperationPublisher(
final Class documentClass, final CodecRegistry codecRegistry, final ReadPreference readPreference,
final ReadConcern readConcern, final WriteConcern writeConcern, final boolean retryWrites, final boolean retryReads,
final UuidRepresentation uuidRepresentation, @Nullable final AutoEncryptionSettings autoEncryptionSettings,
final OperationExecutor executor) {
this(new MongoNamespace("_ignored", "_ignored"), documentClass,
codecRegistry, readPreference, readConcern, writeConcern, retryWrites, retryReads,
uuidRepresentation, autoEncryptionSettings, executor);
}
MongoOperationPublisher(
final MongoNamespace namespace, final Class documentClass, final CodecRegistry codecRegistry,
final ReadPreference readPreference, final ReadConcern readConcern, final WriteConcern writeConcern,
final boolean retryWrites, final boolean retryReads, final UuidRepresentation uuidRepresentation,
@Nullable final AutoEncryptionSettings autoEncryptionSettings,
final OperationExecutor executor) {
this.operations = new Operations<>(namespace, notNull("documentClass", documentClass),
notNull("readPreference", readPreference), notNull("codecRegistry", codecRegistry),
notNull("readConcern", readConcern), notNull("writeConcern", writeConcern),
retryWrites, retryReads);
this.uuidRepresentation = notNull("uuidRepresentation", uuidRepresentation);
this.autoEncryptionSettings = autoEncryptionSettings;
this.executor = notNull("executor", executor);
}
MongoNamespace getNamespace() {
return operations.getNamespace();
}
ReadPreference getReadPreference() {
return operations.getReadPreference();
}
CodecRegistry getCodecRegistry() {
return operations.getCodecRegistry();
}
ReadConcern getReadConcern() {
return operations.getReadConcern();
}
WriteConcern getWriteConcern() {
return operations.getWriteConcern();
}
public boolean getRetryWrites() {
return operations.isRetryWrites();
}
public boolean getRetryReads() {
return operations.isRetryReads();
}
Class getDocumentClass() {
return operations.getDocumentClass();
}
public Operations getOperations() {
return operations;
}
MongoOperationPublisher withDatabase(final String name) {
return withDatabaseAndDocumentClass(name, getDocumentClass());
}
MongoOperationPublisher withDatabaseAndDocumentClass(final String name, final Class documentClass) {
return withNamespaceAndDocumentClass(new MongoNamespace(notNull("name", name), "ignored"),
notNull("documentClass", documentClass));
}
MongoOperationPublisher withNamespace(final MongoNamespace namespace) {
return withNamespaceAndDocumentClass(namespace, getDocumentClass());
}
MongoOperationPublisher withDocumentClass(final Class documentClass) {
return withNamespaceAndDocumentClass(getNamespace(), documentClass);
}
@SuppressWarnings("unchecked")
MongoOperationPublisher withNamespaceAndDocumentClass(final MongoNamespace namespace, final Class documentClass) {
if (getNamespace().equals(namespace) && getDocumentClass().equals(documentClass)) {
return (MongoOperationPublisher) this;
}
return new MongoOperationPublisher<>(notNull("namespace", namespace), notNull("documentClass", documentClass),
getCodecRegistry(), getReadPreference(), getReadConcern(), getWriteConcern(),
getRetryWrites(), getRetryReads(), uuidRepresentation, autoEncryptionSettings, executor);
}
MongoOperationPublisher withCodecRegistry(final CodecRegistry codecRegistry) {
return new MongoOperationPublisher<>(getNamespace(), getDocumentClass(),
withUuidRepresentation(notNull("codecRegistry", codecRegistry), uuidRepresentation),
getReadPreference(), getReadConcern(), getWriteConcern(), getRetryWrites(), getRetryReads(),
uuidRepresentation, autoEncryptionSettings, executor);
}
MongoOperationPublisher withReadPreference(final ReadPreference readPreference) {
if (getReadPreference().equals(readPreference)) {
return this;
}
return new MongoOperationPublisher<>(getNamespace(), getDocumentClass(), getCodecRegistry(),
notNull("readPreference", readPreference),
getReadConcern(), getWriteConcern(), getRetryWrites(), getRetryReads(),
uuidRepresentation, autoEncryptionSettings, executor);
}
MongoOperationPublisher withWriteConcern(final WriteConcern writeConcern) {
if (getWriteConcern().equals(writeConcern)) {
return this;
}
return new MongoOperationPublisher<>(getNamespace(), getDocumentClass(), getCodecRegistry(), getReadPreference(), getReadConcern(),
notNull("writeConcern", writeConcern),
getRetryWrites(), getRetryReads(), uuidRepresentation, autoEncryptionSettings, executor);
}
MongoOperationPublisher withReadConcern(final ReadConcern readConcern) {
if (getReadConcern().equals(readConcern)) {
return this;
}
return new MongoOperationPublisher<>(getNamespace(), getDocumentClass(),
getCodecRegistry(), getReadPreference(), notNull("readConcern", readConcern),
getWriteConcern(), getRetryWrites(), getRetryReads(), uuidRepresentation,
autoEncryptionSettings, executor);
}
Publisher dropDatabase(@Nullable final ClientSession clientSession) {
return createWriteOperationMono(() -> new DropDatabaseOperation(getNamespace().getDatabaseName(), getWriteConcern()),
clientSession);
}
Publisher createCollection(
@Nullable final ClientSession clientSession, final MongoNamespace namespace,
final CreateCollectionOptions options) {
return createWriteOperationMono(() -> {
CreateCollectionOperation operation =
new CreateCollectionOperation(namespace.getDatabaseName(), namespace.getCollectionName(), getWriteConcern())
.capped(options.isCapped())
.sizeInBytes(options.getSizeInBytes())
.maxDocuments(options.getMaxDocuments())
.storageEngineOptions(toBsonDocument(options.getStorageEngineOptions()))
.collation(options.getCollation())
.expireAfter(options.getExpireAfter(TimeUnit.SECONDS))
.timeSeriesOptions(options.getTimeSeriesOptions())
.changeStreamPreAndPostImagesOptions(options.getChangeStreamPreAndPostImagesOptions());
ClusteredIndexOptions clusteredIndexOptions = options.getClusteredIndexOptions();
if (clusteredIndexOptions != null) {
operation.clusteredIndexKey(toBsonDocument(clusteredIndexOptions.getKey()));
operation.clusteredIndexUnique(clusteredIndexOptions.isUnique());
operation.clusteredIndexName(clusteredIndexOptions.getName());
}
Bson encryptedFields = options.getEncryptedFields();
operation.encryptedFields(toBsonDocument(encryptedFields));
if (encryptedFields == null && autoEncryptionSettings != null) {
Map encryptedFieldsMap = autoEncryptionSettings.getEncryptedFieldsMap();
if (encryptedFieldsMap != null) {
operation.encryptedFields(encryptedFieldsMap.getOrDefault(namespace.getFullName(), null));
}
}
IndexOptionDefaults indexOptionDefaults = options.getIndexOptionDefaults();
Bson storageEngine = indexOptionDefaults.getStorageEngine();
if (storageEngine != null) {
operation.indexOptionDefaults(new BsonDocument("storageEngine", toBsonDocument(storageEngine)));
}
ValidationOptions validationOptions = options.getValidationOptions();
Bson validator = validationOptions.getValidator();
if (validator != null) {
operation.validator(toBsonDocument(validator));
}
if (validationOptions.getValidationLevel() != null) {
operation.validationLevel(validationOptions.getValidationLevel());
}
if (validationOptions.getValidationAction() != null) {
operation.validationAction(validationOptions.getValidationAction());
}
return operation;
}, clientSession);
}
Publisher createView(
@Nullable final ClientSession clientSession, final String viewName, final String viewOn,
final List pipeline, final CreateViewOptions options) {
List bsonDocumentPipeline = createBsonDocumentList(notNull("pipeline", pipeline));
return createWriteOperationMono(
() -> new CreateViewOperation(getNamespace().getDatabaseName(),
notNull("viewName", viewName),
notNull("viewOn", viewOn),
bsonDocumentPipeline,
getWriteConcern())
.collation(notNull("options", options).getCollation()),
clientSession);
}
public Publisher runCommand(
@Nullable final ClientSession clientSession, final Bson command,
final ReadPreference readPreference, final Class clazz) {
if (clientSession != null && clientSession.hasActiveTransaction() && !readPreference.equals(ReadPreference.primary())) {
return Mono.error(new MongoClientException("Read preference in a transaction must be primary"));
}
return createReadOperationMono(
() -> new CommandReadOperation<>(getNamespace().getDatabaseName(),
toBsonDocument(notNull("command", command)),
getCodecRegistry().get(notNull("clazz", clazz))),
clientSession, notNull("readPreference", readPreference));
}
Publisher estimatedDocumentCount(final EstimatedDocumentCountOptions options) {
return createReadOperationMono(() -> operations.estimatedDocumentCount(notNull("options", options)), null);
}
Publisher countDocuments(@Nullable final ClientSession clientSession, final Bson filter, final CountOptions options) {
return createReadOperationMono(() -> operations.countDocuments(notNull("filter", filter), notNull("options", options)
), clientSession);
}
Publisher bulkWrite(
@Nullable final ClientSession clientSession,
final List> requests, final BulkWriteOptions options) {
return createWriteOperationMono(() -> operations.bulkWrite(notNull("requests", requests), notNull("options", options)),
clientSession);
}
Publisher insertOne(@Nullable final ClientSession clientSession, final T document, final InsertOneOptions options) {
return createSingleWriteRequestMono(() -> operations.insertOne(notNull("document", document),
notNull("options", options)),
clientSession, WriteRequest.Type.INSERT)
.map(INSERT_ONE_RESULT_MAPPER);
}
Publisher insertMany(
@Nullable final ClientSession clientSession, final List documents,
final InsertManyOptions options) {
return createWriteOperationMono(() -> operations.insertMany(notNull("documents", documents), notNull("options", options)),
clientSession)
.map(INSERT_MANY_RESULT_MAPPER);
}
Publisher deleteOne(@Nullable final ClientSession clientSession, final Bson filter, final DeleteOptions options) {
return createSingleWriteRequestMono(() -> operations.deleteOne(notNull("filter", filter), notNull("options", options)),
clientSession, WriteRequest.Type.DELETE)
.map(DELETE_RESULT_MAPPER);
}
Publisher deleteMany(@Nullable final ClientSession clientSession, final Bson filter, final DeleteOptions options) {
return createSingleWriteRequestMono(() -> operations.deleteMany(notNull("filter", filter), notNull("options", options)),
clientSession, WriteRequest.Type.DELETE)
.map(DELETE_RESULT_MAPPER);
}
Publisher replaceOne(
@Nullable final ClientSession clientSession, final Bson filter, final T replacement,
final ReplaceOptions options) {
return createSingleWriteRequestMono(() -> operations.replaceOne(notNull("filter", filter),
notNull("replacement", replacement),
notNull("options", options)),
clientSession, WriteRequest.Type.REPLACE)
.map(UPDATE_RESULT_MAPPER);
}
Publisher updateOne(
@Nullable final ClientSession clientSession, final Bson filter, final Bson update,
final UpdateOptions options) {
return createSingleWriteRequestMono(() -> operations.updateOne(notNull("filter", filter),
notNull("update", update),
notNull("options", options)),
clientSession, WriteRequest.Type.UPDATE)
.map(UPDATE_RESULT_MAPPER);
}
Publisher updateOne(
@Nullable final ClientSession clientSession, final Bson filter, final List update,
final UpdateOptions options) {
return createSingleWriteRequestMono(() -> operations.updateOne(notNull("filter", filter),
notNull("update", update),
notNull("options", options)),
clientSession, WriteRequest.Type.UPDATE)
.map(UPDATE_RESULT_MAPPER);
}
Publisher updateMany(
@Nullable final ClientSession clientSession, final Bson filter, final Bson update,
final UpdateOptions options) {
return createSingleWriteRequestMono(() -> operations.updateMany(notNull("filter", filter),
notNull("update", update),
notNull("options", options)),
clientSession, WriteRequest.Type.UPDATE)
.map(UPDATE_RESULT_MAPPER);
}
Publisher updateMany(
@Nullable final ClientSession clientSession, final Bson filter, final List update,
final UpdateOptions options) {
return createSingleWriteRequestMono(() -> operations.updateMany(notNull("filter", filter),
notNull("update", update),
notNull("options", options)),
clientSession, WriteRequest.Type.UPDATE)
.map(UPDATE_RESULT_MAPPER);
}
Publisher findOneAndDelete(@Nullable final ClientSession clientSession, final Bson filter, final FindOneAndDeleteOptions options) {
return createWriteOperationMono(() -> operations.findOneAndDelete(notNull("filter", filter),
notNull("options", options)),
clientSession);
}
Publisher findOneAndReplace(
@Nullable final ClientSession clientSession, final Bson filter, final T replacement,
final FindOneAndReplaceOptions options) {
return createWriteOperationMono(() -> operations.findOneAndReplace(notNull("filter", filter),
notNull("replacement", replacement),
notNull("options", options)),
clientSession);
}
Publisher findOneAndUpdate(
@Nullable final ClientSession clientSession, final Bson filter, final Bson update,
final FindOneAndUpdateOptions options) {
return createWriteOperationMono(() -> operations.findOneAndUpdate(notNull("filter", filter),
notNull("update", update),
notNull("options", options)),
clientSession);
}
Publisher findOneAndUpdate(
@Nullable final ClientSession clientSession, final Bson filter,
final List update, final FindOneAndUpdateOptions options) {
return createWriteOperationMono(() -> operations.findOneAndUpdate(notNull("filter", filter),
notNull("update", update),
notNull("options", options)),
clientSession);
}
Publisher dropCollection(@Nullable final ClientSession clientSession, final DropCollectionOptions dropCollectionOptions) {
return createWriteOperationMono(() -> operations.dropCollection(dropCollectionOptions, autoEncryptionSettings), clientSession);
}
Publisher createIndex(@Nullable final ClientSession clientSession, final Bson key, final IndexOptions options) {
return createIndexes(clientSession, singletonList(new IndexModel(notNull("key", key), options)), new CreateIndexOptions());
}
Publisher createIndexes(
@Nullable final ClientSession clientSession, final List indexes,
final CreateIndexOptions options) {
return createWriteOperationMono(() -> operations.createIndexes(notNull("indexes", indexes),
notNull("options", options)), clientSession)
.thenMany(Flux.fromIterable(IndexHelper.getIndexNames(indexes, getCodecRegistry())));
}
Publisher dropIndex(@Nullable final ClientSession clientSession, final String indexName, final DropIndexOptions options) {
return createWriteOperationMono(() -> operations.dropIndex(notNull("indexName", indexName), notNull("options", options)),
clientSession);
}
Publisher dropIndex(@Nullable final ClientSession clientSession, final Bson keys, final DropIndexOptions options) {
return createWriteOperationMono(() -> operations.dropIndex(notNull("keys", keys), notNull("options", options)),
clientSession);
}
Publisher dropIndexes(@Nullable final ClientSession clientSession, final DropIndexOptions options) {
return dropIndex(clientSession, "*", options);
}
Publisher renameCollection(
@Nullable final ClientSession clientSession, final MongoNamespace newCollectionNamespace,
final RenameCollectionOptions options) {
return createWriteOperationMono(() -> operations.renameCollection(notNull("newCollectionNamespace", newCollectionNamespace),
notNull("options", options)),
clientSession);
}
Mono createReadOperationMono(
final Supplier> operation,
@Nullable final ClientSession clientSession) {
return createReadOperationMono(operation, clientSession, getReadPreference());
}
Mono createReadOperationMono(
final Supplier> operation,
@Nullable final ClientSession clientSession,
final ReadPreference readPreference) {
AsyncReadOperation readOperation = operation.get();
return executor.execute(readOperation, readPreference, getReadConcern(), clientSession);
}
Mono createWriteOperationMono(final Supplier> operation, @Nullable final ClientSession clientSession) {
AsyncWriteOperation writeOperation = operation.get();
return executor.execute(writeOperation, getReadConcern(), clientSession);
}
private Mono createSingleWriteRequestMono(
final Supplier> operation,
@Nullable final ClientSession clientSession,
final WriteRequest.Type type) {
return createWriteOperationMono(operation, clientSession)
.onErrorMap(MongoBulkWriteException.class, e -> {
MongoException exception;
WriteConcernError writeConcernError = e.getWriteConcernError();
if (e.getWriteErrors().isEmpty() && writeConcernError != null) {
WriteConcernResult writeConcernResult;
if (type == WriteRequest.Type.INSERT) {
writeConcernResult = WriteConcernResult.acknowledged(e.getWriteResult().getInsertedCount(), false, null);
} else if (type == WriteRequest.Type.DELETE) {
writeConcernResult = WriteConcernResult.acknowledged(e.getWriteResult().getDeletedCount(), false, null);
} else {
writeConcernResult = WriteConcernResult
.acknowledged(e.getWriteResult().getMatchedCount() + e.getWriteResult().getUpserts().size(),
e.getWriteResult().getMatchedCount() > 0,
e.getWriteResult().getUpserts().isEmpty()
? null : e.getWriteResult().getUpserts().get(0).getId());
}
exception = new MongoWriteConcernException(writeConcernError, writeConcernResult, e.getServerAddress());
} else if (!e.getWriteErrors().isEmpty()) {
exception = new MongoWriteException(new WriteError(e.getWriteErrors().get(0)), e.getServerAddress());
} else {
exception = new MongoWriteException(new WriteError(-1, "Unknown write error", new BsonDocument()),
e.getServerAddress());
}
for (final String errorLabel : e.getErrorLabels()) {
exception.addLabel(errorLabel);
}
return exception;
});
}
private static final Function INSERT_ONE_RESULT_MAPPER = result -> {
if (result.wasAcknowledged()) {
BsonValue insertedId = result.getInserts().isEmpty() ? null : result.getInserts().get(0).getId();
return InsertOneResult.acknowledged(insertedId);
} else {
return InsertOneResult.unacknowledged();
}
};
private static final Function INSERT_MANY_RESULT_MAPPER = result -> {
if (result.wasAcknowledged()) {
return InsertManyResult.acknowledged(result.getInserts().stream()
.collect(HashMap::new, (m, v) -> m.put(v.getIndex(), v.getId()), HashMap::putAll));
} else {
return InsertManyResult.unacknowledged();
}
};
private static final Function DELETE_RESULT_MAPPER = result -> {
if (result.wasAcknowledged()) {
return DeleteResult.acknowledged(result.getDeletedCount());
} else {
return DeleteResult.unacknowledged();
}
};
private static final Function UPDATE_RESULT_MAPPER = result -> {
if (result.wasAcknowledged()) {
BsonValue upsertedId = result.getUpserts().isEmpty() ? null : result.getUpserts().get(0).getId();
return UpdateResult.acknowledged(result.getMatchedCount(), (long) result.getModifiedCount(), upsertedId);
} else {
return UpdateResult.unacknowledged();
}
};
private List createBsonDocumentList(final List pipeline) {
if (pipeline.contains(null)) {
throw new IllegalArgumentException("pipeline can not contain a null value");
}
return pipeline.stream().map(this::toBsonDocument).collect(toList());
}
public static SingleResultCallback sinkToCallback(final MonoSink sink) {
return (result, t) -> {
if (t != null) {
sink.error(t);
} else if (result == null) {
sink.success();
} else {
sink.success(result);
}
};
}
@Nullable
private BsonDocument toBsonDocument(@Nullable final Bson document) {
return document == null ? null : document.toBsonDocument(BsonDocument.class, getCodecRegistry());
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy