|
@@ -0,0 +1,412 @@
|
|
|
|
|
+/*
|
|
|
|
|
+ * Copyright 2019-2020 Zheng Jie
|
|
|
|
|
+ *
|
|
|
|
|
+ * Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
|
+ * you may not use this file except in compliance with the License.
|
|
|
|
|
+ * You may obtain a copy of the License at
|
|
|
|
|
+ *
|
|
|
|
|
+ * http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
+ *
|
|
|
|
|
+ * Unless required by applicable law or agreed to in writing, software
|
|
|
|
|
+ * distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
|
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
|
+ * See the License for the specific language governing permissions and
|
|
|
|
|
+ * limitations under the License.
|
|
|
|
|
+ */
|
|
|
|
|
+package com.coffee.common.util;
|
|
|
|
|
+
|
|
|
|
|
+import cn.hutool.core.io.IoUtil;
|
|
|
|
|
+import cn.hutool.core.util.IdUtil;
|
|
|
|
|
+import cn.hutool.poi.excel.BigExcelWriter;
|
|
|
|
|
+import cn.hutool.poi.excel.ExcelUtil;
|
|
|
|
|
+import cn.hutool.poi.excel.ExcelWriter;
|
|
|
|
|
+import com.coffee.common.exception.CustomException;
|
|
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
|
|
+import org.apache.poi.util.IOUtils;
|
|
|
|
|
+import org.springframework.web.multipart.MultipartFile;
|
|
|
|
|
+
|
|
|
|
|
+import javax.servlet.ServletOutputStream;
|
|
|
|
|
+import javax.servlet.http.HttpServletRequest;
|
|
|
|
|
+import javax.servlet.http.HttpServletResponse;
|
|
|
|
|
+import java.io.*;
|
|
|
|
|
+import java.security.MessageDigest;
|
|
|
|
|
+import java.text.DecimalFormat;
|
|
|
|
|
+import java.text.SimpleDateFormat;
|
|
|
|
|
+import java.util.*;
|
|
|
|
|
+import java.util.zip.ZipEntry;
|
|
|
|
|
+import java.util.zip.ZipOutputStream;
|
|
|
|
|
+
|
|
|
|
|
+/**
|
|
|
|
|
+ * File工具类,扩展 hutool 工具包
|
|
|
|
|
+ * @author Zheng Jie
|
|
|
|
|
+ * @date 2018-12-27
|
|
|
|
|
+ */
|
|
|
|
|
+@Slf4j
|
|
|
|
|
+public class FileUtil extends cn.hutool.core.io.FileUtil {
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 定义GB的计算常量
|
|
|
|
|
+ */
|
|
|
|
|
+ private static final int GB = 1024 * 1024 * 1024;
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 定义MB的计算常量
|
|
|
|
|
+ */
|
|
|
|
|
+ private static final int MB = 1024 * 1024;
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 定义KB的计算常量
|
|
|
|
|
+ */
|
|
|
|
|
+ private static final int KB = 1024;
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 格式化小数
|
|
|
|
|
+ */
|
|
|
|
|
+ private static final DecimalFormat DF = new DecimalFormat("0.00");
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * MultipartFile转File
|
|
|
|
|
+ */
|
|
|
|
|
+ public static File toFile(MultipartFile multipartFile){
|
|
|
|
|
+ return toFile(multipartFile,"");
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * MultipartFile转File
|
|
|
|
|
+ */
|
|
|
|
|
+ public static File toFile(MultipartFile multipartFile, String prefixName){
|
|
|
|
|
+ // 获取文件名
|
|
|
|
|
+ String fileName = multipartFile.getOriginalFilename();
|
|
|
|
|
+ // 获取文件后缀
|
|
|
|
|
+ String suffix="."+getExtensionName(fileName);
|
|
|
|
|
+ File file = null;
|
|
|
|
|
+ try {
|
|
|
|
|
+ // 用uuid作为文件名,防止生成的临时文件重复
|
|
|
|
|
+ file = FileUtil.file(prefixName+ IdUtil.simpleUUID()+suffix);
|
|
|
|
|
+ // MultipartFile to File
|
|
|
|
|
+ multipartFile.transferTo(file);
|
|
|
|
|
+ } catch (IOException e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ }
|
|
|
|
|
+ return file;
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 获取文件扩展名,不带 .
|
|
|
|
|
+ */
|
|
|
|
|
+ public static String getExtensionName(String filename) {
|
|
|
|
|
+ if ((filename != null) && (filename.length() > 0)) {
|
|
|
|
|
+ int dot = filename.lastIndexOf('.');
|
|
|
|
|
+ if ((dot >-1) && (dot < (filename.length() - 1))) {
|
|
|
|
|
+ return filename.substring(dot + 1);
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+ return filename;
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * Java文件操作 获取不带扩展名的文件名
|
|
|
|
|
+ */
|
|
|
|
|
+ public static String getFileNameNoEx(String filename) {
|
|
|
|
|
+ if ((filename != null) && (filename.length() > 0)) {
|
|
|
|
|
+ int dot = filename.lastIndexOf('.');
|
|
|
|
|
+ if ((dot >-1) && (dot < (filename.length()))) {
|
|
|
|
|
+ return filename.substring(0, dot);
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+ return filename;
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 文件大小转换
|
|
|
|
|
+ */
|
|
|
|
|
+ public static String getSize(long size){
|
|
|
|
|
+ String resultSize;
|
|
|
|
|
+ if (size / GB >= 1) {
|
|
|
|
|
+ //如果当前Byte的值大于等于1GB
|
|
|
|
|
+ resultSize = DF.format(size / (float) GB) + "GB ";
|
|
|
|
|
+ } else if (size / MB >= 1) {
|
|
|
|
|
+ //如果当前Byte的值大于等于1MB
|
|
|
|
|
+ resultSize = DF.format(size / (float) MB) + "MB ";
|
|
|
|
|
+ } else if (size / KB >= 1) {
|
|
|
|
|
+ //如果当前Byte的值大于等于1KB
|
|
|
|
|
+ resultSize = DF.format(size / (float) KB) + "KB ";
|
|
|
|
|
+ } else {
|
|
|
|
|
+ resultSize = size + "B ";
|
|
|
|
|
+ }
|
|
|
|
|
+ return resultSize;
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 将文件名解析成文件的上传路径
|
|
|
|
|
+ */
|
|
|
|
|
+ public static File upload(MultipartFile file, String filePath) {
|
|
|
|
|
+ Date date = new Date();
|
|
|
|
|
+ SimpleDateFormat format = new SimpleDateFormat("yyyyMMddhhmmssS");
|
|
|
|
|
+ String name = getFileNameNoEx(file.getOriginalFilename());
|
|
|
|
|
+ String suffix = getExtensionName(file.getOriginalFilename());
|
|
|
|
|
+ String nowStr = "-" + format.format(date);
|
|
|
|
|
+ try {
|
|
|
|
|
+ String fileName = name + nowStr + "." + suffix;
|
|
|
|
|
+ String path = filePath + fileName;
|
|
|
|
|
+ // getCanonicalFile 可解析正确各种路径
|
|
|
|
|
+// File dest = new File(path).getCanonicalFile();
|
|
|
|
|
+ log.info("上传文件路径为:{}",path);
|
|
|
|
|
+ File dest= cn.hutool.core.io.FileUtil.touch(path);
|
|
|
|
|
+ // 检测是否存在目录
|
|
|
|
|
+//
|
|
|
|
|
+// if (!dest.getParentFile().exists()) {
|
|
|
|
|
+//
|
|
|
|
|
+//// if (!dest.getParentFile().mkdirs()) {
|
|
|
|
|
+//// System.out.println("was not successful.>>>>>"+dest.getPath());
|
|
|
|
|
+//// }
|
|
|
|
|
+// }
|
|
|
|
|
+ // 文件写入
|
|
|
|
|
+ file.transferTo(dest);
|
|
|
|
|
+ return dest;
|
|
|
|
|
+ } catch (Exception e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ }
|
|
|
|
|
+ return null;
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 导出excel
|
|
|
|
|
+ */
|
|
|
|
|
+ public static void downloadExcel(List<Map<String, Object>> list, HttpServletResponse response) throws IOException {
|
|
|
|
|
+ String tempPath =System.getProperty("java.io.tmpdir") + IdUtil.fastSimpleUUID() + ".xlsx";
|
|
|
|
|
+ File file = new File(tempPath);
|
|
|
|
|
+ BigExcelWriter writer= ExcelUtil.getBigWriter(file);
|
|
|
|
|
+ // 一次性写出内容,使用默认样式,强制输出标题
|
|
|
|
|
+ writer.write(list, true);
|
|
|
|
|
+ //response为HttpServletResponse对象
|
|
|
|
|
+ response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
|
|
|
|
|
+ //test.xls是弹出下载对话框的文件名,不能为中文,中文请自行编码
|
|
|
|
|
+ response.setHeader("Content-Disposition","attachment;filename=file.xlsx");
|
|
|
|
|
+ ServletOutputStream out=response.getOutputStream();
|
|
|
|
|
+ // 终止后删除临时文件
|
|
|
|
|
+ file.deleteOnExit();
|
|
|
|
|
+ writer.flush(out, true);
|
|
|
|
|
+ //此处记得关闭输出Servlet流
|
|
|
|
|
+ IoUtil.close(out);
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 导出excel
|
|
|
|
|
+ */
|
|
|
|
|
+ public static void downloadExcel(ExcelWriter writer, HttpServletResponse response) throws IOException {
|
|
|
|
|
+ //response为HttpServletResponse对象
|
|
|
|
|
+ response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
|
|
|
|
|
+ //test.xls是弹出下载对话框的文件名,不能为中文,中文请自行编码
|
|
|
|
|
+ response.setHeader("Content-Disposition","attachment;filename=file.xlsx");
|
|
|
|
|
+ ServletOutputStream out=response.getOutputStream();
|
|
|
|
|
+ writer.flush(out, true);
|
|
|
|
|
+ //此处记得关闭输出Servlet流
|
|
|
|
|
+ IoUtil.close(out);
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ public static String getFileType(String type) {
|
|
|
|
|
+ String documents = "txt doc pdf ppt pps xlsx xls docx";
|
|
|
|
|
+ String music = "mp3 wav wma mpa ram ra aac aif m4a";
|
|
|
|
|
+ String video = "avi mpg mpe mpeg asf wmv mov qt rm mp4 flv m4v webm ogv ogg";
|
|
|
|
|
+ String image = "bmp dib pcp dif wmf gif jpg tif eps psd cdr iff tga pcd mpt png jpeg";
|
|
|
|
|
+ if(image.contains(type)){
|
|
|
|
|
+ //图片
|
|
|
|
|
+ return "image";
|
|
|
|
|
+ } else if(documents.contains(type)){
|
|
|
|
|
+ //文档
|
|
|
|
|
+ return "doc";
|
|
|
|
|
+ } else if(music.contains(type)){
|
|
|
|
|
+ //音乐
|
|
|
|
|
+ return "music";
|
|
|
|
|
+ } else if(video.contains(type)){
|
|
|
|
|
+ //视频
|
|
|
|
|
+ return "video";
|
|
|
|
|
+ } else {
|
|
|
|
|
+ //其他
|
|
|
|
|
+ return "else";
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ public static void checkSize(long maxSize, long size) {
|
|
|
|
|
+ long len = 1024 * 1024*maxSize;
|
|
|
|
|
+ if(size > (maxSize * len)){
|
|
|
|
|
+ throw new CustomException("文件超出"+maxSize+"M");
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 判断两个文件是否相同
|
|
|
|
|
+ */
|
|
|
|
|
+ public static boolean check(File file1, File file2) {
|
|
|
|
|
+ String img1Md5 = getMd5(file1);
|
|
|
|
|
+ String img2Md5 = getMd5(file2);
|
|
|
|
|
+ return img1Md5.equals(img2Md5);
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 判断两个文件是否相同
|
|
|
|
|
+ */
|
|
|
|
|
+ public static boolean check(String file1Md5, String file2Md5) {
|
|
|
|
|
+ return file1Md5.equals(file2Md5);
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ @SuppressWarnings("resource")
|
|
|
|
|
+ private static byte[] getByte(File file) {
|
|
|
|
|
+ // 得到文件长度
|
|
|
|
|
+ byte[] b = new byte[(int) file.length()];
|
|
|
|
|
+ try {
|
|
|
|
|
+ InputStream in = new FileInputStream(file);
|
|
|
|
|
+ try {
|
|
|
|
|
+ System.out.println(in.read(b));
|
|
|
|
|
+ } catch (IOException e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ }
|
|
|
|
|
+ } catch (FileNotFoundException e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ return null;
|
|
|
|
|
+ }
|
|
|
|
|
+ return b;
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ private static String getMd5(byte[] bytes) {
|
|
|
|
|
+ // 16进制字符
|
|
|
|
|
+ char[] hexDigits = {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
|
|
|
|
|
+ try {
|
|
|
|
|
+ MessageDigest mdTemp = MessageDigest.getInstance("MD5");
|
|
|
|
|
+ mdTemp.update(bytes);
|
|
|
|
|
+ byte[] md = mdTemp.digest();
|
|
|
|
|
+ int j = md.length;
|
|
|
|
|
+ char[] str = new char[j * 2];
|
|
|
|
|
+ int k = 0;
|
|
|
|
|
+ // 移位 输出字符串
|
|
|
|
|
+ for (byte byte0 : md) {
|
|
|
|
|
+ str[k++] = hexDigits[byte0 >>> 4 & 0xf];
|
|
|
|
|
+ str[k++] = hexDigits[byte0 & 0xf];
|
|
|
|
|
+ }
|
|
|
|
|
+ return new String(str);
|
|
|
|
|
+ } catch (Exception e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ }
|
|
|
|
|
+ return null;
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 下载文件
|
|
|
|
|
+ * @param request /
|
|
|
|
|
+ * @param response /
|
|
|
|
|
+ * @param file /
|
|
|
|
|
+ */
|
|
|
|
|
+ public static void downloadFile(HttpServletRequest request, HttpServletResponse response, File file, boolean deleteOnExit){
|
|
|
|
|
+ response.setCharacterEncoding(request.getCharacterEncoding());
|
|
|
|
|
+ response.setContentType("application/octet-stream");
|
|
|
|
|
+ FileInputStream fis = null;
|
|
|
|
|
+ try {
|
|
|
|
|
+ fis = new FileInputStream(file);
|
|
|
|
|
+ response.setHeader("Content-Disposition", "attachment; filename="+file.getName());
|
|
|
|
|
+ IOUtils.copy(fis,response.getOutputStream());
|
|
|
|
|
+ response.flushBuffer();
|
|
|
|
|
+ } catch (Exception e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ } finally {
|
|
|
|
|
+ if (fis != null) {
|
|
|
|
|
+ try {
|
|
|
|
|
+ fis.close();
|
|
|
|
|
+ if(deleteOnExit){
|
|
|
|
|
+ file.deleteOnExit();
|
|
|
|
|
+ }
|
|
|
|
|
+ } catch (IOException e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 下载文件
|
|
|
|
|
+ * @param request /
|
|
|
|
|
+ * @param response /
|
|
|
|
|
+ * @param /
|
|
|
|
|
+ */
|
|
|
|
|
+ public static void downloadFile(HttpServletRequest request, HttpServletResponse response, InputStream in, String filename){
|
|
|
|
|
+ response.setCharacterEncoding(request.getCharacterEncoding());
|
|
|
|
|
+ response.setContentType("application/octet-stream");
|
|
|
|
|
+ BufferedInputStream fis = null;
|
|
|
|
|
+ try {
|
|
|
|
|
+ response.setHeader("Content-Disposition", "attachment; filename="+filename);
|
|
|
|
|
+ IOUtils.copy(in,response.getOutputStream());
|
|
|
|
|
+ response.flushBuffer();
|
|
|
|
|
+ } catch (Exception e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ } finally {
|
|
|
|
|
+ if (in != null) {
|
|
|
|
|
+ try {
|
|
|
|
|
+ in.close();
|
|
|
|
|
+ } catch (IOException e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+ public static String getMd5(File file) {
|
|
|
|
|
+ return getMd5(getByte(file));
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+ public static void zipFiles(File[] srcfile,File zipfile){
|
|
|
|
|
+ byte[] buf=new byte[1024];
|
|
|
|
|
+ try {
|
|
|
|
|
+ //ZipOutputStream类:完成文件或文件夹的压缩
|
|
|
|
|
+ ZipOutputStream out=new ZipOutputStream(new FileOutputStream(zipfile));
|
|
|
|
|
+ for(int i=0;i<srcfile.length;i++){
|
|
|
|
|
+ FileInputStream in=new FileInputStream(srcfile[i]);
|
|
|
|
|
+ out.putNextEntry(new ZipEntry(srcfile[i].getName()));
|
|
|
|
|
+ int len;
|
|
|
|
|
+ while((len=in.read(buf))!=0){
|
|
|
|
|
+ out.write(buf,0,len);
|
|
|
|
|
+ }
|
|
|
|
|
+ out.closeEntry();
|
|
|
|
|
+ in.close();
|
|
|
|
|
+ }
|
|
|
|
|
+ out.close();
|
|
|
|
|
+ } catch (Exception e) {
|
|
|
|
|
+ e.printStackTrace();
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ /**
|
|
|
|
|
+ * 迭代方式进行文件压缩
|
|
|
|
|
+ * @param file
|
|
|
|
|
+ * @param fileName
|
|
|
|
|
+ * @param outputStream
|
|
|
|
|
+ * @throws IOException
|
|
|
|
|
+ */
|
|
|
|
|
+ public static void compressFile(File file, String fileName, final ZipOutputStream outputStream) throws IOException {
|
|
|
|
|
+ //如果是目录
|
|
|
|
|
+ if (file.isDirectory()) {
|
|
|
|
|
+ //创建文件夹
|
|
|
|
|
+ outputStream.putNextEntry(new ZipEntry(fileName + "/"));
|
|
|
|
|
+ //迭代判断,并且加入对应文件路径
|
|
|
|
|
+ File[] files = file.listFiles();
|
|
|
|
|
+ Iterator<File> iterator = Arrays.asList(files).iterator();
|
|
|
|
|
+ while (iterator.hasNext()) {
|
|
|
|
|
+ File f = iterator.next();
|
|
|
|
|
+ compressFile(f, fileName + "/" + f.getName(), outputStream);
|
|
|
|
|
+ }
|
|
|
|
|
+ } else {
|
|
|
|
|
+ //创建文件
|
|
|
|
|
+ outputStream.putNextEntry(new ZipEntry(fileName));
|
|
|
|
|
+ //读取文件并写出
|
|
|
|
|
+ FileInputStream fileInputStream = new FileInputStream(file);
|
|
|
|
|
+ BufferedInputStream bufferedInputStream = new BufferedInputStream(fileInputStream);
|
|
|
|
|
+ byte[] bytes = new byte[1024];
|
|
|
|
|
+ int n;
|
|
|
|
|
+ while ((n = bufferedInputStream.read(bytes)) != -1) {
|
|
|
|
|
+ outputStream.write(bytes, 0, n);
|
|
|
|
|
+ }
|
|
|
|
|
+ //关闭流
|
|
|
|
|
+ fileInputStream.close();
|
|
|
|
|
+ bufferedInputStream.close();
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+}
|