DATAJPA-806 - Add flushAutomatically attribute to @Modifying annotation.

Original pull request: #172.

Author: ciri-cuervo

src/main/java/org/springframework/data/jpa/repository/Modifying.java

@@ -26,12 +26,20 @@
  * 
  * @author Oliver Gierke
  * @author Christoph Strobl
+ * @author Nicolas Cirigliano
  */
 @Retention(RetentionPolicy.RUNTIME)
 @Target({ ElementType.METHOD, ElementType.ANNOTATION_TYPE })
 @Documented
 public @interface Modifying {
 
+	/**
+	 * Defines whether we should flush the underlying persistence context before executing the modifying query.
+	 * 
+	 * @return
+	 */
+	boolean flushAutomatically() default false;
+
 	/**
 	 * Defines whether we should clear the underlying persistence context after executing the modifying query.
 	 * 

src/main/java/org/springframework/data/jpa/repository/query/AbstractJpaQuery.java

@@ -52,6 +52,7 @@
  * @author Thomas Darimont
  * @author Mark Paluch
  * @author Christoph Strobl
+ * @author Nicolas Cirigliano
  * @author Jens Schauder
  * @author Сергей Цыпанов
  */
@@ -145,7 +146,7 @@ protected JpaQueryExecution getExecution() {
 		} else if (method.isPageQuery()) {
 			return new PagedExecution(method.getParameters());
 		} else if (method.isModifyingQuery()) {
-			return method.getClearAutomatically() ? new ModifyingExecution(method, em) : new ModifyingExecution(method, null);
+			return new ModifyingExecution(method, em);
 		} else {
 			return new SingleEntityExecution();
 		}

src/main/java/org/springframework/data/jpa/repository/query/JpaQueryExecution.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2008-2017 the original author or authors.
+ * Copyright 2008-2018 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -54,6 +54,7 @@
  * @author Thomas Darimont
  * @author Mark Paluch
  * @author Christoph Strobl
+ * @author Nicolas Cirigliano
  */
 public abstract class JpaQueryExecution {
 
@@ -218,11 +219,14 @@ protected Object doExecute(AbstractJpaQuery query, Object[] values) {
 	 */
 	static class ModifyingExecution extends JpaQueryExecution {
 
-		private final @Nullable EntityManager em;
+		private final EntityManager em;
+		private final boolean flush;
+		private final boolean clear;
 
 		/**
-		 * Creates an execution that automatically clears the given {@link EntityManager} after execution if the given
-		 * {@link EntityManager} is not {@literal null}.
+		 * Creates an execution that automatically flushes the given {@link EntityManager} before execution and/or
+		 * clears the given {@link EntityManager} after execution if the given {@link EntityManager} is not
+		 * {@literal null}.
 		 * 
 		 * @param em
 		 */
@@ -236,14 +240,20 @@ public ModifyingExecution(JpaQueryMethod method, @Nullable EntityManager em) {
 			Assert.isTrue(isInt || isVoid, "Modifying queries can only use void or int/Integer as return type!");
 
 			this.em = em;
+			this.flush = method.getFlushAutomatically();
+			this.clear = method.getClearAutomatically();
 		}
 
 		@Override
 		protected Object doExecute(AbstractJpaQuery query, Object[] values) {
 
+			if (em != null && flush) {
+				em.flush();
+			}
+
 			int result = query.createQuery(values).executeUpdate();
 
-			if (em != null) {
+			if (em != null && clear) {
 				em.clear();
 			}
 

src/main/java/org/springframework/data/jpa/repository/query/JpaQueryMethod.java

@@ -51,6 +51,7 @@
  * @author Oliver Gierke
  * @author Thomas Darimont
  * @author Christoph Strobl
+ * @author Nicolas Cirigliano
  * @author Mark Paluch
  * @author Сергей Цыпанов
  */
@@ -312,6 +313,15 @@ String getNamedCountQueryName() {
 		return StringUtils.hasText(annotatedName) ? annotatedName : getNamedQueryName() + ".count";
 	}
 
+	/**
+	 * Returns whether we should flush automatically for modifying queries.
+	 * 
+	 * @return
+	 */
+	boolean getFlushAutomatically() {
+		return getMergedOrDefaultAnnotationValue("flushAutomatically", Modifying.class, Boolean.class);
+	}
+
 	/**
 	 * Returns whether we should clear automatically for modifying queries.
 	 *

src/test/java/org/springframework/data/jpa/repository/query/JpaQueryExecutionUnitTests.java

@@ -46,6 +46,7 @@
  * @author Oliver Gierke
  * @author Thomas Darimont
  * @author Mark Paluch
+ * @author Nicolas Cirigliano
  */
 @RunWith(MockitoJUnitRunner.Silent.class)
 public class JpaQueryExecutionUnitTests {
@@ -82,18 +83,37 @@ protected Object doExecute(AbstractJpaQuery query, Object[] values) {
 		}.execute(jpaQuery, new Object[] {}), is(nullValue()));
 	}
 
+	@Test
+	@SuppressWarnings({ "unchecked", "rawtypes" })
+	public void modifyingExecutionFlushesEntityManagerIfSet() {
+
+		when(query.executeUpdate()).thenReturn(0);
+		when(method.getReturnType()).thenReturn((Class) void.class);
+		when(method.getFlushAutomatically()).thenReturn(true);
+		when(jpaQuery.createQuery(Mockito.any(Object[].class))).thenReturn(query);
+		when(jpaQuery.getQueryMethod()).thenReturn(method);
+
+		ModifyingExecution execution = new ModifyingExecution(method, em);
+		execution.execute(jpaQuery, new Object[] {});
+
+		verify(em, times(1)).flush();
+		verify(em, times(0)).clear();
+	}
+
 	@Test
 	@SuppressWarnings({ "unchecked", "rawtypes" })
 	public void modifyingExecutionClearsEntityManagerIfSet() {
 
 		when(query.executeUpdate()).thenReturn(0);
 		when(method.getReturnType()).thenReturn((Class) void.class);
+		when(method.getClearAutomatically()).thenReturn(true);
 		when(jpaQuery.createQuery(Mockito.any(Object[].class))).thenReturn(query);
 		when(jpaQuery.getQueryMethod()).thenReturn(method);
 
 		ModifyingExecution execution = new ModifyingExecution(method, em);
 		execution.execute(jpaQuery, new Object[] {});
 
+		verify(em, times(0)).flush();
 		verify(em, times(1)).clear();
 	}