码迷,mamicode.com
首页 > 编程语言 > 详细

spring 多数据源切换

时间:2016-04-18 18:42:13      阅读:389      评论:0      收藏:0      [点我收藏+]

标签:

spring配置管理多数据源,方便服务访问不同数据库时,数据源切换。多少无意,直接上代码。

1.定义数据源key枚举值(可以直接使用字符串代替,为了代码的整洁及易读性,这里使用枚举)

public enum DataSourceEnum {
	anquan,
	publ,
	tlshow,
}

2.定义数据源持有者,切换数据源使用

public final class DataSourceHolder {
	private static ThreadLocal<DataSourceEnum> currentDBName = new ThreadLocal<DataSourceEnum>();
	
	private static DataSourceEnum DEFAULT_HASH = DataSourceEnum.anquan;

	public static DataSourceEnum determineDefault() {
		return determineDS(DEFAULT_HASH);
	}

	public static DataSourceEnum determineDS(DataSourceEnum key) {
		currentDBName.set(key);
		log.debug("determineDS:" + key);
		return key;
	}
	
	
	public static DataSourceEnum getCurrentDBName() {
		return currentDBName.get();
	}
}

3.实现数据源路由接口

public class DBRoutingDataSource extends AbstractRoutingDataSource {

	@Override
	protected Object determineCurrentLookupKey() {
		return DataSourceHolder.getCurrentDBName();
	}

	@Override
	public void setTargetDataSources(Map targetDataSources) {
		super.setTargetDataSources(targetDataSources);
	}
	
	@Override
	public Object unwrap(Class iface) throws SQLException {
		return null;
	}

	@Override
	public boolean isWrapperFor(Class iface) throws SQLException {
		return false;
	}
}

4.定义spring 切面,在执行service方法前,切换数据源。(首先要保证spring支持注解)

@Aspect
@lombok.extern.log4j.Log4j
public class DetermineAspect {
	@Pointcut("within(com.changyou.*.service.impl.*)")
	public void withController() {}

	@Before("withController()")
	public void beforeAdvice(JoinPoint jp) {
		if(log.isDebugEnabled()) {
			log.debug("Reset db router before " + jp.getSignature().getDeclaringType().getSimpleName() + "." + jp.getSignature().getName() + "()");
		}
		DataSourceMethod datasourceMethod = ((MethodSignature)jp.getSignature()).getMethod().getAnnotation(DataSourceMethod.class);
		DataSourceFile datasourceFile = ((MethodSignature)jp.getSignature()).getMethod().getDeclaringClass().getAnnotation(DataSourceFile.class);
		
		if(datasourceMethod!=null){
			DataSourceHolder.determineDS(datasourceMethod.value());
		}else if(datasourceFile!=null){
			DataSourceHolder.determineDS(datasourceFile.value());
		}else{
			DataSourceHolder.determineDefault();
		}
	}

}

5.spring配置数据源,使用自己实现的数据源路由类(DBRoutingDataSource 

<!-- 配置数据源 -->
	<bean name="dataSource_0" parent="abstractProxoolDataSource">
		<property name="url" value="${jdbc.0.url}" />
		<property name="username" value="${jdbc.0.username}" />
		<property name="password" value="${jdbc.0.password}" />
	</bean>
	
	<!-- 配置数据源tlshow-->
	<bean name="dataSource_1" parent="abstractProxoolDataSource">
		<property name="url" value="${jdbc.1.url}" />
		<property name="username" value="${jdbc.1.username}" />
		<property name="password" value="${jdbc.1.password}" />
	</bean>
	
	<!-- 配置数据源publicaccount-->
	<bean name="dataSource_2" parent="abstractProxoolDataSource">
		<property name="url" value="${jdbc.2.url}" />
		<property name="username" value="${jdbc.2.username}" />
		<property name="password" value="${jdbc.2.password}" />
	</bean>
	
	
	<bean id="dataSource" class="com.changyou.common.util.dbaccess.DBRoutingDataSource">
		<property name="defaultTargetDataSource" ref="dataSource_0"/>
		<property name="targetDataSources">
			<map key-type="com.changyou.common.enums.DataSourceEnum"><!-- 数据源枚举 -->
				<entry key="anquan" value-ref="dataSource_0" />
				<entry key="tlshow" value-ref="dataSource_1" />
				<entry key="publ" value-ref="dataSource_2" />
			</map>
		</property>
	</bean>

6.基础工作完成,下面开始使用,需要在server类或方法上添加注解。(上面的切面会查看是否有指定注解的存在)。

  定义自己的注解:

@Target(value={ElementType.TYPE,ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSourceRouter {
	public DataSourceEnum value() default DataSourceEnum.anquan;
}

7.开始愉快的使用吧,方法上的注解权限高于整个类上的注解,这取决于切面的处理方式。

@Service
@DataSourceRouter(DataSourceEnum.publ)
public class CyjUserServiceImpl{

	@Autowired
	private UserMapper userMapper;
	
	public void saveOrUpdateBlock(CyjUserBlock cyjUserBlock) {
		userMapper.saveOrUpdateBlock(cyjUserBlock);
	}

        @DataSourceRouter(DataSourceEnum.anquan)
	public void deleteBlockById(int blockId) {
		userMapper.deleteBlockById(blockId);
	}
}

  

ok,后期补上demo。。。。。。。

 

spring 多数据源切换

标签:

原文地址:http://www.cnblogs.com/rmsSpring/p/5405403.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!