Allocating BsonArray using capacity when known

JAVA-3613

Signed-off-by: Gaspard Petit <[email protected]>

Author: Gaspard Petit

bson/src/main/org/bson/BsonArray.java

@@ -243,7 +243,7 @@ public String toString() {
 
     @Override
     public BsonArray clone() {
-        BsonArray to = new BsonArray();
+        BsonArray to = new BsonArray(this.size());
         for (BsonValue cur : this) {
             switch (cur.getBsonType()) {
                 case DOCUMENT:

driver-core/src/main/com/mongodb/TaggableReadPreference.java

@@ -545,7 +545,7 @@ protected List<ServerDescription> chooseForReplicaSet(final ClusterDescription c
     }
 
     private BsonArray tagsListToBsonArray() {
-        BsonArray bsonArray = new BsonArray();
+        BsonArray bsonArray = new BsonArray(tagSetList.size());
         for (TagSet tagSet : tagSetList) {
             bsonArray.add(toDocument(tagSet));
         }

driver-core/src/main/com/mongodb/client/model/Filters.java

@@ -648,7 +648,7 @@ public static Bson geoWithinBox(final String fieldName, final double lowerLeftX,
      * @since 3.1
      */
     public static Bson geoWithinPolygon(final String fieldName, final List<List<Double>> points) {
-        BsonArray pointsArray = new BsonArray();
+        BsonArray pointsArray = new BsonArray(points.size());
         for (List<Double> point : points) {
             pointsArray.add(new BsonArray(asList(new BsonDouble(point.get(0)), new BsonDouble(point.get(1)))));
         }
@@ -1231,7 +1231,7 @@ public <TDocument> BsonDocument toBsonDocument(final Class<TDocument> documentCl
                 Map.Entry<String, BsonValue> entry = filterDocument.entrySet().iterator().next();
                 return createFilter(entry.getKey(), entry.getValue());
             } else {
-                BsonArray values = new BsonArray();
+                BsonArray values = new BsonArray(filterDocument.size());
                 for (Map.Entry<String, BsonValue> docs : filterDocument.entrySet()) {
                     values.add(new BsonDocument(docs.getKey(), docs.getValue()));
                 }

driver-core/src/main/com/mongodb/internal/connection/InternalStreamConnectionInitializer.java

@@ -162,7 +162,7 @@ private BsonDocument createIsMasterCommand(final Authenticator authenticator, fi
             isMasterCommandDocument.append("client", clientMetadataDocument);
         }
         if (!requestedCompressors.isEmpty()) {
-            BsonArray compressors = new BsonArray();
+            BsonArray compressors = new BsonArray(this.requestedCompressors.size());
             for (MongoCompressor cur : this.requestedCompressors) {
                 compressors.add(new BsonString(cur.getName()));
             }

driver-core/src/main/com/mongodb/internal/connection/KillCursorProtocol.java

@@ -140,7 +140,7 @@ public void setCommandListener(final CommandListener commandListener) {
     }
 
     private BsonDocument asCommandDocument() {
-        BsonArray array = new BsonArray();
+        BsonArray array = new BsonArray(cursors.size());
         for (long cursor : cursors) {
             array.add(new BsonInt64(cursor));
         }
@@ -149,7 +149,7 @@ private BsonDocument asCommandDocument() {
     }
 
     private BsonDocument asCommandResponseDocument() {
-        BsonArray cursorIdArray = new BsonArray();
+        BsonArray cursorIdArray = new BsonArray(cursors.size());
         for (long cursorId : cursors) {
             cursorIdArray.add(new BsonInt64(cursorId));
         }