策略模式和工厂模式相信大家都比较熟悉,但是大家有没有在springboot中实现策略和工厂模式?
具体策略模式和工厂模式的UML我就不给出来了,使用这个这两个模式主要是防止程序中出现大量的IF ELSE IF ELSE....。接下来咱们直接实现,项目结构图:
工厂类FactoryStrategy负责创建策略的工厂,代码比较简单,比较关键的一点是AutoWired一个Map 这个会在初始化的时候将所有的Strategy自动加载到Map中,是不是很方便。使用concurrentHashMap是防止多线程操作的时候出现问题。同时还要注意@Service注解。
packagecom.hqs.pattern.factory;importcom.hqs.pattern.strategy.Strategy;importorg.springframework.beans.factory.annotation.Autowired;importorg.springframework.stereotype.Service;importjava.util.Map;importjava.util.concurrent.ConcurrentHashMap;/***@authorhuangqingshi
* @Date 2019-01-31*/@Servicepublic classFactoryForStrategy {
@Autowired
Map strategys = new ConcurrentHashMap<>(3);public Strategy getStrategy(String component) throwsException{
Strategy strategy=strategys.get(component);if(strategy == null) {throw new RuntimeException("no strategy defined");
}returnstrategy;
}
}
接下来就是Strategy接口,就一个doOperation方法。
packagecom.hqs.pattern.strategy;/***@authorhuangqingshi
* @Date 2019-01-31*/
public interfaceStrategy {
String doOperation();
}
定义接口的实现,我定义了三个, 都类似,这里我就放出一个来吧。Component里边的one是指定其名字,这个会作为key放到Map strategys里边。
packagecom.hqs.pattern.strategy.impl;importcom.hqs.pattern.strategy.Strategy;importorg.springframework.stereotype.Component;/***@authorhuangqingshi
* @Date 2019-01-31*/@Component("one")public class StrategyOne implementsStrategy {
@OverridepublicString doOperation() {return "one";
}
}
好了,写一个Controller类,用于进行测试,当然我还是使用swagger,使用swagger的时候有个细节,就是注意生产上一定不能打开,否则是个非常可怕的事情。
packagecom.hqs.pattern.controller;importcom.hqs.pattern.factory.FactoryForStrategy;importorg.springframework.beans.factory.annotation.Autowired;importorg.springframework.stereotype.Controller;importorg.springframework.web.bind.annotation.PostMapping;importorg.springframework.web.bind.annotation.RequestParam;importorg.springframework.web.bind.annotation.ResponseBody;/***@authorhuangqingshi
* @Date 2019-01-31*/@Controllerpublic classStrategyController {
@Autowired
FactoryForStrategy factoryForStrategy;
@PostMapping("/strategy")
@ResponseBodypublic String strategy(@RequestParam("key") String key) {
String result;try{
result=factoryForStrategy.getStrategy(key).doOperation();
}catch(Exception e) {
result=e.getMessage();
}returnresult;
}
}
打开swagger进行测试,输入one,返回one。输入four,返回no strategy defined。后续如果有新策略的话,直接实现即可。
好了,这块就这么完成了,你get到了吗?
代码地址:https://github.com/stonehqs/pattern.git
欢迎拍砖~