模板方法模式（继承的优雅使用）

目录

前言

UML

plantuml

类图

实战代码

AbstractRoutingDataSource

DynamicDataSource

DynamicDataSourceContextHolder

---

前言

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

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

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

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

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

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

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

UML

plantuml

@startuml
'https://plantuml.com/class-diagramabstract Template {+ doSomething() : void- step1() : void- step2() : void+ {abstract} step3() : void
}class Concrete {+ step3() : void
}class Client {}Template <|-- ConcreteClient ..> 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();}@Overrideprotected 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();}
}