如果switch/if-else分支不固定，或将来有可能改，可以采用反射+策略模式来替换switch/if-else语句。例如动物，随着种类变化，我们需要不断增加switch分支，这样我们就可以使反射+策略模式，使之更加灵活。
if-else方式：
Animal animal = new Animal();
if("dog".equals(type)) {
animal = new Dog();
} if else("cat".equals(type)) {
animal = new Cat();
}
animal.eat();
switch方式：
Animal animal = new Animal();
switch (type) {
case "dog":
animal = new Dog();
break;
case "cat":
animal = new Cat();
break;
}
animal.eat();
Spring反射+策略模式：
Animal接口(抽象策略角色)：
public interface Animal {
void eat();
}
Cat实现Animal接口(具体实现策略)：
@Component
public class Cat implements Animal {
@Override
public void eat() {
System.out.println("猫吃鱼");
}
}
Dog实现Animal接口(具体实现策略)：
@Component
public class Dog implements Animal {
@Override
public void eat() {
System.out.println("狗啃骨头");
}
}
-
@Component：将bean实例化交给spring容器管理。或者使用.xml配置文件<bean id="" class=""></bean>方式等等。
AnimalFactory类(环境角色):
@Service("animalFactory")
public class AnimalFactory implements ApplicationContextAware, InitializingBean {
private Map<String, Animal> animalMap;
private ApplicationContext applicationContext;
@Override
public void afterPropertiesSet() throws Exception {
Map<String, Animal> animalMap = this.applicationContext.getBeansOfType(Animal.class);
this.animalMap = animalMap;
}
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
this.applicationContext = applicationContext;
}
public Animal getAnimal(String beanName) {
return this.animalMap.get(beanName);
}
}
- 由于
AnimalFactory类实现ApplicationContextAware接口，Spring容器会在创建AnimalFactory类之后，自动调用实现接口的setApplicationContextAware()方法，调用该方法时，会将ApplicationContext(容器本身)作为参数传给该方法，我们可以在该方法中将Spring传入的参数ApplicationContext赋给AnimalFactory对象的applicationContext实例变量，因此接下来可以通过该applicationContext实例变量来访问容器本身。
- 实现
InitializingBean接口，该接口提供了afterPropertiesSet方法。spirng容器在初始化bean的时候会执行afterPropertiesSet方法，我们可以在该方法中调用applicationContext接口提供的getBeansOfType方法获得实现Animal类的Bean，将之存储至map集合中。
存储bean
测试类：
@RunWith(SpringRunner.class)
@SpringBootTest
public class DemoApplicationTests {
@Autowired
private AnimalFactory animalFactory;
@Test
public void test() {
Animal animal = animalFactory.getAnimal("dog");
// Animal animal = animalFactory.getAnimal("cat");
animal.eat();
}
}
输出：
狗啃骨头
以后需求变更，只要增加实现类即可。
