package com.boying.service.impl;
|
|
import cn.hutool.core.io.IoUtil;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.boying.common.SystemConfigProperties;
|
import com.boying.entity.EnterPark;
|
import com.boying.entity.FileInfo;
|
import com.boying.mapper.EnterParkMapper;
|
import com.boying.mapper.FileInfoMapper;
|
import com.boying.service.FileInfoService;
|
import lombok.AllArgsConstructor;
|
import org.springframework.http.MediaType;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.FileCopyUtils;
|
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.net.URLEncoder;
|
import java.text.SimpleDateFormat;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.zip.Adler32;
|
import java.util.zip.CheckedOutputStream;
|
import java.util.zip.ZipEntry;
|
import java.util.zip.ZipOutputStream;
|
|
@Service
|
@AllArgsConstructor
|
public class FileInfoServiceImpl extends ServiceImpl<FileInfoMapper, FileInfo> implements FileInfoService {
|
|
private final SystemConfigProperties systemConfigProperties;
|
private final FileInfoMapper fileInfoMapper;
|
@Override
|
public String downloadPathFile(String downloadPath, HttpServletRequest request, HttpServletResponse response) {
|
//设置文件路径
|
File file = new File(downloadPath);
|
//获取文件名称
|
String fileName = file.getName();
|
//判断文件是否存在
|
if (file.exists()) {
|
response.setContentType("application/force-download");// 设置强制下载不打开
|
response.addHeader("Content-Disposition", "attachment;fileName=" + fileName);// 设置文件名
|
byte[] buffer = new byte[1024];
|
FileInputStream fis = null;
|
BufferedInputStream bis = null;
|
try {
|
fis = new FileInputStream(file);
|
bis = new BufferedInputStream(fis);
|
OutputStream os = response.getOutputStream();
|
int i = bis.read(buffer);
|
while (i != -1) {
|
os.write(buffer, 0, i);
|
i = bis.read(buffer);
|
}
|
file.delete();
|
return "下载成功";
|
} catch (Exception e) {
|
e.printStackTrace();
|
} finally {
|
if (bis != null) {
|
try {
|
bis.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
if (fis != null) {
|
try {
|
fis.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
}
|
return "下载失败";
|
}
|
|
@Override
|
public void downForZip(List<FileInfo> fileInfoList, HttpServletRequest req, HttpServletResponse response) {
|
// 创建临时文件
|
File zipFile = null;
|
FileInputStream fis = null;
|
BufferedInputStream buff = null;
|
try {
|
//临时文件名称
|
zipFile = File.createTempFile("test", ".zip");
|
FileOutputStream fot = new FileOutputStream(zipFile);
|
// 为任何OutputStream产生校验，第一个参数是制定产生校验和的输出流，第二个参数是指定Checksum的类型 (Adler32(较快)和CRC32两种)
|
CheckedOutputStream cos = new CheckedOutputStream(fot, new Adler32());
|
// 用于将数据压缩成Zip文件格式
|
ZipOutputStream zos = new ZipOutputStream(cos);
|
for (FileInfo file: fileInfoList) {
|
//根据文件信息获取输入流。这个看实际的业务目的就是获取当前文件的输入流
|
InputStream inputStream = new FileInputStream(new File(file.getFileBaseDir()+file.getFileName()+file.getFileLast()));
|
if (null == inputStream) {
|
System.out.println("输入流为空");
|
break;
|
}
|
// 对于每一个要被存放到压缩包的文件，都必须调用ZipOutputStream对象的putNextEntry()方法，确保压缩包里面文件不同名
|
//多个文件名称重复时zos.putNextEntry()会报错！！ 可以再文件名称后面加编号等操作
|
zos.putNextEntry(new ZipEntry(file.getFileName()+file.getFileLast()));
|
int bytesRead = 0;
|
// 向压缩文件中输出数据
|
while ((bytesRead = inputStream.read()) != -1) {
|
System.out.println("写入数据");
|
zos.write(bytesRead);
|
}
|
inputStream.close();
|
// 当前文件写完，写入下一个文件
|
zos.closeEntry();
|
}
|
zos.close();
|
ServletOutputStream os = response.getOutputStream();
|
//下载文件,使用spring框架中的FileCopyUtils工具
|
//response.setCharacterEncoding("GB2312");
|
response.setContentType("application/octet-stream");
|
//设置响应头,attachment表示以附件的形式下载，inline表示在线打开
|
SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
|
response.setHeader("Content-Disposition", "attachment;fileName=" + new String((sdf.format(new Date())+".zip").getBytes("utf-8"),"iso8859-1"));
|
fis = new FileInputStream(zipFile);
|
buff = new BufferedInputStream(fis);
|
FileCopyUtils.copy(buff, os);
|
} catch (Exception e1) {
|
System.out.println("批量下载失败");
|
e1.printStackTrace();
|
// 关闭流
|
} finally {
|
try {
|
if (null != fis) {
|
fis.close();
|
}
|
if (null != buff) {
|
buff.close();
|
}
|
} catch (IOException e) {
|
System.out.println("流关闭异常");
|
log.error( "流关闭异常");
|
}
|
// 删除临时文件
|
if (null != zipFile) {
|
zipFile.delete();
|
}
|
}
|
}
|
|
@Override
|
public void downloadPng(FileInfo byId, HttpServletRequest request, HttpServletResponse response) {
|
try {
|
InputStream input = new FileInputStream(new File(byId.getFileBaseDir()+byId.getFileName()+byId.getFileLast()));
|
response.setContentType("application/octet-stream; charset=UTF-8");
|
IoUtil.copy(input, response.getOutputStream());
|
}catch (Exception e){
|
System.out.println("文件读取异常，"+e.getLocalizedMessage());
|
}
|
}
|
|
@Override
|
public void getFile(String downloadPath, HttpServletResponse response) {
|
try {
|
InputStream input = new FileInputStream(new File(systemConfigProperties.getUploadImgPath()+downloadPath));
|
response.setContentType(MediaType.IMAGE_PNG_VALUE);
|
IoUtil.copy(input, response.getOutputStream());
|
}catch (Exception e){
|
System.out.println("文件读取异常，"+e.getLocalizedMessage());
|
}
|
}
|
|
@Override
|
public void getFileById(Long fileId, HttpServletResponse response) {
|
try {
|
FileInfo fileInfo = fileInfoMapper.selectById(fileId);
|
InputStream input = new FileInputStream(new File(fileInfo.getFileBaseDir()+fileInfo.getFileName()+fileInfo.getFileLast()));
|
response.setContentType(MediaType.IMAGE_PNG_VALUE);
|
IoUtil.copy(input, response.getOutputStream());
|
}catch (Exception e){
|
System.out.println("文件读取异常，"+e.getLocalizedMessage());
|
}
|
}
|
}
|
