背景:有时我们想批量地创建一些小文件,用于其他功能的测试,比如测试存储性能,人工地在操作系统创建文件及文件夹显然很费时,我们就需要自动化地对文件和文件夹操作。
1.创建空文件夹
输入是文件夹目的地点,比如想在D盘创建名为smallfile的文件夹,输入参数则应该是 “D:\\smallfile”。
public static String createDirectory(String dir){
/*如果sPath不以文件分隔符结尾,自动添加文件分隔符,对于创建文件夹可以不加这几句
if (!dir.endsWith(File.separator)) {
dir = dir + File.separator;
}*/
File file = new File(dir);
if(!file.exists() ){
file.mkdirs();
}
return file.getPath();
}
2.批量创建文件
参数是目标文件夹,如 “D:\\smallfile”,为了文件名不重复可以用数字,也可以用UUID,本文用数字。
public static void createFilesInPath(String filePath) throws IOException {
if (!filePath.endsWith(File.separator)) {
filePath = filePath + File.separator;
}
for(int i = 0; i<10; i++){
String path = filePath;
//文件放在D:\smallfiles\ 目录下 命名为i.txt
File file = new File(path+String.valueOf(i) +".txt");
//如果文件不存在,创建一个文件
if (!file.exists()) {
file.createNewFile();
}
FileWriter fw = null;
BufferedWriter bw = null;
try {
fw = new FileWriter(file);
String content = String.valueOf(i);
bw = new BufferedWriter(fw);
bw.write(content);
} catch (IOException e) {
} finally {
bw.close();
fw.close();
}
}
System.out.println("create 10 files success");
}
3.删除指定文件
输入是文件绝对路径
public static boolean deleteFile(String filePath)throws IOException {
boolean flag = false;
File file = new File(filePath);
// 路径为文件且不为空则进行删除
if (file.isFile() && file.exists()) {
file.delete();
flag = true;
}
else {
System.out.println("file not found");
}
return flag;
}
4. 列出文件夹所有文件,包括子文件夹里的
输入是文件夹路径和一个用来存放文件名的集合,因为使用的递归的方法,所以集合需是全局变量
//如果文件夹有嵌套,则fileNameList必须是全局变量,否则递归调用后清空了
public static List<String> listAllFiles(String filePath, List<String> fileNameList){
if (!filePath.endsWith(File.separator)) {
filePath = filePath + File.separator;
}
//目标集合fileList
File file = new File(filePath);
if (file.isDirectory()) {
File[] files = file.listFiles();
for (File fileIndex : files) {
//如果这个文件是目录,则进行递归搜索
if (fileIndex.isDirectory()) {
listAllFiles(fileIndex.getPath()+File.separator,fileNameList);
} else {
//把文件名加入列表
// fileList.add(fileIndex.getName());
fileList.add(fileIndex.getAbsolutePath());
}
}
}
return fileList;
}
5.删除文件夹
输入是文件夹目录
public static boolean deleteDirectory(String sPath) throws IOException {
//如果sPath不以文件分隔符结尾,自动添加文件分隔符
if (!sPath.endsWith(File.separator)) {
sPath = sPath + File.separator;
}
File dirFile = new File(sPath);
//如果dir对应的文件不存在,或者不是一个目录,则退出
if (!dirFile.exists() || !dirFile.isDirectory()) {
return false;
}
boolean flag = true;
//删除文件夹下的所有文件(包括子目录)
File[] files = dirFile.listFiles();
for (int i = 0; i < files.length; i++) {
//删除子文件
if (files[i].isFile()) {
flag = deleteFile(files[i].getAbsolutePath());
if (!flag) break;
} //删除子目录
else {
flag = deleteDirectory(files[i].getAbsolutePath());
if (!flag) break;
}
}
if (!flag) return false;
//删除当前目录
if (dirFile.delete()) {
return true;
} else {
return false;
}
}
6.测试主函数
static List<String> fileList = new ArrayList<String>();
public static void main(String[] str) throws IOException {
String filePath = "D:\\smallfiles\\";
//创建目录
String path = createDirectory(filePath);
System.out.println(path);
//在指定目录创建文件
createFilesInPath(path);
//删除1.txt
deleteFile("D:\\smallfiles\\1.txt");
//查看目录里所有文件
List<String> fileName = listAllFiles("D:\\smallfiles",fileList);
//根据绝对路径删除文件
for(String name : fileName){
System.out.println(name);
// deleteaFile(name);
}
deleteDirectory(filePath);
}