利用JDK8的新特性计算某个目录下的文件中包含字符串的次数

来源：互联网发布：成都闪电seo 编辑：程序博客网时间：2024/05/29 12:41

需求：计算某个字符串在某个文件夹中出现的次数。**这篇文章利用了JDK1.8的新特性Stream流和Lambda表达式并结合了线程池的使用。**

package com.zkn.fullstacktraining.seventh;import javafx.util.Pair;import java.io.IOException;import java.nio.file.Files;import java.nio.file.Paths;import java.util.List;import java.util.Optional;import java.util.concurrent.*;import java.util.stream.Collectors;/** * Created by zkn on 2017/2/5. */public class SearchStringByThreadPool {    public static void main(String[] args) {        try {            //创建5个固定线程的线程池            ExecutorService executorService = Executors.newFixedThreadPool(5);            List<Future<Pair<String, Integer>>> listFile =                    //这里是取所传入目录的最多四层，如果不知道这个API的话需要递归去做。                    Files.walk(Paths.get("D:\\CUST\\workspace\\JavaCore\\FullStackTraining\\src\\main\\java\\com\\zkn"), 4)                            .filter(file -> !Files.isDirectory(file) && file.toString().endsWith("java"))//文件文件夹和不是java的文件                            .map(file -> (Callable<Pair<String, Integer>>) () -> {//创建N多个Callable实现类                                Pair<String, Integer> pair = null;//这里的键值对用pair比用Map更好一些                                try {                                    Optional optional = Files.lines(file).map(str -> {                                        int count = 0;                                        int index = str.indexOf("main");                                        if (index >= 0) {                                            //这里需要循环计算，因为可能在某一行中会出现多次                                            do {                                                count++;                                            } while ((index = str.indexOf("main", index + 1)) > 0);                                        }                                        return count;                                    }).reduce(Integer::sum);//合并最终的计算结果                                    int count = optional.isPresent() ? (int) optional.get() :0;                                    pair = new Pair<>(file.toString(),count);                                } catch (IOException e) {                                    e.printStackTrace();                                }                                return pair == null ? new Pair<>("", 0) : pair;                            })                            .map(file -> executorService.submit(file))//提交给线程池进行处理                            .collect(Collectors.toList());            listFile.stream().map(file -> {                Pair<String, Integer> pair = null;                try {                    pair = file.get();                } catch (InterruptedException e) {                    e.printStackTrace();                } catch (ExecutionException e) {                    e.printStackTrace();                }                return pair == null ? new Pair<>("", 0) : pair;            })                    .filter(file -> file.getValue() > 0)//过滤掉不包含字符串的文件                    .sorted((s1, s2) -> Integer.compare(s2.getValue(), s1.getValue()))//从大到小排序                    .forEach(file -> System.out.println(String.format("%d次出现在%s文件中", file.getValue(), file.getKey())));            //关闭线程池            executorService.shutdown();        } catch (Exception e) {            e.printStackTrace();        }    }    public void test() {        String str = "mainmainmainmainmain";    }}

0 0