Skip to content

Commit 169de9d

Browse files
committed
#56 - Polishing.
Replaced calls to `SqlRenderer.render(.)` with `SqlRenderer.toString(.)`. And code formatting. Original pull request: #66.
1 parent c241cd9 commit 169de9d

File tree

3 files changed

+38
-21
lines changed

3 files changed

+38
-21
lines changed

src/main/java/org/springframework/data/r2dbc/function/DefaultReactiveDataAccessStrategy.java

+14-9
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,7 @@ public Sort getMappedSort(Class<?> typeToRead, Sort sort) {
229229
* (non-Javadoc)
230230
* @see org.springframework.data.r2dbc.function.ReactiveDataAccessStrategy#getRowMapper(java.lang.Class)
231231
*/
232+
@SuppressWarnings("unchecked")
232233
@Override
233234
public <T> BiFunction<Row, RowMetadata, T> getRowMapper(Class<T> typeToRead) {
234235
return new EntityRowMapper<T>((RelationalPersistentEntity) getRequiredPersistentEntity(typeToRead),
@@ -262,6 +263,7 @@ private RelationalPersistentEntity<?> getPersistentEntity(Class<?> typeToRead) {
262263
return mappingContext.getPersistentEntity(typeToRead);
263264
}
264265

266+
@SuppressWarnings("unchecked")
265267
private Object getWriteValue(PersistentPropertyAccessor propertyAccessor, RelationalPersistentProperty property) {
266268

267269
TypeInformation<?> type = property.getTypeInformation();
@@ -317,20 +319,23 @@ public BindableOperation insertAndReturnGeneratedKeys(String table, Set<String>
317319
* @see org.springframework.data.r2dbc.function.ReactiveDataAccessStrategy#select(java.lang.String, java.util.Set, org.springframework.data.domain.Sort, org.springframework.data.domain.Pageable)
318320
*/
319321
@Override
320-
public String select(String table, Set<String> columns, Sort sort, Pageable page) {
322+
public String select(String tableName, Set<String> columns, Sort sort, Pageable page) {
321323

322-
Table tableToUse = Table.create(table);
324+
Table table = Table.create(tableName);
323325

324326
Collection<? extends Expression> selectList;
325327

326328
if (columns.isEmpty()) {
327-
selectList = Collections.singletonList(tableToUse.asterisk());
329+
selectList = Collections.singletonList(table.asterisk());
328330
} else {
329-
selectList = tableToUse.columns(columns);
331+
selectList = table.columns(columns);
330332
}
331333

332-
SelectFromAndOrderBy selectBuilder = StatementBuilder.select(selectList).from(table)
333-
.orderBy(createOrderByFields(tableToUse, sort));
334+
SelectFromAndOrderBy selectBuilder = StatementBuilder //
335+
.select(selectList) //
336+
.from(tableName) //
337+
.orderBy(createOrderByFields(table, sort));
338+
334339
OptionalLong limit = OptionalLong.empty();
335340
OptionalLong offset = OptionalLong.empty();
336341

@@ -508,7 +513,7 @@ public void bindId(Statement statement, Object value) {
508513
* @see org.springframework.data.r2dbc.function.BindIdOperation#bindIds(io.r2dbc.spi.Statement, java.lang.Iterable)
509514
*/
510515
@Override
511-
public void bindIds(Statement statement, Iterable<? extends Object> values) {
516+
public void bindIds(Statement statement, Iterable<?> values) {
512517
throw new UnsupportedOperationException();
513518
}
514519

@@ -569,7 +574,7 @@ public void bindId(Statement statement, Object value) {
569574
* @see org.springframework.data.r2dbc.function.BindIdOperation#bindIds(io.r2dbc.spi.Statement, java.lang.Iterable)
570575
*/
571576
@Override
572-
public void bindIds(Statement statement, Iterable<? extends Object> values) {
577+
public void bindIds(Statement statement, Iterable<?> values) {
573578
throw new UnsupportedOperationException();
574579
}
575580

@@ -636,7 +641,7 @@ public void bindId(Statement statement, Object value) {
636641
* @see org.springframework.data.r2dbc.function.BindIdOperation#bindIds(io.r2dbc.spi.Statement, java.lang.Iterable)
637642
*/
638643
@Override
639-
public void bindIds(Statement statement, Iterable<? extends Object> values) {
644+
public void bindIds(Statement statement, Iterable<?> values) {
640645

641646
for (Object value : values) {
642647
bindId(statement, value);

src/main/java/org/springframework/data/r2dbc/repository/support/SimpleR2dbcRepository.java

+23-11
Original file line numberDiff line numberDiff line change
@@ -135,10 +135,13 @@ public Mono<T> findById(ID id) {
135135
BindMarker bindMarker = bindMarkers.next("id");
136136

137137
Table table = Table.create(entity.getTableName());
138-
Select select = StatementBuilder.select(table.columns(columns)).from(table)
139-
.where(Conditions.isEqual(table.column(idColumnName), SQL.bindMarker(bindMarker.getPlaceholder()))).build();
138+
Select select = StatementBuilder //
139+
.select(table.columns(columns)) //
140+
.from(table) //
141+
.where(Conditions.isEqual(table.column(idColumnName), SQL.bindMarker(bindMarker.getPlaceholder()))) //
142+
.build();
140143

141-
return databaseClient.execute().sql(SqlRenderer.render(select)) //
144+
return databaseClient.execute().sql(SqlRenderer.toString(select)) //
142145
.bind(0, id) //
143146
.as(entity.getJavaType()) //
144147
.fetch() //
@@ -167,10 +170,13 @@ public Mono<Boolean> existsById(ID id) {
167170
BindMarker bindMarker = bindMarkers.next("id");
168171

169172
Table table = Table.create(entity.getTableName());
170-
Select select = StatementBuilder.select(table.column(idColumnName)).from(table)
171-
.where(Conditions.isEqual(table.column(idColumnName), SQL.bindMarker(bindMarker.getPlaceholder()))).build();
173+
Select select = StatementBuilder //
174+
.select(table.column(idColumnName)) //
175+
.from(table) //
176+
.where(Conditions.isEqual(table.column(idColumnName), SQL.bindMarker(bindMarker.getPlaceholder()))) //
177+
.build();
172178

173-
return databaseClient.execute().sql(SqlRenderer.render(select)) //
179+
return databaseClient.execute().sql(SqlRenderer.toString(select)) //
174180
.bind(0, id) //
175181
.map((r, md) -> r) //
176182
.first() //
@@ -230,10 +236,13 @@ public Flux<T> findAllById(Publisher<ID> idPublisher) {
230236
}
231237

232238
Table table = Table.create(entity.getTableName());
233-
Select select = StatementBuilder.select(table.columns(columns)).from(table)
234-
.where(Conditions.in(table.column(idColumnName), markers)).build();
239+
Select select = StatementBuilder
240+
.select(table.columns(columns))
241+
.from(table)
242+
.where(Conditions.in(table.column(idColumnName), markers))
243+
.build();
235244

236-
GenericExecuteSpec executeSpec = databaseClient.execute().sql(SqlRenderer.render(select));
245+
GenericExecuteSpec executeSpec = databaseClient.execute().sql(SqlRenderer.toString(select));
237246

238247
for (int i = 0; i < ids.size(); i++) {
239248
executeSpec = executeSpec.bind(i, ids.get(i));
@@ -250,9 +259,12 @@ public Flux<T> findAllById(Publisher<ID> idPublisher) {
250259
public Mono<Long> count() {
251260

252261
Table table = Table.create(entity.getTableName());
253-
Select select = StatementBuilder.select(Functions.count(table.column(getIdColumnName()))).from(table).build();
262+
Select select = StatementBuilder //
263+
.select(Functions.count(table.column(getIdColumnName()))) //
264+
.from(table) //
265+
.build();
254266

255-
return databaseClient.execute().sql(SqlRenderer.render(select)) //
267+
return databaseClient.execute().sql(SqlRenderer.toString(select)) //
256268
.map((r, md) -> r.get(0, Long.class)) //
257269
.first() //
258270
.defaultIfEmpty(0L);

src/main/java/org/springframework/data/r2dbc/support/StatementRenderUtil.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ public abstract class StatementRenderUtil {
4040
*/
4141
public static String render(Select select, OptionalLong limit, OptionalLong offset, Dialect dialect) {
4242

43-
String sql = SqlRenderer.render(select);
43+
String sql = SqlRenderer.toString(select);
4444

4545
// TODO: Replace with proper {@link Dialect} rendering for limit/offset.
4646
if (limit.isPresent()) {

0 commit comments

Comments
 (0)