Примечание: некоторые читатели сообщили, что не видели обработки транзакций.Я проверил ту часть статьи, в которой не указаны транзакции.На самом деле основная обработка транзакций сосредоточена в двух классах MyBatisConfig и CustomSqlSessionTemplate. Если вы хотите увидеть более профессиональный анализ, вы можете прочитать две справочные статьи.
предисловие
Продолжить предыдущую статьюОсновы транзакций Spring, эта статья в основном о том, как обеспечить нормальный откат транзакций в случае использования нескольких источников данных Spring. Здесь также широко используется всемиjta-atomikosПродолжай, я просто делаю некоторые резюме для удобства использования их непосредственно в будущем.
Если вы спешите, вы можете напрямую скачать этот проект и посмотреть:
GitHub.com/Small Nanny Eat That/Wood…
Примечание. Версия здесь очень специфична, не переходите на mysql8.0. mysql-коннектор-java: 5.1.47 mybatis-весенняя загрузка-стартер: 1.3.2 друид-спринг-бут-стартер: 1.1.9
Главная идея
В Интернете уже есть много связанных статей о jta-atomikos.Эта статья может быть немного запутанной и не простой для понимания, поэтому я опишу идею здесь:
1. Настройте mybatis и druid, чтобы они могли подключаться к нескольким источникам данных. 2. Путем настройки источника данных транзакции нескольких источников данных интегрируются в один SqlSession, тем самым реализуя унифицированное управление транзакциями. 3. Используйте АОП и пользовательские аннотации для динамического переключения источников данных (то есть dao A должен подключаться к источнику данных A).
Для получения более подробной информации вы можете просмотреть исходный код или следующее краткое введение.
добавить зависимости
Основная зависимость это jta-atomikos, а остальные зависимости mybatis и druid вставляться не будут.
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>
<!--atomikos transaction management-->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-jta-atomikos</artifactId>
</dependency>
Настройка нескольких источников данных
1. Сначала определите перечисление, чтобы объяснить, что представляют собой ключи текущего экземпляра источника данных.
public class DataSourceKey {
/** 数据库源one*/
public static final String ONE= "one";
/** 数据库源two*/
public static final String TWO= "two";
}
2. Во-вторых, используйте ThreadLocal для хранения ключа текущего использования экземпляра источника данных. THREADLOCAL создается со значением Master по умолчанию, то есть источником данных по умолчанию является основной источник данных.
public class DynamicDataSourceContextHolder {
private static ThreadLocal<Object> CONTEXT_HOLDER = ThreadLocal.withInitial(() -> DataSourceKey.MASTER.getName());
public static List<Object> dataSourceKeys = new ArrayList<Object>();
public static void setDataSourceKey(String key){
CONTEXT_HOLDER.set(key);
}
public static Object getDataSourceKey(){
return CONTEXT_HOLDER.get();
}
public static void clearDataSourceKey(){
CONTEXT_HOLDER.remove();
}
public static Boolean containDataSourceKey(String key){
return dataSourceKeys.contains(key);
}
}
3. Переписать метод defineCurrentLookupKey класса AbstractRoutingDataSource, при обращении к базе данных будет вызываться метод defineCurrentLookupKey() этого класса для получения ключа экземпляра базы данных.
public class DynamicDataSource extends AbstractRoutingDataSource {
/**
* 取得当前使用那个数据源。
*/
@Override
protected Object determineCurrentLookupKey() {
return DataSourceContextHolder.getDatasourceType();
}
}
4. Соберите и интегрируйте несколько источников данных через SqlSessionFactory и, наконец, верните sqlSessionTemplate на уровень dao.
@Configuration
@MapperScan(basePackages = MyBatisConfig.BASE_PACKAGE, sqlSessionTemplateRef = "sqlSessionTemplate")
public class MyBatisConfig extends AbstractDataSourceConfig {
//mapper模式下的接口层
static final String BASE_PACKAGE = "cn.xbmchina.multidatasourceatomikos.mapper";
//对接数据库的实体层
static final String ALIASES_PACKAGE = "ccn.xbmchina.multidatasourceatomikos.domain";
static final String MAPPER_LOCATION = "classpath:mapper/*.xml";
@Primary
@Bean(name = "dataSourceOne")
public DataSource dataSourceOne(Environment env) {
String prefix = "spring.datasource.druid.one.";
return getDataSource(env,prefix,"one");
}
@Bean(name = "dataSourceTwo")
public DataSource dataSourceTwo(Environment env) {
String prefix = "spring.datasource.druid.two.";
return getDataSource(env,prefix,"two");
}
@Bean("dynamicDataSource")
public DynamicDataSource dynamicDataSource(@Qualifier("dataSourceOne")DataSource dataSourceOne, @Qualifier("dataSourceTwo")DataSource dataSourceTwo) {
Map<Object, Object> targetDataSources = new HashMap<>();
targetDataSources.put("one",dataSourceOne);
targetDataSources.put("two",dataSourceTwo);
DynamicDataSource dataSource = new DynamicDataSource();
dataSource.setTargetDataSources(targetDataSources);
dataSource.setDefaultTargetDataSource(dataSourceOne);
return dataSource;
}
@Bean(name = "sqlSessionFactoryOne")
public SqlSessionFactory sqlSessionFactoryOne(@Qualifier("dataSourceOne") DataSource dataSource)
throws Exception {
return createSqlSessionFactory(dataSource);
}
@Bean(name = "sqlSessionFactoryTwo")
public SqlSessionFactory sqlSessionFactoryTwo(@Qualifier("dataSourceTwo") DataSource dataSource)
throws Exception {
return createSqlSessionFactory(dataSource);
}
@Bean(name = "sqlSessionTemplate")
public CustomSqlSessionTemplate sqlSessionTemplate(@Qualifier("sqlSessionFactoryOne")SqlSessionFactory factoryOne, @Qualifier("sqlSessionFactoryTwo")SqlSessionFactory factoryTwo) throws Exception {
Map<Object,SqlSessionFactory> sqlSessionFactoryMap = new HashMap<>();
sqlSessionFactoryMap.put("one",factoryOne);
sqlSessionFactoryMap.put("two",factoryTwo);
CustomSqlSessionTemplate customSqlSessionTemplate = new CustomSqlSessionTemplate(factoryOne);
customSqlSessionTemplate.setTargetSqlSessionFactorys(sqlSessionFactoryMap);
return customSqlSessionTemplate;
}
/**
* 创建数据源
* @param dataSource
* @return
*/
private SqlSessionFactory createSqlSessionFactory(DataSource dataSource) throws Exception{
SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
bean.setDataSource(dataSource);
bean.setVfs(SpringBootVFS.class);
bean.setTypeAliasesPackage(ALIASES_PACKAGE);
bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources(MAPPER_LOCATION));
return bean.getObject();
}
}
5. Используйте АОП для динамического переключения источника данных с помощью метода, в котором находится пользовательская аннотация.
import cn.xbmchina.multidatasourceatomikos.annotations.TargetDataSource;
import cn.xbmchina.multidatasourceatomikos.db.DataSourceContextHolder;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import java.lang.reflect.Method;
public class DataSourceAspect {
protected static final ThreadLocal<String> preDatasourceHolder = new ThreadLocal<>();
/**
* @param clazz
* @param name
* @return
*/
private static Method findUniqueMethod(Class<?> clazz, String name) {
Class<?> searchType = clazz;
while (searchType != null) {
Method[] methods = (searchType.isInterface() ? searchType.getMethods() : searchType.getDeclaredMethods());
for (Method method : methods) {
if (name.equals(method.getName())) {
return method;
}
}
searchType = searchType.getSuperclass();
}
return null;
}
@Pointcut("@annotation(cn.xbmchina.multidatasourceatomikos.annotations.TargetDataSource)")
protected void datasourceAspect() {
}
/**
* 根据@TargetDataSource的属性值设置不同的dataSourceKey,以供DynamicDataSource
*/
@Before("datasourceAspect()")
public void changeDataSourceBeforeMethodExecution(JoinPoint jp) {
String key = determineDatasource(jp);
if (key == null) {
DataSourceContextHolder.setDatasourceType(null);
return;
}
preDatasourceHolder.set(DataSourceContextHolder.getDatasourceType());
DataSourceContextHolder.setDatasourceType(key);
}
/**
* @param jp
* @return
*/
public String determineDatasource(JoinPoint jp) {
String methodName = jp.getSignature().getName();
Class targetClass = jp.getSignature().getDeclaringType();
String dataSourceForTargetClass = resolveDataSourceFromClass(targetClass);
String dataSourceForTargetMethod = resolveDataSourceFromMethod(targetClass, methodName);
String resultDS = determinateDataSource(dataSourceForTargetClass, dataSourceForTargetMethod);
return resultDS;
}
/**
*
*/
@After("datasourceAspect()")
public void restoreDataSourceAfterMethodExecution() {
DataSourceContextHolder.setDatasourceType(preDatasourceHolder.get());
preDatasourceHolder.remove();
}
/**
* @param targetClass
* @param methodName
* @return
*/
private String resolveDataSourceFromMethod(Class targetClass, String methodName) {
Method m = findUniqueMethod(targetClass, methodName);
if (m != null) {
TargetDataSource choDs = m.getAnnotation(TargetDataSource.class);
return resolveDataSourceName(choDs);
}
return null;
}
/**
* @param classDS
* @param methodDS
* @return
*/
private String determinateDataSource(String classDS, String methodDS) {
return methodDS == null ? classDS : methodDS;
}
/**
* @param targetClass
* @return
*/
private String resolveDataSourceFromClass(Class targetClass) {
TargetDataSource classAnnotation = (TargetDataSource) targetClass.getAnnotation(TargetDataSource.class);
return null != classAnnotation ? resolveDataSourceName(classAnnotation) : null;
}
/**
* @param ds
* @return
*/
private String resolveDataSourceName(TargetDataSource ds) {
return ds == null ? null : ds.value();
}
}
6. Для сценария динамической архитектуры с несколькими источниками данных источник данных переключается через aop, но поскольку транзакция управляется до переключения, переключение блокируется транзакцией. Способ обработки: Необходимо переписать SqlSessionTemplate, чтобы изменить SqlSessionFactory для динамического получения источника данных.
targetSqlSessionFactorys.get(DataSourceContextHolder.getDatasourceType());
DataSourceContextHolder обычно представляет собой класс операции контекста источника данных, который вы создали на втором шаге, его нужно только изменить в соответствии с вашими потребностями.
public class CustomSqlSessionTemplate extends SqlSessionTemplate {
//......省略
@Override
public SqlSessionFactory getSqlSessionFactory() {
SqlSessionFactory targetSqlSessionFactory = targetSqlSessionFactorys.get(DataSourceContextHolder.getDatasourceType());
if (targetSqlSessionFactory != null) {
return targetSqlSessionFactory;
} else if (defaultTargetSqlSessionFactory != null) {
return defaultTargetSqlSessionFactory;
} else {
Assert.notNull(targetSqlSessionFactorys, "Property 'targetSqlSessionFactorys' or 'defaultTargetSqlSessionFactory' are required");
Assert.notNull(defaultTargetSqlSessionFactory, "Property 'defaultTargetSqlSessionFactory' or 'targetSqlSessionFactorys' are required");
}
return this.sqlSessionFactory;
}
//......省略
}
7. Последний эффект выглядит следующим образом:
Справочная статья
woo woo woo.cn blog on.com/Xiao Fengfen…
Наконец
Для получения дополнительных статей, пожалуйста, обратите внимание на паблик-аккаунт** [love coding],Ответить2020**Есть актуальные видеоматериалы.