diff --git a/src/main/java/com/ai/da/common/config/PrimaryDataSourceConfig.java b/src/main/java/com/ai/da/common/config/PrimaryDataSourceConfig.java index c12182db..44a4540b 100644 --- a/src/main/java/com/ai/da/common/config/PrimaryDataSourceConfig.java +++ b/src/main/java/com/ai/da/common/config/PrimaryDataSourceConfig.java @@ -1,11 +1,14 @@ package com.ai.da.common.config; +import com.baomidou.mybatisplus.autoconfigure.MybatisPlusProperties; +import com.baomidou.mybatisplus.core.config.GlobalConfig; import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor; import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean; import org.apache.ibatis.session.SqlSessionFactory; import org.mybatis.spring.SqlSessionFactoryBean; import org.mybatis.spring.SqlSessionTemplate; import org.mybatis.spring.annotation.MapperScan; +import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Qualifier; import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.boot.jdbc.DataSourceBuilder; @@ -19,6 +22,8 @@ import javax.sql.DataSource; @Configuration @MapperScan(basePackages = "com.ai.da.mapper.primary", sqlSessionFactoryRef = "primarySqlSessionFactory") public class PrimaryDataSourceConfig { + @Autowired + private MybatisPlusProperties mybatisPlusProperties; @Primary @Bean(name = "primaryDataSource") @@ -36,6 +41,11 @@ public class PrimaryDataSourceConfig { MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean(); bean.setDataSource(dataSource); bean.setMapperLocations(applicationContext.getResources("classpath:mapper/primary/*.xml")); + // 设置 MyBatis Plus 全局配置 + GlobalConfig globalConfig = this.mybatisPlusProperties.getGlobalConfig(); + if (globalConfig != null) { + bean.setGlobalConfig(globalConfig); + } bean.setPlugins(mybatisPlusInterceptor); return bean.getObject(); }