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;
		}
	}