|
20 | 20 |
|
21 | 21 | import javax.sql.DataSource; |
22 | 22 |
|
| 23 | +import org.apache.commons.lang3.reflect.MethodUtils; |
23 | 24 | import org.flywaydb.core.Flyway; |
24 | 25 | import org.flywaydb.core.api.configuration.FluentConfiguration; |
25 | 26 |
|
|
29 | 30 |
|
30 | 31 | public final class FlywayPreparer implements DatabasePreparer { |
31 | 32 |
|
32 | | - private final FluentConfiguration flyway; |
| 33 | + private final FluentConfiguration configuration; |
33 | 34 | private final List<String> locations; |
34 | 35 |
|
35 | 36 | public static FlywayPreparer forClasspathLocation(String... locations) { |
36 | | - FluentConfiguration f = Flyway.configure() |
37 | | - .locations(locations); |
38 | | - return new FlywayPreparer(f, Arrays.asList(locations)); |
| 37 | + FluentConfiguration config = Flyway.configure().locations(locations); |
| 38 | + return new FlywayPreparer(config, Arrays.asList(locations)); |
39 | 39 | } |
40 | 40 |
|
41 | | - private FlywayPreparer(FluentConfiguration flyway, List<String> locations) { |
42 | | - this.flyway = flyway; |
| 41 | + private FlywayPreparer(FluentConfiguration configuration, List<String> locations) { |
| 42 | + this.configuration = configuration; |
43 | 43 | this.locations = locations; |
44 | 44 | } |
45 | 45 |
|
46 | 46 | @Override |
47 | 47 | public void prepare(DataSource ds) throws SQLException { |
48 | | - flyway.dataSource(ds); |
49 | | - flyway.load().migrate(); |
| 48 | + configuration.dataSource(ds); |
| 49 | + Flyway flyway = configuration.load(); |
| 50 | + try { |
| 51 | + MethodUtils.invokeMethod(flyway, "migrate"); |
| 52 | + } catch (Exception e) { |
| 53 | + throw new RuntimeException(e); |
| 54 | + } |
50 | 55 | } |
51 | 56 |
|
52 | 57 | @Override |
|
0 commit comments