Spring 配置多数据源实现

Spring 配置多数据源原理

要使用的技术，我们先看下annotation的定义：

1. @Retention(RetentionPolicy.RUNTIME)  
2. @Target(ElementType.METHOD)  
3. public @interface DataSource {  
4.     String value();  
5. }  

 

    我们还需要实现spring的抽象类AbstractRoutingDataSource，就是实现determineCurrentLookupKey方法：

1. public class DynamicDataSource extends AbstractRoutingDataSource {  
2.   
3.     @Override  
4.     protected Object determineCurrentLookupKey() {  
5.         // TODO Auto-generated method stub  
6.         return DynamicDataSourceHolder.getDataSouce();  
7.     }  
8.   
9. }  
10.   
11.   
12. public class DynamicDataSourceHolder {  
13.     public static final ThreadLocal<String> holder = new ThreadLocal<String>();  
14.   
15.     public static void putDataSource(String name) {  
16.         holder.set(name);  
17.     }  
18.   
19.     public static String getDataSouce() {  
20.         return holder.get();  
21.     }  
22. }  

 

实现的核心部分，也就是AOP部分，DataSourceAspect定义如下:

1. public class DataSourceAspect {  
2.   
3.     public void before(JoinPoint point)  
4.     {  
5.         Object target = point.getTarget();  
6.         String method = point.getSignature().getName();  
7.   
8.         Class<?>[] classz = target.getClass().getInterfaces();  
9.   
10.         Class<?>[] parameterTypes = ((MethodSignature) point.getSignature())  
11.                 .getMethod().getParameterTypes();  
12.         try {  
13.             Method m = classz[0].getMethod(method, parameterTypes);  
14.             if (m != null && m.isAnnotationPresent(DataSource.class)) {  
15.                 DataSource data = m  
16.                         .getAnnotation(DataSource.class);  
17.                 DynamicDataSourceHolder.putDataSource(data.value());  
18.                 System.out.println(data.value());  
19.             }  
20.               
21.         } catch (Exception e) {  
22.             // TODO: handle exception  
23.         }  
24.     }  
25. }