這篇文章給大家介紹如何將Spring的動態(tài)數(shù)據(jù)源進行讀寫分離，內(nèi)容非常詳細，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。

我們提供的服務(wù)有：成都做網(wǎng)站、網(wǎng)站設(shè)計、微信公眾號開發(fā)、網(wǎng)站優(yōu)化、網(wǎng)站認證、吉林ssl等。為成百上千企事業(yè)單位解決了網(wǎng)站和推廣的問題。提供周到的售前咨詢和貼心的售后服務(wù)，是有科學(xué)管理、有技術(shù)的吉林網(wǎng)站制作公司

一、創(chuàng)建基于ThreadLocal的動態(tài)數(shù)據(jù)源容器，保證數(shù)據(jù)源的線程安全性

package com.bounter.mybatis.extension;

/**
 * 基于ThreadLocal實現(xiàn)的動態(tài)數(shù)據(jù)源容器，保證DynamicDataSource的線程安全性
 * @author simon
 *
 */
public class DynamicDataSourceHolder {

 private static final ThreadLocal dataSourceHolder = new ThreadLocal<>();

 public static void setDataSource(String dataSourceKey) {
 dataSourceHolder.set(dataSourceKey);
 }

 public static String getDataSource() {
 return dataSourceHolder.get();
 }

 public static void clearDataSource() {
 dataSourceHolder.remove();
 }
}

二、定義Spring動態(tài)數(shù)據(jù)源擴展類,用來實現(xiàn)Master、Slave數(shù)據(jù)源動態(tài)切換

package com.bounter.mybatis.extension;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

/**
 * 自定義的Spring 動態(tài)數(shù)據(jù)源擴展類,用來實現(xiàn)Master、Slave數(shù)據(jù)源動態(tài)切換
 * @author simon
 *
 */
public class DynamicDataSource extends AbstractRoutingDataSource {

 @Override
 protected Object determineCurrentLookupKey() {
 //使用DynamicDataSourceHolder保證線程安全
 return DynamicDataSourceHolder.getDataSource();
 }

}

三、配置Master、Slave數(shù)據(jù)源

1. db.properties配置Master、Slave數(shù)據(jù)信息

# Master DB
db.master.url=jdbc:MySQL://192.168.168.110:3306/bounter?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&allowMultiQueries=true&serverTimezone=PRC&useSSL=false
db.master.username=bounter
# AES encrypt,Base64 encode
db.master.password=ZNhnEjauk3pecZxxS84ofA==

# Slave DB
db.slave.url=jdbc:mysql://192.168.168.111:3306/database?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&allowMultiQueries=true&serverTimezone=PRC&useSSL=false
db.slave.username=bounter
# AES encrypt,Base64 encode
db.slave.password=jFYmt2f57RHhzItYDhWiSA==

2. Spring 配置文件配置Master、Slave連接池，動態(tài)數(shù)據(jù)源

四、創(chuàng)建數(shù)據(jù)源切面，通過AOP實現(xiàn)根據(jù)Dao層方法前綴動態(tài)選取讀、寫數(shù)據(jù)源

package com.bounter.mybatis.aop;

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.springframework.context.annotation.EnableAspectJAutoProxy;
import org.springframework.stereotype.Component;

import com.bounter.mybatis.extension.DynamicDataSourceHolder;

/**
 * 數(shù)據(jù)源切面，通過dao方法前綴決定訪問讀、寫數(shù)據(jù)源
 * @author simon
 *
 */
@Component
@Aspect
@EnableAspectJAutoProxy(proxyTargetClass = true)
public class DataSourceAspect {

 //讀庫數(shù)據(jù)源key
 private static final String DATASOURCE_KEY_READ = "read";
 //查詢方法清單
 String[] queryMethods = {"find","get","query","count","select"};

 /**
 * dao層方法執(zhí)行前選擇數(shù)據(jù)源
 * @param point
 */
 @Before("execution(* com.bounter.mybatis.dao..*.*(..))")
 public void before(JoinPoint point) {
 // 獲取到當(dāng)前執(zhí)行的方法名
 String methodName = point.getSignature().getName();
 //匹配查詢方法
 for(String queryMethod : queryMethods) {
  if(methodName.startsWith(queryMethod)) {
  //查詢方法設(shè)置數(shù)據(jù)源為讀庫
  DynamicDataSourceHolder.setDataSource(DATASOURCE_KEY_READ);
  break;
  }
 }
 }

 /**
 * dao層方法執(zhí)行完后清空數(shù)據(jù)源選擇
 * @param point
 */
 @After("execution(* com.bounter.mybatis.dao..*.*(..))")
 public void after(JoinPoint point) {
 DynamicDataSourceHolder.clearDataSource();
 }
}

關(guān)于如何將Spring的動態(tài)數(shù)據(jù)源進行讀寫分離就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，可以學(xué)到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。