/**
 * 為什么要用鏈表存儲(準(zhǔn)確的是棧)
 * <pre>
 * 為了支持嵌套切換，如ABC三個service都是不同的數(shù)據(jù)源
 * 其中A的某個業(yè)務(wù)要調(diào)B的方法，B的方法需要調(diào)用C的方法。一級一級調(diào)用切換，形成了鏈。
 * 傳統(tǒng)的只設(shè)置當(dāng)前線程的方式不能滿足此業(yè)務(wù)需求，必須模擬棧，后進先出。
 * </pre>
 */
 private static final ThreadLocal<Deque<String>> LOOKUP_KEY_HOLDER = new ThreadLocal() {
 @Override
 protected Object initialValue() {
  return new ArrayDeque();
 }
 };

package xxx.xxx.xxx.config;

import com.baomidou.dynamic.datasource.provider.DynamicDataSourceProvider;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DataSourceProperty;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
import org.springframework.context.annotation.Primary;
import org.springframework.stereotype.Service;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

/**
 * @author : zuoyu
 * @description : 接管MybatisPlus多數(shù)據(jù)源至Atomikos管理
 * @date : 2020-06-01 16:36
 **/
@Service
@Primary
public class DynamicDataSourceProviderImpl implements DynamicDataSourceProvider {


 /**
  * 配置文件數(shù)據(jù)的松散綁定
  */
 private final DynamicDataSourceProperties properties;

 /**
  * Atomikos驅(qū)動數(shù)據(jù)源創(chuàng)建
  */
 private final AtomikosDataSourceCreator atomikosDataSourceCreator;

 public DynamicDataSourceProviderImpl(DynamicDataSourceProperties properties, AtomikosDataSourceCreator atomikosDataSourceCreator) {
  this.properties = properties;
  this.atomikosDataSourceCreator = atomikosDataSourceCreator;
 }

 @Override
 public Map<String, DataSource> loadDataSources() {
  Map<String, DataSourceProperty> dataSourcePropertiesMap = properties.getDatasource();
  Map<String, DataSource> dataSourceMap = new HashMap<>(dataSourcePropertiesMap.size() * 2);
  for (Map.Entry<String, DataSourceProperty> item : dataSourcePropertiesMap.entrySet()) {
   String pollName = item.getKey();
   DataSourceProperty dataSourceProperty = item.getValue();
   dataSourceProperty.setPollName(pollName);
   dataSourceMap.put(pollName, atomikosDataSourceCreator.createDataSource(dataSourceProperty));
  }
  return dataSourceMap;
 }
}

package xxx.xxx.xxx.config;

import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DataSourceProperty;
import com.mysql.jdbc.jdbc2.optional.MysqlXADataSource;
import org.springframework.boot.jta.atomikos.AtomikosDataSourceBean;
import org.springframework.stereotype.Component;

import javax.sql.DataSource;

/**
 * @author : zuoyu
 * @description : 事務(wù)數(shù)據(jù)源
 * @date : 2020-06-01 17:30
 **/
@Component
public class AtomikosDataSourceCreator {
 /**
  * 創(chuàng)建數(shù)據(jù)源
  *
  * @param dataSourceProperty 數(shù)據(jù)源信息
  * @return 數(shù)據(jù)源
  */
 public DataSource createDataSource(DataSourceProperty dataSourceProperty) {
  MysqlXADataSource mysqlXaDataSource = new MysqlXADataSource();
  mysqlXaDataSource.setUrl(dataSourceProperty.getUrl());
  mysqlXaDataSource.setPassword(dataSourceProperty.getPassword());
  mysqlXaDataSource.setUser(dataSourceProperty.getUsername());
  AtomikosDataSourceBean xaDataSource = new AtomikosDataSourceBean();
  xaDataSource.setXaDataSource(mysqlXaDataSource);
  xaDataSource.setMinPoolSize(5);
  xaDataSource.setBorrowConnectionTimeout(60);
  xaDataSource.setMaxPoolSize(20);
  xaDataSource.setXaDataSourceClassName(dataSourceProperty.getDriverClassName());
  xaDataSource.setTestQuery("SELECT 1 FROM DUAL");
  xaDataSource.setUniqueResourceName(dataSourceProperty.getPollName());
  return xaDataSource;
 }
}

package xxx.xxx.xxx.config;

import com.atomikos.icatch.jta.UserTransactionImp;
import com.atomikos.icatch.jta.UserTransactionManager;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.DependsOn;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.annotation.EnableTransactionManagement;
import org.springframework.transaction.jta.JtaTransactionManager;

import javax.transaction.TransactionManager;
import javax.transaction.UserTransaction;

/**
 * @author : zuoyu
 * @description : 分布式事務(wù)配置
 * @date : 2020-06-01 17:55
 **/
@Configuration
@EnableTransactionManagement
public class TransactionManagerConfig {

 @Bean(name = "userTransaction")
 public UserTransaction userTransaction() throws Throwable {
  UserTransactionImp userTransactionImp = new UserTransactionImp();
  userTransactionImp.setTransactionTimeout(10000);
  return userTransactionImp;
 }

 @Bean(name = "atomikosTransactionManager")
 public TransactionManager atomikosTransactionManager() throws Throwable {
  UserTransactionManager userTransactionManager = new UserTransactionManager();
  userTransactionManager.setForceShutdown(false);
  return userTransactionManager;
 }

 @Bean(name = "transactionManager")
 @DependsOn({"userTransaction", "atomikosTransactionManager"})
 public PlatformTransactionManager transactionManager() throws Throwable {
  return new JtaTransactionManager(userTransaction(), atomikosTransactionManager());
 }
}