使用Java 8 Streams基于常见条件从两个不同的列表创建映射

时间:2018-05-22 18:40:59

标签: java java-8 java-stream

我有两个这样的列表实例:

List<NameAndAge> nameAndAgeList = new ArrayList<>();
nameAndAgeList.add(new NameAndAge("John", "28"));
nameAndAgeList.add(new NameAndAge("Paul", "30"));
nameAndAgeList.add(new NameAndAge("Adam", "31"));

List<NameAndSalary> nameAndSalaryList = new ArrayList<>();
nameAndSalaryList.add(new NameAndSalary("John", 1000));
nameAndSalaryList.add(new NameAndSalary("Paul", 1100));
nameAndSalaryList.add(new NameAndSalary("Adam", 1200));

其中NameAndAge

class NameAndAge {
    public String name;
    public String age;

    public NameAndAge(String name, String age) {
        this.name = name;
        this.age = age;
    }

    @Override
    public String toString() {
        return name + ": " + age;
    }
}

NameAndSalary

private class NameAndSalary {
    private String name;
    private double salary;

    public NameAndSalary(String name, double salary) {
        this.name = name;
        this.salary = salary;
    }

    @Override
    public String toString() {
        return name + ": " + salary;
    }
}

现在,我想创建一个映射,其中第一个列表中的键为NameAndAge对象,而第二个列表中的值为NameAndSalary,其中两个对象中的名称相同。

因此,当我打印地图时,它应该如下所示:

{John: 28=John: 1000.0}
{Paul: 30=Paul: 1100.0}
{Adam: 31=Adam: 1200.0}

我尝试过这样做,但结束返回类型是&#39; void&#39;因为我是Streams的新手,所以我一无所知。

nameAndAgeList
    .forEach(n ->
        nameAndSalaryList
            .stream()
            .filter(ns -> ns.name.equals(n.name))
            .collect(Collectors.toList()));

有人可以建议如何使用Java Streams API实现这一目标吗?

4 个答案:

答案 0 :(得分:8)

首先,假设您要创建HashMap,您的密钥类(NameAndAge)必须覆盖equalshashCode()

其次,为了提高效率,建议您先从第二个Map<String,NameAndSalary>创建一个List

Map<String,NameAndSalary> helper =
    nameAndSalaryList.stream()
                     .collect(Collectors.toMap(NameAndSalary::getName,
                                               Function.identity()));

最后,您可以创建所需的Map

Map<NameAndAge,NameAndSalary> output = 
    nameAndAgeList.stream()
                  .collect(Collectors.toMap(Function.identity(),
                                            naa->helper.get(naa.getName())));

答案 1 :(得分:0)

这也应该可以解决问题:

Map<NameAndAge, NameAndSalary> map = new HashMap<>();
nameAndAgeList.forEach(age -> {
     NameAndSalary salary = nameAndSalaryList.stream().filter(
              s -> age.getName().equals(s.getName())).
              findFirst().
              orElseThrow(IllegalStateException::new);
      map.put(age, salary);
});

请注意,如果找不到匹配的名称,它会抛出IllegalStateException

答案 2 :(得分:-1)

这也应该有用

List<String> commonNames = nameAndAgeList
    .stream()
    .filter(na -> 
         nameAndSalaryList.anyMatch((ns) -> ns.getName().equals(na.getName()))
    .collect(Collectors.toList());

Map<NameAndAge, NameAndSalary> map = 
    commonNames.stream().collect(Collectors.toMap(name -> 
        nameAndAgeList.get(name), name -> nameAndSalaryList.get(name)));

答案 3 :(得分:-2)

不使用forEach创建任何中间对象,只需一个衬垫:

Map<NameAndAge, NameAndSalary> resultMap1 = nameAndAgeList.stream()
            .map(nameAndAge -> nameAndSalaryList.stream()
                    .filter(nameAndSalary -> nameAndAge.getName().equals(nameAndSalary.getName()))
                    .map(nameAndSalary -> new SimpleEntry<>(nameAndAge, nameAndSalary))
                    .collect(Collectors.toList()).get(0))
            .collect(Collectors.toMap(simpleEntry -> simpleEntry.getKey(), simpleEntry -> simpleEntry.getValue()));

您必须将getter函数添加到域类中。直接访问属性可能不是一个好主意。