保存文件到指定位置，读取/删除指定文件夹中文件

使用背景

用户每次上传的文件太多，如果批量处理，服务器性能不够，所以将用户上传的文件保存到服务器上，再启动一个线程，依次处理完成后删除文件

用户上传文件

@PostMapping("/inFile")publicStringinFile(@RequestParam("file")List<MultipartFile>listFile,@RequestParamMap<String,String>params){// 创建临时文件夹FileSaveClearfsc=newFileSaveClear();fsc.createTempDirectory();listFile.forEach(file->{// 保存上传的文件到临时文件夹StringfilePath=fsc.saveFileToTemp(file);});//读取到指定文件夹中的所有文件List<MultipartFile>files=fsc.getAllTempFilesAsMultipartFiles();//启动一个线程处理对应的文件newThread(()->{System.out.println(LocalDateTime.now());files.forEach(file->{inFileRun(file,params);});//删除对应的文件fsc.deleteAllFilesInDirectory();System.out.println(LocalDateTime.now());}).start();return"inFile";}

文件保存/读取/删除代码

importorg.springframework.web.multipart.MultipartFile;importjava.io.ByteArrayInputStream;importjava.io.File;importjava.io.IOException;importjava.io.InputStream;importjava.util.ArrayList;importjava.util.List;publicclassFileSaveClear{finalStringTEMP_FILE_PATH=System.getProperty("user.dir")+"/temp-files/";// 创建临时文件夹publicvoidcreateTempDirectory(){FiletempDir=newFile(TEMP_FILE_PATH);if(!tempDir.exists()){tempDir.mkdirs();}}// 保存文件到临时文件夹publicStringsaveFileToTemp(MultipartFilefile){try{StringfileName=System.currentTimeMillis()+"_"+file.getOriginalFilename();StringfilePath=TEMP_FILE_PATH+fileName;FiledestFile=newFile(filePath);file.transferTo(destFile);returnfilePath;}catch(Exceptione){e.printStackTrace();returnnull;}}/** * 删除指定文件夹中的所有文件 * @return 删除成功的文件数量 */publicintdeleteAllFilesInDirectory(){Filedirectory=newFile(TEMP_FILE_PATH);if(!directory.exists()||!directory.isDirectory()){return0;}File[]files=directory.listFiles();if(files==null){return0;}intdeletedCount=0;for(Filefile:files){if(file.isFile()&&file.delete()){deletedCount++;}}returndeletedCount;}/** * 读取临时文件夹中的所有文件，返回MultipartFile对象列表 * @return MultipartFile对象列表 */publicList<MultipartFile>getAllTempFilesAsMultipartFiles(){FiletempDir=newFile(TEMP_FILE_PATH);if(!tempDir.exists()||!tempDir.isDirectory()){returnnewArrayList<>();}File[]files=tempDir.listFiles();if(files==null){returnnewArrayList<>();}List<MultipartFile>multipartFiles=newArrayList<>();for(Filefile:files){if(file.isFile()){try{// 读取文件内容byte[]fileContent=java.nio.file.Files.readAllBytes(file.toPath());// 创建MultipartFile对象 - 使用自定义实现MultipartFilemultipartFile=newMultipartFile(){@OverridepublicStringgetName(){returnfile.getName();}@OverridepublicStringgetOriginalFilename(){returnfile.getName();}@OverridepublicStringgetContentType(){// 根据文件扩展名确定内容类型StringfileName=file.getName().toLowerCase();if(fileName.endsWith(".csv")){return"text/csv";}elseif(fileName.endsWith(".txt")){return"text/plain";}elseif(fileName.endsWith(".xlsx")||fileName.endsWith(".xls")){return"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";}else{return"application/octet-stream";}}@OverridepublicbooleanisEmpty(){returnfileContent.length==0;}@OverridepubliclonggetSize(){returnfileContent.length;}@Overridepublicbyte[]getBytes()throwsIOException{returnfileContent;}@OverridepublicInputStreamgetInputStream()throwsIOException{returnnewByteArrayInputStream(fileContent);}@OverridepublicvoidtransferTo(Filedest)throwsIOException,IllegalStateException{java.nio.file.Files.write(dest.toPath(),fileContent);}};multipartFiles.add(multipartFile);}catch(Exceptione){System.err.println("读取文件 "+file.getName()+" 时出错: "+e.getMessage());}}}returnmultipartFiles;}}