Spring/spring-data-relational
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

#475 - Polishing.

Browse Source 
Extract mapping target type into field instead of obtaining it each time from EntityMetadata.
This commit is contained in:
Mark Paluch
2020-10-08 15:04:59 +02:00
parent b8f4e8b765
commit 101725c7e5
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/java/org/springframework/data/r2dbc/repository/query/R2dbcQueryCreator.java
View File

@@ -56,6 +56,7 @@ class R2dbcQueryCreator extends RelationalQueryCreator<PreparedOperation<?>> {
private final ReactiveDataAccessStrategy dataAccessStrategy;
private final RelationalEntityMetadata<?> entityMetadata;
private final List<String> projectedProperties;
private final Class<?> entityToRead;
/**
* Creates new instance of this class with the given {@link PartTree}, {@link ReactiveDataAccessStrategy},
@@ -78,6 +79,7 @@ class R2dbcQueryCreator extends RelationalQueryCreator<PreparedOperation<?>> {
this.dataAccessStrategy = dataAccessStrategy;
this.entityMetadata = entityMetadata;
this.projectedProperties = projectedProperties;
this.entityToRead = entityMetadata.getTableEntity().getType();
}
/**
@@ -90,7 +92,7 @@ class R2dbcQueryCreator extends RelationalQueryCreator<PreparedOperation<?>> {
@Override
protected PreparedOperation<?> complete(@Nullable Criteria criteria, Sort sort) {
StatementMapper statementMapper = dataAccessStrategy.getStatementMapper().forType(entityMetadata.getJavaType());
StatementMapper statementMapper = dataAccessStrategy.getStatementMapper().forType(entityToRead);
if (tree.isDelete()) {
return delete(criteria, statementMapper);
@@ -157,7 +159,7 @@ class R2dbcQueryCreator extends RelationalQueryCreator<PreparedOperation<?>> {
} else if (tree.isExistsProjection()) {
expressions = dataAccessStrategy.getIdentifierColumns(entityMetadata.getJavaType()).stream()
expressions = dataAccessStrategy.getIdentifierColumns(entityToRead).stream()
.map(table::column)
.collect(Collectors.toList());
} else if (tree.isCountProjection()) {
@@ -165,7 +167,7 @@ class R2dbcQueryCreator extends RelationalQueryCreator<PreparedOperation<?>> {
SqlIdentifier idColumn = entityMetadata.getTableEntity().getRequiredIdProperty().getColumnName();
expressions = Collections.singletonList(Functions.count(table.column(idColumn)));
} else {
expressions = dataAccessStrategy.getAllColumns(entityMetadata.getJavaType()).stream()
expressions = dataAccessStrategy.getAllColumns(entityToRead).stream()
.map(table::column)
.collect(Collectors.toList());
}