支持配置默认导入模块

ssssssss-team · Jul 26, 2020 · d4d11ce · d4d11ce
1 parent 4c71e61
commit d4d11ce
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/main/java/org/ssssssss/magicapi/functions/DatabaseQuery.java b/src/main/java/org/ssssssss/magicapi/functions/DatabaseQuery.java
@@ -5,6 +5,7 @@
 import org.ssssssss.magicapi.cache.SqlCache;
 import org.ssssssss.magicapi.config.DynamicDataSource;
 import org.ssssssss.magicapi.config.DynamicDataSource.DataSourceNode;
+import org.ssssssss.magicapi.config.MagicModule;
 import org.ssssssss.magicapi.dialect.Dialect;
 import org.ssssssss.magicapi.dialect.DialectUtils;
 import org.ssssssss.magicapi.exception.MagicAPIException;
@@ -24,7 +25,7 @@
 import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.function.Function;
 
-public class DatabaseQuery extends HashMap<String, DatabaseQuery> {
+public class DatabaseQuery extends HashMap<String, DatabaseQuery> implements MagicModule {
 
 	@UnableCall
 	private DynamicDataSource dynamicDataSource;
@@ -287,6 +288,12 @@ public Object selectValue(String sql) {
 
 	private static GenericTokenParser ifParamTokenParser = new GenericTokenParser("?{", ",", true);
 
+	@UnableCall
+	@Override
+	public String getModuleName() {
+		return "db";
+	}
+
 	public static class BoundSql {
 		private String sql;
 		private List<Object> parameters = new ArrayList<>();