動態註解多資料來源
阿新 • • 發佈：2019-02-11
<!-- 配置整合mybatis--> <bean id="propertyConfigurer" class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer"> <property name="locations"> <list> <value>classpath*:jdbc.properties</value> </list> </property> </bean> <!--配置資料庫的連線池--> <bean id="abstractDataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource"> <!-- 配置連結屬性 多資料來源配置--> <!-- <property name="driverClass" value="${driver}"/> <property name="jdbcUrl" value="${url}"/> <property name="user" value="${username}"/> <property name="password" value="${password}"/>--> <!--配置c3p0連線池的私有屬性--> <property name="maxPoolSize" value="30"/> <property name="minPoolSize" value="10"/> <!--關閉連結後不自動commit--> <property name="autoCommitOnClose" value="false"/> <property name="checkoutTimeout" value="1000"/> <!--獲取連線失敗重試次數--> <property name="acquireRetryAttempts" value="2"/> </bean> <!--多資料來源配置 parent指向上面的配置資料庫的連線池abstractDataSource 這裡配置了兩個資料來源--> <bean id="base" parent="abstractDataSource"> <property name="driverClass" value="${driver}"/> <property name="jdbcUrl" value="${url}"/> <property name="user" value="${username}"/> <property name="password" value="${password}"/> </bean> <bean id="server_1" parent="abstractDataSource"> <property name="driverClass" value="${server_1_driver}"/> <property name="jdbcUrl" value="${server_1_url}"/> <property name="user" value="${server_1_username}"/> <property name="password" value="${server_1_password}"/> </bean> <!-- 在RoutingDataSource類中讀取 當前的返回值 並匹配key值 選擇你的資料庫源--> <bean id="dataSource" class="com.cdms.dao.resources.RoutingDataSource"> <property name="targetDataSources"> <map key-type="java.lang.String"> <entry key="base" value-ref="base"></entry> <entry key="server_1" value-ref="server_1"></entry> </map> </property> <property name="defaultTargetDataSource" ref="base"></property> </bean> <!-- 配置sqlSessionFactory物件--> <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean"> <!-- 注入資料庫連線池--> <property name="dataSource" ref="dataSource"/> <!-- 配置Mybatisq全域性檔案：--> <property name="configLocation" value="classpath:mybatis-config.xml"/> <!-- 掃描entity包--> <property name="typeAliasesPackage" value="com.cdms.entity"/> <!-- 掃描sql配置檔案--> <property name="mapperLocations" value="classpath:mapper/*.xml"/> </bean>
另外 我的jdbc：perperties 配置檔案是
driver=com.mysql.jdbc.Driver url=jdbc:mysql://127.0.0.1:3306/course_design?useUnicode=true&characterEncoding=UTF-8 username=root password=root server_1_driver=com.mysql.jdbc.Driver server_1_url=jdbc:mysql://127.0.0.1:3306/course_design_test?useUnicode=true&characterEncoding=UTF-8 server_1_username=root server_1_password=root
RoutingDataSource類和DynamicDataSourceHolder類
public class RoutingDataSource extends AbstractRoutingDataSource{ public final Logger logger = LoggerFactory.getLogger(this.getClass()); @Override protected Object determineCurrentLookupKey() { String dataSourceName = DynamicDataSourceHolder.getDataSourceName(); if(dataSourceName == null){ dataSourceName = DataSourceType.BASE.getDataSource(); } logger.info("選擇的資料庫是："+dataSourceName); return dataSourceName; } }
public class DynamicDataSourceHolder { //解決執行緒安全問題 private static final ThreadLocal<String> holder = new ThreadLocal<String>(); public static void putDataSourceName(String dataName){ holder.set(dataName); } public static String getDataSourceName(){ return holder.get(); } public static class DataSourceName{ public final static String BASE = "base"; } }
我們可以通過在holder中 put資料庫的key值 來實現資料來源的動態載入，其中 ThreadLocal<String> holder 保證了執行緒安全，而 RoutingDataSource
類裡面的determineCurrentLookupKey（）方法返回值 就是資料來源的key值 配置檔案通過這個key值找到了當前的配置源連結。這個是的測試 我就不多演示。
那我們怎麼動態的進行資料鏈接呢，我們可以通過aop切面變成來實現
具體的aop切面程式設計的配置這裡不多做介紹，可以參考之前的記過博文 有介紹
三：AOP注入實現 資料來源的動態配置 首先需要第一一個切點。
package com.cdms.aop; import com.cdms.eunms.DataSourceType; import java.lang.annotation.*; /** * 建立 by 草帽boy on 2017/3/29. */ @Target({ElementType.PARAMETER, ElementType.METHOD}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface ChooseDataSource { /** * 資料庫名稱 預設似base * @return 資料庫名稱 */ String dataSourceName() default "base"; }
然後定義一個切面方法
package com.cdms.aop; import com.cdms.dao.resources.DynamicDataSourceHolder; import com.cdms.eunms.DataSourceType; import org.aspectj.lang.ProceedingJoinPoint; import org.aspectj.lang.annotation.*; import org.springframework.stereotype.Component; import java.lang.reflect.Method; /** * 建立 by 草帽boy on 2017/3/29. */ @Aspect @Component public class GetDataSource { /** * 切面設定要選擇的資料庫 * @param joinPoint * @return * @throws Throwable */ @Around("@annotation(com.cdms.aop.ChooseDataSource)") public Object permission(ProceedingJoinPoint joinPoint) throws Throwable { Object target = joinPoint.getTarget(); Object[] args = joinPoint.getArgs(); Method method = getMethod(joinPoint, args); //獲取資料庫名稱引數 ChooseDataSource chooseDataSource = method.getAnnotation(ChooseDataSource.class); if(chooseDataSource != null){ String dataSourceName = chooseDataSource.dataSourceName(); //堅持名稱是否合法 如果不合法 就用預設資料庫 if(DataSourceType.check(dataSourceName)){ DynamicDataSourceHolder.putDataSourceName(dataSourceName); }else { DynamicDataSourceHolder.putDataSourceName(DataSourceType.BASE.getDataSource()); } } return joinPoint.proceed(); } /** * 獲取切面方法 * @param joinPoint * @param args * @return * @throws NoSuchMethodException */ private Method getMethod(ProceedingJoinPoint joinPoint, Object[] args) throws NoSuchMethodException { // Class[] argsClazz = new Class[args.length]; // for(int i = 0 ; i < args.length; i++){ // argsClazz[i] = args[i].getClass(); // } String methodName = joinPoint.getSignature().getName(); Class clazz = joinPoint.getTarget().getClass(); Method[] methods = clazz.getMethods(); for(Method method : methods) { if (methodName.equals(method.getName())) { return method; } } return null; } }
<!-- 啟動@aspectj的自動代理支援--> <aop:aspectj-autoproxy expose-proxy="true"></aop:aspectj-autoproxy> <!-- 定義aspect類 --> <bean id="logAopAction" class="com.cdms.aop.LogAopAction"></bean> <bean id="aopTest" class="com.cdms.aop.DemoAopTest"></bean> <bean id="getDataSourcesName" class="com.cdms.aop.GetDataSource"></bean>
在之後需要改變資料來源的地方 只需要 @ChooseDataSource(dataSourceName = "server_1") 表明你的資料來源就看了
使用單元測試的結果是 ;(@ChooseDataSource(dataSourceName = "server_1") )
(@ChooseDataSource(dataSourceName = "base") )
這個是注入點：
