Java实现批量导出导入数据及附件文件zip包

第Java实现批量导出导入数据及附件文件zip包目录前言-应用场景一、导出ZIP包1.列表数据导出到本地excel文件2.下载附件信息3.生成压缩文件（浏览器下载）4.删除临时目录二、导入ZIP包1.上传zip包，解压到临时目录2.读取附件信息上传到文件服务器3.读取Excel文件存入数据库4.删除临时文件

前言-应用场景

某系统在不同单位使用时存在两套生产环境，他们的数据不是互通的，所以这些单位的上一级领导部门在统计数据的时候希望将A系统的数据和附件信息导出到一个压缩包里，然后把这个压缩包一键导入到B系统，这样B系统就包含了全部的数据，上级领导就能看到全部的业务信息，便于统计分析。

一、导出ZIP包

1.列表数据导出到本地excel文件

Stringpath=profile+"/temp/"+DateUtils.dateTimeNow();

Filefile=newFile(path);

if(file.mkdirs()){

System.out.println("文件夹创建成功！创建后的文件目录为："+file.getPath());

//1.输出Excel文件

HSSFWorkbookworkbook=newHSSFWorkbook();

HSSFSheetsheet=workbook.createSheet("sheet");

StringfileName="XX数据导出.xls";

StringsavePath=file.getPath()+File.separator+fileName;

OutputStreamos=newFileOutputStream(savePath);

//响应到客户端(即浏览器端直接弹出下载连接的方式)需要用response获取流

//this.setResponseHeader(response,filename);

//OutputStreamos=response.getOutputStream();

ListHashMapdataList=newArrayList();

try{

//表头

this.createExcelTitle(workbook,sheet);

//查询条件

HashMapparam=this.buildQueryParams(params);

dataList=shareRegisterMapper.shareList(param);

if(CollectionUtils.isEmpty(dataList)){

return;

this.dealAssetData(dataList,sheet);

//处理子表数据

this.dealAssetDetailData(dataList,workbook);

workbook.write(os);

os.flush();

os.close();

}catch(Exceptione){

e.printStackTrace();

​​​​​​​}finally{

if(os!=null){

os.flush();

os.close();

workbook.close();

}

2.下载附件信息

在上一步生成的Excel文件路径下新建files文件夹，里面存放附件

publicvoiddownloadFile(ListHashMapdataList)throwsException{

StringurlPrefix=request.getScheme()+"://"+request.getServerName()+":"+request.getServerPort()+request.getContextPath();

StringsavePath=file.getPath()+File.separator+"/files";

OutputStreamos=null;

InputStreamis=null;

inti=0;

try{

for(HashMapdata:dataList){

if(data==null||data.get("FILES")==null){

continue;

ListZsglFileEntityentityList=newArrayList();

String[]fileArray=data.get("FILES").toString().split(",");

ListidList=Arrays.asList(fileArray);

entityList.addAll(fileMapper.selectByIds(idList));

if(CollectionUtils.isNotEmpty(entityList)){

for(ZsglFileEntityfile:entityList){

if(file.getFssFileId()==null){

continue;

StringfileUrl=urlPrefix+"/fss/download/"+file.getFssFileId();

//构造URL

URLurl=newURL(fileUrl);

//打开连接

URLConnectioncon=url.openConnection();

//设置请求超时为5s

con.setConnectTimeout(5*1000);

//输入流

is=con.getInputStream();

FiletempFile=newFile(savePath+"/"+file.getFileName());

//校验文件夹目录是否存在，不存在就创建一个目录

if(!tempFile.getParentFile().exists()){

tempFile.getParentFile().mkdirs();

os=newFileOutputStream(tempFile);

is=con.getInputStream();

con.getHeaderFields();

IOUtils.copy(is,os);

System.out.println("下载完成");

entityList.clear();

}catch(IOExceptione){

System.err.println(e);

}finally{

IOUtils.closeQuietly(is);

IOUtils.closeQuietly(os);

}

3.生成压缩文件（浏览器下载）

response.setCharacterEncoding("UTF-8");

response.setContentType("multipart/form-data");

response.setHeader("content-disposition","attachment;filename="+"XX数据导出.zip");

ZipOutputStreamzos=newZipOutputStream(response.getOutputStream());

try{

File[]sourceFiles=file.listFiles();

if(null==sourceFiles||sourceFiles.length1){

System.out.println("待压缩的文件目录："+"里面不存在文件，无需压缩.");

}else{

for(inti=0;isourceFiles.length;i++){

FilesrcFile=sourceFiles[i];

if(srcFile.isDirectory()){

File[]imageSourceFiles=srcFile.listFiles();

if(null==imageSourceFiles||imageSourceFiles.length1){

continue;

for(FileimageFile:imageSourceFiles){

compress(zos,imageFile,srcFile.getName()+"/");

}else{

compress(zos,srcFile,"");

}catch(Exceptione){

e.printStackTrace();

}finally{

//关闭流

try{

if(null!=zos){

zos.close();

}catch(IOExceptione){

e.printStackTrace();

}

其中的压缩方法如下：

publicvoidcompress(ZipOutputStreamout,FilesourceFile,Stringbase)throwsException

out.putNextEntry(newZipEntry(base+sourceFile.getName()));

FileInputStreamfos=newFileInputStream(sourceFile);

BufferedInputStreambis=newBufferedInputStream(fos);

inttag;

System.out.println(base);

//将源文件写入到zip文件中

while((tag=bis.read())!=-1){

out.write(tag);

out.flush();

out.closeEntry();

bis.close();

fos.close();

}

4.删除临时目录

publicvoiddeleteDirectory(Filefile){

File[]list=file.listFiles();//无法做到list多层文件夹数据

if(list!=null){

for(Filetemp:list){//先去递归删除子文件夹及子文件

deleteDirectory(temp);//注意这里是递归调用

if(!file.delete()){//再删除自己本身的文件夹

logger.error("文件删除失败:%s%n",file);

二、导入ZIP包

1.上传zip包，解压到临时目录

这里开始想着在不解压的情况下读取里面的文件，结果没有走通。因为zip里面包含了子文件夹里面的附件信息需要解析。不解压直接解析文件适用于只需要解析zip包中第一层文件的场景，如果子文件夹下的文件也需要处理的话，最好解压后再处理。

publicvoidunzip(ZipInputStreamzipIn,StringdestDirectory)throwsIOException{

FiledestDir=newFile(destDirectory);

if(!destDir.exists()){

destDir.mkdirs();

ZipEntryentry=zipIn.getNextEntry();

//遍历Zip文件中的条目

while(entry!=null){

StringfilePath=destDirectory+File.separator+entry.getName();

if(!entry.isDirectory()){

intindex=entry.getName().indexOf("/");

if(index-1entry.getName().length()index){

FiletempFile=newFile(destDirectory+File.separator+entry.getName().substring(0,index));

if(!tempFile.exists()){

tempFile.mkdir();

FilecheckFile=newFile(filePath);

if(!checkFile.exists()){

checkFile.createNewFile();//创建目标文件

//如果条目是文件直接解压

BufferedOutputStreambos=newBufferedOutputStream(newFileOutputStream(filePath));

byte[]bytesIn=newbyte[1024];

intread=0;

while((read=zipIn.read(bytesIn))!=-1){

bos.write(bytesIn,0,read);

bos.close();

}else{

Filedir=newFile(filePath);

if(!dir.exists()){

dir.mkdirs();

zipIn.closeEntry();

entry=zipIn.getNextEntry();

zipIn.close();

}

这里解压遇到了一个问题，之前导出生成的zip包直接导入没问题，但是我把导出的包手动解压后修改了部分数据重新压缩后再导入报错：ZipInputStream解压远程文件报错，java.lang.IllegalArgumentException:MALFORMED

原因：文件名含有中文，zip解析出错

解决方案，如下行代码，在生成ZipInputStream的时候指定编码格式。

ZipInputStreamzis=newZipInputStream(newBufferedInputStream(inputStream),Charset.forName(GBK));

2.读取附件信息上传到文件服务器

publicListHashMapreadLocalFile()throwsException{

Filefile=newFile(destDirectory+"/files");

ListHashMapfssList=newArrayList();

if(file.exists()){

File[]sourceFiles=file.listFiles();

if(null==sourceFiles||sourceFiles.length1){

System.out.println(file.getName()+"目录里面不存在文件，无需处理.");

returnfssList;

}else{

for(inti=0;isourceFiles.length;i++){

FilesrcFile=sourceFiles[i];

FileItemFactoryfactory=newDiskFileItemFactory(16,null);

FileItemitem=factory.createItem(srcFile.getName(),"text/plain",true,srcFile.getName());

intbytesRead=0;

byte[]buffer=newbyte[8192];

try{

FileInputStreamfis=newFileInputStream(srcFile);

OutputStreamos=item.getOutputStream();

while((bytesRead=fis.read(buffer,0,8192))!=-1){

os.write(buffer,0,bytesRead);

os.close();

fis.close();

}catch(IOException