地图上的流过滤器和分组列表

Question

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

• CLA_FileName
• CLA_FileName
• CLA_FileName
• CM_FileName
• CM_FileName
• CM_FileName
• SP_FileName
• SP_FileName
• CON_FileName
• CON_FileName

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

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。请您告诉我如何在地图上过滤和加载这些文件？

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 文件保存在地图内的列表中。这是正确的，但我需要添加其余文件。

Answer 1

最后我找到了过滤和分组的解决方案。这里是代码。 谢谢大家的帮助

    List<Predicate<File>> allPredicates = new ArrayList<Predicate<File>>();
    allPredicates.add(str -> str.getName().startsWith("CM"));
    allPredicates.add(str -> str.getName().startsWith("CLA"));
    allPredicates.add(str -> str.getName().startsWith("IDP"));
    allPredicates.add(str -> str.getName().startsWith("SP"));
    allPredicates.add(str -> str.getName().startsWith("CON"));

    return Arrays.asList(csvFiles)
                    .stream()
                    .filter(allPredicates.stream().reduce(f->true, Predicate::or))
                    .collect(Collectors.groupingBy(f-> f.getName().substring(0,f.getName().indexOf("_"))));

Answer 2

试试这个。任何不以 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]