众所周知,策略模式的实现 其中一个核心思想就是Java的多态,即在有多个接口的实现类,调用不同的实现类对象,从而达到执行不同的策略逻辑。
所以想要实现代码,需要注意这几个点:
- 设计一个通用的入参出参抽象方法;
- 将所有实现类都注册进Spring容器中;
- 准备好一个工厂类,用于根据策略名称获取策略对象;
而解决方案很简单,用@Service注解即可注册进入Spring容器中,注册进去后,我们可以在工厂类的构造方法中将所有bean实例拿出来,并且维护一个map用于随时获取。给出如下代码:
- 给出抽象方法:
public interface TestStrategy {
String getStrategyName();
void doStrategy();
}
- 给出两个实现类:
@Service
public class Test1Strategy implements TestStrategy{
@Override
public String getStrategyName() {
return "test1";
}
@Override
public void doStrategy() {
System.out.println("执行test1策略方法");
}
}
@Service
public class Test2Strategy implements TestStrategy{
@Override
public String getStrategyName() {
return "test2";
}
@Override
public void doStrategy() {
System.out.println("执行test2策略方法");
}
}
- 给出工厂类
@Component
public class TestStrategyFactory {
private Map<String, TestStrategy> testStrategyMap = new ConcurrentHashMap<>();
public TestStrategyFactory(ObjectProvider<List<TestStrategy>> objectProvider) {
List<TestStrategy> testStrategies = objectProvider.getIfAvailable();
if (Objects.nonNull(testStrategies) && testStrategies.size() > 0) {
testStrategyMap = testStrategies.stream()
.collect(Collectors.toMap(TestStrategy::getStrategyName, objSelf -> objSelf, (k1, k2) -> k1));
}
}
public TestStrategy getTestStrategy(String name) {
return testStrategyMap.get(name);
}
}