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

Example 1 with UpdateFieldNameValidator

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

the class UpdateMessage method encodeMessageBodyWithMetadata.

@Override
protected EncodingMetadata encodeMessageBodyWithMetadata(final BsonOutput bsonOutput, final int messageStartPosition) {
    // reserved
    bsonOutput.writeInt32(0);
    bsonOutput.writeCString(getCollectionName());
    UpdateRequest updateRequest = updates.get(0);
    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) {
        addCollectibleDocument(updateRequest.getUpdate(), bsonOutput, new CollectibleDocumentFieldNameValidator());
    } else {
        int bufferPosition = bsonOutput.getPosition();
        addDocument(updateRequest.getUpdate(), bsonOutput, new UpdateFieldNameValidator());
        if (bsonOutput.getPosition() == bufferPosition + 5) {
            throw new IllegalArgumentException("Invalid BSON document for an update");
        }
    }
    if (updates.size() == 1) {
        return new EncodingMetadata(null, firstDocumentStartPosition);
    } else {
        return new EncodingMetadata(new UpdateMessage(getCollectionName(), updates.subList(1, updates.size()), getSettings()), firstDocumentStartPosition);
    }
}

Example 2 with UpdateFieldNameValidator

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

the class FindAndUpdateOperation method getValidator.

private FieldNameValidator getValidator() {
    Map<String, FieldNameValidator> map = new HashMap<String, FieldNameValidator>();
    map.put("update", new UpdateFieldNameValidator());
    return new MappedFieldNameValidator(new NoOpFieldNameValidator(), map);
}