springboot 事务统一配置_springboot基于注解动态配置多数据源以及多数据源的事务统一...

一、动态注入多数据源

1、配置多数据源配置文件(application-db.properties)

######多数据源配置文件####################

###第一个####

spring.datasource.first.name=first

spring.datasource.first.url=jdbc:oracle:thin:@127.0.0.1:1521:orcl

spring.datasource.first.username=sepcore

spring.datasource.first.password=sepcore

spring.datasource.first.driverClassName=oracle.jdbc.driver.OracleDriver

spring.datasource.first.mapperLocations=classpath:mappers/*Mapper.xml

####第二个####

spring.datasource.second.name=second

spring.datasource.second.url=jdbc:mysql://127.0.0.1:3306/test?serverTimezone=UTC&characterEncoding=utf8&useUnicode=true&useSSL=false

spring.datasource.second.username=root

spring.datasource.second.password=123456

spring.datasource.second.driverClassName=com.mysql.jdbc.Driver

spring.datasource.second.mapperLocations=classpath:mappers/*Mapper.xml

#####mapper接口所在包#######

scanner.mapperInterfacePackage=com.example.demo.mappers

2、读取配置文件类(DataSourceConfig)

public classDataSourceConfig {/*** 存储dataSource、SqlSessionTemplate、DataSourceTransactionManager*/

private Map>mapMap;/*** 获取mybatis扫描的指定接口包(所有数据源的接口放在同一的父包下面)*/

privateString mapperInterfacePackage;publicDataSourceConfig(){

mapMap= new HashMap<>();

InputStream in= DataSourceConfig.class.getClassLoader().

getResourceAsStream("application-db.properties");

Properties properties= newProperties();try{

properties.load(in);

}catch(IOException e) {

e.printStackTrace();

}

Set set =properties.stringPropertyNames();for(String s : set) {//判断是否是mapper接口指定包路径

if (s.contains("mapperInterfacePackage")){

mapperInterfacePackage=properties.get(s).toString();continue;

}

String key= s.substring(0, s.lastIndexOf("."));if(mapMap.containsKey(key)){

Map map =mapMap.get(key);

map.put(s,properties.get(s));

}else{

Mapmap = new HashMap<>();

map.put(s,properties.get(s));

mapMap.put(key,map);

}

}

}publicString getMapperInterfacePackage() {returnmapperInterfacePackage;

}/*** 获取SqlSessionTemplate

*@return*@throwsException*/

public MapgetSqlSessionTemplateAndDataSource() throwsException {

Set>> entries = this.mapMap.entrySet();

Mapresult = new HashMap<>(entries.size());for (Map.Entry>entry : entries) {

String key=entry.getKey();

Map map =entry.getValue();

DataSource dataSource= DataSourceBuilder.create().url(map.get(key+".url").toString()).

username(map.get(key+".username").toString()).password(map.get(key+".password").toString()).

driverClassName(map.get(key+".driverClassName").toString()).

build();//为每个数据源设置事务

DataSourceTransactionManager dataSourceTransactionManager = newDataSourceTransactionManager();

dataSourceTransactionManager.setDataSource(dataSource);

SqlSessionFactoryBean sqlSessionFactoryBean= newSqlSessionFactoryBean();//设置dataSource数据源

sqlSessionFactoryBean.setDataSource(dataSource);//设置*mapper.xml路径

sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources(map.get(key+".mapperLocations").toString()));

String s= map.get(key + ".name").toString();

result.put(s+"SqlSessionTemplate",newSqlSessionTemplate(sqlSessionFactoryBean.getObject()));

result.put(s+"DataSource",dataSource);

result.put(s+"DataSourceTransactionManager",dataSourceTransactionManager);

}returnresult;

}

}

3、使用注解(DataSourceRoute)，确定每个mapper接口使用哪个数据源

@Target(ElementType.TYPE)

@Retention(RetentionPolicy.RUNTIME)public @interfaceDataSourceRoute {

String name()default "first";

}

@DataSourceRoutepublic interfaceSepUserMapper {

List>findByUserLevel(Long userLevel);void insert(Mapmap);

}

@DataSourceRoute(name = "second")public interfaceIDiDataItemMapper {

@Select("SELECT dataitem_id,name FROM di_dataitem WHERE dataitem_id=#{dataItemId}")

MapselectOne(Long dataItemId);void insert(Mapmap);

}

4、扫描指定包下面的mapper接口

