diff --git a/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/config/MybatisFlexConfig.java b/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/config/MybatisFlexConfig.java index f8161146..2ea4d5e4 100644 --- a/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/config/MybatisFlexConfig.java +++ b/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/config/MybatisFlexConfig.java @@ -50,6 +50,8 @@ public class MybatisFlexConfig { protected final Properties properties = new Properties(); public MybatisFlexConfig(Filer filer) { + // 先从 resources 目录读取 mybatis-flex.config 配置文件 + loadConfigFromResource(); try { //target/classes/ FileObject resource = filer.createResource(StandardLocation.CLASS_OUTPUT, "", "mybatis-flex"); @@ -97,6 +99,21 @@ public class MybatisFlexConfig { } } + private void loadConfigFromResource() { + Properties resourceProp = new Properties(); + try (InputStream resourceStream = MybatisFlexConfig.class.getClassLoader().getResourceAsStream(APT_FILE_NAME)) { + if (resourceStream == null) { + return; + } + resourceProp.load(resourceStream); + if (resourceProp != null && !resourceProp.isEmpty()) { + properties.putAll(resourceProp); + } + } catch (Throwable e) { + e.printStackTrace(); + } + } + public String get(ConfigurationKey key) { return properties.getProperty(key.getConfigKey(), key.getDefaultValue()); }