From fe42deb18c51c3bf61bba1bdc6f21d1882fb0ecf Mon Sep 17 00:00:00 2001 From: ruansheng Date: Mon, 3 Jun 2024 16:43:02 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20APT=20=E6=94=AF=E6=8C=81=E4=BB=8E=20Res?= =?UTF-8?q?ources=20=E7=9B=AE=E5=BD=95=E8=BF=9B=E8=A1=8C=E9=85=8D=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../processor/config/MybatisFlexConfig.java | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) 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()); }