Java 8函数,它接受List <V>并返回HashMap <K,List <V >>

Question

我试图在Java 8中创建一个函数,它将一个列表作为参数并返回HashMap一个结果.

我已经在Java 7中完成了它,但我想通过使用Java 8流来实现它:

这是Java 7中的代码:

public static HashMap<String, List<Eleve>> getMentions(List<Eleve> liste){

    HashMap<String, List<Eleve>> listeMentions = new HashMap<String, List<Eleve>>();

    List<Eleve> noMention = new ArrayList<Eleve>();
    List<Eleve> mentionAB = new ArrayList<Eleve>();
    List<Eleve> mentionB  = new ArrayList<Eleve>();
    List<Eleve> mentionTB = new ArrayList<Eleve>();


    for (Eleve eleve: liste) {
        if (eleve.getAverage()>=12 && eleve.getAverage()<14) {mentionAB.add(eleve);}
        else if(eleve.getAverage()>=14 && eleve.getAverage()<16) {mentionB.add(eleve);}
        else if (eleve.getAverage()>=16) {mentionTB.add(eleve);}
        else{noMention.add(eleve);}
    }

    listeMentions.put("No mention", noMention);
    listeMentions.put("Mention AB", mentionAB); 
    listeMentions.put("Mention B",  mentionB);
    listeMentions.put("Mention TB", mentionTB);

    return listeMentions;
}

我尝试了collect(Collectors.toMap)流,但我没有得到估计的结果.

Answer 1

您可以创建一个接受int平均值的方法,并返回该平均值的相应组.

public static String getGroup (int average) {
    if (average >= 12 && average < 14) {
        return "Mention AB";
    } else if(average >= 14 && average < 16) {
        return "Mention B";
    } else if (average >= 16) {
        return "Mention TB";
    } else {
        return "No mention";
    }
}

现在,您可以使用Collectors.groupingBy此条件对实例进行分组:

Map<String, List<Eleve>> listeMentions =
    liste.stream()
         .collect(Collectors.groupingBy(e -> getGroup(e.getAverage())));

或者,您可以将Eleve实例传递给getGroup()方法,因此流管道将变为:

Map<String, List<Eleve>> listeMentions =
    liste.stream()
         .collect(Collectors.groupingBy(e -> getGroup(e)));

或者,如果你制作(即)getGroup()的实例方法:Elevepublic String getGroup() {...}

Map<String, List<Eleve>> listeMentions =
    liste.stream()
         .collect(Collectors.groupingBy(Eleve::getGroup));