From abb6f6d9a7d38ed7b74fb68b9f944b2185ebd750 Mon Sep 17 00:00:00 2001
From: XiaoLin <>
Date: Fri, 23 Jun 2023 22:16:26 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=20mybatis-flex-processor=20=E8=AF=BB?=
 =?UTF-8?q?=E5=8F=96=20mybatis-flex.properties=20=E6=B7=BB=E5=8A=A0?=
 =?UTF-8?q?=E5=85=9C=E5=BA=95=E7=AD=96=E7=95=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../processor/config/MybatisFlexConfig.java           | 11 +++++++++++
 1 file changed, 11 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 c3d668f3..ea5e92e8 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
@@ -16,6 +16,8 @@
 
 package com.mybatisflex.processor.config;
 
+import com.mybatisflex.processor.util.FileUtil;
+
 import javax.annotation.processing.Filer;
 import javax.tools.FileObject;
 import javax.tools.StandardLocation;
@@ -67,6 +69,15 @@ public class MybatisFlexConfig {
                 inputStream = Files.newInputStream(propertiesFile.toPath());
             }
 
+            // 兜底，如果还是没找到，就找项目根目录下的 mybatis-flex.properties
+            if (inputStream == null) {
+                final String projectRootPath = FileUtil.getProjectRootPath(propertiesFileObject.toUri().getPath());
+                final File filePath = new File(projectRootPath, MYBATIS_FLEX);
+                if (filePath.exists()) {
+                    inputStream = Files.newInputStream(filePath.toPath());
+                }
+            }
+
             if (inputStream != null) {
                 try (InputStreamReader reader = new InputStreamReader(inputStream, StandardCharsets.UTF_8)) {
                     properties.load(reader);