spring mvc 多數據源切換,不支持事務控制[一]

一個項目中需要使用兩個資料庫,Oracle 和Mysql ,於是參考各個blog,實現此功能.寫好後才發現,原來的事務失效了,我去...

spring-mybatis.xml 配置

    <bean id="configReader" class="org.springframework.beans.factory.config.PreferencesPlaceholderConfigurer">
        <property name="locations">
            <list>
                <value>classpath:spring/db.properties</value>
            </list>
        </property>
        <property name="ignoreResourceNotFound" value="true"/>
    </bean>

    <bean id="dataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource">
        <property name="driverClass" value="${jdbc.oracle.DriverClassName}"></property>
        <property name="jdbcUrl" value="${jdbc.oracle.Url}"></property>
        <property name="user" value="${jdbc.oracle.UserName}"></property>
        <property name="password" value="${jdbc.oracle.UserPassword}"></property>

        <property name="acquireIncrement" value="5"></property>
        <property name="initialPoolSize" value="5"></property>
        <property name="maxIdleTime" value="60"></property>
        <property name="maxPoolSize" value="100"></property>
        <property name="minPoolSize" value="5"></property>
    </bean>


    <!-- 配置數據源:MySQL start -->
    <bean name="mySqlDataSource" class="com.alibaba.druid.pool.DruidDataSource"
          init-method="init" destroy-method="close">
        <property name="driverClassName" value="${jdbc.mysql.DriverClassName}"/>
        <property name="url" value="${jdbc.mysql.Url}"/>
        <property name="username" value="${jdbc.mysql.UserName}"/>
        <property name="password" value="${jdbc.mysql.UserPassword}"/>

        <!-- 初始化連接大小 -->
        <property name="initialSize" value="5"/>
        <!-- 連接池最大使用連接數量 -->
        <property name="maxActive" value="30"/>
        <!-- 連接池最小空閑 -->
        <property name="minIdle" value="2"/>
        <!-- 獲取連接最大等待時間 -->
        <property name="maxWait" value="300"/>

        <property name="validationQuery" value="SELECT 1"/>
        <property name="testOnBorrow" value="false"/>
        <property name="testOnReturn" value="false"/>
        <property name="testWhileIdle" value="true"/>

        <!-- 配置間隔多久才進行一次檢測，檢測需要關閉的空閑連接，單位是毫秒 -->
        <property name="timeBetweenEvictionRunsMillis" value="10000"/>
        <!-- 配置一個連接在池中最小生存的時間，單位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="30000"/>

        <!-- 打開removeAbandoned功能 -->
        <property name="removeAbandoned" value="true"/>
        <!-- 1800秒，也就是30分鐘 -->
        <property name="removeAbandonedTimeout" value="1800"/>
        <!-- 關閉abanded連接時輸出錯誤日誌 -->
        <property name="logAbandoned" value="true"/>

        <!-- 監控資料庫 -->
        <property name="filters" value="stat"/>
    </bean>


    <bean id="multipleDataSource" class="com.we.database.MultipleDataSource">
        <property name="defaultTargetDataSource" ref="dataSource"/>
        <property name="targetDataSources">
            <map>
                <entry key="oracleDataSource" value-ref="dataSource"/>
                <entry key="mySqlDataSource" value-ref="mySqlDataSource"/>
            </map>
        </property>
    </bean>

    <!-- oracle myBatis file -->
    <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="dataSource" ref="multipleDataSource"/>
        <!--<property name="configLocation" value="classpath:configuration.xml" /> -->
        <property name="mapperLocations" value="classpath:com/we/dao/mapper/*.xml"/>
    </bean>

    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="basePackage" value="com.we.dao"/>
        <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory"/>
    </bean>


    <!-- configure transaction -->
    <bean id="transactionManager"
          class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSource"/>
    </bean>

    <!-- annotation transaction -->
    <tx:annotation-driven transaction-manager="transactionManager" proxy-target-class="true"/>


    <!-- interception transatcion -->
    <tx:advice id="transactionAdvice" transaction-manager="transactionManager">
        <tx:attributes>
            <tx:method name="add*" propagation="REQUIRED"/>
        </tx:attributes>
    </tx:advice>

    <!-- 配置資料庫註解aop -->
    <bean id="dataSourceAspect" class="com.we.database.DataSourceAspect"/>

    <aop:config>
        <aop:pointcut id="transactionPointcut" expression="execution(* com.wewe.licai.service..*Impl.*(..))"/>
        <aop:advisor pointcut-ref="transactionPointcut" advice-ref="transactionAdvice" order="2"/>

        <!--數據源選擇切麵,保證在事務開始之前執行-->
        <aop:advisor pointcut-ref="transactionPointcut" advice-ref="dataSourceAspect"  order="1" />
    </aop:config>

@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD,ElementType.TYPE})

public @interface DataSource {
     String name() default DataSource.oracleDataSource;
     String mySqlDataSource = "mySqlDataSource";
     String oracleDataSource = "oracleDataSource";
}

/**
 * Created by eastday on 2017/9/21.
 */
public class DataSourceAspect implements MethodBeforeAdvice,AfterReturningAdvice
{

    @Override
    public void afterReturning(Object returnValue, Method method,
                               Object[] args, Object target) throws Throwable {

        MultipleDataSource.clearDataSource();
    }

    @Override
    public void before(Method method, Object[] args, Object target)
            throws Throwable {

        //首先取類上的數據源
        if(method.getDeclaringClass().isAnnotationPresent(DataSource.class) && !method.isAnnotationPresent(DataSource.class)) {

            DataSource datasource = method.getDeclaringClass().getAnnotation(DataSource.class);
            MultipleDataSource.setDataSource(datasource.name());

            //方法上的數據源 優先順序高於類上的
        } else if (method.isAnnotationPresent(DataSource.class)) {

            DataSource datasource = method.getAnnotation(DataSource.class);
            MultipleDataSource.setDataSource(datasource.name());
        }
        else
        {
            MultipleDataSource.setDataSource(DataSource.oracleDataSource);
        }
    }
}

public class MultipleDataSource extends AbstractRoutingDataSource {
    private static final ThreadLocal<String> dataSources = new InheritableThreadLocal<String>();

    public static void setDataSource(String dataSource) {
        dataSources.set(dataSource);
    }

    //清除數據源
    public static void clearDataSource() {
        dataSources.remove();
    }

    @Override
    protected Object determineCurrentLookupKey() {
        return dataSources.get();
    }
}

@DataSource(name = DataSource.mySqlDataSource)
public class ContentServiceImpl implements IContentService {

    @Autowired
    private IContentDao contentDao;

    @Override
    public Content queryOne(String type) {
        return contentDao.queryOne(type);
    }
}