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

Example 31 with BsonDocumentWriter

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

the class BsonHelper method toBson.

@SuppressWarnings("unchecked")
static BsonDocument toBson(final Bson bson) {
    BsonDocumentWriter writer = new BsonDocumentWriter(new BsonDocument());
    ((Codec<Bson>) CODEC_REGISTRY.get(bson.getClass())).encode(writer, bson, EncoderContext.builder().build());
    return writer.getDocument();
}

Example 32 with BsonDocumentWriter

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

the class Updates method pullByFilter.

/**
 * Creates an update that removes from an array all elements that match the given filter.
 *
 * @param filter the query filter
 * @return the update
 * @mongodb.driver.manual reference/operator/update/pull/ $pull
 */
public static Bson pullByFilter(final Bson filter) {
    return new Bson() {

        @Override
        public <TDocument> BsonDocument toBsonDocument(final Class<TDocument> tDocumentClass, final CodecRegistry codecRegistry) {
            BsonDocumentWriter writer = new BsonDocumentWriter(new BsonDocument());
            writer.writeStartDocument();
            writer.writeName("$pull");
            encodeValue(writer, filter, codecRegistry);
            writer.writeEndDocument();
            return writer.getDocument();
        }
    };
}

Example 33 with BsonDocumentWriter

use of org.bson.BsonDocumentWriter in project drill by apache.

the class TestBsonRecordReader method testRecursiveDocuments.

@Test
public void testRecursiveDocuments() throws IOException {
    BsonDocument topDoc = new BsonDocument();
    final int count = 3;
    for (int i = 0; i < count; ++i) {
        BsonDocument bsonDoc = new BsonDocument();
        BsonWriter bw = new BsonDocumentWriter(bsonDoc);
        bw.writeStartDocument();
        bw.writeName("k1" + i);
        bw.writeString("drillMongo1" + i);
        bw.writeName("k2" + i);
        bw.writeString("drillMongo2" + i);
        bw.writeEndDocument();
        bw.flush();
        topDoc.append("doc" + i, bsonDoc);
    }
    writer.reset();
    bsonReader.write(writer, new BsonDocumentReader(topDoc));
    SingleMapReaderImpl mapReader = (SingleMapReaderImpl) writer.getMapVector().getReader();
    for (int i = 0; i < count; ++i) {
        SingleMapReaderImpl reader = (SingleMapReaderImpl) mapReader.reader("doc" + i);
        assertEquals("drillMongo1" + i, reader.reader("k1" + i).readText().toString());
        assertEquals("drillMongo2" + i, reader.reader("k2" + i).readText().toString());
    }
}

Example 34 with BsonDocumentWriter

use of org.bson.BsonDocumentWriter in project drill by apache.

the class TestBsonRecordReader method testArrayType.

@Test
public void testArrayType() throws IOException {
    BsonDocument bsonDoc = new BsonDocument();
    BsonWriter bw = new BsonDocumentWriter(bsonDoc);
    bw.writeStartDocument();
    bw.writeName("arrayKey");
    bw.writeStartArray();
    bw.writeInt32(1);
    bw.writeInt32(2);
    bw.writeInt32(3);
    bw.writeEndArray();
    bw.writeEndDocument();
    bw.flush();
    bsonReader.write(writer, new BsonDocumentReader(bsonDoc));
    SingleMapReaderImpl mapReader = (SingleMapReaderImpl) writer.getMapVector().getReader();
    FieldReader reader = mapReader.reader("arrayKey");
    assertEquals(3, reader.size());
}