added cache performance test

Author: Mark

src/test/java/com/arangodb/ArangoDriverCacheTest.java

@@ -0,0 +1,133 @@
+package com.arangodb;
+
+import java.util.Map;
+
+import org.junit.AfterClass;
+import org.junit.Assert;
+import org.junit.BeforeClass;
+import org.junit.Ignore;
+import org.junit.Test;
+
+import com.arangodb.entity.QueryCachePropertiesEntity;
+import com.arangodb.entity.QueryCachePropertiesEntity.CacheMode;
+import com.arangodb.util.AqlQueryOptions;
+import com.arangodb.util.MapBuilder;
+
+/**
+ * @author Mark - mark@arangodb.com
+ *
+ */
+@Ignore
+public class ArangoDriverCacheTest {
+
+	private static final String COLLECTION_NAME = "unitTestCollection";
+	private static final String DATABASE_NAME = "unitTestDatabase";
+	private static ArangoConfigure configure;
+	private static ArangoDriver driver;
+
+	@BeforeClass
+	public static void setup() throws ArangoException {
+
+		configure = new ArangoConfigure();
+		configure.init();
+		driver = new ArangoDriver(configure);
+
+		// create test database
+		try {
+			driver.createDatabase(DATABASE_NAME);
+		} catch (ArangoException e) {
+		}
+		driver.setDefaultDatabase(DATABASE_NAME);
+
+		// create test collection
+		try {
+			driver.createCollection(COLLECTION_NAME);
+		} catch (final ArangoException e) {
+		}
+		driver.truncateCollection(COLLECTION_NAME);
+
+		// create some test data
+		for (int i = 0; i < 1000000; i++) {
+			final TestEntity value = new TestEntity("user_" + (i % 10), "desc" + (i % 10), i);
+			driver.createDocument(COLLECTION_NAME, value, null);
+		}
+
+	}
+
+	@AfterClass
+	public static void shutdown() {
+		try {
+			driver.deleteDatabase(DATABASE_NAME);
+		} catch (final ArangoException e) {
+		}
+		configure.shutdown();
+	}
+
+	private AqlQueryOptions createAqlQueryOptions(
+		final Boolean count,
+		final Integer batchSize,
+		final Boolean fullCount,
+		final Boolean cache) {
+		return new AqlQueryOptions().setCount(count).setBatchSize(batchSize).setFullCount(fullCount).setCache(cache);
+	}
+
+	@Test
+	public void test_withoutCache() throws ArangoException {
+		// set cache mode off
+		final QueryCachePropertiesEntity properties = new QueryCachePropertiesEntity();
+		properties.setMode(CacheMode.off);
+		driver.setQueryCacheProperties(properties);
+
+		final AqlQueryOptions aqlQueryOptions = createAqlQueryOptions(true, 1000, null, false);
+
+		exceuteQuery(aqlQueryOptions);
+	}
+
+	@Test
+	public void test_withCache() throws ArangoException {
+		// set cache mode on
+		final QueryCachePropertiesEntity properties = new QueryCachePropertiesEntity();
+		properties.setMode(CacheMode.on);
+		driver.setQueryCacheProperties(properties);
+
+		// set caching to true for the query
+		final AqlQueryOptions aqlQueryOptions = createAqlQueryOptions(true, 1000, null, true);
+
+		exceuteQuery(aqlQueryOptions);
+	}
+
+	private void exceuteQuery(AqlQueryOptions aqlQueryOptions) throws ArangoException {
+
+		final String query = "FOR t IN " + COLLECTION_NAME + " FILTER t.age >= @age SORT t.age RETURN t";
+		final Map<String, Object> bindVars = new MapBuilder().put("age", 90).get();
+
+		DocumentCursor<TestEntity> rs = driver.executeDocumentQuery(query, bindVars, aqlQueryOptions, TestEntity.class);
+		// first time, the query isn't cached
+		Assert.assertEquals(false, rs.isCached());
+
+		final long start = System.currentTimeMillis();
+
+		// query the cached value
+		rs = driver.executeDocumentQuery(query, bindVars, aqlQueryOptions, TestEntity.class);
+		Assert.assertEquals(aqlQueryOptions.getCache(), rs.isCached());
+
+		// load all results
+		rs.asEntityList();
+
+		final long time = System.currentTimeMillis() - start;
+		System.out.println(time);
+	}
+
+	private static class TestEntity {
+		private String user;
+		private String desc;
+		private Integer age;
+
+		public TestEntity(String user, String desc, Integer age) {
+			super();
+			this.user = user;
+			this.desc = desc;
+			this.age = age;
+		}
+	}
+}