Add CollectionCreateOptions.distributeShardsLike(String) (Issue #170)

Author: mpv1989

ChangeLog

@@ -1,5 +1,6 @@
 v4.3.3 (2018-xx-xx)
 ---------------------------
+* added CollectionCreateOptions.distributeShardsLike(String) (Issue #170)
 * fixed inconsistency of getDocument() variants (Issue #168)
 * added AqlQueryOptions.memoryLimit(Long)
 * added AqlQueryOptions.failOnWarning(Boolean)

src/main/java/com/arangodb/model/CollectionCreateOptions.java

@@ -44,6 +44,7 @@ public class CollectionCreateOptions {
 	private Boolean isSystem;
 	private CollectionType type;
 	private Integer indexBuckets;
+	private String distributeShardsLike;
 
 	public CollectionCreateOptions() {
 		super();
@@ -257,4 +258,22 @@ public CollectionCreateOptions indexBuckets(final Integer indexBuckets) {
 		return this;
 	}
 
+	public String getDistributeShardsLike() {
+		return distributeShardsLike;
+	}
+
+	/**
+	 * @param distributeShardsLike
+	 *            (The default is ""): in an enterprise cluster, this attribute binds the specifics of sharding for the
+	 *            newly created collection to follow that of a specified existing collection. Note: Using this parameter
+	 *            has consequences for the prototype collection. It can no longer be dropped, before sharding imitating
+	 *            collections are dropped. Equally, backups and restores of imitating collections alone will generate
+	 *            warnings, which can be overridden, about missing sharding prototype.
+	 * @return options
+	 */
+	public CollectionCreateOptions distributeShardsLike(final String distributeShardsLike) {
+		this.distributeShardsLike = distributeShardsLike;
+		return this;
+	}
+
 }

src/test/java/com/arangodb/ArangoDatabaseTest.java

@@ -52,6 +52,7 @@
 import com.arangodb.entity.AqlParseEntity;
 import com.arangodb.entity.AqlParseEntity.AstNode;
 import com.arangodb.entity.ArangoDBVersion;
+import com.arangodb.entity.ArangoDBVersion.License;
 import com.arangodb.entity.BaseDocument;
 import com.arangodb.entity.BaseEdgeDocument;
 import com.arangodb.entity.CollectionEntity;
@@ -197,6 +198,18 @@ public void createCollectionWithNumberOfShardsAndShardKeys() {
 		}
 	}
 
+	@Test
+	public void createCollectionWithDistributeShardsLike() {
+		if (arangoDB.getVersion().getLicense() == License.ENTERPRISE && arangoDB.getRole() != ServerRole.SINGLE) {
+			final Integer numberOfShards = 3;
+			db.createCollection(COLLECTION_NAME, new CollectionCreateOptions().numberOfShards(numberOfShards));
+			db.createCollection(COLLECTION_NAME + "2",
+				new CollectionCreateOptions().distributeShardsLike(COLLECTION_NAME));
+			assertThat(db.collection(COLLECTION_NAME).getProperties().getNumberOfShards(), is(numberOfShards));
+			assertThat(db.collection(COLLECTION_NAME + "2").getProperties().getNumberOfShards(), is(numberOfShards));
+		}
+	}
+
 	@Test
 	public void deleteCollection() {
 		db.createCollection(COLLECTION_NAME, null);