com.jeesuite.springboot.starter.mybatis.JeesuiteMybatisConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jeesuite-springboot-starter Show documentation
There is a newer version: 1.4.0
Show newest version
package com.jeesuite.springboot.starter.mybatis;

import java.util.List;
import java.util.Properties;

import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.boot.autoconfigure.MybatisAutoConfiguration;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Configuration;

import com.jeesuite.mybatis.datasource.MutiRouteDataSource;
import com.jeesuite.mybatis.parser.EntityInfo;
import com.jeesuite.mybatis.parser.MybatisMapperParser;
import com.jeesuite.mybatis.plugin.JeesuiteMybatisInterceptor;
import com.jeesuite.mybatis.plugin.PluginConfig;

import tk.mybatis.mapper.entity.Config;
import tk.mybatis.mapper.mapperhelper.MapperHelper;

@Configuration
@EnableConfigurationProperties(MybatisPluginProperties.class)
@ConditionalOnClass(MutiRouteDataSource.class)
@AutoConfigureAfter(MybatisAutoConfiguration.class)
public class JeesuiteMybatisConfiguration implements InitializingBean {

	@Autowired
	private SqlSessionFactory sqlSessionFactory;
	@Autowired
	private MybatisPluginProperties properties;

	@Value("${mybatis.mapper-locations}")
	private String mapperLocations;

	@Override
	public void afterPropertiesSet() throws Exception {
		//

		String interceptorHandlers = null;
		if (properties.isCacheEnabled()) {
			interceptorHandlers = "cache";
		}

		if (properties.isRwRouteEnabled()) {
			interceptorHandlers = interceptorHandlers == null ? "rwRoute" : interceptorHandlers + ",rwRoute";
		}

		if (properties.isDbShardEnabled()) {
			interceptorHandlers = interceptorHandlers == null ? "dbShard" : interceptorHandlers + ",dbShard";
		}
		
		if (properties.isPaginationEnabled()) {
			interceptorHandlers = interceptorHandlers == null ? "page" : interceptorHandlers + ",page";
		}

		if (interceptorHandlers != null) {
			JeesuiteMybatisInterceptor interceptor = new JeesuiteMybatisInterceptor();
			interceptor.setMapperLocations(mapperLocations);
			interceptor.setInterceptorHandlers(interceptorHandlers);
			sqlSessionFactory.getConfiguration().addInterceptor(interceptor);
			
			Properties p = new Properties();
			p.setProperty(PluginConfig.CRUD_DRIVER, properties.getCrudDriver());
			p.setProperty(PluginConfig.DB_TYPE, properties.getDbType());
			p.setProperty(PluginConfig.CACHE_NULL_VALUE, String.valueOf(properties.isNullValueCache()));
			
			interceptor.setProperties(p);
			interceptor.afterPropertiesSet();
		}else{
			MybatisMapperParser.setMapperLocations(mapperLocations);
		}

		//
		MapperHelper mapperHelper = new MapperHelper();
		Config config = new Config();
		config.setNotEmpty(false);
		mapperHelper.setConfig(config);
		List entityInfos = MybatisMapperParser.getEntityInfos();
		for (EntityInfo entityInfo : entityInfos) {
			mapperHelper.registerMapper(entityInfo.getMapperClass());
		}

		mapperHelper.processConfiguration(sqlSessionFactory.getConfiguration());

	}

}