只是服务员公司内部架构所开发,只是服务于公司项目的架构,无法通用全部项目,若要通用,则还需要进行通用化改造,本文只是提供一个思路
package com.iskytrip.mybatis.inject.mybatis.dataSource;
import com.iskytrip.mybatis.inject.mybatis.dataSource.config.ISkyTripDataSourceConfiguration;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
@Configuration
@Import(ISkyTripDataSourceConfiguration.class)
public class DataSourceAutoConfig {
}
package com.iskytrip.mybatis.inject.mybatis.dataSource.config;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.annotation.TableName;
import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
import com.ctrip.framework.apollo.internals.DefaultConfig;
import com.ctrip.framework.apollo.spring.config.ConfigPropertySource;
import com.ctrip.framework.apollo.spring.config.ConfigPropertySourceFactory;
import com.ctrip.framework.apollo.spring.util.SpringInjector;
import com.iskytrip.framework.util.db.DBConfigEntity;
import com.iskytrip.framework.util.exception.BizServiceException;
import com.iskytrip.mybatis.inject.util.CodeUtil;
import com.iskytrip.mybatis.inject.util.PropertyUtil;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.mapper.MapperScannerConfigurer;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.context.annotation.ImportBeanDefinitionRegistrar;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.core.type.AnnotationMetadata;
import org.springframework.core.type.classreading.CachingMetadataReaderFactory;
import org.springframework.core.type.classreading.MetadataReader;
import org.springframework.core.type.classreading.MetadataReaderFactory;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.stereotype.Repository;
import org.springframework.util.ClassUtils;
import org.springframework.util.StringUtils;
import javax.sql.DataSource;
import java.util.*;
import java.util.stream.Collectors;
public class ISkyTripDataSourceConfiguration implements ImportBeanDefinitionRegistrar {
@Override
public void registerBeanDefinitions(AnnotationMetadata annotationMetadata, BeanDefinitionRegistry beanFactory) {
String dbConfig = PropertyUtil.getValue("db.config");
if (StrUtil.isEmpty(dbConfig)) {
return;
}
List<ConfigPropertySource> configPropertySources = SpringInjector.getInstance(ConfigPropertySourceFactory.class).getAllConfigPropertySources();
if (CollUtil.isEmpty(configPropertySources)) {
return;
}
Collections.sort(configPropertySources, (o1, o2) -> o1.getName().compareToIgnoreCase(o2.getName()));
Map<String, String> valueMap = new HashMap<>();
for (ConfigPropertySource configPropertySource : configPropertySources) {
DefaultConfig defaultConfig = (DefaultConfig) configPropertySource.getSource();
for (String name : defaultConfig.getPropertyNames()) {
String value;
if (!valueMap.containsKey(name) && StrUtil.isNotEmpty(value = defaultConfig.getProperty(name, ""))) {
valueMap.put(name, value);
}
}
}
String[] strArr = dbConfig.split(",");
for (String str : strArr) {
String[] strArr1 = str.split("=");
registerDataSourceBean(beanFactory, valueMap.get(strArr1[0]), strArr1[1]);
}
/*PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
paginationInterceptor.setLimit(-1);
paginationInterceptor.setCountSqlParser(new JsqlParserCountOptimize(true));
beanFactory.registerBeanDefinition("paginationInterceptor", BeanDefinitionBuilder.genericBeanDefinition(paginationInterceptor.getClass()).getBeanDefinition());*/
}
private void registerDataSourceBean(BeanDefinitionRegistry beanFactory, String value0, String value1) {
try {
DataSource dataSource = DbBaseConfig.getDbDataSource(JSON.parseObject(value0, DBConfigEntity.class));
beanFactory.registerBeanDefinition(value1 + "#dataSource", BeanDefinitionBuilder.genericBeanDefinition(dataSource.getClass()).getBeanDefinition());
MybatisSqlSessionFactoryBean mybatisSqlSessionFactoryBean = DbBaseConfig.getMybatisSqlSessionFactoryBean(dataSource, getEntityPath(value1), getMapperPath(value1));
SqlSessionFactory sqlSessionFactory = mybatisSqlSessionFactoryBean.getObject();
registerMybatisSqlSessionFactoryBean(value1, beanFactory, mybatisSqlSessionFactoryBean, getEntityPath(value1), getMapperPath(value1), sqlSessionFactory, dataSource);
beanFactory.registerBeanDefinition(value1 + "#dataSourceTransactionManager", BeanDefinitionBuilder.genericBeanDefinition(new DataSourceTransactionManager(dataSource).getClass()).addPropertyValue("dataSource", dataSource).getBeanDefinition());
beanFactory.registerBeanDefinition(value1 + "#sqlSessionTemplate", BeanDefinitionBuilder.genericBeanDefinition(SqlSessionTemplate.class).addConstructorArgValue(sqlSessionFactory).getBeanDefinition());
List<String> basePackages = new ArrayList();
basePackages.addAll(Arrays.stream(new String[]{getDaoPath(value1)}).filter(StringUtils::hasText).collect(Collectors.toList()));
beanFactory.registerBeanDefinition(value1 + "#mybatis", BeanDefinitionBuilder.genericBeanDefinition(MapperScannerConfigurer.class).addPropertyValue("sqlSessionTemplateBeanName", value1 + "#sqlSessionTemplate").addPropertyValue("basePackage", StringUtils.collectionToCommaDelimitedString(basePackages)).getBeanDefinition());
} catch (Exception e) {
throw new BizServiceException(e);
}
}
private String getEntityPath(String value) {
return getRealPath("entity", value);
}
private String getDaoPath(String value) {
return getRealPath("mapper", value);
}
private String getMapperPath(String value) {
return "classpath*:mapper/" + CodeUtil.underscoreName(value) + "/*.xml";
}
private String getRealPath(String type, String value) {
try {
ResourcePatternResolver resourcePatternResolver = new PathMatchingResourcePatternResolver();
String pattern = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX +
ClassUtils.convertClassNameToResourcePath("com.iskytrip.**.repository." + type + "." + value) + "/**/*.class";
Resource[] resources = resourcePatternResolver.getResources(pattern);
//MetadataReader 的工厂类
MetadataReaderFactory readerFactory = new CachingMetadataReaderFactory(resourcePatternResolver);
for (Resource resource : resources) {
//用于读取类信息
MetadataReader reader = readerFactory.getMetadataReader(resource);
//扫描到的class
Class<?> clazz = Class.forName(reader.getClassMetadata().getClassName());
//判断是否有指定主解
//判断是否有指定主解
if (clazz.getAnnotation(TableName.class) != null || clazz.getAnnotation(Repository.class) != null) {
return clazz.getName().substring(0, clazz.getName().lastIndexOf("."+value+"."))+"."+value;
}
}
} catch (Exception e) {
return StrUtil.EMPTY;
}
return StrUtil.EMPTY;
}
private void registerMybatisSqlSessionFactoryBean(String value, BeanDefinitionRegistry beanFactory, MybatisSqlSessionFactoryBean mybatisSqlSessionFactoryBean, String typeAliasesPackage, String mapperLocations, SqlSessionFactory sqlSessionFactory, DataSource dataSource) throws Exception {
beanFactory.registerBeanDefinition(value + "#mybatisSqlSessionFactoryBean", BeanDefinitionBuilder.genericBeanDefinition(mybatisSqlSessionFactoryBean.getClass()).
addPropertyValue("dataSource", dataSource).
addPropertyValue("configuration", sqlSessionFactory.getConfiguration()).
addPropertyValue("sqlSessionFactory", sqlSessionFactory).
addPropertyValue("vfs", mybatisSqlSessionFactoryBean.getVfs()).
addPropertyValue("typeAliasesPackage", typeAliasesPackage).
addPropertyValue("mapperLocations", new PathMatchingResourcePatternResolver().getResources(mapperLocations))
.getBeanDefinition());
}
}
package com.iskytrip.mybatis.inject.mybatis.dataSource.config;
import com.baomidou.mybatisplus.autoconfigure.SpringBootVFS;
import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.core.MybatisXMLLanguageDriver;
import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
import com.iskytrip.framework.util.db.DBConfigEntity;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import javax.sql.DataSource;
import java.io.IOException;
public class DbBaseConfig {
/**
*
* @param dataSource
* @param typeAliasesPackage
* @param mapperLocations
* @return
*/
public static MybatisSqlSessionFactoryBean getMybatisSqlSessionFactoryBean(DataSource dataSource, String typeAliasesPackage, String mapperLocations) throws IOException {
MybatisSqlSessionFactoryBean mybatisPlus = new ISkyTripMybatisSqlSessionFactoryBean();
mybatisPlus.setDataSource(dataSource);
mybatisPlus.setVfs(SpringBootVFS.class);
MybatisConfiguration mc = new MybatisConfiguration();
mc.setDefaultScriptingLanguage(MybatisXMLLanguageDriver.class);
mc.setMapUnderscoreToCamelCase(true);// 数据库和java都是驼峰,就不需要
mybatisPlus.setConfiguration(mc);
//设置mapper.xml文件的路径
mybatisPlus.setTypeAliasesPackage(typeAliasesPackage);
mybatisPlus.setMapperLocations(new PathMatchingResourcePatternResolver().getResources(mapperLocations));
return mybatisPlus;
}
/**
* 获取数据源
* @param dbConfigEntity
* @return
*/
public static DataSource getDbDataSource(DBConfigEntity dbConfigEntity) {
HikariConfig config = new HikariConfig();
config.setJdbcUrl(dbConfigEntity.getUrl()); //数据源
config.setUsername(dbConfigEntity.getUserName()); //用户名
config.setPassword(dbConfigEntity.getPassword()); //密码
config.addDataSourceProperty("cachePrepStmts", "true"); //是否自定义配置,为true时下面两个参数才生效
config.addDataSourceProperty("prepStmtCacheSize", "250"); //连接池大小默认25,官方推荐250-500
config.addDataSourceProperty("prepStmtCacheSqlLimit", "2048"); //单条语句最大长度默认256,官方推荐2048
config.addDataSourceProperty("useServerPrepStmts", "true"); //新版本MySQL支持服务器端准备,开启能够得到显著性能提升
config.addDataSourceProperty("useLocalSessionState", "true");
config.addDataSourceProperty("useLocalTransactionState", "true");
config.addDataSourceProperty("rewriteBatchedStatements", "true");
config.addDataSourceProperty("cacheResultSetMetadata", "true");
config.addDataSourceProperty("cacheServerConfiguration", "true");
config.addDataSourceProperty("elideSetAutoCommits", "true");
config.addDataSourceProperty("maintainTimeStats", "false");
HikariDataSource ds = new HikariDataSource(config);
return ds;
}
}
package com.iskytrip.mybatis.inject.mybatis.dataSource.config;
import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
import org.apache.ibatis.session.SqlSessionFactory;
public class ISkyTripMybatisSqlSessionFactoryBean extends MybatisSqlSessionFactoryBean {
private SqlSessionFactory sqlSessionFactory;
public SqlSessionFactory getSqlSessionFactory() {
return sqlSessionFactory;
}
public void setSqlSessionFactory(SqlSessionFactory sqlSessionFactory) {
this.sqlSessionFactory = sqlSessionFactory;
}
@Override
public SqlSessionFactory buildSqlSessionFactory() throws Exception {
if (this.sqlSessionFactory == null) {
this.sqlSessionFactory = super.buildSqlSessionFactory();
}
return this.sqlSessionFactory;
}
}
配置的时候只要在项目的配置文件里加上这个配置就好,第一个红框里的是applo里的配置,第二个是项目工程里的package名称,多个数据源用,隔开