在实际业务代码中,我们经常会碰到这样的代码:
String type = actualService.getRealtype(uid); if(type.equals("typeA")){ // do func A }else if(type.equals("typeB")){ // do func B }else if(type.equals("typeC")){ // do func C }else[ //... }
这种 if-else 或者 switch-case 代码在每个分支都会判断分支类型,然后执行不同的方法获取结果,当代码分支比较少并且确定不会增加时,使用这种方式也是完全 ok 的,但是当分支比较多,并且后面可能会增加分支判断条件时,这种方式就违反了单一职责和开闭原则,因此对于我们开发工作中遇到这种情况,首先想到的是应该去优化这种代码中的“坏味道”,其中的方法之一就是考虑能不能用策略模式去重写,将代码和业务逻辑解耦,这样才有利于后续的维护工作。
策略模式,简单来说就是通过实现接口来重写不同的方法,从而通过上下文自动获取选择的策略方法并执行。
以下基于 SpringBoot 的依赖注入实现策略模式。假设场景如下:某个客户需要订购多个资源,每个资源在不同资源池中,不同资源池下的资源也都不一样,在此处把原始的 if-else 代码逻辑优化为策略模式。
首先我们实现一个 ResourceStrategy 接口,并定义选择资源的抽象方法:
public interface ResourceStrategy { String orderInformation(String id); }
然后根据 if-else 中的判断条件,构造三个资源类实现 ResourceStrategy 接口:
@Component("A") public class ResourceA implements ResourceStrategy { @override public String orderInformation(String id){ System.out.println("策略选择:Strategy A"); return "A"; } }
@Component("B") public class ResourceB implements ResourceStrategy { @override public String orderInformation(String id){ System.out.println("策略选择:Strategy B"); return "B"; } }
@Component("C") public class ResourceC implements ResourceStrategy { @override public String orderInformation(String id){ System.out.println("策略选择:Strategy C"); return "C"; } }
注意其中每个类都需要标注策略类别名称。
然后我们需要写一个 SimpleContext 类来存储我们的策略类别,这时候就用到了 Spring 的依赖注入和自动发现。
@Service public class SimpleContextService { @Autowired private final MapstrategyMap = new ConcurrentHashMap<>(); public SimpleContext(Map strategyMap) { this.strategyMap.clear(); strategyMap.forEach(strategyMap::put); } public String getResource(String poolId){ return strategyMap.get(poolId).orderInformation(poolId); } }
接下来就是我们的实际调用场景了~,如下:
@RestController @RequestMapping("/test") public class TestController { @Autowired private SimpleContextService contextService; @GetMapping("/choose") public String choose(@RequestParam String poolId){ return simpleContext.getResource(poolId); } }
那么当我们的入参 poolId 传入 “A” 时,返回的结果如下:
策略选择:Strategy A A
同理,不同传参都会进入不同的策略执行方法。过这个简单的 demo,就可以看到通过获取输入不同的资源池 id,可以自动的拿到不同的资源。
通过实践总结下来,使用策略模式的好处就是通过一个封装的上下文可以自由的切换不同的算法,省去多重判断,同时可以具有很好的扩展性。
从上面可以看出,策略模式的优缺点十分明显,在我们实际的业务中,也需要看情况使用。
以上就是SpringBoot策略模式的实践使用的详细内容,更多关于SpringBoot策略模式的使用的资料请关注其它相关文章!