From 3c1748aff13402e069ce24d73da17d4e05d7d9ea Mon Sep 17 00:00:00 2001 From: YunaiV Date: Mon, 24 Nov 2025 11:24:52 +0800 Subject: [PATCH] =?UTF-8?q?!208=20=E4=BD=BF=E7=94=A8=E5=A4=96=E9=83=A8?= =?UTF-8?q?=E4=B8=AD=E9=97=B4=E4=BB=B6=E6=97=B6=EF=BC=8C=E5=A6=82=EF=BC=9A?= =?UTF-8?q?=E9=87=91=E8=9D=B6=E3=80=81=E4=B8=9C=E6=96=B9=E9=80=9A=20?= =?UTF-8?q?=E5=9C=A8=E4=B8=8D=E9=87=8D=E5=90=AF=E6=95=B4=E4=B8=AA=E4=B8=AD?= =?UTF-8?q?=E9=97=B4=E4=BB=B6=E7=9A=84=E6=83=85=E5=86=B5=E4=B8=8B=EF=BC=8C?= =?UTF-8?q?=E4=BA=8C=E6=AC=A1=E9=83=A8=E7=BD=B2=E6=88=96=E5=A4=9A=E4=B8=AA?= =?UTF-8?q?=E6=9C=8D=E5=8A=A1=E5=90=8C=E6=97=B6=E9=83=A8=E7=BD=B2=E5=9C=A8?= =?UTF-8?q?=E4=B8=80=E4=B8=AA=E8=99=9A=E6=8B=9F=E6=9C=BA=E4=B8=8B=EF=BC=88?= =?UTF-8?q?JVM=EF=BC=89=20IdTypeEnvir=E2=80=A6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../config/IdTypeEnvironmentPostProcessor.java | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/yudao-framework/yudao-spring-boot-starter-mybatis/src/main/java/cn/iocoder/yudao/framework/mybatis/config/IdTypeEnvironmentPostProcessor.java b/yudao-framework/yudao-spring-boot-starter-mybatis/src/main/java/cn/iocoder/yudao/framework/mybatis/config/IdTypeEnvironmentPostProcessor.java index 3a67b905f..0ea6bf252 100644 --- a/yudao-framework/yudao-spring-boot-starter-mybatis/src/main/java/cn/iocoder/yudao/framework/mybatis/config/IdTypeEnvironmentPostProcessor.java +++ b/yudao-framework/yudao-spring-boot-starter-mybatis/src/main/java/cn/iocoder/yudao/framework/mybatis/config/IdTypeEnvironmentPostProcessor.java @@ -9,7 +9,10 @@ import lombok.extern.slf4j.Slf4j; import org.springframework.boot.SpringApplication; import org.springframework.boot.env.EnvironmentPostProcessor; import org.springframework.core.env.ConfigurableEnvironment; +import org.springframework.core.env.MapPropertySource; +import java.util.HashMap; +import java.util.Map; import java.util.Set; /** @@ -56,11 +59,19 @@ public class IdTypeEnvironmentPostProcessor implements EnvironmentPostProcessor } public IdType getIdType(ConfigurableEnvironment environment) { - return environment.getProperty(ID_TYPE_KEY, IdType.class); + String value = environment.getProperty(ID_TYPE_KEY); + try { + return StrUtil.isNotBlank(value) ? IdType.valueOf(value) : IdType.NONE; + } catch (IllegalArgumentException ex) { + log.error("[getIdType][无法解析 id-type 配置值({})]", value, ex); + return IdType.NONE; + } } public void setIdType(ConfigurableEnvironment environment, IdType idType) { - environment.getSystemProperties().put(ID_TYPE_KEY, idType); + Map map = new HashMap<>(); + map.put(ID_TYPE_KEY, idType); + environment.getPropertySources().addFirst(new MapPropertySource("mybatisPlusIdType", map)); log.info("[setIdType][修改 MyBatis Plus 的 idType 为({})]", idType); }