|
@@ -20,6 +20,7 @@ import com.baomidou.mybatisplus.core.MybatisConfiguration;
|
|
|
import com.baomidou.mybatisplus.core.config.GlobalConfig;
|
|
|
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
|
|
|
import com.baomidou.mybatisplus.core.incrementer.IKeyGenerator;
|
|
|
+import com.baomidou.mybatisplus.core.incrementer.IdGenerator;
|
|
|
import com.baomidou.mybatisplus.core.injector.ISqlInjector;
|
|
|
import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
|
|
|
import org.apache.ibatis.annotations.Mapper;
|
|
@@ -217,6 +218,11 @@ public class MybatisPlusAutoConfiguration implements InitializingBean {
|
|
|
ISqlInjector iSqlInjector = this.applicationContext.getBean(ISqlInjector.class);
|
|
|
globalConfig.setSqlInjector(iSqlInjector);
|
|
|
}
|
|
|
+ if (this.applicationContext.getBeanNamesForType(IdGenerator.class, false,
|
|
|
+ false).length > 0) {
|
|
|
+ IdGenerator idGenerator = this.applicationContext.getBean(IdGenerator.class);
|
|
|
+ globalConfig.setIdGenerator(idGenerator);
|
|
|
+ }
|
|
|
// TODO 设置 GlobalConfig 到 MybatisSqlSessionFactoryBean
|
|
|
factory.setGlobalConfig(globalConfig);
|
|
|
return factory.getObject();
|