代码行数统计
递归遍历项目的文件夹,找出代码文件,统计代码的行数并输出。其实这是一个比较简单的功能,如果能实现一个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方法中添加你的错误处理代码。