public classClassScanner {public static Map>getMapperInterface(String mapperInterfacePackage) throwsException {

Map>classMap = new HashMap<>();

ClassLoader loader=Thread.currentThread().getContextClassLoader();//将"."替换成"/"

String packagePath = mapperInterfacePackage.replace(".", "/");

URL url=loader.getResource(packagePath);

List fileNames = null;if (url != null) {

String type=url.getProtocol();if ("file".equals(type)) {

fileNames= getClassNameByFile(url.getPath(), null, true);

}

}for(String classPath : fileNames) {

classMap.putAll(getClassByPath(classPath));

}returnclassMap;

}/*** 读取package下的所有类文件

*@paramfilePath

*@paramclassName

*@paramchildPackage

*@return

*/

private static List getClassNameByFile(String filePath, List className, booleanchildPackage) {

List myClassName = new ArrayList<>();

File file= newFile(filePath);

File[] childFiles=file.listFiles();for(File childFile : childFiles) {if(childFile.isDirectory()) {if(childPackage) {

myClassName.addAll(getClassNameByFile(childFile.getPath(), myClassName, childPackage));

}

}else{

String childFilePath=childFile.getPath();if (childFilePath.endsWith(".class")) {

childFilePath= childFilePath.substring(childFilePath.indexOf("\\classes") + 9,

childFilePath.lastIndexOf("."));

childFilePath= childFilePath.replace("\\", ".");

myClassName.add(childFilePath);

}

}

}returnmyClassName;

}/*** 将Mapper的标准文件，转成 Mapper Class

*@paramclassPath

*@return*@throwsException*/

private static Map>getClassByPath(String classPath)throwsException{

ClassLoader loader=Thread.currentThread().getContextClassLoader();

Map> classMap = new HashMap<>();

classMap.put(getClassAlias(classPath),loader.loadClass(getFullClassName(classPath)));returnclassMap;

}/*** 将Mapper的标准文件，转成java标准的类名称

*@paramclassPath

*@return*@throwsException*/

private staticString getFullClassName(String classPath)throwsException{int comIndex = classPath.indexOf("com");

classPath=classPath.substring(comIndex);

classPath= classPath.replaceAll("\\/", ".");returnclassPath;

}/*** 根据类地址，获取类的Alais，即根据名称，按照驼峰规则，生成可作为变量的名称

*@paramclassPath

*@return*@throwsException*/

private staticString getClassAlias(String classPath)throwsException{

String split= "\\/";

String[] classTmp=classPath.split(split);

String className= classTmp[classTmp.length-1];returntoLowerFisrtChar(className);

}/*** 将字符串的第一个字母转小写

*@paramclassName

*@return

*/

private staticString toLowerFisrtChar(String className){

String fisrtChar= className.substring(0,1);

fisrtChar=fisrtChar.toLowerCase();return fisrtChar+className.substring(1);

}

5、使用BeanFactoryPostProcessor动态插入数据源

@Componentpublic class DataSourceBean implementsBeanFactoryPostProcessor{

@Overridepublic void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throwsBeansException {

System.out.println("在spring处理bean前，将自定义的bean注册到容器中======================"); DataSourceConfig dataSourceConfig = newDataSourceConfig();try{

Map sqlSessionTemplateAndDataSource =dataSourceConfig.getSqlSessionTemplateAndDataSource();

Map> mapperInterface =ClassScanner.getMapperInterface(dataSourceConfig.getMapperInterfacePackage());

Set>> entries =mapperInterface.entrySet();for (Map.Entry>entry : entries) {

MapperFactoryBean mapperFactoryBean= newMapperFactoryBean();

Class> value =entry.getValue();

DataSourceRoute dataSourceRoute= value.getAnnotation(DataSourceRoute.class);if (null==dataSourceConfig){continue;

}

String name=dataSourceRoute.name();

SqlSessionTemplate template= (SqlSessionTemplate) sqlSessionTemplateAndDataSource.get(name + "SqlSessionTemplate");

mapperFactoryBean.setMapperInterface(value);

mapperFactoryBean.setSqlSessionTemplate(template);

mapperFactoryBean.afterPropertiesSet();

configurableListableBeanFactory.registerSingleton(name+"MapperFactory",mapperFactoryBean.getObject());

configurableListableBeanFactory.registerSingleton(name+"DataSource",sqlSessionTemplateAndDataSource.get(name + "DataSource"));

configurableListableBeanFactory.registerSingleton(name+"SqlSessionTemplate",template);

configurableListableBeanFactory.registerSingleton(name+"DataSourceTransactionManager",sqlSessionTemplateAndDataSource.get(name+"DataSourceTransactionManager"));

}

}catch(Exception e) {

System.err.println(e.getMessage());

}

}

}

至此多数据源动态加载就完成了。

二、多数据源统一事务控制

当使用多数据源时，单一的事务会出现问题(当在service层同时操作两个数据源时，当发生异常，只会回滚离抛出异常最近的数据源的数据)

1、自定义事务注解

@Target({ElementType.METHOD,ElementType.PARAMETER})

@Retention(RetentionPolicy.RUNTIME)public @interfaceCustomTransaction {

String[] name()default {"firstDataSourceTransactionManager"};

}

2、创建aop切面进行事务控制

@Component

@Aspectpublic classTransactionAop {

@Pointcut(value= "@annotation(com.example.demo.annon.CustomTransaction)")public voidpointCut(){}

@Around(value= "pointCut()&&@annotation(annotation)")public Object twiceAsOld(ProceedingJoinPoint point, CustomTransaction annotation) throwsThrowable {

Stack dataSourceTransactionManagerStack = new Stack();

Stack transactionStatuStack = new Stack();try{if (!openTransaction(dataSourceTransactionManagerStack, transactionStatuStack, annotation)) {return null;

}

Object ret=point.proceed();

commit(dataSourceTransactionManagerStack, transactionStatuStack);returnret;

}catch(Throwable e) {

rollback(dataSourceTransactionManagerStack, transactionStatuStack);throwe;

}

}/*** 开启事务处理方法

*

*@paramdataSourceTransactionManagerStack

*@paramtransactionStatuStack

*@parammultiTransactional

*@return

*/

private boolean openTransaction(StackdataSourceTransactionManagerStack,

StacktransactionStatuStack, CustomTransaction multiTransactional) {

String[] transactionMangerNames=multiTransactional.name();if(ArrayUtils.isEmpty(multiTransactional.name())) {return false;

}for(String beanName : transactionMangerNames) {//根据事务名称获取具体的事务

DataSourceTransactionManager dataSourceTransactionManager =(DataSourceTransactionManager) SpringContextUtil

.getBean(beanName);

TransactionStatus transactionStatus=dataSourceTransactionManager

.getTransaction(newDefaultTransactionDefinition());

transactionStatuStack.push(transactionStatus);

dataSourceTransactionManagerStack.push(dataSourceTransactionManager);

}return true;

}/*** 提交处理方法

*

*@paramdataSourceTransactionManagerStack

*@paramtransactionStatuStack*/

private void commit(StackdataSourceTransactionManagerStack,

StacktransactionStatuStack) {while (!dataSourceTransactionManagerStack.isEmpty()) {

dataSourceTransactionManagerStack.pop().commit(transactionStatuStack.pop());

}

}/*** 回滚处理方法

*

*@paramdataSourceTransactionManagerStack

*@paramtransactionStatuStack*/

private void rollback(StackdataSourceTransactionManagerStack,

StacktransactionStatuStack) {while (!dataSourceTransactionManagerStack.isEmpty()) {

dataSourceTransactionManagerStack.pop().rollback(transactionStatuStack.pop());

}

}

}

3、在service层指定使用哪个事务

//注意事务的命名规则

@CustomTransaction(name = {"firstDataSourceTransactionManager","secondDataSourceTransactionManager"})public voidsetSepUserMapper(){//操作数据源2

Mapmm = new HashMap<>(2);

mm.put("dataitemId",1L);

mm.put("name","测试");

diDataItemMapper.insert(mm);//操作数据源1

Mapmap = new HashMap<>(3);

map.put("userId",1L);

map.put("userName","张三");

map.put("name","平台管理员");

sepUserMapper.insert(map);throw new RuntimeException("sfsa");

}

辅助类：SpringContextUtil

@Componentpublic class SpringContextUtil implementsApplicationContextAware{private staticApplicationContext applicationContext;

@Overridepublic void setApplicationContext(ApplicationContext applicationContext) throwsBeansException {

SpringContextUtil.applicationContext=applicationContext;

}/*** @Description: 获取spring容器中的bean,通过bean名称获取

*@parambeanName bean名称

*@return: Object 返回Object,需要做强制类型转换

*@author: zongf

* @time: 2018-12-26 10:45:07*/

public staticObject getBean(String beanName){returnapplicationContext.getBean(beanName);

}/*** @Description: 获取spring容器中的bean, 通过bean类型获取

*@parambeanClass bean 类型

*@return: T 返回指定类型的bean实例

*@author: zongf

* @time: 2018-12-26 10:46:31*/

public static T getBean(ClassbeanClass) {returnapplicationContext.getBean(beanClass);

}/*** @Description: 获取spring容器中的bean, 通过bean名称和bean类型精确获取

*@parambeanName bean 名称

*@parambeanClass bean 类型

*@return: T 返回指定类型的bean实例

*@author: zongf

* @time: 2018-12-26 10:47:45*/

public static T getBean(String beanName, ClassbeanClass){returnapplicationContext.getBean(beanName,beanClass);

}