收集时将泛型类型传递给 HashSet

Question

假设我有一张公司部门地图到该部门的员工记录列表。

              Map<String, List<Employee>> departmentToEmployeesList = getMyMap();   

              Set<String> uniqueFirstNames = departmentToEmployeesList.
                    values()
                    .stream()
                    .collect(HashSet::new, HashSet::addAll, HashSet::addAll)
                    .stream()
                    .map(employee -> ((Employee)employee).getFirstName())
                    .collect(Collectors.toSet());

上面的代码有效，但我不想让((Employee)employee) 演员。如果没有强制转换，map 调用只会认为每个参数都是 Object 而不是 Employee。

我们可以像这样把它分成两个调用：

                HashSet<Employees> flattenedEmployees = departmentToEmployeesList.
                    values()
                    .stream()
                    .collect(HashSet::new, HashSet::addAll, HashSet::addAll);

                Set<String> uniqueFirstNames = flattenedEmployees
                    .stream()
                    .map(Employee::getFirstName)
                    .collect(Collectors.toSet());

现在我们不必强制转换，HashSet 知道它的类型为Employee。

有没有办法在collect 调用中将类型信息传递给HashSet::new？我是不是走错路了？

Answer 1

您可以使用HashSet<Employee>::new。

但我宁愿重构您的代码以使用 flatMap：

Set<String> uniqueFirstNames = 
    departmentToEmployeesList.values()
        .stream()
        .flatMap(List::stream)
        .map‌​(Employee::getFirstN‌​ame)
        .collect(toSet()‌​);