Java查找指定文件

/*
* 查找文件夹中的指定类型文件,本次指定问电影文件
*/
public static void main(String[] args) {

File dir = getDir();
getJavaFile(dir);
}

public static void getJavaFile(File dir) {
File[] subFiles = dir.listFiles();
for (File subFile : subFiles) {
String name = subFile.getName();
if (subFile.isFile() && name.endsWith(".mkv")
|| name.endsWith("mp4") || name.endsWith("rmvb")
|| name.endsWith("flv")) {
System.out.println(subFile.getName());
} else if (subFile.isDirectory()) {
getJavaFile(subFile);
}
}
}

@SuppressWarnings("resource")
public static File getDir() {
Scanner sc = new Scanner(System.in);
System.out.println("请输入文件夹路径:");
while (true) {
String pathname = sc.nextLine();
File dir = new File(pathname);
if (!dir.exists()) {
System.out.println("您输入的路径不存在,请重新输入.");
} else if (dir.isFile()) {
System.out.println("您输入的是一个文件路径,请输入一个文件夹路径.");
} else {
return dir;
}
}
}