在实际的接口测试中，有时需要根据情况进行文件的上传和下载。在文件数量比较小的时候，我们当然可以直接处理（比如若干个接口都用一个文件）。但是，如果我们上传的文件需要使用不同文件夹里不同的文件，而且数量又比较大的情况下，直接写路径就比较麻烦了。
那么，怎样来处理这样的问题呢？还是需要先整理一下思路。
首先，我们需要根据某些特征，找到这些文件。例如，我们可以在指定的文件夹下面，按照文件后缀名来找寻文件。通常，我们在DOS下查询该文件夹下寻找同一后缀名文件的时候，会使用通配符“*”来代替文件名，例如：*.jpg,*.xls等等。那么，我们可以指定一个文件夹，然后使用将通配符变为正则的方式，来匹配查找文件。
另外，如果在我们指定的文件夹中，存在子文件夹的情况，我们需要进行递归处理。即，进入子文件夹后，再次对子文件夹下的对象进行遍历，以此类推。
先来看一段代码：
private static String generatePattern(String fileConf) {
fileConf = fileConf.trim();
// 根据配置生成正确的正则
fileConf = fileConf.replace(‘*‘, ‘#‘);
// 将*号之前加上.
fileConf = fileConf.replaceAll("#", ".*");
return fileConf;
}
这里，我们定义了一个生成正则的方法。就是将"*.XXX"变成“.*.XXX”（仔细看，就是在之前加了一个"."）。我们可以利用这个正则表达式来匹配我们想要寻找的文件。
接下来，我们来看一下递归查找的问题。首先，我们需要定义一个要查找的文件夹。而在java中，我们可以直接定义一个主路径作为file的对象。
例如：
String c = "k:/abc/";
File file = new File(c);
这样的话，可以使用getName()方法获取文件或文件夹的名称（不包含上级路径）。另外，也可以使用listFiles() 返回目录下的文件或者目录对象（File类实例），包含隐藏文件。当然，对于文件，这样操作会返回null。
下面，我们来看一下递归的操作。思路是这样的：
1.判断当前的file对象是文件还是文件夹
2.如果是文件，若匹配正则，则将文件加入list。
3.如果是文件夹，则获取该文件夹下面所有的文件或者目录对象，若匹配正则，则将文件加入list。
private static ArrayList<File> filePattern(File file, Pattern p) {
if (file == null) {
return null;
}
// 如果是文件,则加入fileList
if (file.isFile()) {
Matcher fMatcher = p.matcher(file.getName());
if (fMatcher.matches()) {
ArrayList<File> list = new ArrayList<File>();
list.add(file);
return list;
}
// 如果是目录，则进行递归查找
} else if (file.isDirectory()) {
File[] files = file.listFiles();
if (files != null && files.length > 0) {
ArrayList<File> list = new ArrayList<File>();
for (File f : files) {
//此处进行递归
ArrayList<File> rlist = filePattern(f, p);
if (rlist != null) {
//将查找结果都加入fileList
list.addAll(rlist);
}
}
return list;
}
}
return null;
}
那么，根据上面的代码，我们可以将路径下所有的匹配正则的文件都找到，并放入fileList中。
接下来，我们来定义一个方法，使用主路径以及通配符作为参数，来进行文件的查找。在通配符参数的指定中，我们使用“;”来进行分隔，来找到我们所需要的文件。
例如：
String dirs = "k:/abc/"; String file_con = "*.doc;*.xls";
这里会出现一个新的问题，如果我按照上面所写，会把“k:/abc/”下的所有匹配的文件都找出来。
假如，我是需要寻找主路经下test1文件夹下的*.doc,和test2文件夹下的*.xls怎么办？
这里在指定参数的部分，我们需要对文件夹作出判断，即对最后一个“/”进行分隔处理。
请看下面一段代码：
public static List<File> getFilesByConf(String dir, String fileConf) {
String[] fileConfArr = fileConf.split(";");// 多个配置‘;’分开
List<File> list = new ArrayList<File>();
if (fileConfArr != null && fileConfArr.length > 0) {
for (String conftemp : fileConfArr) {
int at = conftemp.lastIndexOf("/");
File file = null;
String fileContextPath = "";
// 绝对目录路径
String contextPath = dir;
// 针对目录部分的处理
if (at > 0) {
fileContextPath = fileConf.substring(0, at);
}
if (StringUtil.isNotEmpty(fileContextPath)) {
contextPath = contextPath + fileContextPath;
}
file = new File(contextPath);
String fileNameConf = conftemp.substring(at + 1,conftemp.length());// 文件名配置
String fileConfTemp = generatePattern(fileNameConf);
Pattern p = Pattern.compile(fileConfTemp);
ArrayList<File> listtemp = filePattern(file, p);
list.addAll(listtemp);
}
}
return removeDuplicate(list);// 去重
}
这里，我们就完成了针对路径的处理。顺便说一下，这里的去重，使用的是利用HashSet的特性来进行去重。
public static List removeDuplicate(List list) {
if (list == null) {
return null;
}
Set set = new HashSet();
List newList = new ArrayList();
for (Iterator iter = list.iterator(); iter.hasNext();) {
Object obj = iter.next();
if (set.add(obj)) {
newList.add(obj);
}
}
return newList;
}
至此，我们就完成了对于文件查找的处理。
总结一下，我们只需要提供两个参数，一个是主路经，一个匹配的通配符的字符串参数（可以带路径），即可批量查找到我们所需要的文件。
接下来，我们看一下，下载文件的处理。
下载文件一般比较好处理，基本思路就是使用输入流来写入，然后保存即可。
需要注意的是，针对比较大的文件时，我们需要设置Buffer（即缓冲区）。不应用缓冲区的时候，每次读取一个字节，写入一个字节，由于操作磁盘比内存慢的很多，所以不应用缓冲区效率很低；应用缓冲区，可以一次读取多个字节，先不写入磁盘，而是放入内存之中，到了缓冲区大小的时候，在写入磁盘，减少了对磁盘的操作，效率较高。
public static boolean writeFile(InputStream is, String filePath) {
File file = new File(filePath);
if (!file.getParentFile().exists()) {
file.getParentFile().mkdirs();
}
FileOutputStream fileout;
try {
fileout = new FileOutputStream(file);
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
return false;
}
/**
* 根据实际运行效果 设置缓冲区大小
*/
byte[] buffer = new byte[10 * 1024];
int ch = 0;
try {
while ((ch = is.read(buffer)) != -1) {
fileout.write(buffer, 0, ch);
}
return true;
} catch (IOException e) {
e.printStackTrace();
return false;
} finally {
try {
is.close();
fileout.flush();
fileout.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
这里根据实际情况，来设置缓冲区的大小。
这样就可以根据接口返回的内容，写入文件到指定的路径中。
原文地址：https://www.cnblogs.com/generalli2019/p/12247715.html