DATAJDBC-334 - Fixes broken parameter names for quoted column names.

When a column has a quoted name it contains characters illegal for parameter names.
Therefore the parameter names now get sanitised.

Compared to the version on master testing is more lenient.
This accounts for the SqlGenerator behaving slightly different then on master.

Original pull request: #120.
See also: https://jira.spring.io/browse/DATAJDBC-262.

Author: schlagi123

src/main/java/org/springframework/data/jdbc/core/SqlGenerator.java

@@ -21,6 +21,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
+import java.util.regex.Pattern;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
@@ -40,6 +41,7 @@
  *
  * @author Jens Schauder
  * @author Yoichi Imai
+ * @author Bastian Wilhelm
  */
 class SqlGenerator {
 
@@ -61,6 +63,8 @@ class SqlGenerator {
 	private final Lazy<String> deleteByListSql = Lazy.of(this::createDeleteByListSql);
 	private final SqlGeneratorSource sqlGeneratorSource;
 
+	private final Pattern parameterPattern = Pattern.compile("\\W");
+
 	SqlGenerator(RelationalMappingContext context, RelationalPersistentEntity<?> entity,
 			SqlGeneratorSource sqlGeneratorSource) {
 
@@ -263,6 +267,7 @@ private String createInsertSql(Set<String> additionalColumns) {
 		String tableColumns = String.join(", ", columnNamesForInsert);
 
 		String parameterNames = columnNamesForInsert.stream()//
+				.map(this::columnNameToParameterName)
 				.map(n -> String.format(":%s", n))//
 				.collect(Collectors.joining(", "));
 
@@ -274,10 +279,11 @@ private String createUpdateSql() {
 		String updateTemplate = "UPDATE %s SET %s WHERE %s = :%s";
 
 		String setClause = columnNames.stream()//
-				.map(n -> String.format("%s = :%s", n, n))//
+				.map(n -> String.format("%s = :%s", n, columnNameToParameterName(n))) //
 				.collect(Collectors.joining(", "));
 
-		return String.format(updateTemplate, entity.getTableName(), setClause, entity.getIdColumn(), entity.getIdColumn());
+		return String.format(updateTemplate, entity.getTableName(), setClause, entity.getIdColumn(),
+				columnNameToParameterName(entity.getIdColumn()));
 	}
 
 	private String createDeleteSql() {
@@ -349,4 +355,8 @@ private String cascadeConditions(String innerCondition, PersistentPropertyPath<R
 				entity.getTableName(), innerCondition //
 		);
 	}
+
+	private String columnNameToParameterName(String columnName){
+		return parameterPattern.matcher(columnName).replaceAll("");
+	}
 }

src/test/java/org/springframework/data/jdbc/core/SqlGeneratorUnitTests.java

@@ -38,6 +38,7 @@
  *
  * @author Jens Schauder
  * @author Greg Turnquist
+ * @author Bastian Wilhelm
  */
 public class SqlGeneratorUnitTests {
 
@@ -189,15 +190,34 @@ public void getInsertForEmptyColumnList() {
 		assertThat(insert).endsWith("()");
 	}
 
+	@Test // DATAJDBC-334
+	public void getInsertForQuotedColumnName() {
+		SqlGenerator sqlGenerator = createSqlGenerator(EntityWithQuotedColumnName.class);
+
+		String insert = sqlGenerator.getInsert(emptySet());
+
+		assertThat(insert)
+				.isEqualTo("INSERT INTO entity_with_quoted_column_name " + "(\"test_@123\") " + "VALUES (:test_123)");
+	}
+
+	@Test // DATAJDBC-334
+	public void getUpdateForQuotedColumnName() {
+		SqlGenerator sqlGenerator = createSqlGenerator(EntityWithQuotedColumnName.class);
+
+		String update = sqlGenerator.getUpdate();
+
+		assertThat(update).startsWith("UPDATE entity_with_quoted_column_name " + "SET ")
+				.endsWith("\"test_@123\" = :test_123 " + "WHERE \"test_@id\" = :test_id");
+	}
+
 	private PersistentPropertyPath<RelationalPersistentProperty> getPath(String path, Class<?> base) {
 		return PersistentPropertyPathTestUtils.getPath(context, path, base);
 	}
 
 	@SuppressWarnings("unused")
 	static class DummyEntity {
 
-		@Column("id1")
-		@Id Long id;
+		@Column("id1") @Id Long id;
 		String name;
 		ReferencedEntity ref;
 		Set<Element> elements;
@@ -239,4 +259,8 @@ static class IdOnlyEntity {
 		@Id Long id;
 	}
 
+	static class EntityWithQuotedColumnName {
+		@Id @Column("\"test_@id\"") Long id;
+		@Column("\"test_@123\"") String name;
+	}
 }