Spring 配置多数据源实现

淡淡的烟草味﹌ 2022-07-13 01:52 275阅读 0赞

Spring 配置多数据源原理![Image 1][]

![SouthEast][]

要使用的技术，我们先看下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. \}

[Image 1]: 
[SouthEast]: /images/20220713/cbd9d117e52d4a2f84e57e2c9e6e1e6e.png