Examples with ReplacingDocumentFieldNameValidator com.mongodb.internal.validator.ReplacingDocumentFieldNameValidator used on opensource projects

Example 1 with ReplacingDocumentFieldNameValidator

use of com.mongodb.internal.validator.ReplacingDocumentFieldNameValidator in project mongo-java-driver by mongodb.

the class UpdateMessage method encodeMessageBodyWithMetadata.

@Override
protected EncodingMetadata encodeMessageBodyWithMetadata(final BsonOutput bsonOutput) {
    // reserved
    bsonOutput.writeInt32(0);
    bsonOutput.writeCString(getCollectionName());
    int flags = 0;
    if (updateRequest.isUpsert()) {
        flags |= 1;
    }
    if (updateRequest.isMulti()) {
        flags |= 2;
    }
    bsonOutput.writeInt32(flags);
    int firstDocumentStartPosition = bsonOutput.getPosition();
    addDocument(updateRequest.getFilter(), bsonOutput, new NoOpFieldNameValidator());
    if (updateRequest.getType() == REPLACE && updateRequest.getUpdateValue().isDocument()) {
        addDocument(updateRequest.getUpdateValue().asDocument(), bsonOutput, new ReplacingDocumentFieldNameValidator());
    } else {
        int bufferPosition = bsonOutput.getPosition();
        BsonValue update = updateRequest.getUpdateValue();
        if (update.isDocument()) {
            addDocument(update.asDocument(), bsonOutput, new UpdateFieldNameValidator());
        } else {
            throw new IllegalArgumentException("Invalid update filter in update request. The filter must be a document.");
        }
        if (bsonOutput.getPosition() == bufferPosition + 5) {
            throw new IllegalArgumentException("Invalid BSON document for an update");
        }
    }
    return new EncodingMetadata(firstDocumentStartPosition);
}

Example 2 with ReplacingDocumentFieldNameValidator

use of com.mongodb.internal.validator.ReplacingDocumentFieldNameValidator in project mongo-java-driver by mongodb.

the class FindAndReplaceOperation method getFieldNameValidator.

@Override
protected FieldNameValidator getFieldNameValidator() {
    Map<String, FieldNameValidator> map = new HashMap<String, FieldNameValidator>();
    map.put("update", new ReplacingDocumentFieldNameValidator());
    return new MappedFieldNameValidator(new NoOpFieldNameValidator(), map);
}