package org.jeecg.modules.utils.file; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.StringUtils; import org.jeecg.common.api.vo.FileUploadResult; import org.jeecg.common.util.SHA256Util; import org.jeecg.modules.utils.DateUtil; import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Component; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; import java.io.*; import java.net.URLEncoder; import java.util.Date; @Slf4j @Component public class FileUtils { private static String fileUploadFolder; @Value("${jeecg.path.upload}") public void setFileUploadFolder(String fileUploadFolder) { FileUtils.fileUploadFolder = fileUploadFolder; } /** * 文件路径上传保存 * @param filePath * @return */ public static FileUploadResult uploadFilePath(String filePath) { File file = new File(filePath); if(file == null) { return null; } String fileName = file.getName(); String suffix = getFileSuffix(fileName); Date currentDate = DateUtil.getNow(); String monthStr = DateUtil.format(currentDate, DateUtil.STR_YEAR_MONTH); //相对路径 String relativePath = "/" + monthStr + "/" + DateUtil.getDayStr(currentDate) + "/"; //绝对路径 String absolutePath = fileUploadFolder + "/" + monthStr + "/" + DateUtil.getDayStr(currentDate) + "/"; String fileNameNonSuffix = getFilenameNonSuffix(fileName); if(fileNameNonSuffix == null) return null; String encodeFileName = SHA256Util.getSHA256Str(fileNameNonSuffix + System.currentTimeMillis()); if(StringUtils.isNotBlank(suffix)) { encodeFileName = encodeFileName + "." + suffix; } Long b = uploadFileInput(file, absolutePath, encodeFileName); FileUploadResult dto = new FileUploadResult(); dto.setFileName(fileNameNonSuffix); dto.setFilePath(encodeFileName); dto.setFilePath(relativePath); dto.setFileSize(b); dto.setFileSuffix(suffix); return dto; } /** * 文件File上传保存 * @param file * @return */ public static FileUploadResult uploadFileInFile(File file) { if(file == null) { return null; } String fileName = file.getName(); String suffix = getFileSuffix(fileName); Date currentDate = DateUtil.getNow(); String monthStr = DateUtil.format(currentDate, DateUtil.STR_YEAR_MONTH); //相对路径 String relativePath = "/" + monthStr + "/" + DateUtil.getDayStr(currentDate) + "/"; //绝对路径 String absolutePath = fileUploadFolder + "/" + monthStr + "/" + DateUtil.getDayStr(currentDate) + "/"; String fileNameNonSuffix = getFilenameNonSuffix(fileName); if(fileNameNonSuffix == null) { return null; } String encodeFileName = SHA256Util.getSHA256Str(fileNameNonSuffix + System.currentTimeMillis()); if(StringUtils.isNotBlank(suffix)) { encodeFileName = encodeFileName + "." + suffix; } Long b = uploadFileInput(file, absolutePath, encodeFileName); FileUploadResult dto = new FileUploadResult(); dto.setFileName(fileNameNonSuffix); dto.setFileEncodeName(encodeFileName); dto.setFilePath(relativePath); dto.setFileSize(b); dto.setFileSuffix(suffix); return dto; } /** * MultipartFile 方式保存 * @param file * @return */ public static FileUploadResult uploadMultipartFile(MultipartFile file) { if(file == null || file.isEmpty()) { return null; } String fileName = file.getOriginalFilename(); String suffix = getFileSuffix(fileName); Date currentDate = DateUtil.getNow(); String monthStr = DateUtil.format(currentDate, DateUtil.STR_YEAR_MONTH); //相对路径 String relativePath = "/" + monthStr + "/" + DateUtil.getDayStr(currentDate) + "/"; //绝对路径 String absolutePath = fileUploadFolder + "/" + monthStr + "/" + DateUtil.getDayStr(currentDate) + "/"; String fileNameNonSuffix = getFilenameNonSuffix(fileName); if(fileNameNonSuffix == null) return null; String encodeFileName = SHA256Util.getSHA256Str(fileNameNonSuffix + System.currentTimeMillis()); Long fileSize = file.getSize(); if(StringUtils.isNotBlank(suffix)) { encodeFileName = encodeFileName + "." + suffix; } boolean b = uploadFile(file, absolutePath, encodeFileName); if(!b) return null; FileUploadResult dto = new FileUploadResult(); dto.setFileName(fileNameNonSuffix); dto.setFilePath(encodeFileName); dto.setFilePath(relativePath); dto.setFileSize(fileSize); dto.setFileSuffix(suffix); return dto; } public static void downLoadFile(HttpServletResponse response, String fileName, String filePath, String toFileName) { String absolutePath = fileUploadFolder + filePath; File file = new File(absolutePath , fileName); if(file.exists()) { byte[] buffer = new byte[1024]; FileInputStream fis = null; BufferedInputStream bis = null; try { response.setHeader("Content-Type", "application/octet-stream;charset=utf-8"); // 告诉浏览器输出内容为流 response.setHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode(toFileName, "UTF-8")); 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); } }catch (Exception e) { log.error(e.getMessage(), e.getStackTrace()); }finally { if(bis != null) { try { bis.close(); } catch (IOException e) { } } if(fis != null) { try { fis.close(); } catch (IOException e) { } } } }else { } } /** * 上传文件工具类 * @param file * @param path * @param fileNewName 新的文件名 * @return */ public static Long uploadFileInput(File file, String path, String fileNewName) { File targetFile = new File(path, fileNewName); if(!targetFile.getParentFile().exists()){ targetFile.getParentFile().mkdirs(); } try { FileInputStream fis = new FileInputStream(file); FileOutputStream fos = new FileOutputStream(targetFile); //FileInputStream fis = new FileInputStream(file); byte[] bytes = new byte[1024 * 4]; int index = 0; long total = 0; while ((index = fis.read(bytes)) != -1) { fos.write(bytes, 0, index); total =+ index; } fis.close(); fos.flush(); fos.close(); return total; } catch (Exception e) { log.error(e.getMessage(), e.getStackTrace()); return 0L; } } /** * 上传文件工具类 * @param multipartFile * @param path * @param fileNewName 新的文件名 * @return */ public static boolean uploadFile(MultipartFile multipartFile, String path, String fileNewName) { File targetFile = new File(path, fileNewName); if(!targetFile.getParentFile().exists()){ targetFile.getParentFile().mkdirs(); } try { multipartFile.transferTo(targetFile); return true; } catch (Exception e) { log.error(e.getMessage(), e.getStackTrace()); return false; } } /** * 获取文件后缀 无后缀文件返回NULL * @param fileName * @return */ public static String getFileSuffix(String fileName) { if (fileName.contains(".")) { String suffix = fileName.substring(fileName.lastIndexOf('.') + 1); return suffix; }else { return null; } } /** * 获取文件名 不带后缀和点号 * @param fileName * @return */ public static String getFilenameNonSuffix(String fileName) { if (fileName.contains(".")) { String filename = fileName.substring(0, fileName.lastIndexOf('.')); return filename; }else { return fileName; } } public static String checkFileEncode(File file) { String charset = "GBK"; byte[] first3Bytes = new byte[3]; BufferedInputStream bis = null; try { boolean checked = false; bis = new BufferedInputStream(new FileInputStream(file)); bis.mark(0); int read = bis.read(first3Bytes, 0, 3); if (read == -1) return charset; if(first3Bytes[0] == (byte) 0xFF && first3Bytes[1] == (byte) 0xFE ) { charset = "UTF-16LE"; checked = true; }else if ( first3Bytes[0] == (byte) 0xFE && first3Bytes[1] == (byte) 0xFF ) { charset = "UTF-16BE"; checked = true; }else if (first3Bytes[0] == (byte) 0xEF && first3Bytes[1] == (byte) 0xBB && first3Bytes[2] == (byte) 0xBF ) { charset = "UTF-8"; checked = true; } bis.reset(); if (!checked) { while ((read = bis.read()) != -1 ) { if ( read >= 0xF0 ) break; if ( 0x80 <= read && read <= 0xBF ) // 单独出现BF以下的，也算是GBK break; if ( 0xC0 <= read && read <= 0xDF ) { read = bis.read(); if ( 0x80 <= read && read <= 0xBF ) // 双字节 (0xC0 - 0xDF) (0x80 - 0xBF),也可能在GBK编码内   continue; else break; } else if ( 0xE0 <= read && read <= 0xEF ) { // 也有可能出错，但是几率较小 read = bis.read(); if ( 0x80 <= read && read <= 0xBF ) { read = bis.read(); if ( 0x80 <= read && read <= 0xBF ) { charset = "UTF-8"; break; } else break; } else { break; } } } } return charset; }catch (Exception e) { return null; }finally { if(bis != null) { try { bis.close(); } catch (IOException e) { } } } } }