spring aop基础上得分库分表(1)

1.环境说明：spring,mybtis,mysql;

2.分两种方式实现自定义切库

3.第一种方式通过自定义注解得方式实现

基本思路：我们在使用mybatis得时候，一个数据源配套一套mybatis配置。若要链接多个数据源，那么就需要多套mybatis配置。通过不同的配置操作相应的数据库。在实际操作中，我们更多就是通过mapper来操作DB。

(1)定义多个数据源，这里定义两个数据源

jdbc.master.driver=com.mysql.jdbc.Driver
jdbc.slave.driver=com.mysql.jdbc.Driver
jdbc.master.url=jdbc:mysql://127.0.0.1:3306/xxx?characterEncoding=utf-8
jdbc.slave.url=jdbc:mysql://127.0.0.1:3306/xxx?characterEncoding=utf-8
jdbc.slave.username=zzz
jdbc.slave.password=zzz
jdbc.master.username=zzz
jdbc.master.password=zzz

(2)生成不同的mapper（例如mybatis插件生成dao,mapper,xml工具）

(3)配置数据源

<!-- 数据库连接池 -->
<bean id="abstractDataSource" abstract="true" class="com.alibaba.druid.pool.DruidDataSource"
destroy-method="close">
<property name="maxActive" value="30" />
<property name="minIdle" value="10" />
</bean>
<bean id="master" parent="abstractDataSource">
<property name="url" value="${jdbc.master.url}" />
<property name="username" value="${jdbc.master.username}" />
<property name="password" value="${jdbc.master.password}" />
<property name="driverClassName" value="${jdbc.master.driver}" />
</bean>
<bean id="slave" parent="abstractDataSource">
<property name="url" value="${jdbc.slave.url}" />
<property name="username" value="${jdbc.slave.username}" />
<property name="password" value="${jdbc.slave.password}" />
<property name="driverClassName" value="${jdbc.slave.driver}" />
</bean>

       <!-- 配置动态数据源，这targetDataSource就是路由数据源所对应的名称 -->
<bean id="dataSource" class="com.lai.lailaishare.mapper.split.DynamicDataSource">
<property name="targetDataSources">
<map>
<entry value-ref="master" key="master"></entry>
<entry value-ref="slave" key="slave"></entry>
</map>
</property>

</bean>

 mybatis抽象数据源路由

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource{

	@Override
	protected Object determineCurrentLookupKey() {
		return DynamicDataSourceHolder.getDb();
	}
	
}

(3)在请求的方法上面加注解

@Aspect 
@Component 
@Slf4j 
public class DynamicDataSourceAspect { 
	@Before("@annotation(RoutingDataSource)") 
	public void beforeSwitchDS(JoinPoint point){ 
		//获得当前访问的
		class Class<?> className = point.getTarget().getClass(); 
		//获得访问的方法名 
		String methodName = point.getSignature().getName(); //得到方法的参数的类型 
		Class[] argClass = ((MethodSignature)point.getSignature()).getParameterTypes(); 
		String dataSource = DataSourceContextHolder.DEFAULT_DATASOURCE; 
		try { 
			// 得到访问的方法对象 
			Method method = className.getMethod(methodName, argClass); // 判断是否存在@DS注解 
			if (method.isAnnotationPresent(RoutingDataSource.class)) { 
				RoutingDataSource annotation = method.getAnnotation(RoutingDataSource.class); 
				// 取出注解中的数据源名 
				dataSource = annotation.value(); 
			}
		} catch (Exception e) {
			log.error("routing datasource exception, " + methodName, e); 
		} 
		// 切换数据源 
		DataSourceContextHolder.setDB(dataSource); 
	} 
	@After("@annotation(RoutingDataSource)") 
	public void afterSwitchDS(JoinPoint point){ 
		DataSourceContextHolder.clearDB(); 
	} 
}

在这里使用threadLocal来避免多线程之间互相影响。同时也需要注意请求完后清空threadlocal避免内存泄露

使用：在service层配置：

        @Override
	@RoutingDataSource(DataSources.SLAVE_DB)
	public ServerResponse<String> updateUser(User user) {
		System.out.println(1);
		ServerResponse<String> checkValid = checkValid(user.getUsername(), Const.USERNAME);
		if(checkValid.isSuccess()) {
			User updateUser = new User();
			updateUser.setId(user.getId());
			updateUser.setPhone(user.getPhone());
			userMapper.updateByPrimaryKeySelective(updateUser);
			return ServerResponse.createBySuccessMessage(null);
		} else {
			return checkValid;
		}
	}