#80 use options with transaction id where ever possible

Author: aburmeis

src/main/java/com/arangodb/springframework/repository/SimpleArangoRepository.java

@@ -22,6 +22,9 @@
 
 import com.arangodb.ArangoCursor;
 import com.arangodb.model.AqlQueryOptions;
+import com.arangodb.model.DocumentDeleteOptions;
+import com.arangodb.model.DocumentExistsOptions;
+import com.arangodb.model.DocumentReadOptions;
 import com.arangodb.springframework.core.ArangoOperations;
 import com.arangodb.springframework.core.mapping.ArangoMappingContext;
 import com.arangodb.springframework.core.util.AqlUtils;
@@ -77,7 +80,7 @@ public SimpleArangoRepository(final ArangoOperations arangoOperations, final Cla
 	 */
 	@Override
 	public <S extends T> S save(final S entity) {
-		arangoOperations.repsert(entity);
+		arangoOperations.repsert(entity, defaultQueryOptions());
 		return entity;
 	}
 
@@ -90,7 +93,7 @@ public <S extends T> S save(final S entity) {
 	 */
 	@Override
 	public <S extends T> Iterable<S> saveAll(final Iterable<S> entities) {
-		arangoOperations.repsert(entities, domainClass);
+		arangoOperations.repsert(entities, domainClass, defaultQueryOptions());
 		return entities;
 	}
 
@@ -102,7 +105,7 @@ public <S extends T> Iterable<S> saveAll(final Iterable<S> entities) {
 	 */
 	@Override
 	public Optional<T> findById(final ID id) {
-		return arangoOperations.find(id, domainClass);
+		return arangoOperations.find(id, domainClass, defaultReadOptions());
 	}
 
 	/**
@@ -113,7 +116,7 @@ public Optional<T> findById(final ID id) {
 	 */
 	@Override
 	public boolean existsById(final ID id) {
-		return arangoOperations.exists(id, domainClass);
+		return arangoOperations.exists(id, domainClass, defaultExistsOptions());
 	}
 
 	/**
@@ -123,7 +126,7 @@ public boolean existsById(final ID id) {
 	 */
 	@Override
 	public Iterable<T> findAll() {
-		return arangoOperations.findAll(domainClass);
+		return arangoOperations.findAll(domainClass, defaultReadOptions());
 	}
 
 	/**
@@ -135,7 +138,7 @@ public Iterable<T> findAll() {
 	 */
 	@Override
 	public Iterable<T> findAllById(final Iterable<ID> ids) {
-		return arangoOperations.find(ids, domainClass);
+		return arangoOperations.find(ids, domainClass, defaultReadOptions());
 	}
 
 	/**
@@ -156,7 +159,7 @@ public long count() {
 	 */
 	@Override
 	public void deleteById(final ID id) {
-		arangoOperations.delete(id, domainClass);
+		arangoOperations.delete(id, domainClass, defaultDeleteOptions());
 	}
 
 	/**
@@ -174,15 +177,15 @@ public void delete(final T entity) {
 		} catch (final IllegalAccessException e) {
 			e.printStackTrace();
 		}
-		arangoOperations.delete(id, domainClass);
+		arangoOperations.delete(id, domainClass, defaultDeleteOptions());
 	}
 
     /**
      * Deletes all instances of the type {@code T} with the given IDs.
 	 * @implNote do not add @Override annotation to keep backwards compatibility with spring-data-commons 2.4
      */
     public void deleteAllById(Iterable<? extends ID> ids) {
-        arangoOperations.delete((Iterable<Object>) ids, domainClass);
+        arangoOperations.delete((Iterable<Object>) ids, domainClass, defaultDeleteOptions());
     }
 
 	/**
@@ -319,7 +322,7 @@ public <S extends T> long count(final Example<S> example) {
 		final String filter = predicate.length() == 0 ? "" : " FILTER " + predicate;
 		final String query = String.format("FOR e IN @@col %s COLLECT WITH COUNT INTO length RETURN length", filter);
 		arangoOperations.collection(domainClass);
-		final ArangoCursor<Long> cursor = arangoOperations.query(query, bindVars, null, Long.class);
+		final ArangoCursor<Long> cursor = arangoOperations.query(query, bindVars, defaultQueryOptions(), Long.class);
 		return cursor.next();
 	}
 
@@ -345,7 +348,7 @@ private <S extends T> ArangoCursor<T> findAllInternal(final Sort sort, @Nullable
 		final String query = String.format("FOR e IN @@col %s %s RETURN e",
 				buildFilterClause(example, bindVars), buildSortClause(sort, "e"));
 		arangoOperations.collection(domainClass);
-		return arangoOperations.query(query, bindVars, null, domainClass);
+		return arangoOperations.query(query, bindVars, defaultQueryOptions(), domainClass);
 	}
 
 	private <S extends T> ArangoCursor<T> findAllInternal(final Pageable pageable, @Nullable final Example<S> example,
@@ -355,7 +358,7 @@ private <S extends T> ArangoCursor<T> findAllInternal(final Pageable pageable, @
 				buildFilterClause(example, bindVars), buildPageableClause(pageable, "e"));
 		arangoOperations.collection(domainClass);
 		return arangoOperations.query(query, bindVars,
-				pageable != null ? new AqlQueryOptions().fullCount(true) : null, domainClass);
+				pageable != null ? new AqlQueryOptions().fullCount(true) : defaultQueryOptions(), domainClass);
 	}
 
 	private <S extends T> String buildFilterClause(final Example<S> example, final Map<String, Object> bindVars) {
@@ -383,4 +386,36 @@ private String buildSortClause(final Sort sort, final String varName) {
         return sort == null ? "" : AqlUtils.buildSortClause(AqlUtils.toPersistentSort(sort, mappingContext, domainClass), varName);
     }
 
+	private DocumentReadOptions defaultReadOptions() {
+		DocumentReadOptions options = new DocumentReadOptions();
+		if (transactionBridge != null) {
+			options.streamTransactionId(transactionBridge.getCurrentTransaction(Collections.singleton(getCollectionName())));
+		}
+		return options;
+	}
+
+	private AqlQueryOptions defaultQueryOptions() {
+		AqlQueryOptions options = new AqlQueryOptions();
+		if (transactionBridge != null) {
+			options.streamTransactionId(transactionBridge.getCurrentTransaction(Collections.singleton(getCollectionName())));
+		}
+		return options;
+	}
+
+	private DocumentExistsOptions defaultExistsOptions() {
+		DocumentExistsOptions options = new DocumentExistsOptions();
+		if (transactionBridge != null) {
+			options.streamTransactionId(transactionBridge.getCurrentTransaction(Collections.singleton(getCollectionName())));
+		}
+		return options;
+	}
+
+	private DocumentDeleteOptions defaultDeleteOptions() {
+		DocumentDeleteOptions options = new DocumentDeleteOptions();
+		if (transactionBridge != null) {
+			options.streamTransactionId(transactionBridge.getCurrentTransaction(Collections.singleton(getCollectionName())));
+		}
+		return options;
+	}
+
 }

src/test/java/com/arangodb/springframework/transaction/ArangoTransactionManagerRepositoryTest.java

@@ -22,8 +22,8 @@ public class ArangoTransactionManagerRepositoryTest extends AbstractArangoTest {
 	private HumanBeingRepository humanBeingRepository;
 
 	@Before
-	public void setUp() {
-		humanBeingRepository.deleteAll();
+	public void cleanupDatabase() {
+		template.collection(HumanBeing.class).truncate();
 	}
 
 	@Test