Spring MyBatis多数据源(同包)

傷城~ 2022-04-06 03:26 50阅读 0赞

分包的同学点这里:[Spring MyBatis多数据源(分包)][Spring MyBatis]

创建基本的包entity service dao 为了区分多数据源 一个用的是Mysql 一个是Oracle 方便测试，

![Center][]

创建MyBatis dao 映射 xml 文件

![Center 1][]

创建db.properties

![Center 2][]

我这里设置了2个数据源

创建Mybatis.xml 文件

建议创建2个，也可以创建一个.但是2个 耦合性降低

还需要用到工具类:

package com.cp.util.datasources;
    
    import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
    
    /**
     * 动态切换数据库
     * 
     * @author Joey
     * @project:SSM_MultiDataSource_CP
     * @date：2017年4月27日
     * 
     */
    public class DynamicDataSource extends AbstractRoutingDataSource {
    
    	@Override
    	protected Object determineCurrentLookupKey() {
    		return DataSourceContextHolder.getDbType();
    	}
    
    }

package com.cp.util.datasources;
    
    /**
     * 数据库切换的工具类
     * 
     * @author Joey
     * @project:SSM_MultiDataSource_CP
     * @date：2017年4月27日
     * 
     */
    public class DataSourceContextHolder {
    
    	public static final String DATA_SOURCE_A = "dataSource1";
    	public static final String DATA_SOURCE_B = "dataSource2";
    
    	/** 数据源类型 */
    	private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
    
    	/**
    	 * 设置数据源类型
    	 * 
    	 * @param dbType
    	 *            数据源类型
    	 */
    	public static void setDbType(String dbType) {
    		contextHolder.set(dbType);
    	}
    
    	/**
    	 * 获取数据源类型
    	 * 
    	 * @return String
    	 */
    	public static String getDbType() {
    		return ((String) contextHolder.get());
    	}
    
    	/**
    	 * 清除数据源类型
    	 */
    	public static void clearDbType() {
    		contextHolder.remove();
    	}
    
    }

package com.cp.util.datasources;
    
    import java.lang.reflect.Method;
    
    import org.springframework.aop.AfterReturningAdvice;
    import org.springframework.aop.MethodBeforeAdvice;
    
    /**
     * 数据源前置增强
     * 
     * @author Joey
     * @project:SSM_MultiDataSource_CP
     * @date：2017年4月28日
     * 
     */
    public class DataSourceAspect implements MethodBeforeAdvice,
    		AfterReturningAdvice {
    
    	@Override
    	public void afterReturning(Object returnValue, Method method,
    			Object[] args, Object target) throws Throwable {
    		DataSourceContextHolder.clearDbType();
    	}
    
    	@Override
    	public void before(Method method, Object[] args, Object target)
    			throws Throwable {
    		if (method.isAnnotationPresent(DataSource.class)) {
    			DataSource datasource = method.getAnnotation(DataSource.class);
    			DataSourceContextHolder.setDbType(datasource.value());
    		} else {
    			DataSourceContextHolder
    					.setDbType(DataSourceContextHolder.DATA_SOURCE_A);
    		}
    
    	}
    }

package com.cp.util.datasources;
    
    import java.lang.annotation.Documented;
    import java.lang.annotation.ElementType;
    import java.lang.annotation.Retention;
    import java.lang.annotation.RetentionPolicy;
    import java.lang.annotation.Target;
    
    /**
     * 动态数据源(default:DataSourceContextHolder.DATA_SOURCE_A)
     * 
     * @author Joey
     * @project:SSM_MultiDataSource_CP
     * @date：2017年4月28日
     * 
     */
    @Retention(RetentionPolicy.RUNTIME)
    @Target(ElementType.METHOD)
    @Documented
    public @interface DataSource {
    
    	String value() default DataSourceContextHolder.DATA_SOURCE_A;
    
    }

Spring配置文件 :

![Center 3][]

配置扫描包.

导入数据源连接信息

设置2个数据源

![Center 4][]

我使用的是Druid 个人觉得不错。推荐一下

![Center 5][]

Druid的日志记录Filter

![Center 6][]

DynamicDataSource类是你刚刚自定义的工具类，上面有

导入2个数据源，默认数据源是DataSource1

![Center 7][]

事物管理 和 开启事物自动扫描注解

![Center 8][]

加上前置增强 DataSourceAspect类也是你刚刚自定义的工具类，上面有

最重要的一步: 开启AOP注解支持

![Center 9][]

方法注解使用多数据源

在service里的方法要使用不同的数据源  
  
方法直接注解:  
@DataSource(DataSourceContextHolder.DATA\_SOURCE\_B)

默认的使用是A 可以自己配置

注意: DataSourceContextHolder类里的DATA\_SOURCE\_A 和 DATA\_SOURCE\_B 的值要和spring配置文件对应

![Center 10][]

可以自己去试试

附上楼主的源码

[亲，点我去下载][Link 1]

chuagn 创建

[Spring MyBatis]: http://blog.csdn.net/qq_30930805/article/details/71467177
[Center]: /images/20220406/3a7c0695e8d64f2cadbfc9ebc2a182ad.png
[Center 1]: /images/20220406/5e8284e7bb3e4ec8bbfa38e84e446b06.png
[Center 2]: /images/20220406/49a4bbcc3e1f4d4f8a83f14c4a7bc18c.png
[Center 3]: /images/20220406/1eba8499b4174aedb1de566374407f50.png
[Center 4]: /images/20220406/eb165262db264f1ab9916b634370c4c6.png
[Center 5]: /images/20220406/1929300ef5444c8f9b0376b8b5d6115b.png
[Center 6]: /images/20220406/c222e31d28b24b8c94a98070032cb1b2.png
[Center 7]: /images/20220406/56d8e5fe8620428fb1e5090f8a57b7bd.png
[Center 8]: /images/20220406/1d9c4a3d83254c1eb7b2270a7935e779.png
[Center 9]: /images/20220406/eb46407946b6442aa956bec7406900c4.png
[Center 10]: /images/20220406/8b5a8960fc5c448b8171f6512297d68f.png
[Link 1]: http://download.csdn.net/download/qq_30930805/9837560