67
67
import org .springframework .boot .flyway .autoconfigure .FlywayAutoConfiguration .PostgresqlFlywayConfigurationCustomizer ;
68
68
import org .springframework .boot .flyway .autoconfigure .FlywayAutoConfiguration .SqlServerFlywayConfigurationCustomizer ;
69
69
import org .springframework .boot .jdbc .DataSourceBuilder ;
70
- import org .springframework .boot .jdbc .EmbeddedDatabaseConnection ;
71
70
import org .springframework .boot .jdbc .SchemaManagement ;
72
71
import org .springframework .boot .jdbc .autoconfigure .DataSourceProperties ;
73
72
import org .springframework .boot .jdbc .autoconfigure .EmbeddedDataSourceConfiguration ;
74
73
import org .springframework .boot .jdbc .autoconfigure .JdbcConnectionDetails ;
75
- import org .springframework .boot .jpa .EntityManagerFactoryBuilder ;
76
74
import org .springframework .boot .test .context .FilteredClassLoader ;
77
75
import org .springframework .boot .test .context .assertj .AssertableApplicationContext ;
78
76
import org .springframework .boot .test .context .runner .ApplicationContextRunner ;
@@ -1065,10 +1063,11 @@ FlywayMigrationInitializer customFlywayMigrationInitializer(Flyway flyway) {
1065
1063
1066
1064
@ Bean
1067
1065
LocalContainerEntityManagerFactoryBean entityManagerFactoryBean (DataSource dataSource ) {
1068
- return new EntityManagerFactoryBuilder (new HibernateJpaVendorAdapter (), (ds ) -> configureJpaProperties (),
1069
- null )
1070
- .dataSource (dataSource )
1071
- .build ();
1066
+ LocalContainerEntityManagerFactoryBean localContainerEntityManagerFactoryBean = new LocalContainerEntityManagerFactoryBean ();
1067
+ localContainerEntityManagerFactoryBean .setDataSource (dataSource );
1068
+ localContainerEntityManagerFactoryBean .setJpaVendorAdapter (new HibernateJpaVendorAdapter ());
1069
+ localContainerEntityManagerFactoryBean .setJpaPropertyMap (configureJpaProperties ());
1070
+ return localContainerEntityManagerFactoryBean ;
1072
1071
}
1073
1072
1074
1073
}
@@ -1089,50 +1088,11 @@ Flyway customFlyway() {
1089
1088
1090
1089
@ Bean
1091
1090
LocalContainerEntityManagerFactoryBean entityManagerFactoryBean () {
1092
- return new EntityManagerFactoryBuilder (new HibernateJpaVendorAdapter (),
1093
- (datasource ) -> configureJpaProperties (), null )
1094
- .dataSource (this .dataSource )
1095
- .build ();
1096
- }
1097
-
1098
- }
1099
-
1100
- @ Configuration (proxyBeanMethods = false )
1101
- static class JpaWithMultipleDataSourcesConfiguration {
1102
-
1103
- @ Bean
1104
- @ Primary
1105
- DataSource normalDataSource () {
1106
- return new EmbeddedDatabaseBuilder ().setType (EmbeddedDatabaseConnection .HSQLDB .getType ())
1107
- .generateUniqueName (true )
1108
- .build ();
1109
- }
1110
-
1111
- @ Bean
1112
- @ Primary
1113
- LocalContainerEntityManagerFactoryBean normalEntityManagerFactory (EntityManagerFactoryBuilder builder ,
1114
- DataSource normalDataSource ) {
1115
- Map <String , Object > properties = new HashMap <>();
1116
- properties .put ("configured" , "normal" );
1117
- properties .put ("hibernate.transaction.jta.platform" , NoJtaPlatform .INSTANCE );
1118
- return builder .dataSource (normalDataSource ).properties (properties ).build ();
1119
- }
1120
-
1121
- @ Bean
1122
- @ FlywayDataSource
1123
- DataSource flywayDataSource () {
1124
- return new EmbeddedDatabaseBuilder ().setType (EmbeddedDatabaseConnection .HSQLDB .getType ())
1125
- .generateUniqueName (true )
1126
- .build ();
1127
- }
1128
-
1129
- @ Bean
1130
- LocalContainerEntityManagerFactoryBean flywayEntityManagerFactory (EntityManagerFactoryBuilder builder ,
1131
- @ FlywayDataSource DataSource flywayDataSource ) {
1132
- Map <String , Object > properties = new HashMap <>();
1133
- properties .put ("configured" , "flyway" );
1134
- properties .put ("hibernate.transaction.jta.platform" , NoJtaPlatform .INSTANCE );
1135
- return builder .dataSource (flywayDataSource ).properties (properties ).build ();
1091
+ LocalContainerEntityManagerFactoryBean localContainerEntityManagerFactoryBean = new LocalContainerEntityManagerFactoryBean ();
1092
+ localContainerEntityManagerFactoryBean .setDataSource (this .dataSource );
1093
+ localContainerEntityManagerFactoryBean .setJpaVendorAdapter (new HibernateJpaVendorAdapter ());
1094
+ localContainerEntityManagerFactoryBean .setJpaPropertyMap (configureJpaProperties ());
1095
+ return localContainerEntityManagerFactoryBean ;
1136
1096
}
1137
1097
1138
1098
}
0 commit comments