<转>Spring AOP根据JdbcTemplate方法名动态设置数据源

转载: Spring AOP根据JdbcTemplate方法名动态设置数据源

http://blog.youkuaiyun.com/yubaoma2014/article/details/12427885
作者: yubaoma2014

 

有删节.

 

目的: 

1. 配置多个数据源

2. 根据不同的数据源执行不同的数据操作方法

3. 事务管理?

 

• 多数据源配置

      <!-- 主数据源, 用于执行写入操作 -->
      <bean id="masterDataSource"
          class="org.apache.commons.dbcp.BasicDataSource"
          destroy-method="close">
          <property name="driverClassName"
              value="${jdbc.driverClassName}" />
          <property name="url" value="${jdbc.url}" />
          <property name="username" value="${jdbc.username}" />
          <property name="password" value="${jdbc.password}" />
          <property name="poolPreparedStatements" value="true" />
          <property name="defaultAutoCommit" value="true" />
      </bean>
      <!-- 从数据源, 用于执行查询操作 -->
      <bean id="slaveDataSource"
          class="org.apache.commons.dbcp.BasicDataSource"
          destroy-method="close">
          <property name="driverClassName"
              value="${jdbc.driverClassName}" />
          <property name="url" value="${jdbc.url2}" />
          <property name="username" value="${jdbc.username}" />
          <property name="password" value="${jdbc.password}" />
          <property name="poolPreparedStatements" value="true" />
          <property name="defaultAutoCommit" value="true" />
      </bean>
  
      <!-- 动态数据源 -->
      <bean id="dataSource"
          class="test.my.serivce.ds.DynamicDataSource">
          <!-- 配置一个数据源Map, 注意这里的key值, 后面很多地方都使用到 -->
          <property name="targetDataSources">
              <map>
                  <entry key="master" value-ref="masterDataSource" />
                  <entry key="slave" value-ref="slaveDataSource" />
              </map>
          </property>
          <!-- 动态数据源默认使用的数据源 -->
          <property name="defaultTargetDataSource" ref="masterDataSource" />
      </bean>
  
  <!-- 配置JdbcTemplate -->
  <bean id="jdbcTemplate"
          class="org.springframework.jdbc.core.JdbcTemplate">
      <property name="dataSource" ref="dataSource" />
  </bean>

  BEAN "dataSource" 是自定义类DynamicDataSource的一个实例, 而DynamicDataSource则是继承了Spring的抽象类AbstractRoutingDataSource, 顾名思义, 这是一个数据源的路由/查找类.

  import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
  // 自定义类
  public class DynamicDataSource extends AbstractRoutingDataSource {
      
      @Override
      // 返回当前需要使用的数据源的key
      protected Object determineCurrentLookupKey() {
          // 
          return CustomerContextHolder.getCustomerType();
      }
  }

   

  //  AbstractRoutingDataSource 的部分源码
  
  public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {
          
          // 重写了 getConnection() 方法
       	@Override
  	public Connection getConnection() throws SQLException {
  		return determineTargetDataSource().getConnection();
  	}
  
          // 设置当前使用的目标数据源
  	protected DataSource determineTargetDataSource() {
  		Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
                  // 目标数据源的key.
  		Object lookupKey = determineCurrentLookupKey();
                  // 获取数据源
  		DataSource dataSource = this.resolvedDataSources.get(lookupKey);
  		if (dataSource == null && (this.lenientFallback || lookupKey == null)) {
  			dataSource = this.resolvedDefaultDataSource;
  		}
  		if (dataSource == null) {
  			throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");
  		}
  		return dataSource;
  	}
          
          // 抽象方法, 返回目标数据源对应的key
          // 这里的key对应于在BEAN "dataSource"中配置的属性"targetDataSources"中的key
          protected abstract Object determineCurrentLookupKey();
  }

   

  // CustomerContextHolder, 一个ThreadLocal实现, 用于管理key
  // ThreadLocal是线程安全的, 且线程之间不能共享
  public class CustomerContextHolder {
      private static final ThreadLocal contextHolder = new ThreadLocal();
      public static void setCustomerType(String customerType) {
          contextHolder.set(customerType);
      }
      public static String getCustomerType() {
          return (String) contextHolder.get();
      }
      public static void clearCustomerType() {
          contextHolder.remove();
      }
  }

   
  
• 利用AOP, 实现不同的数据源执行不同的数据操作方法

      <bean id="ba" class="test.my.serivce.ds.BeforeAdvice" />
      <!-- 强制使用CGLIB代理 -->
      <aop:config proxy-target-class="true">
  
          <!-- 切面一 主数据源执行写入操作 -->
          <aop:aspect ref="ba">
              <!-- 切点, 与数据库update相关的函数 -->
              <aop:pointcut id="update"
                  expression="execution(* org.springframework.jdbc.core.JdbcTemplate.update*(..)) || execution(* org.springframework.jdbc.core.JdbcTemplate.batchUpdate(..))" />
              <!-- 前置通知, 执行数据源设置操作 -->
              <aop:before method="setMasterDataSource"
                  pointcut-ref="update" />
          </aop:aspect>
  
          <!-- 切面二 从数据源执行查询操作 -->
          <aop:aspect ref="ba">
              <aop:before method="setSlaveDataSource"
                  pointcut="execution(* org.springframework.jdbc.core.JdbcTemplate.query*(..)) || execution(* org.springframework.jdbc.core.JdbcTemplate.execute(..))" />
          </aop:aspect>
      </aop:config>

      
   BEAN "ba"相关的类

  public class BeforeAdvice {
      public void setMasterDataSource() {
          CustomerContextHolder.setCustomerType("master");
      }
      public void setSlaveDataSource() {
          CustomerContextHolder.setCustomerType("slave");
      }
  }