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);