Java 如何使用泛型或函数类避免代码中的重复

发布于07月04日

private <Y> void meth(
            MyObj ds, MultiValueMap<String, List> mvm, Class<Y> data) {

            

        if(data.isAssignableFrom(Employee.class)) {
            for (Employee rd : (List<Employee>) mvm.get(0).get(1)) {
               
                for (String cName : (List<String>) mvm.get(0).get(0)) {
                    ds.setCellValue((String)rd.getDataElement(cName));
                   
                }
               
            }
        }

        if(data.isAssignableFrom(Department.class)) {
            for (Department rd : (List<Department>) mvm.get(0).get(1)) {
               
                for (String cName : (List<String>) mvm.get(0).get(0)) {
                    ds.setCellValue((String)rd.getDataElement(cName));
                    
                }
               
            }
        }
        
        
        //some more similar if conditions as above
        
}

在上面，我有类似的10个if条件，如何避免上面重复的代码？我是否需要使用任何Java 8函数类作为参数以避免重复代码(或)必须使用任何额外的泛型代码？

步骤1:定义接口

不管具体的类是什么，你的 map 都包含有getDataElement()个对象的列表，因此制作一个界面来捕捉:

interface HasDataElement {
    String getDataElement();
}

步骤2:类实现接口

为了得到这个答案，我编写了几个简单的类——a、B和C，它们实现了接口，但在调用getDataElement()时只返回一个字符串.在代码中，您将修改员工、部门等以实现新界面.

class A implements HasDataElement {
    @Override
    public String getDataElement() {
        return "A";
    }
}

class B implements HasDataElement {
    @Override
    public String getDataElement() {
        return "B";
    }
}

class C implements HasDataElement {
    @Override
    public String getDataElement() {
        return "C";
    }
}

第三步:处理 map

我使用的是内置类型，所以是Map<String, List>，而不是你发布的使用MultiValueMap<String, List>的代码.我认为这一差异并不显著，但还是指出了这一点.

下面的方法签名指定映射不仅仅是<String, List>，还进一步指定列表本身必须包含扩展HasDataElement接口List<? extends HasDataElement>的内容.

在方法内部，相同的List<? extends HasDataElement>类型显示在第一个循环中.

一旦进入，item的具体类就不相关了——我们知道它符合HasDataElement接口，所以我们可以调用item.getDataElement().

private static void processAllDataElements(Map<String, List<? extends HasDataElement>> map) {
    for (List<? extends HasDataElement> list : map.values()) {
        for (HasDataElement item : list) {
            System.out.println(item.getDataElement());
        }
    }
}

示例用法

下面是一个简单的示例，以及输出，它创建了不同类a、B和C的几个不同列表.然后创建一个映射并将所有三个列表相加.

List<A> listOfA = List.of(new A[]{new A()});
List<B> listOfB = List.of(new B[]{new B(), new B()});
List<C> listOfC = List.of(new C[]{new C(), new C(), new C()});

Map<String, List<? extends HasDataElement>> map = new HashMap<>();
map.put("A", listOfA);
map.put("B", listOfB);
map.put("C", listOfC);

processAllDataElements(map);

A
B
B
C
C
C