是否有办法简化或优化这个if-else结构？

Question

你好，作为Java 11的新手，我正在尝试优化以下代码段，看看是否有更好的实现方式：

if(code == 1){
    return ClassName.static_list1.contains(string_object);
}else if(code == 2){
    return ClassName.static_list2.contains(string_object);
}else if(code == 3){
    return ClassName.static_list3.contains(string_object);
}else if(code == 4){
    return ClassName.static_list4.contains(string_object);
}else{
    return false;
}

该方法返回一个布尔值，并且这些列表源自一个常量类。请告知我是否有任何可以优化的方式。

Pedro Lobito · Answer

除了使用switch语句（如前所述），您还应该考虑根据传入的code创建一个仅返回列表的方法。您目前的代码在每个分支上都重复了“contains(string_object)”这一语句。我会这样处理（伪代码）：

var list = ClassName.getList(code); // 实现使用switch语句的静态"getList"方法
if (list != null)
    return list.contains(string_object);
else
    return false;

Dimitre Novatchev · Answer

我建议进行两个独立的重构。

首先，我建议将int code替换为一个enum Code，这样我们就能拥有带有语义名称的值。

其次，我建议使用一个Map>，然后我们可以用Code的值来访问它。如果我们只想进行这项优化，也可以使用Map>。

将这些结合在一起，解决方案可能看起来像这样：

public enum Code {
  FOO, BAR
}

Map> map = ...; // 这里需要初始化映射关系

boolean contains(Code code, String stringObject) {
  return map.getOrDefault(code, List.of()).contains(stringObject);
}

Steve · Answer

尝试像这样实现。只需使用一个映射（Map）并将列表放入其中。

public class ClassName {
    static List static_list1 = List.of("A", "B");
    static List static_list2 = List.of("C", "D");
    static List static_list3 = List.of("E", "F");
    static List static_list4 = List.of("G", "H");
    static Map> map = Map.of(
            1, ClassName.static_list1,
            2, ClassName.static_list2,
            3, ClassName.static_list3,
            4, ClassName.static_list4
    );

    public static void main(String[] args) {
        System.out.println(contains(10, "A"));
        System.out.println(contains(1, "A"));
        System.out.println(contains(1, "C"));
        System.out.println(contains(2, "D"));
        System.out.println(contains(3, "A"));
    }
    // 除非代码在映射中且对象在列表中，否则将返回false

    public static boolean contains(int code, String object) {
        return map.containsKey(code) && map.get(code).contains(object);
    }
}

输出为：

false
true
false
true
false

Michael Berkowski · Answer

您可以提取一个方法，该方法根据需要调用 contains 方法的列表（或集合）。这可以通过 switch/case 语句、数组或 Map> 来实现。回退是返回一个空集合（永远不会包含任何内容，因此返回 false）。

static List listByCode(final int code) {
  switch (code) {
    case 1: return ClassName.static_list1;
    case 2: return ClassName.static_list2;
    case 3: return ClassName.static_list3;
    case 4: return ClassName.static_list4;
    default: return Collections.emptyList();
  }
}

// 调用点（您的现有代码）：
return listByCode(code).contains(string_object);

使用数组的替代解决方案：

private static List[] lists = {
  ClassName.static_list1,
  ClassName.static_list2,
  ClassName.static_list3,
  ClassName.static_list4
};

static List listByCode(final int code) {
  return code > 0 && code < lists.length
      ? lists[code - 1]
      : Collections.emptyList();
  // 或者：不检查，直接抛出 ArrayIndexOutOfBoundsException
}

使用 Map 的解决方案：

private static Map> listsByCode = Map.of(
    1, ClassName.static_list1,
    2, ClassName.static_list2,
    3, ClassName.static_list3,
    4, ClassName.static_list4
);

static List listByCode(final int code) {
  return listsByCode.getOrDefault(code, Collections.emptyList());
}

请注意，基于数组和 Map 的解决方案不允许您替换类中的列表引用，而基于 switch 的解决方案则可以。但是，也许您不需要有单独的引用，数组/Map 可以成为访问这些列表的标准方式。

Peter Lyons · Answer

Switch 表达式

自从 Java 14 起，可以使用switch 表达式：

return switch (code) {
    case 1 -> ClassName.staticList1.contains(stringObject);
    case 2 -> ClassName.staticList2.contains(stringObject);
    case 3 -> ClassName.staticList3.contains(stringObject);
    case 4 -> ClassName.staticList4.contains(stringObject);
    default -> false;
};

contains() 方法的调用可以从 switch 语句中提取出来，但不确定这样做是否值得（为 default 情况添加一个空的哑列表，可能不够简洁。）

顺带一提，Java 命名规范建议 string_object 改为 stringObject，static_list1 改为 staticList1，以此类推。

Gishu · Answer

在我看来，这是最好的方法：

static List searchByCode(final int code) {
  switch (code) {
    case 1: 
        return ClassName.staticList1;
    case 2: 
        return ClassName.staticList2;
    case 3: 
        return ClassName.staticList3;
    case 4: 
        return ClassName.staticList4;
    default: 
        return Collections.emptyList();
  }
}

return searchByCode(code).contains(stringObject);