目录

前言

UML

plantuml

类图

实战代码

AbstractRoutingDataSource

DynamicDataSource

DynamicDataSourceContextHolder

---

前言

在设计类时，一般优先考虑使用组合来替代继承，能够让程序更加的灵活，但这并不意味着要完全抛弃掉继承。

继承在面向对象编程中用来减少代码冗余和提高复用性，是面向对象编程的一大特性。

只不过在使用继承时必须满足一些条件，才能让我们更好地利用继承，设计出更易维护和扩展的程序。

一般情况下，使用继承需要满足一下两个条件

1. 父类所有的属性和方法，都能在子类中适用
2. 子类无需复用其他类的方法，并且不会覆写父类已有的方法

模板方法模式是继承使用的优雅示例，一般会设计一个抽象类，在类中定义了一个操作的算法结构，其中一些步骤被设计为抽象方法，需要子类去实现，这些方法被称为模板方法。

模板方法允许子类在不改变算法结构的情况下，重新定义算法的某些特定步骤。

UML

plantuml

@startuml
'https://plantuml.com/class-diagram

abstract Template {
    + doSomething() : void
    - step1() : void
    - step2() : void
    + {abstract} step3() : void
}

class Concrete {
    + step3() : void
}

class Client {}

Template <|-- Concrete

Client ..> Concrete
@enduml

类图

实战代码

AbstractRoutingDataSource

业务上需要用到动态数据源，可以继承 spring 框架提供的抽象类 AbstractRoutingDataSource 来实现运行中动态切换数据源功能。

初始化动态数据源时，将所有的数据源都保存在 private Map<Object, DataSource> resolvedDataSources 中，每一个数据源对应一个唯一标识。

抽象类定义了决定目标数据源的方法（determineTargetDataSource），用来决定当前操作要使用动态数据源中的哪一个数据源，方法中调用了模板方法（determineCurrentLookupKey），子类只需要实现 determineCurrentLookupKey 这个模板方法，动态地返回数据源唯一标识，便能够实现动态切换数据源了

determineTargetDataSource 与 determineCurrentLookupKey

DynamicDataSource

public class DynamicDataSource extends AbstractRoutingDataSource {

    public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
        super.setDefaultTargetDataSource(defaultTargetDataSource);
        super.setTargetDataSources(targetDataSources);
        super.afterPropertiesSet();
    }

    @Override
    protected Object determineCurrentLookupKey() {
        return DynamicDataSourceContextHolder.getDataSourceType();
    }
}

DynamicDataSourceContextHolder

public class DynamicDataSourceContextHolder {

    /**
     * 使用ThreadLocal维护变量，ThreadLocal为每个使用该变量的线程提供独立的变量副本，
     *  所以每一个线程都可以独立地改变自己的副本，而不会影响其它线程所对应的副本。
     */
    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

    /**
     * 设置数据源变量
     * @param dataSourceType
     */
    public static void setDataSourceType(String dataSourceType){
        System.out.printf("切换到{%s}数据源", dataSourceType);
        CONTEXT_HOLDER.set(dataSourceType);
    }

    /**
     * 获取数据源变量
     * @return
     */
    public static String getDataSourceType(){
        return CONTEXT_HOLDER.get();
    }

    /**
     * 清空数据源变量
     */
    public static void clearDataSourceType(){
        CONTEXT_HOLDER.remove();
    }
}