Skip to content

Commit c36c615

Browse files
committed
spring-projects#433 - Polishing.
Replace bean declarations with AbstractR2dbcConfiguration class.
1 parent caedc4c commit c36c615

File tree

1 file changed

+9
-31
lines changed

1 file changed

+9
-31
lines changed

r2dbc/example/src/test/java/example/springdata/r2dbc/basics/InfrastructureConfiguration.java

Lines changed: 9 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -17,50 +17,28 @@
1717

1818
import io.r2dbc.postgresql.PostgresqlConnectionConfiguration;
1919
import io.r2dbc.postgresql.PostgresqlConnectionFactory;
20-
import io.r2dbc.spi.ConnectionFactory;
20+
21+
import javax.annotation.PreDestroy;
2122

2223
import org.springframework.context.annotation.Bean;
2324
import org.springframework.context.annotation.Configuration;
24-
import org.springframework.data.r2dbc.function.DatabaseClient;
25-
import org.springframework.data.r2dbc.repository.support.R2dbcRepositoryFactory;
26-
import org.springframework.data.relational.core.mapping.RelationalMappingContext;
25+
import org.springframework.data.r2dbc.repository.config.AbstractR2dbcConfiguration;
26+
import org.springframework.data.r2dbc.repository.config.EnableR2dbcRepositories;
2727
import org.testcontainers.containers.PostgreSQLContainer;
2828

29-
import javax.annotation.PreDestroy;
30-
3129
/**
3230
* @author Oliver Gierke
31+
* @author Mark Paluch
3332
*/
3433
@Configuration
35-
class InfrastructureConfiguration {
34+
@EnableR2dbcRepositories
35+
class InfrastructureConfiguration extends AbstractR2dbcConfiguration {
3636

3737
private PostgreSQLContainer postgres = new PostgreSQLContainer();
3838

3939
@Bean
40-
CustomerRepository customerRepository(R2dbcRepositoryFactory factory) {
41-
return factory.getRepository(CustomerRepository.class);
42-
}
43-
44-
@Bean
45-
R2dbcRepositoryFactory repositoryFactory(DatabaseClient client) {
46-
47-
RelationalMappingContext context = new RelationalMappingContext();
48-
context.afterPropertiesSet();
49-
50-
return new R2dbcRepositoryFactory(client, context);
51-
}
52-
53-
@Bean
54-
DatabaseClient databaseClient(ConnectionFactory factory) {
55-
56-
return DatabaseClient.builder() //
57-
.connectionFactory(factory) //
58-
.build();
59-
}
60-
61-
@Bean
62-
PostgresqlConnectionFactory connectionFactory() {
63-
40+
@Override
41+
public PostgresqlConnectionFactory connectionFactory() {
6442

6543
postgres.start();
6644

0 commit comments

Comments
 (0)