DATAJDBC-107 - Implement naming strategy

Created NamingStrategy and a default implementation that replicates the current solution. Several unit tests illustrates how to override the default and plugin a custom solution including a ThreadLocal, contextual one that could be user-based if, for example, Spring Security's SecurityContextHolder was used.

Author: gregturn

src/main/java/org/springframework/data/jdbc/mapping/model/BasicJdbcPersistentProperty.java

@@ -74,7 +74,7 @@ protected Association<JdbcPersistentProperty> createAssociation() {
  * @see org.springframework.data.jdbc.mapping.model.JdbcPersistentProperty#getColumnName()
  */
 public String getColumnName() {
-return getName();
+return this.context.getNamingStrategy().getColumnName(this);
 }
 
 /**

src/main/java/org/springframework/data/jdbc/mapping/model/Context.java

@@ -0,0 +1,26 @@
+/*
+ * Copyright 2017 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.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.jdbc.mapping.model;
+
+/**
+ * Token definition of context.
+ * 
+ * @author Greg Turnquist
+ */
+public interface Context {
+
+String getContext();
+}

src/main/java/org/springframework/data/jdbc/mapping/model/DefaultNamingStrategy.java

@@ -0,0 +1,61 @@
+/*
+ * Copyright 2017 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.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.jdbc.mapping.model;
+
+/**
+ * Basic implementation of {@link NamingStrategy} with no schema, table based on {@link Class} and
+ * column name based on {@link JdbcPersistentProperty}.
+ *
+ * NOTE: Can also be used as an adapter. Create an anonymous subclass and override any settings to implement
+ * a different strategy on the fly.
+ *
+ * @author Greg Turnquist
+ */
+public class DefaultNamingStrategy implements NamingStrategy {
+
+/**
+ * No schema at all!
+ * 
+ * @return
+ */
+@Override
+public String getSchema() {
+return "";
+}
+
+/**
+ * Look up the {@link Class}'s simple name.
+ * 
+ * @param type
+ * @return
+ */
+@Override
+public String getTableName(Class<?> type) {
+return type.getSimpleName();
+}
+
+
+/**
+ * Look up the {@link JdbcPersistentProperty}'s name.
+ * 
+ * @param property
+ * @return
+ */
+@Override
+public String getColumnName(JdbcPersistentProperty property) {
+return property.getName();
+}
+}

src/main/java/org/springframework/data/jdbc/mapping/model/JdbcMappingContext.java

@@ -17,6 +17,9 @@
 
 import static java.util.Arrays.*;
 
+import lombok.Getter;
+import lombok.experimental.Wither;
+
 import java.math.BigDecimal;
 import java.math.BigInteger;
 import java.time.temporal.Temporal;
@@ -46,7 +49,11 @@ public class JdbcMappingContext extends AbstractMappingContext<JdbcPersistentEnt
 Temporal.class //
 ));
 
