开发者社区> 琴岛蛏子> 正文

项目中如何优雅的消除if-else

简介: 通过一些设计模式可以消除if-else 提高系统的扩展性
+关注继续查看

项目中如何优雅的消除if-else

消除if-else的方法有很多种,本文结合spring中ArgumentResolver的应用,提供一种优雅地方式。

先看ArgumentResolver的应用方式

public class UserArgumentResolver implements HandlerMethodArgumentResolver {
  private static final String TOKEN_KEY = "token";

  @Override
  public boolean supportsParameter(MethodParameter parameter) {
      return parameter.hasParameterAnnotation(LoginUser.class);
  }

  @Override
  public Object resolveArgument(MethodParameter parameter,
                                ModelAndViewContainer mavContainer,
                                NativeWebRequest webRequest,
                                WebDataBinderFactory binderFactory) throws Exception {
      // 通过session取userid
      String token = webRequest.getHeader(TOKEN_KEY);
      User user = null;
      // 从redis中获取user json
      // user  = JSON.parse(redis.get(token),User.class);

      return user;
  }
}

controller 需要注入参数的地方

@RequestMapping(value = "/doSth", method = RequestMethod.POST)
public String feed(@LoginUser User user) {
  System.out.println("userId: " + JSON.toJsonString(user));
  return "success";
}

声明注解LoginUser

@Target(ElementType.TYPE,ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface RestController {

 String value() default "";

}

配置类添加

@Configuration
public class WebConfiguration implements WebMvcConfigurer {

  @Override
  public void addArgumentResolvers (List<HandlerMethodArgumentResolver> resolvers) {
    resolvers.add(new UserArgumentResolver());
  }
}

有多个ArgumentResolver在WebConfiguration中配置即可。

if-else消除

常用代码 如交通方式有 bike, bus,subway,train,plan

if (bike) {
    
} else if (bus) {
    
} else if (subway) {
    
} else if (train) {
    
} else if (plan) {
    
}else {
    
}

编写不同的reslover

定义统一的接口

public interface TrafficResolver {

    boolean support(String type);

    Object resolver(Object argument);
}

定义实现类BusTrafficResolver,其他省略

@Compoment
public class BusTrafficResolver implements TrafficResolver{

  @Override
  public boolean support (String type) {
    return "BUS".equals(type);
  }

  @Override
  public Object resolver (Object argument) {
    // doSth
    return "bus resolver";
  }
}

真正消除if-else的服务类

List 若没有bean则需要required 比提供默认值new ArrayList<>() ,有bean 可以去掉。

@Service
public class TrafficService {

  // required = false
  @Autowired(required = false)
  List<TrafficResolver> trafficResolverList = new ArrayList<>();

  public Object traffic(String type,Object argument){
    System.out.printf("trafficResolverList "+trafficResolverList.size());
    TrafficResolver resolver = getFirstResolver(type);
    if(resolver!=null){
      return resolver.resolver(argument);
    }

    return "traffic not find. type: "+type;
  }

  private TrafficResolver getFirstResolver(String type){
    TrafficResolver resolver = trafficResolverList.stream()
        .filter(item -> item.support(type))
        .findFirst()
        .orElse(null);
    return resolver;
  }

}

通过此种方式完成了对if-else调用类的解耦,实现了高扩展性。遵循了面向修改关闭,面向扩展开放的原则。

消除if-else完成,当然可以用注解的方式,通过注解类型判断是否支持。

其实if-else挺好用,除非考虑扩展性很强的地方,否则不要过度设计。

版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。

相关文章
RPC框架(1 - 实现服务端注册一个服务)
RPC框架(1 - 实现服务端注册一个服务)
4 0
轻量级RPC框架(0 - 项目概述)
轻量级RPC框架(0 - 项目概述)
4 0
python中的迭代对象、迭代器、生成器
在python中对于list和tuple,有很多的方式得到里面的元素,如果要输出里面的所有元素,可以采用for循环的形式,得到list或tuple中的每个元素,这就是迭代,凡是能够进行for循环的对象就是可迭代对象。 在java中的迭代方式有两种,一种直接for循环,一种采用迭代器 for循环的使用:
5 0
算法必知 --- 归并排序(优化与案例)
算法必知 --- 归并排序(优化与案例)
4 0
+关注
琴岛蛏子
潮来潮去,打洞卧沙
62
文章
0
问答
文章排行榜
最热
最新
相关电子书
更多
OceanBase 入门到实战教程
立即下载
阿里云图数据库GDB,加速开启“图智”未来.ppt
立即下载
实时数仓Hologres技术实战一本通2.0版(下)
立即下载
http://www.vxiaotou.com