java - map 上的流过滤器和分组列表

Question

我正在制作一个过滤器，允许我生成一个值为文件列表的 map ，为此，我尝试使用 Streams。我有以下文件列表示例:

• CLA_FileName
• CLA_FileName
• CLA_FileName
• CM_FileName
• CM_FileName
• CM_FileName
• SP_FileName
• SP_FileName
• CON_文件名
• CON_文件名

然后我需要获取该文件列表并将它们带到一个 map ，该 map 的键将所有类型的文件分组在一个列表中；

Map >:所以这是预期的结果。

 "CLA_", List <File> 3 CLA Files
 "CM_",  List <File> 3 CM Files
 "SP_",  List <File> 2 SP Files
 "CON_", List <File> 2 CON Files

我有以下代码，它只为我分组了一种文件类型，CLA。请告诉我如何在 map 上过滤和加载这些文件？

List<File> csvList = getFiles();

Function<File, String> filterCLA = new Function<File, String>() {
    @Override
    public String apply(File file) {
        return String.valueOf(file.getName().startsWith("CLA_"));
    }
};

Map<String, List<File>> map = csvList.stream()
        .collect(groupingBy(f-> filterCLA.apply(f), toCollection(ArrayList::new)));


for(Map.Entry<String, List<File>> entry: map.entrySet()){
    System.out.println(entry.getKey()+" - "+ entry.getValue());
}

此代码仅使用 CLA_ 键将 CLA 文件保存在 map 中的列表中。这是正确的，但我需要添加其余文件。

Answer 1

试试这个。任何不以类似 XXX_ 开头的文件名都将映射到类型为 UnknownType

的 catch all 键

List<File> csvList = List.of(
new File("CLA_FileName"),
new File("CLA_FileName"),
new File("CLA_FileName"),
new File("CM_FileName"),
new File("CM_FileName"),
new File("CM_FileName"),
new File("SP_FileName"),
new File("SP_FileName"),
new File("CON_FileName"),
new File("CON_FileName"),
new File("BadFileName1"),
new File("BadFileName2"));
        

// get the prefix (e.g. CM_)
Function<File, String> getPrefix  = f-> {
       String name = f.getName();
    // location of delimiter
    int index = name.indexOf('_');
    if (index < 0) {
      return "UNKNOWN_TYPE";
    }
    return name.substring(0,index+1);
};
    
// create the map
Map<String, List<File>> map = csvList.stream()
        .collect(Collectors.groupingBy(getPrefix));

//print the map
map.entrySet().forEach(System.out::println);

打印

UNKNOWN_TYPE - [BadFileName1, BadFileName2]
CON_ - [CON_FileName, CON_FileName]
SP_ - [SP_FileName, SP_FileName]
CM_ - [CM_FileName, CM_FileName, CM_FileName]
CLA_ - [CLA_FileName, CLA_FileName, CLA_FileName]