Update FlywayFactory to support custom schema (#2055)

Signed-off-by: Michael Collado <collado.mike@gmail.com> Co-authored-by: Willy Lulciuc <willy@datakin.com>
MarquezProject · Aug 9, 2022 · 1330831 · 1330831
1 parent 005b418
commit 1330831
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();
   }
 }