-public JdbcMappingContext() {
+private final @Getter @Wither NamingStrategy namingStrategy;
+
+public JdbcMappingContext(NamingStrategy namingStrategy) {
+
+this.namingStrategy = namingStrategy;
 setSimpleTypeHolder(new SimpleTypeHolder(CUSTOM_SIMPLE_TYPES, true));
 }
 
@@ -80,7 +87,7 @@ public List<PropertyPath> referencedEntities(Class<?> rootType, PropertyPath pat
  */
 @Override
 protected <T> JdbcPersistentEntity<T> createPersistentEntity(TypeInformation<T> typeInformation) {
-return new JdbcPersistentEntityImpl<>(typeInformation);
+return new JdbcPersistentEntityImpl<>(typeInformation, this.namingStrategy);
 }
 
 /*

src/main/java/org/springframework/data/jdbc/mapping/model/JdbcPersistentEntityImpl.java

@@ -29,18 +29,19 @@
 class JdbcPersistentEntityImpl<T> extends BasicPersistentEntity<T, JdbcPersistentProperty>
 implements JdbcPersistentEntity<T> {
 
+private final NamingStrategy namingStrategy;
 private final @Getter String tableName;
 
 /**
  * Creates a new {@link JdbcPersistentEntityImpl} for the given {@link TypeInformation}.
  * 
  * @param information must not be {@literal null}.
  */
-JdbcPersistentEntityImpl(TypeInformation<T> information) {
+JdbcPersistentEntityImpl(TypeInformation<T> information, NamingStrategy namingStrategy) {
 
 super(information);
-
-tableName = getType().getSimpleName();
+this.namingStrategy = namingStrategy;
+this.tableName = this.namingStrategy.getSchema() + this.namingStrategy.getTableName(getType());
 }
 
 /* 
@@ -49,7 +50,7 @@ class JdbcPersistentEntityImpl<T> extends BasicPersistentEntity<T, JdbcPersisten
  */
 @Override
 public String getIdColumn() {
-return getRequiredIdProperty().getName();
+return this.namingStrategy.getColumnName(getRequiredIdProperty());
 }
 
 @Override

src/main/java/org/springframework/data/jdbc/mapping/model/NamingStrategy.java

@@ -0,0 +1,29 @@
+/*
+ * Copyright 2017 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.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.jdbc.mapping.model;
+
+/**
+ * @author Greg Turnquist
+ */
+public interface NamingStrategy {
+
+String getSchema();
+
+String getTableName(Class<?> type);
+
+String getColumnName(JdbcPersistentProperty property);
+
+}

src/main/java/org/springframework/data/jdbc/repository/support/JdbcRepositoryFactory.java

@@ -15,14 +15,13 @@
  */
 package org.springframework.data.jdbc.repository.support;
 
-import lombok.RequiredArgsConstructor;
-
 import org.springframework.context.ApplicationEventPublisher;
 import org.springframework.data.jdbc.core.JdbcEntityTemplate;
 import org.springframework.data.jdbc.mapping.model.BasicJdbcPersistentEntityInformation;
 import org.springframework.data.jdbc.mapping.model.JdbcMappingContext;
 import org.springframework.data.jdbc.mapping.model.JdbcPersistentEntity;
 import org.springframework.data.jdbc.mapping.model.JdbcPersistentEntityInformation;
+import org.springframework.data.jdbc.mapping.model.NamingStrategy;
 import org.springframework.data.jdbc.repository.SimpleJdbcRepository;
 import org.springframework.data.repository.core.EntityInformation;
 import org.springframework.data.repository.core.RepositoryInformation;
@@ -34,13 +33,19 @@
  * @author Jens Schauder
  * @since 2.0
  */
-@RequiredArgsConstructor
 public class JdbcRepositoryFactory extends RepositoryFactorySupport {
 
-private final JdbcMappingContext context = new JdbcMappingContext();
+private final JdbcMappingContext context;
 private final NamedParameterJdbcOperations jdbcOperations;
 private final ApplicationEventPublisher publisher;
 
+public JdbcRepositoryFactory(NamedParameterJdbcOperations namedParameterJdbcOperations, ApplicationEventPublisher publisher, NamingStrategy namingStrategy) {
+
+this.jdbcOperations = namedParameterJdbcOperations;
+this.publisher = publisher;
+this.context = new JdbcMappingContext(namingStrategy);
+}
+
 @SuppressWarnings("unchecked")
 @Override
 public <T, ID> EntityInformation<T, ID> getEntityInformation(Class<T> aClass) {

src/main/java/org/springframework/data/jdbc/repository/support/JdbcRepositoryFactoryBean.java

@@ -23,6 +23,8 @@
 
 import org.springframework.context.ApplicationContext;
 import org.springframework.context.ApplicationEventPublisher;
+import org.springframework.data.jdbc.mapping.model.DefaultNamingStrategy;
+import org.springframework.data.jdbc.mapping.model.NamingStrategy;
 import org.springframework.data.repository.Repository;
 import org.springframework.data.repository.core.support.RepositoryFactorySupport;
 import org.springframework.data.repository.core.support.TransactionalRepositoryFactoryBeanSupport;
@@ -45,9 +47,12 @@ public class JdbcRepositoryFactoryBean<T extends Repository<S, ID>, S, ID extend
 "No unique NamedParameterJdbcOperation could be found, " //
 + "nor JdbcOperations or DataSource to construct one from.";
 
+private static final String NO_NAMING_STRATEGY_ERROR_MESSAGE = "No unique NamingStrategy could be found.";
+
 private static final String NAMED_PARAMETER_JDBC_OPERATIONS_BEAN_NAME = "namedParameterJdbcTemplate";
 private static final String JDBC_OPERATIONS_BEAN_NAME = "jdbcTemplate";
 private static final String DATA_SOURCE_BEAN_NAME = "dataSource";
+private static final String NAMING_STRATEGY_BEAN_NAME = "namingStrategy";
 
 private final ApplicationEventPublisher applicationEventPublisher;
 private final ApplicationContext context;
@@ -62,7 +67,7 @@ public class JdbcRepositoryFactoryBean<T extends Repository<S, ID>, S, ID extend
 
 @Override
 protected RepositoryFactorySupport doCreateRepositoryFactory() {
-return new JdbcRepositoryFactory(findOrCreateJdbcOperations(), applicationEventPublisher);
+return new JdbcRepositoryFactory(findOrCreateJdbcOperations(), applicationEventPublisher, findOrCreateNamingStrategy());
 }
 
 private NamedParameterJdbcOperations findOrCreateJdbcOperations() {
@@ -75,6 +80,12 @@ private NamedParameterJdbcOperations findOrCreateJdbcOperations() {
 .orElseThrow(() -> new IllegalStateException(NO_NAMED_PARAMETER_JDBC_OPERATION_ERROR_MESSAGE));
 }
 
+private NamingStrategy findOrCreateNamingStrategy() {
+
+return getNamingStrategy()
+.orElse(new DefaultNamingStrategy());
+}
+
 private Optional<NamedParameterJdbcOperations> getNamedParameterJdbcOperations() {
 return getBean(NamedParameterJdbcOperations.class, NAMED_PARAMETER_JDBC_OPERATIONS_BEAN_NAME);
 }
@@ -87,6 +98,10 @@ private Optional<DataSource> getDataSource() {
 return getBean(DataSource.class, DATA_SOURCE_BEAN_NAME);
 }
 
+private Optional<NamingStrategy> getNamingStrategy() {
+return getBean(NamingStrategy.class, NAMING_STRATEGY_BEAN_NAME);
+}
+
 private <R> Optional<R> getBean(Class<R> type, String name) {
 
 Map<String, R> beansOfType = context.getBeansOfType(type);