瀏覽代碼

add tableExists method, use it

tags/2.0.1
Jonathan Cobb 4 年之前
父節點
當前提交
9c6caddb0e
共有 2 個文件被更改,包括 11 次插入3 次删除
  1. +9
    -0
      wizard-server/src/main/java/org/cobbzilla/wizard/server/config/PgRestServerConfiguration.java
  2. +2
    -3
      wizard-server/src/main/java/org/cobbzilla/wizard/server/listener/FlywayMigrationListener.java

+ 9
- 0
wizard-server/src/main/java/org/cobbzilla/wizard/server/config/PgRestServerConfiguration.java 查看文件

@@ -417,4 +417,13 @@ public class PgRestServerConfiguration extends RestServerConfiguration implement
.generateConstraintSql(includeIndexes).toArray(new String[0]);
}

public boolean tableExists(String tableName) {
try {
execSql("select count(*) from "+tableName);
return true;
} catch (Exception e) {
return false;
}
}

}

+ 2
- 3
wizard-server/src/main/java/org/cobbzilla/wizard/server/listener/FlywayMigrationListener.java 查看文件

@@ -33,10 +33,9 @@ public class FlywayMigrationListener<C extends RestServerConfiguration> extends
final PgRestServerConfiguration configuration = (PgRestServerConfiguration) server.getConfiguration();
if (configuration.getDatabase().isMigrationEnabled()) {
// does the flyway table exist? If not, this is a first-time setup
try {
configuration.execSql("select count(*) from "+getFlywayTableName());
if (configuration.tableExists(getFlywayTableName())) {
migrate(configuration);
} catch (Exception e) {
} else {
log.warn("beforeStart: database schema does not seem to exist, will create it");
configuration.getDatabase().getHibernate().setHbm2ddlAuto("create");
}


Loading…
取消
儲存