基于Mybatis继承AbstractRoutingDataSource使用自定义注解实现动态数据源

一：实现

方式一：继承AbstractRoutingDataSource使用自定义注解实现

环境：springboot3 + MyBatis3 + mysql-connector8

DataSourceKeyEnum枚举类

有几个数据源就配置几个枚举类，和数据源数量一一对应

class DataSourceKeyEnum{
    DEFAULT,
    SLAVE
}

DataSourceAnnotation注解

用来标记在Service方法或Mapper上，用于辅助AOP切换数据源操作。

@Target({ElementType.METHOD,ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSourceAnnotation {

    DataSourceKeyEnum value() default DataSourceKeyEnum.DEFAULT;

}

AbstractRoutingDataSource 的作用

AbstractRoutingDataSource的核心功能是根据用户定义的规则，动态决定在特定的操作中使用哪个数据源。这通常是通过重写determineCurrentLookupKey方法来实现的，该方法用于确定当前线程应该使用哪个数据源的键。

public class DynamicDataSource extends AbstractRoutingDataSource {

    /**
     *  MyBatis内部根据实现AbstractRoutingDataSource类的determineCurrentLookupKey实现数据源的动态路由
     */
    @Override
    protected Object determineCurrentLookupKey() {

        return DynamicDataSourceContextHolder.getDataSourceType();
    }

    /**
     *  静态内部类，用于从线程中获取当前要使用的数据源
     */
    public static class DynamicDataSourceContextHolder {

        private static final ThreadLocal<DataSourceKeyEnum> contextHolder = new ThreadLocal<>();

        public static DataSourceKeyEnum getDataSourceType() {
            return contextHolder.get();
        }

        public static void setDataSourceType(DataSourceKeyEnum dataSourceKey) {
            if (dataSourceKey == null) {
                dataSourceKey = DataSourceKeyEnum.DEFAULT;