Polishing.

See #2555.

Author: gregturn

spring-data-jpa/src/main/java/org/springframework/data/jpa/repository/query/JSqlParserQueryEnhancer.java

@@ -63,16 +63,18 @@ public class JSqlParserQueryEnhancer implements QueryEnhancer {
 	 * @param query the query we want to enhance. Must not be {@literal null}.
 	 */
 	public JSqlParserQueryEnhancer(DeclaredQuery query) {
+
 		this.query = query;
 		this.parsedType = detectParsedType();
 	}
 
 	/**
 	 * Detects what type of query is provided.
-	 * 
+	 *
 	 * @return the parsed type
 	 */
 	private ParsedType detectParsedType() {
+
 		try {
 			Statement statement = CCJSqlParserUtil.parse(this.query.getQueryString());
 
@@ -85,14 +87,14 @@ private ParsedType detectParsedType() {
 			} else {
 				return ParsedType.SELECT;
 			}
-
 		} catch (JSQLParserException e) {
 			throw new IllegalArgumentException("The query you provided is not a valid SQL Query!", e);
 		}
 	}
 
 	@Override
 	public String applySorting(Sort sort, @Nullable String alias) {
+
 		String queryString = query.getQueryString();
 		Assert.hasText(queryString, "Query must not be null or empty!");
 
@@ -168,9 +170,11 @@ Set<String> getSelectionAliases() {
 	 * @return a {@literal Set} of aliases used in the query. Guaranteed to be not {@literal null}.
 	 */
 	private Set<String> getJoinAliases(String query) {
+
 		if (this.parsedType != ParsedType.SELECT) {
 			return new HashSet<>();
 		}
+
 		return getJoinAliases((PlainSelect) parseSelectStatement(query).getSelectBody());
 	}
 
@@ -377,16 +381,17 @@ private boolean onlyASingleColumnProjection(List<SelectItem> projection) {
 	public DeclaredQuery getQuery() {
 		return this.query;
 	}
-}
 
-/**
- * An enum to represent the top level parsed statement of the provided query.
- * <ul>
- * <li>{@code ParsedType.DELETE}: means the top level statement is {@link Delete}</li>
- * <li>{@code ParsedType.UPDATE}: means the top level statement is {@link Update}</li>
- * <li>{@code ParsedType.SELECT}: means the top level statement is {@link Select}</li>
- * </ul>
- */
-enum ParsedType {
-	DELETE, UPDATE, SELECT;
+	/**
+	 * An enum to represent the top level parsed statement of the provided query.
+	 * <ul>
+	 * <li>{@code ParsedType.DELETE}: means the top level statement is {@link Delete}</li>
+	 * <li>{@code ParsedType.UPDATE}: means the top level statement is {@link Update}</li>
+	 * <li>{@code ParsedType.SELECT}: means the top level statement is {@link Select}</li>
+	 * </ul>
+	 */
+	enum ParsedType {
+		DELETE, UPDATE, SELECT;
+	}
+
 }

spring-data-jpa/src/test/java/org/springframework/data/jpa/repository/UserRepositoryTests.java

@@ -2867,6 +2867,7 @@ void existsWithSpec() {
 
 	@Test // GH-2555
 	void modifyingUpdateNativeQueryWorksWithJSQLParser() {
+
 		flushTestUsers();
 
 		Optional<User> byIdUser = repository.findById(firstUser.getId());

spring-data-jpa/src/test/java/org/springframework/data/jpa/repository/query/QueryEnhancerUnitTests.java

@@ -733,6 +733,7 @@ void countQueryUsesCorrectVariable() {
 
 	@Test // GH-2555
 	void modifyingQueriesAreDetectedCorrectly() {
+
 		String modifyingQuery = "update userinfo user set user.is_in_treatment = false where user.id = :userId";
 
 		String aliasNotConsideringQueryType = QueryUtils.detectAlias(modifyingQuery);