From b9b8574b3487c39a257c9356f16798859eb47176 Mon Sep 17 00:00:00 2001 From: caoli5288 Date: Thu, 7 Jan 2021 10:48:24 +0800 Subject: [PATCH] improve IHandlerInitializer.java --- src/main/java/com/mengcraft/simpleorm/EbeanManager.java | 7 ++++++- .../mengcraft/simpleorm/provider/IHandlerInitializer.java | 4 ++++ 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/mengcraft/simpleorm/EbeanManager.java b/src/main/java/com/mengcraft/simpleorm/EbeanManager.java index f00eabd..f773469 100644 --- 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 index b4cca32..25d385d 100644 --- 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; + } }