Spring 多資料來源 AOP 動態切換

百聯達發表於2017-04-06
Spring
一：新增多資料來源類


點選(此處)摺疊或開啟

  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<String> contextHolder = new ThreadLocal<String>();

  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<Object, Object> 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. }

來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/28624388/viewspace-2136769/，如需轉載，請註明出處，否則將追究法律責任。

相關文章