大家好，我是鋒哥。今天分享關于【MyBatis如何實現動態數據源切換？】面試題。希望對大家有幫助；
MyBatis如何實現動態數據源切換？
超硬核AI學習資料，現在永久免費了！
在MyBatis中實現動態數據源切換，通常需要依賴Spring的動態數據源管理機制，結合一些自定義的切換策略。以下是實現動態數據源切換的基本步驟：
1.?配置數據源
首先，在配置文件中配置多個數據源。例如，配置兩個數據源：masterDataSource 和 slaveDataSource。
<bean id="masterDataSource" class="org.apache.commons.dbcp2.BasicDataSource"><property name="driverClassName" value="com.mysql.cj.jdbc.Driver"/><property name="url" value="jdbc:mysql://localhost:3306/master_db"/><property name="username" value="root"/><property name="password" value="root"/>
</bean><bean id="slaveDataSource" class="org.apache.commons.dbcp2.BasicDataSource"><property name="driverClassName" value="com.mysql.cj.jdbc.Driver"/><property name="url" value="jdbc:mysql://localhost:3306/slave_db"/><property name="username" value="root"/><property name="password" value="root"/>
</bean>2.?定義動態數據源
創建一個自定義的數據源類，它能夠切換不同的數據源。AbstractRoutingDataSource 是Spring提供的一個抽象類，適用于實現動態數據源。
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;public class DynamicDataSource extends AbstractRoutingDataSource {@Overrideprotected Object determineCurrentLookupKey() {// 返回當前的數據源標識（如：master 或 slave）return DataSourceContextHolder.getDataSourceType();}
}3.?數據源上下文管理
為了管理當前使用的數據源，可以創建一個 DataSourceContextHolder 類，用來存儲和獲取當前線程的數據源標識。
public class DataSourceContextHolder {private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();public static void setDataSourceType(String dataSourceType) {contextHolder.set(dataSourceType);}public static String getDataSourceType() {return contextHolder.get();}public static void clearDataSourceType() {contextHolder.remove();}
}
4.?配置動態數據源
在Spring配置中使用 DynamicDataSource 替代傳統的單一數據源。
<bean id="dataSource" class="com.example.DynamicDataSource"><property name="defaultTargetDataSource" ref="masterDataSource"/><property name="targetDataSources"><map><entry key="master" value-ref="masterDataSource"/><entry key="slave" value-ref="slaveDataSource"/></map></property>
</bean>5.?AOP實現數據源切換
使用Spring AOP（面向切面編程）來切換數據源。可以通過注解或者在方法執行前動態修改當前線程的 DataSource。
例如，創建一個注解 @DataSource 來標記需要切換數據源的方法：
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface DataSource {String value() default "master"; // 默認使用master數據源
}然后，創建一個切面（Aspect），根據注解切換數據源：
@Aspect
@Component
public class DataSourceAspect {@Before("@annotation(dataSource)")public void changeDataSource(DataSource dataSource) {// 切換數據源DataSourceContextHolder.setDataSourceType(dataSource.value());}@After("@annotation(dataSource)")public void restoreDataSource(DataSource dataSource) {// 切換回默認數據源DataSourceContextHolder.clearDataSourceType();}
}6.?在服務層使用注解切換數據源
在需要切換數據源的方法上使用 @DataSource 注解：
@Service
public class UserService {@DataSource("master") // 使用master數據源public void addUser(User user) {// 執行數據庫操作}@DataSource("slave") // 使用slave數據源public List<User> getUsers() {// 執行數據庫查詢return userRepository.findAll();}
}7.?使用完畢后清理數據源
在每次請求結束后，確保清理 ThreadLocal 中存儲的數據源標識，避免線程池復用時出現錯誤。
public class DataSourceContextHolder {private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();public static void setDataSourceType(String dataSourceType) {contextHolder.set(dataSourceType);}public static String getDataSourceType() {return contextHolder.get();}public static void clearDataSourceType() {contextHolder.remove();}
}總結：
- 數據源管理：使用?
DynamicDataSource?來管理多個數據源。 - 數據源切換：通過?
DataSourceContextHolder?類和Spring AOP機制，在方法執行時切換數據源。 - 使用場景：動態切換數據源通常用于讀寫分離，或者多數據庫操作的場景。
這樣，你就可以在 MyBatis 中根據需要動態地切換數據源了。
-04)
