ssm框架多数据源配置详解

最新推荐文章于 2024-07-31 19:38:21 发布

蚂蚁一般黑

最新推荐文章于 2024-07-31 19:38:21 发布

阅读量556

点赞数

分类专栏：框架文章标签： java 多数据源

本文链接：https://blog.csdn.net/qq_34295227/article/details/103583927

版权

框架专栏收录该内容

2 篇文章 0 订阅

订阅专栏

文章说明：本文为懒汉式教程，只需要粘贴复制就可以实现。

废话不多说，直接代码走起

这里已ssm框架为例，项目所需的相关依赖jar都已经集成在框架中了。。。

1、数据源连接参数

jdbc.properties

jdbc.driverClassName=com.mysql.jdbc.Driver
#mysql----1
jdbc.url=jdbc\:mysql\://47.96.67.113\:3306/redpacketforwarding?useUnicode\=true&characterEncoding\=UTF-8
jdbc.username=root
jdbc.password=A028c78165fb18d40165fb18d4c30D0f

#mysql----2

jdbc.url2=jdbc\:mysql\://47.96.67.113\:3306/jiatui?useUnicode\=true&characterEncoding\=UTF-8
jdbc.username2=root
jdbc.password2=A028c78165fb18d40165fb18d4c30D0f

2、配置数据源

spring-mvc-context.xml

<bean id="dynamicDataSource" class="com.ssay.util.dynamicDataSource.DynamicDataSource">
       <property name="targetDataSources">
       <map key-type="java.lang.String">
           <entry value-ref="dataSource" key="dataSource"></entry>
           <entry value-ref="dataSource2" key="dataSource2"></entry>
           </map>
       </property>
       
       <property name="defaultTargetDataSource" ref="dataSource"></property>
   </bean>

<aop:aspectj-autoproxy proxy-target-class="true"></aop:aspectj-autoproxy>

3、自定义数据源

package com.ssay.util.dynamicDataSource;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
/**
* 动态数据源
* */
public class DynamicDataSource extends AbstractRoutingDataSource{
@Override
protected Object determineCurrentLookupKey() {
return DBContextHolder.getDbType();
}
}

package com.ssay.util.dynamicDataSource;

/**
* 数据源类型
*/
public class DBContextHolder {
   /**
   * 注意：数据源标识保存在线程变量中，避免多线程操作数据源时互相干扰
   */
   private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

   // 设置当前数据源
   public static void setDbType(String dbType) {
       contextHolder.set(dbType);
   }

   // 获取当前数据源
   public static String getDbType() {
       return ((String) contextHolder.get());
   }

   // 清空上下文数据源
   public static void clearDbType() {
       contextHolder.remove();
   }
}

4、自定义@DataSource注解

package com.ssay.util.dynamicDataSource;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;
import java.lang.annotation.ElementType;
import java.lang.annotation.RetentionPolicy;

/**自定义@DataSource注解 */
@Target({ ElementType.TYPE,ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSource {
String value();
}

5、定义DataSourceAspect切面

package com.ssay.util.dynamicDataSource;

import java.lang.reflect.Method;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import com.ssay.util.dynamicDataSource.DBContextHolder;
import com.ssay.util.dynamicDataSource.DataSource;

/**
* DataSource切面
*/
@Component
@Aspect
public class DataSourceAspect {

private static final Logger logger = LoggerFactory.getLogger(DataSourceAspect.class);

   // 指定切入点表单式
   @Pointcut("execution(* com.ssay.*.*.dao.mapper.*.*(..))")
   public void pointCut_() {
   }

   // 前置通知 : 在执行目标方法之前执行
   @Before("pointCut_()")
   public void begin(JoinPoint point) {
       System.err.println("正在设置数据源。。。");
       //获取由@DataSource指定的数据源标识，设置到线程存储中以便切换数据源
       Class<?> target = point.getTarget().getClass();
       MethodSignature signature = (MethodSignature) point.getSignature();
       // 默认使用目标类型的注解，如果没有则使用其实现接口的注解
       for (Class<?> clazz : target.getInterfaces()) {
           resolveDataSource(clazz,signature.getMethod());
       }
       resolveDataSource(target,signature.getMethod());
   }

   // 后置/最终通知：在执行目标方法之后执行【无论是否出现异常最终都会执行】
   @After("pointCut_()")
   public void after(JoinPoint point) {
       DBContextHolder.clearDbType();
       System.out.println("切换默认数据源成功");
   }

   /**
   * 提取目标对象方法注解和类型注解中的数据源标识
     * @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);
               DBContextHolder.setDbType(source.value());
           }
           // 方法注解可以覆盖类型注解
           Method m = clazz.getMethod(method.getName(),types);
           if (m != null && m.isAnnotationPresent(DataSource.class)) {
               DataSource source = m.getAnnotation(DataSource.class);
               DBContextHolder.setDbType(source.value());
               logger.debug("当前数据源："+source.value());
           }
       } catch (Exception e) {
           logger.error(clazz +":" + e.getMessage());
           }
   }
}

代码截图

6.在dao层加@DataSource("dataSource2")注解

这个注解可以在类上加，也可以在方法上加

public interface TestAreaInfoMapper {
@DataSource("dataSource2")
AreaInfoTest selectAreaInfoByCode(@Param("areaCode") String areaCode);
}

调用成功后的截图如下

蚂蚁一般黑

关注

0
点赞
踩
3

收藏

觉得还不错? 一键收藏
0
评论
ssm框架多数据源配置详解

文章说明：本文为懒汉式教程，只需要粘贴复制就可以实现。废话不多说，直接代码走起这里已ssm框架为例，项目所需的相关依赖jar都已经集成在框架中了。。。1、数据源连接参数jdbc.propertiesjdbc.driverClassName=com.mysql.jdbc.Driver#mysql----1jdbc.url=jdbc\:mysql\://47.96.67.113...
复制链接

扫一扫

专栏目录