diff --git a/src/main/java/org/springframework/data/r2dbc/repository/support/SimpleR2dbcRepository.java b/src/main/java/org/springframework/data/r2dbc/repository/support/SimpleR2dbcRepository.java index b3f05fdfe..edf61603f 100644 --- a/src/main/java/org/springframework/data/r2dbc/repository/support/SimpleR2dbcRepository.java +++ b/src/main/java/org/springframework/data/r2dbc/repository/support/SimpleR2dbcRepository.java @@ -30,6 +30,7 @@ import org.springframework.data.r2dbc.core.PreparedOperation; import org.springframework.data.r2dbc.core.ReactiveDataAccessStrategy; import org.springframework.data.r2dbc.core.StatementMapper; import org.springframework.data.r2dbc.query.Criteria; +import org.springframework.data.relational.core.mapping.RelationalPersistentProperty; import org.springframework.data.relational.core.sql.Functions; import org.springframework.data.relational.core.sql.Select; import org.springframework.data.relational.core.sql.StatementBuilder; @@ -129,12 +130,12 @@ public class SimpleR2dbcRepository implements ReactiveCrudRepository columns = this.accessStrategy.getAllColumns(this.entity.getJavaType()); - String idColumnName = getIdColumnName(); + String idProperty = getIdProperty().getName(); StatementMapper mapper = this.accessStrategy.getStatementMapper().forType(this.entity.getJavaType()); StatementMapper.SelectSpec selectSpec = mapper.createSelect(this.entity.getTableName()) // .withProjection(columns) // - .withCriteria(Criteria.where(idColumnName).is(id)); + .withCriteria(Criteria.where(idProperty).is(id)); PreparedOperation operation = mapper.getMappedObject(selectSpec); @@ -160,12 +161,12 @@ public class SimpleR2dbcRepository implements ReactiveCrudRepository operation = mapper.getMappedObject(selectSpec); @@ -217,12 +218,12 @@ public class SimpleR2dbcRepository implements ReactiveCrudRepository columns = this.accessStrategy.getAllColumns(this.entity.getJavaType()); - String idColumnName = getIdColumnName(); + String idProperty = getIdProperty().getName(); StatementMapper mapper = this.accessStrategy.getStatementMapper().forType(this.entity.getJavaType()); StatementMapper.SelectSpec selectSpec = mapper.createSelect(this.entity.getTableName()) // .withProjection(columns) // - .withCriteria(Criteria.where(idColumnName).in(ids)); + .withCriteria(Criteria.where(idProperty).in(ids)); PreparedOperation operation = mapper.getMappedObject(selectSpec); @@ -238,7 +239,7 @@ public class SimpleR2dbcRepository implements ReactiveCrudRepository implements ReactiveCrudRepository implements ReactiveCrudRepository implements ReactiveCrudRepository