spring:
  datasource:
    master:
      driver-class-name: com.mysql.cj.jdbc.Driver
      url: jdbc:mysql://master-host:3306/db_name
      username: root
      password: master-pass
    slave:
      driver-class-name: com.mysql.cj.jdbc.Driver
      url: jdbc:mysql://slave-host:3306/db_name
      username: root
      password: slave-pass

public class DataSourceContextHolder {
    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();
    
    public static void setDataSourceType(String dataSourceType) {
        CONTEXT_HOLDER.set(dataSourceType);
    }
    
    public static String getDataSourceType() {
        return CONTEXT_HOLDER.get();
    }
    
    public static void clearDataSourceType() {
        CONTEXT_HOLDER.remove();
    }
}

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSourceType();
    }
}

@Aspect
@Component
public class DataSourceAspect {
    
    // 攔截所有Mapper層方法
    @Before("execution(* com.yourpackage.mapper..*.*(..))")
    public void beforeMethod(JoinPoint joinPoint) {
        String methodName = joinPoint.getSignature().getName();
        // 根據(jù)方法名判斷是讀還是寫
        if (methodName.startsWith("get") || methodName.startsWith("select") || 
            methodName.startsWith("find") || methodName.startsWith("query")) {
            // 讀方法切換到從庫
            DataSourceContextHolder.setDataSourceType("slave");
        } else {
            // 寫方法切換到主庫
            DataSourceContextHolder.setDataSourceType("master");
        }
    }
    
    // 方法執(zhí)行后清理數(shù)據(jù)源設(shè)置
    @After("execution(* com.yourpackage.mapper..*.*(..))")
    public void afterMethod() {
        DataSourceContextHolder.clearDataSourceType();
    }
}

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Master {
}