【Spring AOP + 自定义注解 + 动态数据源 实现主从库切换&读写分离】—— 案例实战

以下是一个简化的示例，展示如何使用Spring AOP、自定义注解和动态数据源实现数据库的主从库切换和读写分离。

// 自定义注解
@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSource {
    String value();
}
 
// 数据源路由
public class DataSourceRouter {
    private final Map<Object, Object> dataSourceMap = new HashMap<>();
 
    public void addDataSource(Object key, Object value) {
        dataSourceMap.put(key, value);
    }
 
    public Object getDataSource(Object key) {
        return dataSourceMap.get(key);
    }
}
 
// AOP切面实现
@Aspect
@Component
public class DataSourceAspect {
    @Autowired
    private DataSourceRouter dataSourceRouter;
 
    @Before("@annotation(dataSource)")
    public void switchDataSource(JoinPoint point, DataSource dataSource) {
        // 动态设置数据源
        String dataSourceName = dataSource.value();
        DataSourceType dataSourceType = DataSourceType.valueOf(dataSourceName);
        dataSourceRouter.addDataSource("dataSource", dataSourceType.getDataSource());
    }
}
 
// 使用示例
@Service
public class SomeService {
    @DataSource("MASTER")
    public void insertRecord() {
        // 写入操作
    }
 
    @DataSource("SLAVE")
    public void selectRecord() {
        // 读取操作
    }
}

在这个示例中，我们定义了一个DataSource注解来标注方法使用的数据源类型，并创建了一个简单的DataSourceRouter类来管理数据源的路由。DataSourceAspect是一个Spring AOP切面，它会在方法执行前根据注解的值切换数据源。SomeService中的方法通过@DataSource注解指定了操作的数据源类型。

注意：这只是一个概念性的示例，实际应用中需要根据实际情况进行数据源的初始化和管理。