Spring多数据源AOP动态切换怎么实现-创新互联

这篇文章主要讲解了“Spring多数据源AOP动态切换怎么实现”，文中的讲解内容简单清晰，易于学习与理解，下面请大家跟着小编的思路慢慢深入，一起来研究和学习“Spring多数据源AOP动态切换怎么实现”吧！

一：新增多数据源类

1. public class DynamicDataSource extends AbstractRoutingDataSource {

3.     @Override

4.     protected Object determineCurrentLookupKey() {

5.         return DataSourceContextHolder.getDataSource();

6.     }

8. }

点击(此处)折叠或打开

1. public class DataSourceContextHolder {

3.     private static final ThreadLocal contextHolder = new ThreadLocal();

5.     public static void setDataSource(String dataSource) {

6.         contextHolder.set(dataSource);

7.     }

9.     public static String getDataSource() {

10.         return contextHolder.get();

11.     }

13. }

二：新增注解

点击(此处)折叠或打开

1. @Retention(RetentionPolicy.RUNTIME)

2. @Target(ElementType.METHOD)

3. @Documented

4. public @interface DataSource {

5.     String value();

6. }

三：新增AOP切面

点击(此处)折叠或打开

1. @Aspect

2. @Component

3. public class DataSourceAspect {

5.     @Pointcut("@annotation(com.gemdale.ghome.business.async.deal.center.demo.datasource.DataSource)")

6.     public void dataSourcePointCut() {

7.     };

9.     @Before("dataSourcePointCut()")

10.     public void before(JoinPoint joinPoint) {

12.         System.out.println("=============dataSourcePointCut:before=============");

14.         Object target = joinPoint.getTarget();

15.         String method = joinPoint.getSignature().getName();

17.         // Class[] classz = target.getClass().getInterfaces();

18.         Class classz = target.getClass();

19.         Class[] parameterTypes = ((MethodSignature) joinPoint.getSignature()).getMethod().getParameterTypes();

21.         try {

22.             // Method m = classz[0].getMethod(method, parameterTypes);

23.             Method m = classz.getMethod(method, parameterTypes);

24.             if (null != m && m.isAnnotationPresent(DataSource.class)) {

25.                 DataSource dataSource = m.getAnnotation(DataSource.class);

26.                 DataSourceContextHolder.setDataSource(dataSource.value());

28.                 System.out.println("=============dataSource:" + dataSource.value());

29.             }

30.         }

31.         catch (Exception e) {

32.             e.printStackTrace();

33.         }

34.     }

36. }

四：数据源配置

点击(此处)折叠或打开

1. @Configuration

2. public class DynamicTransactionManagerElConfig {

4.     @Autowired

5.     @Qualifier("platformTomcat")

6.     private DataSource platformTomcat;

8.     @Autowired

9.     @Qualifier("platformReadTomcat")

10.     private DataSource platformReadTomcat;

12.     @Bean(name = "dataSource")

13.     public DynamicDataSource dataSource() {

14.         DynamicDataSource dataSource = new DynamicDataSource();

15.         Map targetDataSources = new HashMap<>();

16.         targetDataSources.put("master", platformTomcat);

17.         targetDataSources.put("slave", platformReadTomcat);

18.         dataSource.setTargetDataSources(targetDataSources);

19.         dataSource.setDefaultTargetDataSource(platformTomcat);

20.         return dataSource;

21.     }

22.     

23.     

24.     @Bean(name = "jdbcTemplate")

25.     public JdbcTemplate jdbcTemplate(DynamicDataSource dataSource) {

26.         JdbcTemplate jdbcTemplate = new JdbcTemplate();

27.         jdbcTemplate.setDataSource(dataSource);

28.         return jdbcTemplate;

29.     }

31.     @Bean(name = "jdbcReadTemplate")

32.     public JdbcTemplate jdbcReadTemplate(DynamicDataSource dataSource) {

33.         JdbcTemplate jdbcReadTemplate = new JdbcTemplate();

34.         jdbcReadTemplate.setDataSource(dataSource);

35.         return jdbcReadTemplate;

36.     }

37.     

39.     @Bean(name = "transactionManager")

40.     public DataSourceTransactionManager transactionManager(DynamicDataSource dataSource) {

41.         DataSourceTransactionManager transactionManager = new DataSourceTransactionManager();

42.         transactionManager.setDataSource(dataSource);

43.         return transactionManager;

44.     }

46. }

五：应用举例

点击(此处)折叠或打开

1. @Service("gmcSmsInfoBo")

2. public class GmcSmsInfoBo extends AbstractBusinessObject {

4.     @Autowired

5.     private GmcSmsInfoDAO gmcSmsInfoDaoImpl;

7.     // @CachePut(value = "GmcSmsInfoCache", key = "'GmcSmsInfo_'+#result.smsId")

8.     // @Transactional(rollbackFor={Exception.class,RuntimeException.class})

9.     @DataSource("master")

10.     public GmcSmsInfo add(GmcSmsInfo smsInfo) throws BusinessServiceException {

11.         System.out.println("=============add==========");

12.         try {

13.             smsInfo.setSmsId(gmcSmsInfoDaoImpl.save(smsInfo));

14.         }

15.         catch (FrameworkDAOException e) {

16.             throw new BusinessServiceException(e);

17.         }

18.         return smsInfo;

19.     }

21.     // @Cacheable(value="GmcSmsInfoCache",key="'GmcSmsInfo_'+#smsId")

22.     @DataSource("slave")

23.     public GmcSmsInfo query(Integer smsId) throws BusinessServiceException {

24.         System.out.println("=============query==========");

25.         try {

26.             return gmcSmsInfoDaoImpl.findById(GmcSmsInfo.class, smsId);

27.         }

28.         catch (Exception e) {

29.             throw new BusinessServiceException(e);

30.         }

31.     }

33. }

感谢各位的阅读，以上就是“Spring多数据源AOP动态切换怎么实现”的内容了，经过本文的学习后，相信大家对Spring多数据源AOP动态切换怎么实现这一问题有了更深刻的体会，具体使用情况还需要大家实践验证。这里是创新互联，小编将为大家推送更多相关知识点的文章，欢迎关注！