improve IHandlerInitializer.java

caoli5288 · Jan 7, 2021 · b9b8574 · b9b8574
1 parent fb0a0f3
commit b9b8574
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/main/java/com/mengcraft/simpleorm/EbeanManager.java b/src/main/java/com/mengcraft/simpleorm/EbeanManager.java
@@ -57,7 +57,7 @@ public EbeanHandler getHandler(String name) {
     }
 
     private EbeanHandler build(JavaPlugin plugin, boolean shared, IHandlerInitializer initializer) {
-        if (shared || plugin.getConfig().getBoolean("dataSource.disabled", false)) {
+        if (shared || !initializer.isEnabled(plugin)) {
             return new EbeanHandler(plugin, true, ORM.getSharedDs());
         }
 
@@ -130,5 +130,10 @@ public void initialize(Plugin plugin, EbeanHandler handler) {
             handler.setUser(user);
             handler.setPassword(password);
         }
+
+        @Override
+        public boolean isEnabled(Plugin plugin) {
+            return !plugin.getConfig().getBoolean("dataSource.disabled", false);
+        }
     }
 }
diff --git a/src/main/java/com/mengcraft/simpleorm/provider/IHandlerInitializer.java b/src/main/java/com/mengcraft/simpleorm/provider/IHandlerInitializer.java
@@ -6,4 +6,8 @@
 public interface IHandlerInitializer {
 
     void initialize(Plugin plugin, EbeanHandler handler);
+
+    default boolean isEnabled(Plugin plugin) {
+        return true;
+    }
 }