Update FlywayFactory to support custom schema

Signed-off-by: Michael Collado <collado.mike@gmail.com>

api/src/main/java/marquez/db/FlywayFactory.java

@@ -10,12 +10,14 @@
 import java.nio.charset.StandardCharsets;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import javax.annotation.Nullable;
 import javax.sql.DataSource;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
 import lombok.NonNull;
 import lombok.Setter;
+import org.apache.commons.lang3.StringUtils;
 import org.flywaydb.core.Flyway;
 
 @NoArgsConstructor
@@ -74,6 +76,15 @@ public final class FlywayFactory {
   @Getter @Setter
   private String repeatableSqlMigrationPrefix = DEFAULT_REPEATABLE_SQL_MIGRATION_PREFIX;
 
+  @Getter private Optional<String> schema = Optional.empty();
+
+  public void setSchema(@NonNull String schema) {
+    if (StringUtils.isEmpty(schema)) {
+      throw new IllegalArgumentException("schema must not be blank");
+    }
+    this.schema = Optional.of(schema);
+  }
+
   public Flyway build(@NonNull DataSource source) {
     return Flyway.configure()
         .dataSource(source)
@@ -102,6 +113,7 @@ public Flyway build(@NonNull DataSource source) {
         .placeholderSuffix(placeholderSuffix)
         .sqlMigrationPrefix(sqlMigrationPrefix)
         .repeatableSqlMigrationPrefix(repeatableSqlMigrationPrefix)
+        .schemas(schema.map(s -> new String[] {s}).orElse(new String[] {}))
         .load();
   }
 }