java 通过继承类AbstractRoutingDataSource 而实现的 多数据源切换 的 缺陷（二）

最近在项目开发中，出现了一些见鬼的问题。保存或删除某些操作，bug的出现是偶发性出现，bug 还讲 概率事件啊，伤不起啊。在刚开始开发中，也偶尔出现，但是当时还不以为然。可是，项目上线后，用户也反映这个问题。 唉，这不是严重影响用户体验么。看来，这已经成了一个必然的bug， 不改不行了。

看来， Java中 偶发性 bug，多半跟 多线程 有关系。

先贴代码

spring-context.xml spring 的配置文件

<bean id="yyzy" class="org.springframework.jndi.JndiObjectFactoryBean">
<property name="jndiName">
<value>yyzy_bwcs</value>
<!-- <value>java:comp/env/jndi/yyzy_db2</value> -->
</property>
</bean>
<bean id="metadata" class="org.springframework.jndi.JndiObjectFactoryBean">
<property name="jndiName">
<value>meta_db2</value>
<!-- <value>java:comp/env/jndi/meta_db2</value> -->
</property>
</bean>

<!-- 交换平台 -->
<bean id="jhpt" class="org.springframework.jndi.JndiObjectFactoryBean">
<property name="jndiName">
<value>jhpt_ds</value>
<!-- <value>java:comp/env/jndi/jhpt_ds</value> -->
</property>
</bean>

<bean id="dynamicDataSource" class="com.conserv.tsas.common.bean.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry value-ref="yyzy" key="yyzy"></entry>
<entry value-ref="metadata" key="metadata"></entry>
<entry value-ref="jhpt" key="jhpt"></entry>
</map>
</property>
<property name="defaultTargetDataSource" ref="yyzy">
</property>
</bean>

<!-- sessionFactory -->
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="dynamicDataSource"/>
<property name="configLocation" value="classpath:mybatis-content.xml"></property>
<property name="mapperLocations"  value="classpath:com/conserv/tsas/**/mapper/*.xml"/>
<property name="typeAliasesPackage"  value="com.conserv.tsas.test.entity,com.conserv.tsas.front.entity"/>
</bean>

com.conserv.tsas.common.bean.DynamicDataSource.java

package com.conserv.tsas.common.bean;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DatabaseContextHolder.getCustomerType();
}

}

DatabaseContextHolder.java

package com.conserv.tsas.common.bean;

public class DatabaseContextHolder {
private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

public static final String YYZY = "yyzy";
public static final String METADATA = "metadata";
public static final String JHPT = "jhpt";

public enum DS {
YYZY, METADATA,JHPT
};

public static void setCustomerType(String customerType) {
contextHolder.set(customerType);
}

public static String getCustomerType() {
return contextHolder.get();
}

public static void clearCustomerType() {
contextHolder.remove();
}
}

上面的代码就是 在web 程序执行的时候，通过 private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();这个变量来获取 数据源名称。

那么有如何给 private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>(); 中设置值呢？

采用面向切面的方式 进行 java 的自定义注释方式 给 contextHolder 中存储值。

首先实现Java的自定义注释。

DataSource.java

package com.conserv.tsas.common.bean;

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;

import com.conserv.tsas.common.bean.DatabaseContextHolder.DS;

@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.TYPE,ElementType.METHOD})
public @interface DataSource {
public DS value()default DS.YYZY;
}

注释如何实用呢？

在 DAO 层中，在需要切换数据源的方法（method）前面加上 @DataSource(value=DS.YYZY) ， DS.YYZY就是我要的数据源

例子：

@DataSource(value=DS.JHPT)
@Override
public void deleteNotWriteOff_DDS(Map params) {
externalProcessMapper.deleteNotWriteOff_DDS(params);
}

先在再来实现上面注释的拦截方法，这里采用面向切面的方式。
DynamicDataSourceAspect.java

package com.conserv.tsas.common.bean;

import java.lang.annotation.Annotation;
import java.lang.reflect.Method;

import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

import com.conserv.tsas.common.bean.DatabaseContextHolder.DS;

@Component
@Aspect
@Order(1)
public class DynamicDataSourceAspect {
@Pointcut("execution(* com.conserv.tsas.*.dao..*.*(..))")
public void serviceExecution(){}

@Before("serviceExecution()")
public void setDynamicDataSource(JoinPoint jp) throws SecurityException, NoSuchMethodException {
String methodName = jp.getSignature().getName();
Method [] ms = jp.getTarget().getClass().getMethods();
Method method = null;
for(int i=0;i<ms.length;i++){
if(ms[i].getName().equals(methodName)){
method = ms[i];
break;
}
}
DataSource ds = method.getAnnotation(DataSource.class);
if(ds != null){
if(ds.value() == DS.YYZY){
DatabaseContextHolder.setCustomerType(DatabaseContextHolder.YYZY);
}else if(ds.value() == DS.METADATA){
DatabaseContextHolder.setCustomerType(DatabaseContextHolder.METADATA);
} else if (ds.value() == DS.JHPT) {
DatabaseContextHolder.setCustomerType(DatabaseContextHolder.JHPT);
}
}else{
DatabaseContextHolder.setCustomerType(DatabaseContextHolder.YYZY);
}

}
}

本文重点， 缺陷。

这种方式实现多数据源的切换，是通过 ThreadLocal 方式来共享 数据。 如果在service 层中使用 事务 @Transactional

例:

@Override
@Transactional
public void deleteT_YYZY_WJG_YL_ZCB() {
externalProcessDao.deleteT_YYZY_WJG_YL_CB();
externalProcessDao.deleteT_YYZY_WJG_YL_ZB();
}

此时，数据源容易混淆， 即本该执行 A 数据源，却执行了 C 数据源。

这是因为 ThreadLocal 类的内部

public void set(T value) {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null)
map.set(this, value);
else
createMap(t, value);
}

public T get() {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null) {
ThreadLocalMap.Entry e = map.getEntry(this);
if (e != null)
return (T)e.value;
}
return setInitialValue();
}

无论是set() 还是 get()，都是通过Thread.currentThread() 来作为 键值 进行存储。

而在 Java web 开发中， 在 service 层使用了事务，对 DAO 的 数据源拦截造成错乱，也即是ThreadLocal 中获取当前线程，出现错乱。