Examples with BsonDocument org.bson.BsonDocument used on opensource projects

Example 81 with BsonDocument

use of org.bson.BsonDocument in project mongo-java-driver by mongodb.

the class DBCollection method distinct.

/**
     * Find the distinct values for a specified field across a collection and returns the results in an array.
     *
     * @param fieldName  Specifies the field for which to return the distinct values
     * @param options    the options to apply for this operation
     * @return A {@code List} of the distinct values
     * @mongodb.driver.manual reference/command/distinct Distinct Command
     * @since 3.4
     */
@SuppressWarnings("unchecked")
public List distinct(final String fieldName, final DBCollectionDistinctOptions options) {
    notNull("fieldName", fieldName);
    return new OperationIterable<BsonValue>(new DistinctOperation<BsonValue>(getNamespace(), fieldName, new BsonValueCodec()).readConcern(options.getReadConcern() != null ? options.getReadConcern() : getReadConcern()).filter(wrapAllowNull(options.getFilter())).collation(options.getCollation()), options.getReadPreference() != null ? options.getReadPreference() : getReadPreference(), executor).map(new Function<BsonValue, Object>() {

        @Override
        public Object apply(final BsonValue bsonValue) {
            if (bsonValue == null) {
                return null;
            }
            BsonDocument document = new BsonDocument("value", bsonValue);
            DBObject obj = getDefaultDBObjectCodec().decode(new BsonDocumentReader(document), DecoderContext.builder().build());
            return obj.get("value");
        }
    }).into(new ArrayList<Object>());
}

Example 82 with BsonDocument

use of org.bson.BsonDocument in project mongo-java-driver by mongodb.

the class WriteConcernConnectionStringTest method getExpectedWriteConcern.

private WriteConcern getExpectedWriteConcern() {
    BsonDocument writeConcernDocument = definition.getDocument("writeConcern");
    BsonValue wValue = writeConcernDocument.get("w");
    WriteConcern retVal;
    if (wValue == null) {
        retVal = WriteConcern.ACKNOWLEDGED;
    } else if (wValue instanceof BsonNumber) {
        retVal = new WriteConcern(wValue.asNumber().intValue());
    } else if (wValue instanceof BsonString) {
        retVal = new WriteConcern(wValue.asString().getValue());
    } else {
        throw new IllegalArgumentException("Unexpected w value: " + wValue);
    }
    if (writeConcernDocument.containsKey("wtimeoutMS")) {
        retVal = retVal.withWTimeout(writeConcernDocument.getNumber("wtimeoutMS", new BsonInt32(0)).intValue(), TimeUnit.MILLISECONDS);
    }
    if (writeConcernDocument.containsKey("journal")) {
        retVal = retVal.withJournal(writeConcernDocument.getBoolean("journal", BsonBoolean.FALSE).getValue());
    }
    return retVal;
}

Example 83 with BsonDocument

use of org.bson.BsonDocument in project mongo-java-driver by mongodb.

the class AbstractServerDiscoveryAndMonitoringTest method applyResponse.

protected void applyResponse(final BsonArray response) {
    ServerAddress serverAddress = new ServerAddress(response.get(0).asString().getValue());
    BsonDocument isMasterResult = response.get(1).asDocument();
    ServerDescription serverDescription;
    if (isMasterResult.isEmpty()) {
        serverDescription = ServerDescription.builder().type(ServerType.UNKNOWN).state(CONNECTING).address(serverAddress).build();
    } else {
        serverDescription = createServerDescription(serverAddress, isMasterResult, getVersion(new BsonDocument("versionArray", new BsonArray(asList(new BsonInt32(2), new BsonInt32(6), new BsonInt32(0))))), 5000000);
    }
    factory.sendNotification(serverAddress, serverDescription);
}

Example 84 with BsonDocument

use of org.bson.BsonDocument in project mongo-java-driver by mongodb.

the class MaxDocumentSizeTest method setUp.

@Before
public void setUp() {
    message = new InsertMessage("test.test", true, ACKNOWLEDGED, asList(new InsertRequest(new BsonDocument("bytes", new BsonBinary(new byte[2048])))), MessageSettings.builder().maxDocumentSize(1024).build());
    buffer = new ByteBufferBsonOutput(new SimpleBufferProvider());
}

Example 85 with BsonDocument

use of org.bson.BsonDocument in project mongo-java-driver by mongodb.

the class MaxMessageSizeTest method setUp.

@Before
@SuppressWarnings("unchecked")
public void setUp() {
    BsonBinary binary = new BsonBinary(new byte[2048]);
    message = new InsertMessage("test.test", true, ACKNOWLEDGED, Arrays.asList(new InsertRequest(new BsonDocument("bytes", binary)), new InsertRequest(new BsonDocument("bytes", binary)), new InsertRequest(new BsonDocument("bytes", binary))), MessageSettings.builder().maxMessageSize(4500).build());
    buffer = new ByteBufferBsonOutput(new SimpleBufferProvider());
}