Skip to content

Commit 40caca5

Browse files
ctailor2ctailor2
committed
Extract insert jdbc operation and key generation logic from DefaultDataAccessStrategy into InsertStrategyFactory.
1 parent a8c3d6c commit 40caca5

File tree

11 files changed

+488
-288
lines changed

11 files changed

+488
-288
lines changed

‎spring-data-jdbc/src/main/java/org/springframework/data/jdbc/core/convert/BatchJdbcOperations.java

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
import org.springframework.jdbc.core.namedparam.NamedParameterUtils;
1212
import org.springframework.jdbc.core.namedparam.ParsedSql;
1313
import org.springframework.jdbc.core.namedparam.SqlParameterSource;
14-
import org.springframework.jdbc.support.GeneratedKeyHolder;
1514
import org.springframework.jdbc.support.JdbcUtils;
1615
import org.springframework.jdbc.support.KeyHolder;
1716
import org.springframework.lang.Nullable;
@@ -22,16 +21,16 @@
2221
import java.util.List;
2322
import java.util.Map;
2423

25-
class BatchJdbcOperations {
24+
public class BatchJdbcOperations {
2625
private final JdbcOperations jdbcOperations;
2726

28-
BatchJdbcOperations(JdbcOperations jdbcOperations) {
27+
public BatchJdbcOperations(JdbcOperations jdbcOperations) {
2928
this.jdbcOperations = jdbcOperations;
3029
}
3130

3231
@Nullable
33-
int[] insert(String insertSql, SqlParameterSource[] sqlParameterSources,
34-
GeneratedKeyHolder generatedKeyHolder) {
32+
int[] batchUpdate(String insertSql, SqlParameterSource[] sqlParameterSources,
33+
KeyHolder generatedKeyHolder) {
3534
// TODO: This is largely duplicated from spring-jdbc and should be replaced with a call into
3635
// NamedParameterJdbcTemplate#batchUpdate once a method taking KeyHolder is added there.
3736
ParsedSql parsedSql = NamedParameterUtils.parseSqlStatement(insertSql);
@@ -80,8 +79,8 @@ public int getBatchSize() {
8079
}
8180

8281
@Nullable
83-
public int[] insert(String insertSql, SqlIdentifierParameterSource[] sqlParameterSources,
84-
GeneratedKeyHolder generatedKeyHolder, String[] keyColumnNames) {
82+
public int[] batchUpdate(String insertSql, SqlParameterSource[] sqlParameterSources,
83+
KeyHolder generatedKeyHolder, String[] keyColumnNames) {
8584
// TODO: This is largely duplicated from spring-jdbc and should be replaced with a call into
8685
// NamedParameterJdbcTemplate#batchUpdate once a method taking KeyHolder is added there.
8786
ParsedSql parsedSql = NamedParameterUtils.parseSqlStatement(insertSql);

‎spring-data-jdbc/src/main/java/org/springframework/data/jdbc/core/convert/DefaultDataAccessStrategy.java

Lines changed: 13 additions & 121 deletions
Original file line numberDiff line numberDiff line change
@@ -22,21 +22,15 @@
2222
import java.util.ArrayList;
2323
import java.util.Collections;
2424
import java.util.List;
25-
import java.util.Map;
26-
import java.util.function.Predicate;
25+
import java.util.Optional;
2726

28-
import org.springframework.dao.DataRetrievalFailureException;
2927
import org.springframework.dao.EmptyResultDataAccessException;
30-
import org.springframework.dao.InvalidDataAccessApiUsageException;
3128
import org.springframework.dao.OptimisticLockingFailureException;
3229
import org.springframework.data.domain.Pageable;
3330
import org.springframework.data.domain.Sort;
3431
import org.springframework.data.jdbc.core.mapping.JdbcValue;
3532
import org.springframework.data.jdbc.support.JdbcUtil;
36-
import org.springframework.data.mapping.PersistentProperty;
37-
import org.springframework.data.mapping.PersistentPropertyAccessor;
3833
import org.springframework.data.mapping.PersistentPropertyPath;
39-
import org.springframework.data.relational.core.dialect.IdGeneration;
4034
import org.springframework.data.relational.core.mapping.PersistentPropertyPathExtension;
4135
import org.springframework.data.relational.core.mapping.RelationalMappingContext;
4236
import org.springframework.data.relational.core.mapping.RelationalPersistentEntity;
@@ -47,9 +41,7 @@
4741
import org.springframework.jdbc.core.RowMapper;
4842
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcOperations;
4943
import org.springframework.jdbc.core.namedparam.SqlParameterSource;
50-
import org.springframework.jdbc.support.GeneratedKeyHolder;
5144
import org.springframework.jdbc.support.JdbcUtils;
52-
import org.springframework.jdbc.support.KeyHolder;
5345
import org.springframework.lang.Nullable;
5446
import org.springframework.util.Assert;
5547

@@ -75,8 +67,8 @@ public class DefaultDataAccessStrategy implements DataAccessStrategy {
7567
private final RelationalMappingContext context;
7668
private final JdbcConverter converter;
7769
private final NamedParameterJdbcOperations operations;
78-
private final BatchJdbcOperations batchOperations;
7970
private final SqlParametersFactory sqlParametersFactory;
71+
private final InsertStrategyFactory insertStrategyFactory;
8072

8173
/**
8274
* Creates a {@link DefaultDataAccessStrategy}
@@ -88,26 +80,21 @@ public class DefaultDataAccessStrategy implements DataAccessStrategy {
8880
* @since 1.1
8981
*/
9082
public DefaultDataAccessStrategy(SqlGeneratorSource sqlGeneratorSource, RelationalMappingContext context,
91-
JdbcConverter converter, NamedParameterJdbcOperations operations, SqlParametersFactory sqlParametersFactory) {
92-
this(sqlGeneratorSource, context, converter, operations, new BatchJdbcOperations(operations.getJdbcOperations()), sqlParametersFactory);
93-
}
94-
95-
DefaultDataAccessStrategy(SqlGeneratorSource sqlGeneratorSource, RelationalMappingContext context,
96-
JdbcConverter converter, NamedParameterJdbcOperations operations,
97-
BatchJdbcOperations batchOperations, SqlParametersFactory sqlParametersFactory) {
98-
83+
JdbcConverter converter, NamedParameterJdbcOperations operations, SqlParametersFactory sqlParametersFactory,
84+
InsertStrategyFactory insertStrategyFactory) {
9985
Assert.notNull(sqlGeneratorSource, "SqlGeneratorSource must not be null");
10086
Assert.notNull(context, "RelationalMappingContext must not be null");
10187
Assert.notNull(converter, "JdbcConverter must not be null");
10288
Assert.notNull(operations, "NamedParameterJdbcOperations must not be null");
10389
Assert.notNull(sqlParametersFactory, "SqlParametersFactory must not be null");
90+
Assert.notNull(insertStrategyFactory, "InsertStrategyFactory must not be null");
10491

10592
this.sqlGeneratorSource = sqlGeneratorSource;
10693
this.context = context;
10794
this.converter = converter;
10895
this.operations = operations;
109-
this.batchOperations = batchOperations;
11096
this.sqlParametersFactory = sqlParametersFactory;
97+
this.insertStrategyFactory = insertStrategyFactory;
11198
}
11299

113100
/*
@@ -121,14 +108,7 @@ public <T> Object insert(T instance, Class<T> domainType, Identifier identifier,
121108

122109
String insertSql = sql(domainType).getInsert(parameterSource.getIdentifiers());
123110

124-
if (!includeId) {
125-
RelationalPersistentEntity<T> persistentEntity = getRequiredPersistentEntity(domainType);
126-
return executeInsertAndReturnGeneratedId(persistentEntity, parameterSource, insertSql);
127-
} else {
128-
129-
operations.update(insertSql, parameterSource);
130-
return null;
131-
}
111+
return insertStrategyFactory.insertStrategy(!includeId, getIdColumn(domainType)).execute(insertSql, parameterSource);
132112
}
133113

134114
@Override
@@ -142,68 +122,7 @@ public <T> Object[] insert(List<RecordDescriptor<T>> recordDescriptors, Class<T>
142122

143123
String insertSql = sql(domainType).getInsert(sqlParameterSources[0].getIdentifiers());
144124

145-
if (includeId) {
146-
operations.batchUpdate(insertSql, sqlParameterSources);
147-
return new Object[sqlParameterSources.length];
148-
}
149-
GeneratedKeyHolder holder = new GeneratedKeyHolder();
150-
151-
IdGeneration idGeneration = sqlGeneratorSource.getDialect().getIdGeneration();
152-
153-
RelationalPersistentEntity<T> persistentEntity = getRequiredPersistentEntity(domainType);
154-
if (idGeneration.driverRequiresKeyColumnNames()) {
155-
156-
String[] keyColumnNames = getKeyColumnNames(persistentEntity.getType());
157-
if (keyColumnNames.length == 0) {
158-
batchOperations.insert(insertSql, sqlParameterSources, holder);
159-
} else {
160-
batchOperations.insert(insertSql, sqlParameterSources, holder, keyColumnNames);
161-
}
162-
} else {
163-
batchOperations.insert(insertSql, sqlParameterSources, holder);
164-
}
165-
// TODO: Is this needed?
166-
if (!persistentEntity.hasIdProperty()) {
167-
return new Object[sqlParameterSources.length];
168-
}
169-
// TODO: Duplicated in #getIdFromHolder - consider refactoring
170-
Object[] ids = new Object[sqlParameterSources.length];
171-
List<Map<String, Object>> keyList = holder.getKeyList();
172-
for (int i = 0; i < keyList.size(); i++) {
173-
Map<String, Object> keys = keyList.get(i);
174-
Object id;
175-
if (keys.size() > 1) {
176-
id = keys.get(persistentEntity.getIdColumn().getReference(getIdentifierProcessing()));
177-
} else {
178-
id = keys.entrySet().stream().findFirst() //
179-
.map(Map.Entry::getValue) //
180-
.orElseThrow(() -> new IllegalStateException("KeyHolder contains an empty key list."));
181-
}
182-
ids[i] = id;
183-
}
184-
return ids;
185-
}
186-
187-
@Nullable
188-
private <T> Object executeInsertAndReturnGeneratedId(RelationalPersistentEntity<T> persistentEntity, SqlIdentifierParameterSource parameterSource, String insertSql) {
189-
190-
KeyHolder holder = new GeneratedKeyHolder();
191-
192-
IdGeneration idGeneration = sqlGeneratorSource.getDialect().getIdGeneration();
193-
194-
if (idGeneration.driverRequiresKeyColumnNames()) {
195-
196-
String[] keyColumnNames = getKeyColumnNames(persistentEntity.getType());
197-
if (keyColumnNames.length == 0) {
198-
operations.update(insertSql, parameterSource, holder);
199-
} else {
200-
operations.update(insertSql, parameterSource, holder, keyColumnNames);
201-
}
202-
} else {
203-
operations.update(insertSql, parameterSource, holder);
204-
}
205-
206-
return getIdFromHolder(holder, persistentEntity);
125+
return insertStrategyFactory.insertStrategy(!includeId, getIdColumn(domainType)).execute(insertSql, sqlParameterSources);
207126
}
208127

209128
/*
@@ -476,26 +395,6 @@ public <T> Iterable<T> findAll(Class<T> domainType, Pageable pageable) {
476395
return operations.query(sql(domainType).getFindAll(pageable), (RowMapper<T>) getEntityRowMapper(domainType));
477396
}
478397

479-
@Nullable
480-
private <S> Object getIdFromHolder(KeyHolder holder, RelationalPersistentEntity<S> persistentEntity) {
481-
482-
try {
483-
// MySQL just returns one value with a special name
484-
return holder.getKey();
485-
} catch (DataRetrievalFailureException | InvalidDataAccessApiUsageException e) {
486-
// Postgres returns a value for each column
487-
// MS SQL Server returns a value that might be null.
488-
489-
Map<String, Object> keys = holder.getKeys();
490-
491-
if (keys == null || persistentEntity.getIdProperty() == null) {
492-
return null;
493-
}
494-
495-
return keys.get(persistentEntity.getIdColumn().getReference(getIdentifierProcessing()));
496-
}
497-
}
498-
499398
private EntityRowMapper<?> getEntityRowMapper(Class<?> domainType) {
500399
return new EntityRowMapper<>(getRequiredPersistentEntity(domainType), converter);
501400
}
@@ -588,17 +487,10 @@ private SqlGenerator sql(Class<?> domainType) {
588487
return sqlGeneratorSource.getSqlGenerator(domainType);
589488
}
590489

591-
private <T> String[] getKeyColumnNames(Class<T> domainType) {
592-
593-
RelationalPersistentEntity<?> requiredPersistentEntity = context.getRequiredPersistentEntity(domainType);
594-
595-
if (!requiredPersistentEntity.hasIdProperty()) {
596-
return new String[0];
597-
}
598-
599-
SqlIdentifier idColumn = requiredPersistentEntity.getIdColumn();
600-
601-
return new String[] { idColumn.getReference(getIdentifierProcessing()) };
490+
@Nullable
491+
private <T> SqlIdentifier getIdColumn(Class<T> domainType) {
492+
return Optional.ofNullable(context.getRequiredPersistentEntity(domainType).getIdProperty())
493+
.map(RelationalPersistentProperty::getColumnName)
494+
.orElse(null);
602495
}
603-
604496
}

0 commit comments

Comments
 (0)