文章詳情頁

Spring-基于Spring使用自定義注解及Aspect實現數據庫切換操作

瀏覽：86日期：2023-08-13 11:44:07

實現思路

重寫Spring的AbstractRoutingDataSource抽象類的determineCurrentLookupKey方法。

我們來看下Spring-AbstractRoutingDataSource的源碼

AbstractRoutingDataSource獲取數據源之前會先調用determineCurrentLookupKey方法查找當前的lookupKey。

Object lookupKey = determineCurrentLookupKey();DataSource dataSource = this.resolvedDataSources.get(lookupKey);.......return dataSource;

lookupKey為數據源標識，因此通過重寫這個查找數據源標識的方法就可以讓spring切換到指定的數據源.

從變量定義中可以知道resolvedDataSources為Map類型的對象。

private Map<Object, DataSource> resolvedDataSources;

示例

Spring-基于Spring使用自定義注解及Aspect實現數據庫切換操作

步驟一新建Maven工程

依賴如下： pom.xml

<project xmlns='http://maven.apache.org/POM/4.0.0' xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance'xsi:schemaLocation='http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd'><modelVersion>4.0.0</modelVersion><groupId>com.artisan</groupId><artifactId>dynamicDataSource</artifactId><version>0.0.1-SNAPSHOT</version><packaging>jar</packaging><name>dynamicDataSource</name><url>http://maven.apache.org</url><properties><project.build.sourceEncoding>UTF-8</project.build.sourceEncoding><file.encoding>UTF-8</file.encoding><spring.version>4.3.9.RELEASE</spring.version><servlet.version>3.1.0</servlet.version><aspectj.version>1.8.1</aspectj.version><commons-dbcp.version>1.4</commons-dbcp.version><jetty.version>8.1.8.v20121106</jetty.version><log4j.version>1.2.17</log4j.version><log4j2.version>2.8.2</log4j2.version><testng.version>6.8.7</testng.version><oracle.version>11.2.0.4.0</oracle.version><jstl.version>1.2</jstl.version></properties><dependencies><dependency><groupId>org.springframework</groupId><artifactId>spring-beans</artifactId><version>${spring.version}</version></dependency><dependency><groupId>org.springframework</groupId><artifactId>spring-context</artifactId><version>${spring.version}</version></dependency><dependency><groupId>org.springframework</groupId><artifactId>spring-context-support</artifactId><version>${spring.version}</version></dependency><dependency><groupId>org.springframework</groupId><artifactId>spring-jdbc</artifactId><version>${spring.version}</version></dependency><dependency><groupId>org.springframework</groupId><artifactId>spring-webmvc</artifactId><version>${spring.version}</version></dependency><dependency><groupId>commons-dbcp</groupId><artifactId>commons-dbcp</artifactId><version>${commons-dbcp.version}</version></dependency><dependency><groupId>org.aspectj</groupId><artifactId>aspectjweaver</artifactId><version>${aspectj.version}</version></dependency><dependency><groupId>org.testng</groupId><artifactId>testng</artifactId><version>${testng.version}</version><scope>test</scope></dependency><dependency><groupId>org.springframework</groupId><artifactId>spring-test</artifactId><version>${spring.version}</version><scope>test</scope></dependency><dependency><groupId>com.oracle</groupId><artifactId>ojdbc6</artifactId><version>${oracle.version}</version></dependency><dependency><groupId>org.testng</groupId><artifactId>testng</artifactId><version>${testng.version}</version><scope>test</scope></dependency><dependency><groupId>org.springframework</groupId><artifactId>spring-test</artifactId><version>${spring.version}</version><scope>test</scope></dependency><dependency><groupId>org.apache.logging.log4j</groupId><artifactId>log4j-api</artifactId><version>${log4j2.version}</version></dependency><dependency><groupId>org.apache.logging.log4j</groupId><artifactId>log4j-core</artifactId><version>${log4j2.version}</version></dependency></dependencies><build><plugins><plugin><groupId>org.apache.maven.plugins</groupId><artifactId>maven-compiler-plugin</artifactId><version>3.1</version><configuration><source>1.7</source><target>1.7</target></configuration></plugin></plugins></build></project>

