Add AqlQueryOptions.memoryLimit(Long)

Author: mpv1989

ChangeLog

@@ -1,3 +1,7 @@
+v4.3.3 (2018-xx-xx)
+---------------------------
+* added AqlQueryOptions.memoryLimit(Long)
+
 v4.3.2 (2017-11-30)
 ---------------------------
 * fixed redirect header (uppercase)

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

@@ -35,6 +35,7 @@ public class AqlQueryOptions {
 	private Integer ttl;
 	private Integer batchSize;
 	private Boolean cache;
+	private Long memoryLimit;
 	private Map<String, Object> bindVars;
 	private String query;
 	private Options options;
@@ -92,6 +93,22 @@ public AqlQueryOptions batchSize(final Integer batchSize) {
 		return this;
 	}
 
+	public Long getMemoryLimit() {
+		return memoryLimit;
+	}
+
+	/**
+	 * @param memoryLimit
+	 *            the maximum number of memory (measured in bytes) that the query is allowed to use. If set, then the
+	 *            query will fail with error "resource limit exceeded" in case it allocates too much memory. A value of
+	 *            0 indicates that there is no memory limit.
+	 * @return options
+	 */
+	public AqlQueryOptions memoryLimit(final Long memoryLimit) {
+		this.memoryLimit = memoryLimit;
+		return this;
+	}
+
 	public Boolean getCache() {
 		return cache;
 	}

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

@@ -632,6 +632,16 @@ public void queryWithCache() throws InterruptedException {
 		}
 	}
 
+	@Test
+	public void queryWithMemoryLimit() {
+		try {
+			db.query("RETURN 'bla'", null, new AqlQueryOptions().memoryLimit(1L), String.class);
+			fail();
+		} catch (final ArangoDBException e) {
+			assertThat(e.getErrorNum(), is(32));
+		}
+	}
+
 	@Test
 	public void queryCursor() {
 		try {