spring中怎么通過注解切換多數據源

spring中怎么通過注解切換多數據源，針對這個問題，這篇文章詳細介紹了相對應的分析和解答，希望可以幫助更多想解決這個問題的小伙伴找到更簡單易行的方法。

第一步，配置數據源

<bean id="masterDataSource" class="com.alibaba.druid.pool.DruidDataSource">
   <property name="username" value="root" />
   <property name="password" value="spring" />
   <property name="url" value="jdbc:mysql://localhost:3306/taotao?characterEncoding=utf-8" />
   <!-- 最大并發連接數 -->
   <property name="maxActive" value="30" />
   <!-- 最小空閑連接數 -->
   <property name="minIdle" value="5" />
   <!-- 用于顯示數據源監控中的sql語句監控 -->
   <property name="filters" value="stat" />
</bean>
 
<bean id="slaveDataSource" class="com.alibaba.druid.pool.DruidDataSource">
   <property name="username" value="root" />
   <property name="password" value="spring" />
   <property name="url" value="jdbc:mysql://localhost:3306/taobao?characterEncoding=utf-8" />
   <!-- 最大并發連接數 -->
   <property name="maxActive" value="30" />
   <!-- 最小空閑連接數 -->
   <property name="minIdle" value="5" />
   <!-- 用于顯示數據源監控中的sql語句監控 -->
   <property name="filters" value="stat" />
</bean>

第二步，定義用來切庫的注解，和枚舉類

@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface Dataswitch {
    Datatype value() default Datatype.master;
}
public enum Datatype {
    master("masterDataSource"),slave("slaveDataSource");
    private String value;
    Datatype(String name){
        this.value = name;
    }
 
    public String getValue() {
        return value;
    }
 
    public void setValue(String value) {
        this.value = value;
    }
}

第三步，定義一個當前線程的變量的工具類，用于設置對應的數據源名稱

public  class DynamicDataSourceHolder {
    private static final ThreadLocal<String> threadLocal = new ThreadLocal<String>();
 
    public static String getThreadLocal() {
        return threadLocal.get();
    }
 
    public static void setThreadLocal(String name) {
        threadLocal.set(name);
    }
    public static void clear(){
        threadLocal.remove();
    }
}

第四步，創建AbstactRoutingDataSource的子類，重寫determineCurrentLockupKey方法

public class DynamicDataSource extends AbstractRoutingDataSource{
    protected Object determineCurrentLookupKey() {
        System.out.println(DynamicDataSourceHolder.getThreadLocal());
        return DynamicDataSourceHolder.getThreadLocal();
    }
}

第五步，將多數據源配置到用我們創建的DynamicDataSource

<bean id="dynamicDataSource" class="xin.youhuila.sorceswitch.process.DynamicDataSource">
   <property name="targetDataSources">
      <map key-type="java.lang.String">
         <entry key="masterDataSource" value-ref="masterDataSource"></entry>
         <entry key="slaveDataSource" value-ref="slaveDataSource"></entry>
      </map>
   </property>
   <property name="defaultTargetDataSource" ref="masterDataSource"></property>
</bean>

第六步，配置切面，在操作數據庫方法之前，獲取注解配置的數據源名稱，返回

@Component
@Aspect
@Order(0)
public class DataSourceAspect {
    @Pointcut("execution (* xin.youhuila.sorceswitch.service..*(..))")
    public void aspect(){
 
    }
    @Before("aspect()")
    public void before(JoinPoint joinPoint){
        Class<?> clazz = joinPoint.getTarget().getClass();
        Method[] method = clazz.getMethods();
        Dataswitch dataswitch = null;
        boolean is = false;
        for(Method m:method){//此處最好改成通過簽名獲取調用方法是否含有注解，而不是遍歷每個方法，可參考http://www.gitout.cn/?p=2398
            if(m.isAnnotationPresent(Dataswitch.class)){
                dataswitch = m.getAnnotation(Dataswitch.class);
                DynamicDataSourceHolder.setThreadLocal(dataswitch.value().getValue());
                is = true;
            }
        }
        if(!is){
            DynamicDataSourceHolder.setThreadLocal(Datatype.master.getValue());
        }
    }
    @After("aspect()")
    public void after(){
        DynamicDataSourceHolder.clear();
    }
}

第七步，使用

@Service
public class DemoService {
    @Autowired
    DemoMapper demoMapper;
    @Dataswitch(Datatype.master)
    public void select(){
        List<Demo> d = demoMapper.select();
        for(Demo demo:d){
            System.out.println(demo);
        }
    }
}

--------------------------------http://www.gitout.cn/?p=2398文章實現-----------------------

/**
 * 數據源切面
 */
@Aspect
@Component
public class DynamicDataSourceAspect {
	
	@Pointcut("@annotation(com...datasource.DynamicDataSourceAnnotation)") 
	public void pointCut() {
	}
	
	@Before("pointCut()")
	public void testBefore(JoinPoint point) {
		// 獲得當前訪問的class
		Class<?> className = point.getTarget().getClass();
		DynamicDataSourceAnnotation dataSourceAnnotation = className.getAnnotation(DynamicDataSourceAnnotation.class);
		String dataSource = DataSourceConst.DB_ACTIVITY;
		// 優先級: 方法 > 類  > DB_ACTIVITY
		if(dataSourceAnnotation != null) {
			dataSource = dataSourceAnnotation.dataSource();
		}
		String methodName = point.getSignature().getName();
		// 得到方法的參數的類型
		Class<?>[] argClass = ((MethodSignature) point.getSignature()).getParameterTypes();
		try {
			Method method = className.getMethod(methodName, argClass);
			if (method.isAnnotationPresent(DynamicDataSourceAnnotation.class)) {
				DynamicDataSourceAnnotation annotation = method.getAnnotation(DynamicDataSourceAnnotation.class);
				dataSource = annotation.dataSource();
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
		DataSourceContextHolder.setDataSourceType(dataSource);

	}

	@After("pointCut()")
	public void testAfter(JoinPoint point) {
		// 獲得當前訪問的class
		Class<?> className = point.getTarget().getClass();
		DynamicDataSourceAnnotation dataSourceAnnotation = className.getAnnotation(DynamicDataSourceAnnotation.class);
		if (dataSourceAnnotation != null) {
			// 獲得訪問的方法名
			String methodName = point.getSignature().getName();
			// 得到方法的參數的類型
			Class<?>[] argClass = ((MethodSignature) point.getSignature()).getParameterTypes();
			String dataSource = DataSourceConst.DB_ACTIVITY;
			try {
				Method method = className.getMethod(methodName, argClass);
				if (method.isAnnotationPresent(DynamicDataSourceAnnotation.class)) {
					DynamicDataSourceAnnotation annotation = method.getAnnotation(DynamicDataSourceAnnotation.class);
					dataSource = annotation.dataSource();
				}
			} catch (Exception e) {
				e.printStackTrace();
			}
			if (dataSource != null && !DataSourceConst.DB_ACTIVITY.equals(dataSource)) {
				DataSourceContextHolder.clearDataSourceType();
			}
		}
	}
}

關于spring中怎么通過注解切換多數據源問題的解答就分享到這里了，希望以上內容可以對大家有一定的幫助，如果你還有很多疑惑沒有解開，可以關注億速云行業資訊頻道了解更多相關知識。