Java实现统计项目代码行数

代码行数统计

递归遍历项目的文件夹，找出代码文件，统计代码的行数并输出。其实这是一个比较简单的功能，如果能实现一个ui窗口，将文件夹拖拽到窗口上，获得拖拽的路径进行统计，也是可以实现的。

package learn2;

import java.io.File;
import java.io.IOException;
import java.nio.file.*;
import java.nio.file.attribute.BasicFileAttributes;

public class FileLineCount {
    static int totalLineCount = 0;

    public static void main(String[] args) {
        String warOfTank = "/home/xxx/IdeaProjects/Hello/src";
        File directory = new File(warOfTank); // 替换为你的目录路径
        try {
            Files.walkFileTree(Paths.get(directory.getPath()), new SimpleFileVisitor<Path>() {
                @Override
                public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
                    if (file.toString().endsWith(".kt")) {
                        int lines = countLines(file);
                        System.out.println(file.toString() + " lines:" + lines);
                        totalLineCount += lines;
                    }
                    return FileVisitResult.CONTINUE;
                }

                @Override
                public FileVisitResult visitFileFailed(Path file, IOException exc) throws IOException {
                    return FileVisitResult.CONTINUE;
                }
            });
        } catch (IOException e) {
            e.printStackTrace();
        }
        System.out.println("Total line count: " + totalLineCount);
    }

    private static int countLines(Path file) throws IOException {
        return (int) Files.lines(file).count();
    }
}

在这个代码中，我们首先创建一个File对象，代表你要遍历的目录。然后，我们使用Files.walkFileTree方法来递归遍历这个目录和它的子目录。我们传递一个自定义的FileVisitor给这个方法，该FileVisitor将会在遍历到每个文件时调用visitFile方法。在visitFile方法中，我们检查文件名是否以.kt结尾，如果是，我们就调用countLines方法来计算这个文件的行数，并把这个行数加到totalLineCount变量上。最后，我们打印出totalLineCount的值。

注意，这个代码不会处理遍历过程中发生的错误，如果你想要处理错误，你可以在visitFileFailed方法中添加你的错误处理代码。