spring 多数据源-实现

紧接上篇 - 实现

1 创建 DynamicDataSource 继承 AbstractRoutingDataSource 重写 determineCurrentLookupKey 方法

                /**
                 * 
                 * @Author: ll
                 * @Date: 2018年5月29日 上午10:32:32
                 * @Description: 多数据源
                 * @Version: 1.0 
                 */
            public class DynamicDataSource extends AbstractRoutingDataSource {
             @Override
             protected Object determineCurrentLookupKey() {
    return DynamicDataSourceHolder.getDatasource();  
            }

                   }

 

                 /**
                 * 
                 * @Author: ll
                 * @Date: 2018年5月29日 上午10:32:32
                 * @Description: 多数据源
                 * @Version: 1.0 
                 */
            public class DynamicDataSource extends AbstractRoutingDataSource {
                 @Override
                protected Object determineCurrentLookupKey() {
            return DynamicDataSourceHolder.getDatasource();  
                   }

 

                        }

2 实现 DynamicDataSourceHolder 类

        /**
             * 
             * @Author: ll
             * @Date: 2018年5月29日 下午10:32:32
             * @Description: 多数据源
             * @Version: 1.0 
             */

            public class DynamicDataSourceHolder {

                    /** 数据源标识保存在线程变量中，避免多线程操作数据源时互相干扰 */  

 

                    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();  
      
                        /** 
                             * 设置数据源名称 
                             * @param dastsource 数据源名称 
                             */  
                    public static void setDatasource(String datasource) {  
                        contextHolder.set(datasource);  
                        }  
      
                /** 
                 * 获取数据源名称 
                 * @return 数据源名称 
                 */  
                    public static String getDatasource() {  
                        return contextHolder.get();  
                    }  
      
                    /** 
                     * 清除标志 
                     */  
                    public static void clearDatasource() {  
                        contextHolder.remove();  
                    }   

 

                }

3 配置多数据源 

 由于代码较多 在此粘出关键代码 

    <bean id="dataSource" class="com.enter.net.frame.datasources.DynamicDataSource">  
        <property name="targetDataSources">  
            <map key-type="java.lang.String">  
                <entry key="fxdataSource" value-ref="fxdataSource"></entry>  
                <entry key="gblzdataSource" value-ref="gblzdataSource"></entry>  
            </map>  

        </property>  

<!-- 默认使用的数据库-->

 

        <property name="defaultTargetDataSource" ref="fxdataSource"></property>  

 

    </bean> 

4 使用数据源 

本人准备使用注解的方式同时结合spring的AOP  

首先定义个一个注解 DataSource 

/**
 * 
 * @Author: ll
 * @Date: 2018年5月29日 上午10:32:32
 * @Description: 多数据源注解
 * @Version: 1.0 
 */
@Target({ElementType.TYPE, ElementType.METHOD})  
@Retention(RetentionPolicy.RUNTIME)  
public @interface DataSource {  
    String value();  

}

2 定义一个切面

/**
 * 
 * @Author: ll
 * @Date: 2018年5月29日 上午10:32:32
 * @Description: 多数据源切面
 * @Version: 1.0 
 */
@Aspect
@Component
public class DynamicDataSourceInterceptor {
/** 
     * 拦截目标方法，获取由@DataSource指定的数据源标识，设置到线程存储中以便切换数据源 
     * @param point 
     * @throws Exception 
     */  
@Before("execution(* com.enter.net.fhbusiness..*Controller.*(..)) or execution(* com.enter.net.system..*Controller.*(..))")
    public void intercept(JoinPoint point) throws Exception {  
        Class<?> target = point.getTarget().getClass();  
        MethodSignature signature = (MethodSignature) point.getSignature();  
        resolveDataSource(target, signature.getMethod());  
    }  
  
    /** 
     * 提取目标对象方法注解和类型注解中的数据源标识 
     * @param clazz 
     * @param method 
     */  
    private void resolveDataSource(Class<?> clazz, Method method) {  
        try {  
            Class<?>[] types = method.getParameterTypes();
            // 默认使用类型注解  
            if (clazz.isAnnotationPresent(DataSource.class)) {  
                DataSource source = clazz.getAnnotation(DataSource.class);  
                DynamicDataSourceHolder.setDatasource(source.value());  
            }  
            // 方法注解可以覆盖类型注解  
            if(types.length!=0) {
            Method m = clazz.getMethod(method.getName(), types);  
                 if (m != null && m.isAnnotationPresent(DataSource.class)) {  
                     DataSource source = m.getAnnotation(DataSource.class);  
                     DynamicDataSourceHolder.setDatasource(source.value());  
                 } 
            }
        } catch (Exception e) {  
            e.printStackTrace();  
        }  
    }  

}

这时候切面 会根据配置的表达式 进行拦截查看是否使用DataSource 注解 如果使用 则获取注解的值 来设定指定的s

下一篇 https://blog.csdn.net/lilongwangyamin/article/details/80507121 总结