乐趣区

关于java:如何解决代码中过多的-if-else


先来一张镇楼图感受一下 if else 的魔法吧。

一、由一个几百行 if 引发的思考

有个场景,50 张字典表,须要为其余服务提供一个对立的接口来校验用户输出的字典表 id 是否非法。

校验逻辑曾经很清晰了,依据参数抉择对应的表校验 id 是否存在。

if("table_a".equals(table)) {// check id}
    if("table_b".equals(table)) {// check id}
    if("table_c".equals(table)) {// check id}...

再加上参数校验,函数调用,@Autowired bean 等等,一坨几百行的代码 ok 了。再新加表再加 if else 就行了,???? 完满。

如此,N 年后另一个可怜的小伙伴就看到这坨货色。

二、KO 这些 if else

回忆下面的场景,实际上就是要依据表名去确定 id 是否存在表中,那么只有将表名与操作对应起来就行了。故而采纳哈希表的模式,将表名与操作对应起来。局部代码如下:

// 用于保留表与 Function 的对应关系 
 private final Map<String, Function<Object, Object>> actionMappings = new ConcurrentHashMap<>(50);

 @PostConstruct
 private void init() {
    // map 初始化
    actionMappings.put(TableConstants.TABLE_A, (params) -> tableAManager.getById(params));
  }

/**
 * 校验逻辑
 *
 *@param table
 *@param id
 */
 public boolean valid(String table, Long id) {Object object = actionMappings.get(table).apply(id);
    // 不存在则校验失败
    return !Objects.isNull(object);
 }

如此,N 多行 if 被打消了,这种编程形式也叫做表驱动。尽管 if 没有了,然而在初始化 actionMappings 的时候还是很多行反复代码。上面采纳注解形式解决:

/**
 * 标记此注解的 bean 会退出根底数据校验全局 Function Map
 *
 * @author aysaml
 * @date 2020/5/7
 */
@Documented
@Inherited
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface ValidHandler {TABLE_ENUM value();
}

value 是表名枚举,在须要的类下面加上此注解即可。同时定义一个 context 用来专门存储 actionMappings。

/**
 * 数据校验上下文对象,用于保留各表的 Function Map
 *
 * @author aysaml
 * @date 2020/5/7
 */
@Component
public class CommonDataValidContext {private static final Logger LOGGER = LoggerFactory.getLogger(CommonDataValidContext.class);

  private final Map<String, Function<Object, Object>> actionMappings = new ConcurrentHashMap<>(50);

  /**
   * 办法退出 mappings
   *
   * @param model 表名
   * @param action 办法
   */
  public void putAction(String model, Function<Object, Object> action) {if (!Objects.isNull(action)) {actionMappings.put(model, action);
      LOGGER.info("[{}] add to CommonDataValidContext actionMappings, actionMappings size : {}",
          model,
          actionMappings.size());
    }
  }

  /**
   * 执行办法获取返回后果
   *
   * @param model
   * @param param
   * @return
   */
  public <P, R> R apply(String model, P param) {if (actionMappings.containsKey(model)) {return (R) actionMappings.get(model).apply(param);
    } else {LOGGER.error("执行数据校验时 model={} 不存在!", model);
      throw new RuntimeException("根底数据校验时产生谬误:" + model + "表不存在!");
    }
  }

  /**
   * 判断 mappings 中是否含有给定 model 的解决办法
   *
   * @param model
   * @return
   */
  public boolean containsKey(String model) {return actionMappings.containsKey(model);
  }

  /**
   * 校验执行办法的返回值是否为空
   *
   * @param model
   * @param param
   * @param <P>
   * @return
   */
  public <P> boolean validResultIsNull(String model, P param) {return Objects.isNull(this.apply(model, param));
  }
}

而后通过监听器的形式,将含有 ValidHandler 注解的办法退出 actionMappings。

/**
 * 根底数据校验解决办法监听器
 *
 * @author aysaml
 * @date 2020/5/7
 */
@Component
public class CommonValidActionListener implements ApplicationListener<ContextRefreshedEvent> {

  @Override
  public void onApplicationEvent(ContextRefreshedEvent event) {
    Map<String, Object> beans =
        event.getApplicationContext().getBeansWithAnnotation(ValidHandler.class);
    CommonDataValidContext commonDataValidContext =
        event.getApplicationContext().getBean(CommonDataValidContext.class);
    beans.forEach((name, bean) -> {ValidHandler validHandler = bean.getClass().getAnnotation(ValidHandler.class);
          commonDataValidContext.putAction(validHandler.value().code(),
              (param) -> {
                try {return bean.getClass().getMethod("getById", Long.class).invoke(bean, param);
                } catch (Exception e) {e.printStackTrace();
                }
                return null;
              });
        });
  }
}

三、更多打消 if else 的办法。

1. 提前 return

这样能够使代码在逻辑表白上会更清晰,如下:

if (condition) {// do something} else {return xxx;}

依照逆向思维来,优化如下:

if (!condition) {return xxx;} 
// do something

还有一种常见的傻瓜编程(如有触犯,敬请见谅,对码不对人????):

if(a > 0) {return true;} else {return false;}

话不多说了,间接 return a > 0; 不香吗?

2. 策略模式

简略来说就是依据不同的参数执行不同的业务逻辑。
如下:

if (status == 0) {// 业务逻辑解决 0} else if (status == 1) {// 业务逻辑解决 1} else if (status == 2) {// 业务逻辑解决 2} else if (status == 3) {// 业务逻辑解决 3}...

优化如下:

  • 多态
interface A {void run() throws Exception;
}

class A0 implements A {
    @Override
    void run() throws Exception {// 业务逻辑解决 0}
}

class A1 implements A {
    @Override
    void run() throws Exception {// 业务逻辑解决 1}
}
// ...

而后策略对象寄存在一个 Map 中,如下:

A a = map.get(param);
a.run();

2.2 枚举

public enum Status {NEW(0) {
      @Override
      void run() {//do something}
    },
    RUNNABLE(1) {
      @Override
       void run() {//do something}
    };

    public int statusCode;

    abstract void run();

    Status(int statusCode){this.statusCode = statusCode;}
}

从新定义策略枚举

public enum Aenum {
    A_0 {
      @Override
      void run() {//do something}
    },
    A_1 {
      @Override
       void run() {//do something}
    };
    //...
    abstract void run();}

通过枚举优化之后的代码如下

Aenum a = Aenum.valueOf(param);
a.run();

3. Java 8 的 Optional

Optional 次要用于非空判断,是 Java 8 提供的新个性。

应用之前:

if (user == null) {//do action 1} else {//do action2}

如果登录用户为空,执行 action1,否则执行 action 2,应用 Optional 优化之后,让非空校验更加优雅,间接的缩小 if 操作

Optional<User> userOptional = Optional.ofNullable(user);
userOptional.map(action1).orElse(action2);

4. 决策表

就是下面的表驱动编程办法。

欢送拜访集体博客 获取更多常识分享。

退出移动版