步驟二 繼承AbstractRoutingDataSource并重寫determineCurrentLookupKey方法獲取特定數據源

package com.artisan.dynamicDB;import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;/** * * * @ClassName: DynamicDataSource * * @Description: * AbstractRoutingDataSource中的抽象方法determineCurrentLookupKey是實現數據源的route的核心 * .需要重寫該方法 * * @author: Mr.Yang * * @date: 2017年7月24日下午8:28:46 */public class DynamicDataSource extends AbstractRoutingDataSource {@Overrideprotected Object determineCurrentLookupKey() {return DynamicDataSourceHolder.getDataSource();}}

步驟三 創建DynamicDataSourceHolder用于持有當前線程中使用的數據源標識

package com.artisan.dynamicDB;/** * * * @ClassName: DynamicDataSourceHolder * * @Description:創建DynamicDataSourceHolder用于持有當前線程中使用的數據源標識 * * @author: Mr.Yang * * @date: 2017年7月24日下午8:23:50 */public class DynamicDataSourceHolder { /** * 數據源標識保存在線程變量中，避免多線程操作數據源時互相干擾 */ private static final ThreadLocal<String> dataSourceHolder = new ThreadLocal<String>(); /** * * * @Title: setDataSource * * @Description: 設置數據源 * * @param dataSource * * @return: void */ public static void setDataSource(String dataSource) { dataSourceHolder.set(dataSource); } /** * * * @Title: getDataSource * * @Description: 獲取數據源 * * @return * * @return: String */ public static String getDataSource() { return dataSourceHolder.get(); } /** * * * @Title: clearDataSource * * @Description: 清除數據源 * * * @return: void */ public static void clearDataSource() { dataSourceHolder.remove(); }}

步驟四 配置多個數據源和DynamicDataSource的bean

<?xml version='1.0' encoding='UTF-8' ?><beans xmlns='http://www.springframework.org/schema/beans' xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance' xmlns:p='http://www.springframework.org/schema/p' xmlns:context='http://www.springframework.org/schema/context' xmlns:aop='http://www.springframework.org/schema/aop' xmlns:tx='http://www.springframework.org/schema/tx' xmlns:util='http://www.springframework.org/schema/util' xsi:schemaLocation='http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context.xsd http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx.xsd http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop.xsd http://www.springframework.org/schema/util http://www.springframework.org/schema/util/spring-util.xsd'>  <context:component-scan base-package='com.artisan'/>  <context:property-placeholder location='classpath:jdbc.properties' />   <bean destroy-method='close' p:driverClassName='${jdbc.driverClassNamePR}' p:url='${jdbc.urlPR}' p:username='${jdbc.usernamePR}' p:password='${jdbc.passwordPR}' />  <bean destroy-method='close' p:driverClassName='${jdbc.driverClassNameDR}' p:url='${jdbc.urlDR}' p:username='${jdbc.usernameDR}' p:password='${jdbc.passwordDR}' />  <bean destroy-method='close' p:driverClassName='${jdbc.driverClassNameCC}' p:url='${jdbc.urlCC}' p:username='${jdbc.usernameCC}' p:password='${jdbc.passwordCC}' /> <bean class='com.artisan.dynamicDB.DynamicDataSource'> <property name='targetDataSources' ref='dynamicDatasourceMap' />  <property name='defaultTargetDataSource' ref='dataSourcePR' /> </bean>  <util:map key-type='java.lang.String'> <entry key='dataSourcePR' value-ref='dataSourcePR' /> <entry key='dataSourceDR' value-ref='dataSourceDR' /> <entry key='dataSourceCC' value-ref='dataSourceCC' /> </util:map>  <bean p:dataSource-ref='dynamicDataSource' />  <bean /> <aop:config> <aop:aspect ref='dataSourceAspect'>  <aop:pointcut expression='execution(* com.artisan..*(..))'/> <aop:before pointcut-ref='dataSourcePointcut' method='intercept' /> </aop:aspect> </aop:config>  <bean p:dataSource-ref='dynamicDataSource' />  <aop:config proxy-target-class='true'> <aop:pointcut expression='(execution(* com.artisan..*(..))) and (@annotation(org.springframework.transaction.annotation.Transactional))' /> <aop:advisor pointcut-ref='serviceMethod' advice-ref='txAdvice' /> </aop:config> <tx:advice transaction-manager='transactionManager'> <tx:attributes> <tx:method name='*' /> </tx:attributes> </tx:advice></beans>

