jeesite 多資料來源切換
阿新 • • 發佈:2019-01-10
jeesite 多資料來源切換
PS:本文參考網上資料後並加以修改,真實有效
先配置資料來源
jeesite.properties
#mysql database setting db1.type=mysql db1.driver=com.mysql.jdbc.Driver db1.url=jdbc:mysql://IP地址:埠/資料庫名?useUnicode=true&allowMultiQueries=true&characterEncoding=utf-8&allowMultiQueries=true db1.username=root db1.password=root #pool settings db1.pool.init=1 db1.pool.minIdle=3 db1.pool.maxActive=20 #jdbc.testSql=SELECT 'x' db1.testSql=SELECT 'x' FROM DUAL #mysql database setting db2.type=mysql db2.driver=com.mysql.jdbc.Driver db2.url=jdbc:mysql://IP地址:埠/資料庫名?useUnicode=true&allowMultiQueries=true&characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull db2.username=root db2.password=root #pool settings db2.pool.init=1 db2.pool.minIdle=3 db2.pool.maxActive=20 #jdbc.testSql=SELECT 'x' db2.testSql=SELECT 'x' FROM DUAL
spring-context.xml
<!-- 資料來源1配置, 使用 BoneCP 資料庫連線池 --> <bean id="dataSource1" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close"> <!-- 資料來源驅動類可不寫,Druid預設會自動根據URL識別DriverClass --> <property name="driverClassName" value="${jdbc.driver}" /> <!-- 基本屬性 url、user、password --> <property name="url" value="${jdbc.url}" /> <property name="username" value="${jdbc.username}" /> <property name="password" value="${jdbc.password}" /> <!-- 配置初始化大小、最小、最大 --> <property name="initialSize" value="${jdbc.pool.init}" /> <property name="minIdle" value="${jdbc.pool.minIdle}" /> <property name="maxActive" value="${jdbc.pool.maxActive}" /> <!-- 配置獲取連線等待超時的時間 --> <property name="maxWait" value="60000" /> <!-- 配置間隔多久才進行一次檢測,檢測需要關閉的空閒連線,單位是毫秒 --> <property name="timeBetweenEvictionRunsMillis" value="60000" /> <!-- 配置一個連線在池中最小生存的時間,單位是毫秒 --> <property name="minEvictableIdleTimeMillis" value="300000" /> <property name="validationQuery" value="${jdbc.testSql}" /> <property name="testWhileIdle" value="true" /> <property name="testOnBorrow" value="false" /> <property name="testOnReturn" value="false" /> <!-- 開啟PSCache,並且指定每個連線上PSCache的大小(Oracle使用) <property name="poolPreparedStatements" value="true" /> <property name="maxPoolPreparedStatementPerConnectionSize" value="20" /> --> <!-- 配置監控統計攔截的filters --> <property name="filters" value="stat" /> </bean> <!-- 資料來源2配置, 使用 BoneCP 資料庫連線池 --> <bean id="dataSource2" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close"> <!-- 資料來源驅動類可不寫,Druid預設會自動根據URL識別DriverClass --> <property name="driverClassName" value="${db.driver}" /> <!-- 基本屬性 url、user、password --> <property name="url" value="${db.url}" /> <property name="username" value="${db.username}" /> <property name="password" value="${db.password}" /> <!-- 配置初始化大小、最小、最大 --> <property name="initialSize" value="${db.pool.init}" /> <property name="minIdle" value="${db.pool.minIdle}" /> <property name="maxActive" value="${db.pool.maxActive}" /> <!-- 配置獲取連線等待超時的時間 --> <property name="maxWait" value="60000" /> <!-- 配置間隔多久才進行一次檢測,檢測需要關閉的空閒連線,單位是毫秒 --> <property name="timeBetweenEvictionRunsMillis" value="60000" /> <!-- 配置一個連線在池中最小生存的時間,單位是毫秒 --> <property name="minEvictableIdleTimeMillis" value="300000" /> <property name="validationQuery" value="${db.testSql}" /> <property name="testWhileIdle" value="true" /> <property name="testOnBorrow" value="false" /> <property name="testOnReturn" value="false" /> <!-- 開啟PSCache,並且指定每個連線上PSCache的大小(Oracle使用) <property name="poolPreparedStatements" value="true" /> <property name="maxPoolPreparedStatementPerConnectionSize" value="20" /> --> <!-- 配置監控統計攔截的filters --> <property name="filters" value="stat" /> </bean> <!-- 設定資料來源key值,切換資料來源時使用 --> <bean id="dataSource" class="com.dataera.modules.datamigration.ChooseDataSource"> <property name="targetDataSources"> <map key-type="java.lang.String"> <entry value-ref="dataSource1" key="first"></entry> <entry value-ref="dataSource2" key="second"></entry> </map> </property> <property name="defaultTargetDataSource" ref="dataSource1"> </property> </bean>
新建四個操作資料來源的類
選擇資料來源
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; public class ChooseDataSource extends AbstractRoutingDataSource{ private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>(); public static String getCurrentLookupKey() { return (String) contextHolder.get(); } public static void setCurrentLookupKey(String currentLookupKey) { contextHolder.set(currentLookupKey); } @Override protected Object determineCurrentLookupKey() { return getCurrentLookupKey(); } }
確定要操作的資料來源
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* 用註解來反射讀取註解中輸入的資料來源以便確定要操作的資料來源
*
*
*/
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface DataSource {
String value();
}
獲取資料來源
import java.lang.reflect.Method;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.reflect.MethodSignature;
public class DataSourceAspect {
public void pointCut(){};
/**
* 在事務開始之前插入到service上,通過反射獲取註解上的資料庫
* @param point
*/
@Before(value="pointCut()")
public void before(JoinPoint point){
//攔截的實體類
Object target = point.getTarget();
//攔截的方法名稱
String methodName = point.getSignature().getName();
Class<?>[] classes = target.getClass().getInterfaces();
//攔截的放參數型別
Class<?>[] parameterTypes = ((MethodSignature) point.getSignature())
.getMethod().getParameterTypes();
try {
Method m = classes[0].getMethod(methodName, parameterTypes);
if (m != null && m.isAnnotationPresent(DataSource.class)) {
DataSource data = m.getAnnotation(DataSource.class);
HandleDataSource.setDataSource(data.value());
}
} catch (Exception e) {
e.printStackTrace();
}
}
}
資料來源操作
public class HandleDataSource {
public static final ThreadLocal<String> holder = new ThreadLocal<String>();
public static void setDataSource(String dataSource){
holder.set(dataSource);
}
public static String getDataSource(){
return holder.get();
}
public static void clearDataSource() {
holder.remove();
}
}
切換資料來源
controller層
ChooseDataSource.setCurrentLookupKey("second"); //spring-context.xml裡的資料來源key值