如何避免此场景中的if-else语句

时间:2016-06-03 08:06:36

标签: java if-statement enums

我有一个具有以下身体的班级成绩:

public class Grades {

    public static Grade getGrades(int marks) {
        if (marks < 10) {
            return Grade.FAILED;
        } else if (marks < 13) {
            return Grade.DISTINCTION;
        } else if (marks < 15) {
            return Grade.GREAT;
        } else {
            return Grade.GREATEST;
        }
    }

    public static void main(String args[]){
        System.out.println(getGrades(6));
    }
}

一个枚举类:

public enum Grade {

    FAILED("F"),
    DISTINCTION("D"),
    GREAT("G"),
    GREATEST("G");


    private String code;

    Grade(String code){
        this.code = code;
    }

    public String getCode() {
        return code;
    }

}

如何在调用getGrades(int marks)

时避免使用if else构造

1 个答案:

答案 0 :(得分:6)

我会为此实施决策表。它只是Predicate(一个输入并返回truefalse的函数)与Predicate被评估为{{true的相应结果之间的映射。 1}}。

请注意,Java8引入了Predicate<T>接口,这对您的情况非常有帮助。此外,它可以表示为lambda,这将减少代码的冗长。

所以你有:

Map<Predicate<Integer>, Grade> table = new LinkedHashMap<Predicate<Integer>, Grade>() {{
      put (x -> x < 10, Grade.FAILED);
      put (x -> x < 13, Grade.DISTINCTION);
      put (x -> x < 15, Grade.GREAT);
}};

然后,只需迭代决策表并评估Predicate,直到找到评估为true的那个,就应该给你答案:

for (Entry<Predicate<Integer>, Grade> entry : table.entrySet()) {
    if (entry.getKey().test(marks)) {
        return entry.getValue();
    }
}
return Grade.GREATEST;