Make ddlAuto configurable (#587)

openraven · Mar 11, 2024 · 0a720d1 · 0a720d1
1 parent ab5a869
commit 0a720d1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/magpie-persist/src/main/java/io/openraven/magpie/plugins/persist/PersistConfig.java b/magpie-persist/src/main/java/io/openraven/magpie/plugins/persist/PersistConfig.java
@@ -25,6 +25,7 @@ public class PersistConfig {
   private String user;
   private String password;
   private boolean migrateDB = true;
+  private String ddlAuto;
 
 
   public String getSchema() {
@@ -67,4 +68,12 @@ public boolean shouldMigrateDB() {
     public void setMigrateDB(boolean migrateDB) {
         this.migrateDB = migrateDB;
     }
+
+  public String getDdlAuto() {
+    return this.ddlAuto;
+  }
+
+  public void setDdlAuto(String ddlAuto) {
+    this.ddlAuto = ddlAuto;
+  }
 }
diff --git a/...src/main/java/io/openraven/magpie/plugins/persist/config/PostgresPersistenceProvider.java b/...src/main/java/io/openraven/magpie/plugins/persist/config/PostgresPersistenceProvider.java
@@ -23,9 +23,9 @@
 import org.hibernate.cfg.Configuration;
 import org.hibernate.cfg.Environment;
 import org.hibernate.service.ServiceRegistry;
-import org.slf4j.LoggerFactory;
 
 import javax.persistence.EntityManager;
+import java.util.Optional;
 import java.util.Properties;
 
 import static io.openraven.magpie.data.utils.EntityTypeResolver.getSubClasses;
@@ -43,7 +43,7 @@ public static EntityManager getEntityManager(PersistConfig config) {
     settings.put(Environment.PASS, config.getPassword());
     settings.put(Environment.DIALECT, "io.openraven.magpie.plugins.persist.config.PostgreSQL10StringDialect");
     settings.put(Environment.SHOW_SQL, "false");
-    settings.put(Environment.HBM2DDL_AUTO, "validate");
+    settings.put(Environment.HBM2DDL_AUTO, Optional.ofNullable(config.getDdlAuto()).orElse("validate"));
     settings.put(Environment.DEFAULT_SCHEMA, config.getSchema());
 
     Configuration configuration = new Configuration();