配置到這里，我們就可以使用多個數據源了，只需要在操作數據庫之前只要DynamicDataSourceHolder.setDataSource(“dataSourcePR”)即可切換到數據源dataSourcePR并對數據庫dataSourcePR進行操作了。

問題：每次使用都需要調用DynamicDataSourceHolder#setDataSource，十分繁瑣，并且難以維護。

我們可以通過Spring的AOP和注解，直接通過注解的方式指定需要訪問的數據源。繼續改進下吧

步驟五 定義名為@DataSource的注解

package com.artisan.dynamicDB;import java.lang.annotation.Documented;import java.lang.annotation.ElementType;import java.lang.annotation.Retention;import java.lang.annotation.RetentionPolicy;import java.lang.annotation.Target;/** * * * @ClassName: DataSource * * * @Description: 注解@DataSource既可以加在方法上，也可以加在接口或者接口的實現類上，優先級別：方法>實現類>接口。 * 如果接口、接口實現類以及方法上分別加了@DataSource注解來指定數據源，則優先以方法上指定的為準。 * * @author: Mr.Yang * * @date: 2017年7月24日下午9:59:29 */@Target({ ElementType.METHOD, ElementType.TYPE })@Retention(RetentionPolicy.RUNTIME)@Documentedpublic @interface DataSource { // 和配置文件中 dynamicDatasourceMap中的key保持一致 public static String PR_RB = 'dataSourcePR'; public static String DR_RB = 'dataSourceDR'; public static String PR_CC = 'dataSourceCC'; /** * * * @Title: name * * @Description: 如果僅標注@DataSource 默認為PR_RB數據庫實例 * * @return * * @return: String */ String name() default DataSource.PR_RB;}

步驟六 定義AOP切面以便攔截所有帶有注解@DataSource的方法，取出注解的值作為數據源標識放到DynamicDataSourceHolder的線程變量中

package com.artisan.dynamicDB;import java.lang.reflect.Method;import org.aspectj.lang.JoinPoint;import org.aspectj.lang.reflect.MethodSignature;/** * * * @ClassName: DataSourceAspect * * @Description: * 定義AOP切面以便攔截所有帶有注解@DataSource的方法，取出注解的值作為數據源標識放到DBContextHolder的線程變量中 * * @author: Mr.Yang * * @date: 2017年7月25日上午10:51:41 */public class DataSourceAspect { /** * * * @Title: intercept * * @Description: 攔截目標方法，獲取由@DataSource指定的數據源標識，設置到線程存儲中以便切換數據源 * * @param point * @throws Exception * * @return: void */ public void intercept(JoinPoint point) throws Exception { Class<?> target = point.getTarget().getClass(); MethodSignature signature = (MethodSignature) point.getSignature(); // 默認使用目標類型的注解，如果沒有則使用其實現接口的注解 for (Class<?> clazz : target.getInterfaces()) { resolveDataSource(clazz, signature.getMethod()); } resolveDataSource(target, signature.getMethod()); } /** * * * @Title: resolveDataSource * * @Description: 提取目標對象方法注解和類型注解中的數據源標識 * * @param clazz * @param method * * @return: void */ private void resolveDataSource(Class<?> clazz, Method method) { try { Class<?>[] types = method.getParameterTypes(); // 默認使用類型注解 if (clazz.isAnnotationPresent(DataSource.class)) { DataSource source = clazz.getAnnotation(DataSource.class); DynamicDataSourceHolder.setDataSource(source.name()); } // 方法注解可以覆蓋類型注解 Method m = clazz.getMethod(method.getName(), types); if (m != null && m.isAnnotationPresent(DataSource.class)) { DataSource source = m.getAnnotation(DataSource.class); DynamicDataSourceHolder.setDataSource(source.name()); } } catch (Exception e) { System.out.println(clazz + ':' + e.getMessage()); } }}

步驟七 在spring配置文件中配置攔截規則

<bean /> <aop:config> <aop:aspect ref='dataSourceAspect'>  <aop:pointcut expression='execution(* com.artisan..*(..))'/> <aop:before pointcut-ref='dataSourcePointcut' method='intercept' /> </aop:aspect> </aop:config>

步驟八 使用注解切換多數據源

ExtractDataService.javapackage com.artisan.extractService;import java.sql.ResultSet;import java.sql.SQLException;import org.apache.logging.log4j.LogManager;import org.apache.logging.log4j.Logger;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.jdbc.core.JdbcTemplate;import org.springframework.jdbc.core.RowCallbackHandler;import org.springframework.stereotype.Service;import com.artisan.dynamicDB.DataSource;/** * * * @ClassName: ExtractDataService * * @Description: 業務類，這里暫時作為測試多數據源切換用 * * @author: Mr.Yang * * @date: 2017年7月24日下午9:07:38 */@Servicepublic class ExtractDataService { private static final Logger logger = LogManager .getLogger(ExtractDataService.class.getName()); private JdbcTemplate jdbcTemplate; @Autowired public void setJdbcTemplate(JdbcTemplate jdbcTemplate) { this.jdbcTemplate = jdbcTemplate; } /** * * * @Title: selectDataFromPR * * @Description: * * * @return: void */ @DataSource(name = DataSource.PR_RB) public void selectDataFromPR_RB() { String sql = 'select subs_id from owe_event_charge where event_inst_id = 10229001 '; jdbcTemplate.query(sql, new RowCallbackHandler() { @Override public void processRow(ResultSet rs) throws SQLException { logger.info(rs.getInt('subs_id')); } }); } @DataSource(name = DataSource.DR_RB) public void selectDataFromDR_RB() { // 改為通過注解指定DB // DynamicDataSourceHolder.setDataSource(DBContextHolder.DATA_SOURCE_DR); String sql = ' select a.task_comments from nm_task_type a where a.task_name = ’ALARM_LOG_LEVEL’ '; jdbcTemplate.query(sql, new RowCallbackHandler() { @Override public void processRow(ResultSet rs) throws SQLException { logger.info(rs.getString('task_comments')); } }); } @DataSource(name = DataSource.PR_CC) public void selectDataFromPR_CC() { // DBContextHolder.setDataSource(DBContextHolder.DATA_SOURCE_CC); String sql = 'select acc_nbr from acc_nbr where acc_nbr_id = 82233858 '; jdbcTemplate.query(sql, new RowCallbackHandler() { @Override public void processRow(ResultSet rs) throws SQLException { logger.info(rs.getString('acc_nbr')); } }); }}

步驟九測試

package com.artisan;import java.io.IOException;import org.apache.logging.log4j.LogManager;import org.apache.logging.log4j.core.LoggerContext;import org.springframework.context.ApplicationContext;import org.springframework.context.support.ClassPathXmlApplicationContext;import org.springframework.core.io.Resource;import org.springframework.core.io.ResourceLoader;import org.springframework.core.io.support.PathMatchingResourcePatternResolver;import com.artisan.extractService.ExtractDataService;/** * * * @ClassName: App * * @Description: 入口類 * * @author: Mr.Yang * * @date: 2017年7月24日下午8:50:25 */public class App { public static void main(String[] args) { try { // 加載日志框架 log4j2 LoggerContext context = (LoggerContext) LogManager .getContext(false); ResourceLoader loader = new PathMatchingResourcePatternResolver(); Resource resource = loader.getResource('classpath:log4j2.xml'); context.setConfigLocation(resource.getFile().toURI()); // 加載spring配置信息 ApplicationContext ctx = new ClassPathXmlApplicationContext( 'classpath:spring-context.xml'); // 從容器中獲取Bean ExtractDataService service = ctx.getBean('extractDataService', ExtractDataService.class); // 從PR的RB實例中獲取數據 service.selectDataFromPR_RB(); // 從DR的RB實例中獲取數據 service.selectDataFromDR_RB(); // 從PR的CC實例中獲取數據 service.selectDataFromPR_CC(); } catch (IOException e) { e.printStackTrace(); } }}

其他代碼

log4j2.xml

<?xml version='1.0' encoding='UTF-8'?><configuration status='info' monitorInterval='180'>  <Properties> <Property name='backupFilePatch'>D:/workspace/workspace-sts/backupOracle/log/</Property> <Property name='fileName'>backupOracle.log</Property> </Properties>  <appenders>  <Console name='Console' target='SYSTEM_OUT'>  <ThresholdFilter level='trace' onMatch='ACCEPT' onMismatch='DENY' />  <PatternLayout pattern='%d{HH:mm:ss.SSS} %-5level %class{36} %L %M - %msg%xEx%n' /> </Console>  <RollingFile name='RollingFile' fileName='${backupFilePatch}${fileName}' filePattern='${backupFilePatch}$${date:yyyy-MM}/app-%d{yyyyMMddHHmmssSSS}.log.gz'> <PatternLayout pattern='%d{yyyy.MM.dd ’at’ HH:mm:ss.SSS z} %-5level %class{36} %L %M - %msg%xEx%n' />  <SizeBasedTriggeringPolicy size='20MB' />  <DefaultRolloverStrategy max='20'/> </RollingFile> </appenders>  <loggers>  <logger name='org.springframework' level='INFO'></logger> <logger name='org.mybatis' level='INFO'></logger> <root level='trace'> <appender-ref ref='RollingFile'/> <appender-ref ref='Console'/> </root> </loggers></configuration>

jdbc.properties

################################ dbcp datasource pool ,basic configuration first.## the other parameters keep default for now , you can change them if you want ###############################Database in Lapazjdbc.driverClassNamePR=oracle.jdbc.driver.OracleDriverjdbc.urlPR=jdbc:oracle:thin:@172.25.243.4:1521:xxjdbc.usernamePR=xxxjdbc.passwordPR=xxxxxxxx#Database in Scluzjdbc.driverClassNameDR=oracle.jdbc.driver.OracleDriverjdbc.urlDR=jdbc:oracle:thin:@172.25.246.1:1521:xxjdbc.usernameDR=xxxjdbc.passwordDR=xxxxxxx#Database in Lapazjdbc.driverClassNameCC=oracle.jdbc.driver.OracleDriverjdbc.urlCC=jdbc:oracle:thin:@172.25.243.3:1521:xxjdbc.usernameCC=xxxjdbc.passwordCC=xxxxxx

運行結果：

Spring-基于Spring使用自定義注解及Aspect實現數據庫切換操作

代碼

https://github.com/yangshangwei/DynamicDataSource

以上這篇Spring-基于Spring使用自定義注解及Aspect實現數據庫切換操作就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持好吧啦網。

Spring

上一條：Spring Security 中細化權限粒度的方法下一條：spring Retryable注解實現重試詳解

相關文章：

1. XHTML 1.0：標記新的開端2. 詳解php如何合并身份證正反面圖片為一張圖片3. SpringBoot+SpringCache實現兩級緩存(Redis+Caffeine)4. Android SeekBar實現平滑滾動5. java必懂的冷知識點之Base64加密與解密6. 詳解android是如何管理內存的7. Java基于redis和mysql實現簡單的秒殺(附demo)8. AJAX實現省市縣三級聯動效果9. php設計模式之模板模式實例分析【星際爭霸游戲案例】10. ASP.NET MVC視圖頁使用jQuery傳遞異步數據的幾種方式詳解

排行榜

					
					java必懂的冷知識點之Base64加密與解密
Java基于redis和mysql實現簡單的秒殺(附demo)
php設計模式之模板模式實例分析【星際爭霸游戲案例】
詳解springBoot啟動時找不到或無法加載主類解決辦法
AJAX實現省市縣三級聯動效果
SpringBoot+SpringCache實現兩級緩存(Redis+Caffeine)
Android SeekBar實現平滑滾動
XHTML 1.0：標記新的開端
springboot集成與使用Sentinel的方法
IntelliJ IDEA 2020最新激活碼(親測有效，可激活至 2089 年)
ASP.NET MVC視圖頁使用jQuery傳遞異步數據的幾種方式詳解
				