public class DataSourceContextHolder {
    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();
    /**
     * 設置當前線程的數據源Key
     */
    public static void setDataSourceKey(String key) {
        CONTEXT_HOLDER.set(key);
    }
    /**
     * 獲取當前線程的數據源Key
     */
    public static String getDataSourceKey() {
        return CONTEXT_HOLDER.get();
    }
    /**
     * 清除當前線程的數據源Key（非常重要，防止內存泄漏和污染）
     */
    public static void clearDataSourceKey() {
        CONTEXT_HOLDER.remove();
    }
}

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        // 直接從上下文中獲取Key
        return DataSourceContextHolder.getDataSourceKey();
    }
}

@Configuration
public class DataSourceConfig {
    // 1. 配置主數據源
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.master")
    public DataSource masterDataSource() {
        return DataSourceBuilder.create().build();
    }
    // 2. 配置從數據源
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.slave")
    public DataSource slaveDataSource() {
        return DataSourceBuilder.create().build();
    }
    // 3. 配置動態(tài)數據源（Primary）
    @Bean
    @Primary
    public DynamicDataSource dynamicDataSource(
            @Qualifier("masterDataSource") DataSource masterDataSource,
            @Qualifier("slaveDataSource") DataSource slaveDataSource) {
        
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("master", masterDataSource);
        targetDataSources.put("slave", slaveDataSource);
        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        // 設置目標數據源映射
        dynamicDataSource.setTargetDataSources(targetDataSources);
        // 設置默認數據源（通常為主庫）
        dynamicDataSource.setDefaultTargetDataSource(masterDataSource);
        
        return dynamicDataSource;
    }
}

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface TargetDataSource {
    String value() default "master"; // 默認主庫
}

@Aspect
@Component
public class DataSourceAspect {
    // 攔截帶有 @TargetDataSource 注解的方法
    @Before("@annotation(targetDataSource)")
    public void before(JoinPoint point, TargetDataSource targetDataSource) {
        String dataSourceKey = targetDataSource.value();
        if (StringUtils.isNotBlank(dataSourceKey)) {
            DataSourceContextHolder.setDataSourceKey(dataSourceKey);
            System.out.println("切換數據源到: " + dataSourceKey);
        }
    }
    // 執(zhí)行完畢后清除上下文
    @After("@annotation(targetDataSource)")
    public void after(TargetDataSource targetDataSource) {
        DataSourceContextHolder.clearDataSourceKey();
    }
}

@Service
public class UserService {
    @Autowired
    private UserMapper userMapper;
    // 寫操作走主庫
    @TargetDataSource("master")
    public void addUser(User user) {
        userMapper.insert(user);
    }
    // 讀操作走從庫
    @TargetDataSource("slave")
    public User getUserById(Long id) {
        return userMapper.selectById(id);
    }
}

public void processOrder(Long userId, Order order) {
    try {
        // 1. 根據用戶ID計算分片key，切換到對應的用戶庫
        String shardKey = "shard_" + (userId % 4);
        DataSourceContextHolder.setDataSourceKey(shardKey);
        // 2. 執(zhí)行業(yè)務操作（此時會路由到指定分片）
        orderMapper.insert(order);
        
    } finally {
        // 3. 務必在finally中清理，避免影響后續(xù)操作
        DataSourceContextHolder.clearDataSourceKey();
    }
}

@Transactional // 開啟事務，此時獲取了 Master 的連接
public void updateAndRead() {
    userMapper.updateUser(user); // 使用 Master 連接
    
    // 嘗試切換數據源
    DataSourceContextHolder.setDataSourceKey("slave"); 
    
    // 問題：事務管理器仍持有 Master 的連接，這里可能并不會真的切換，
    // 或者拋出異常，因為 Connection 已經綁定在事務中了。
    userMapper.selectUser(id); 
}