1.pom引入
<dependency>
<groupId>net.coobird</groupId>
<artifactId>thumbnailator</artifactId>
<version>0.4.8</version>
</dependency>
2.编写工具类
@Slf4j
public class ZipUtils {
private static final Integer ZERO = 0;
private static final Integer ONE_ZERO_TWO_FOUR = 1024;
private static final Integer NINE_ZERO_ZERO = 900;
private static final Integer THREE_TWO_SEVEN_FIVE = 3275;
private static final Integer TWO_ZERO_FOUR_SEVEN = 2047;
private static final Double ZERO_EIGHT_FIVE = 0.85;
private static final Double ZERO_SIX = 0.6;
private static final Double ZERO_FOUR_FOUR = 0.44;
private static final Double ZERO_FOUR = 0.4;
private static final String JPG = ".jpg";
/**
* 导出Zip文件
* @param fileName 文件名称
* @param fileUrl 需要打包的文件地址
*
*/
public static void exportZipFile(String fileName, List<String> fileUrl, HttpServletResponse response, HttpServletRequest request,String dowUrl) {
try {
// 浏览器处理乱码问题
String userAgent = request.getHeader("User-Agent");
// filename.getBytes("UTF-8")处理safari的乱码问题
byte[] bytes = userAgent.contains("MSIE") ? fileName.getBytes() : fileName.getBytes("UTF-8");
// 各浏览器基本都支持ISO编码
fileName = new String(bytes, "ISO-8859-1");
// 文件名外的双引号处理firefox的空格截断问题
response.setHeader("Content-disposition", String.format("attachment; filename=\"%s\"", fileName));
response.setContentType("application/x-msdownload");
response.setCharacterEncoding("UTF-8");
generateZipFile(response, fileUrl,dowUrl);
} catch (Exception e) {
log.info("下载失败!-->" + e);
}
}
/**
* 生成Zip文件
*/
public static void generateZipFile(HttpServletResponse response, List<String> fileUrl,String dowUrl) throws IOException {
ZipOutputStream zipOut = new ZipOutputStream(response.getOutputStream());
for (int i = 0; i < fileUrl.size(); i++) {
File file = new File(fileUrl.get(i));
if(file.exists()){
String url = fileUrl.get(i);
//截取文件名称
String name = url.substring(url.lastIndexOf(File.separator), url.length());
name = name.replace(File.separator,"");
zipOut.putNextEntry(new ZipEntry(name));
InputStream streams = getInputStreamByUrl(fileUrl.get(i));
String fileName = url.substring(url.lastIndexOf("."),url.length());
//转byte字节
byte[] imageBytes = toByteArray(streams);
//压缩为最大512
byte[] compressImg = compressPicForScale(imageBytes,512,fileName);
InputStream stream = new ByteArrayInputStream(compressImg);
if (null == stream) {
continue;
}
int ch;
byte[] buffer=new byte[1024];
while((ch=stream.read(buffer))!=-1){
zipOut.write(buffer,0,ch);
}
stream.close();
}
}
zipOut.closeEntry();
zipOut.close();
//删除临时文件
deleteDirectory(new File(dowUrl));
}
/**
* 删除目录(文件夹)以及目录下的文件
* @param file :
*
* @return void
* @date 2021/10/12
**/
public static void deleteDirectory(File file){
if(file.isFile()){
file.delete();//清理文件
}else{
File[] list = file.listFiles();
if(list!=null){
for(File f: list){
f.delete();
}
}
}
}
/**
* 通过url读取文件信息
*/
public static InputStream getInputStreamByUrl(String url) {
InputStream ins = null;
try {
ins = new FileInputStream(new File(url));
} catch (IOException e) {
log.info("文件读取失败!-->" + e);
}
return ins;
}
/**
* 将inputstream输入流转换成byte[]字节
* @param input :
*
* @return byte[]
* @date 2021/10/10
**/
public static byte[] toByteArray(InputStream input) throws IOException {
ByteArrayOutputStream output = new ByteArrayOutputStream();
byte[] buffer = new byte[1024*4];
int n = 0;
while (-1 != (n = input.read(buffer))) {
output.write(buffer, 0, n);
}
return output.toByteArray();
}
/**
* 根据指定大小压缩图片
*
* @param imageBytes 源图片字节数组
* @param desFileSize 指定图片大小,单位kb
* @return 压缩质量后的图片字节数组
*/
public static byte[] compressPicForScale(byte[] imageBytes, long desFileSize,String fileName) {
if (imageBytes == null || imageBytes.length <= ZERO || imageBytes.length < desFileSize * ONE_ZERO_TWO_FOUR || !fileName.equals(JPG)) {
return imageBytes;
}
long srcSize = imageBytes.length;
double accuracy = getAccuracy(srcSize / ONE_ZERO_TWO_FOUR);
try {
while (imageBytes.length > desFileSize * ONE_ZERO_TWO_FOUR) {
ByteArrayInputStream inputStream = new ByteArrayInputStream(imageBytes);
ByteArrayOutputStream outputStream = new ByteArrayOutputStream(imageBytes.length);
Thumbnails.of(inputStream)
.scale(accuracy)
.outputQuality(accuracy)
.toOutputStream(outputStream);
imageBytes = outputStream.toByteArray();
}
log.info("图片原大小={}kb | 压缩后大小={}kb",
srcSize / ONE_ZERO_TWO_FOUR, imageBytes.length / ONE_ZERO_TWO_FOUR);
} catch (Exception e) {
log.error("【图片压缩】msg=图片压缩失败!", e);
}
return imageBytes;
}
/**
* 自动调节精度(经验数值)
*
* @param size 源图片大小
* @return 图片压缩质量比
*/
private static double getAccuracy(long size) {
double accuracy;
if (size < NINE_ZERO_ZERO) {
accuracy = ZERO_EIGHT_FIVE;
} else if (size < TWO_ZERO_FOUR_SEVEN) {
accuracy = ZERO_SIX;
} else if (size < THREE_TWO_SEVEN_FIVE) {
accuracy = ZERO_FOUR_FOUR;
} else {
accuracy = ZERO_FOUR;
}
return accuracy;
}
3.在方法中调用ZipUtils
@Override
public void test(HttpServletResponse response, HttpServletRequest request) {
ZipUtils.exportZipFile("资料图片" + DateUtil.format(new Date(), DatePattern.PURE_DATETIME_PATTERN) + ".zip", fileUrl,
response, request, dowUrl);
}