Update FlywayFactory to support custom schema

Signed-off-by: Michael Collado <collado.mike@gmail.com>
MarquezProject · Aug 8, 2022 · 1379bb3 · 1379bb3
1 parent 90feb2e
commit 1379bb3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/src/main/java/marquez/db/FlywayFactory.java b/api/src/main/java/marquez/db/FlywayFactory.java
@@ -74,6 +74,8 @@ public final class FlywayFactory {
   @Getter @Setter
   private String repeatableSqlMigrationPrefix = DEFAULT_REPEATABLE_SQL_MIGRATION_PREFIX;
 
+  @Getter @Setter private String schema;
+
   public Flyway build(@NonNull DataSource source) {
     return Flyway.configure()
         .dataSource(source)
@@ -102,6 +104,7 @@ public Flyway build(@NonNull DataSource source) {
         .placeholderSuffix(placeholderSuffix)
         .sqlMigrationPrefix(sqlMigrationPrefix)
         .repeatableSqlMigrationPrefix(repeatableSqlMigrationPrefix)
+        .defaultSchema(schema)
         .load();
   }
 }