走upload方法进行文件的保存,第一个参数为上传文件的类型,头像or照片
第一步检查是否可以上传,是否是jpg,png等类型
第二步根据日期,文件名,uuid等生成文件名称。
第三步将文件保存到服务器
最后return的是一个文件的相对地址,根据subDir和fileName+文件名返回的相对路径,比如profile/avatar/test.jpg
相对路径最后会返回给前端,而不是存入本地的路径!并且相对路径会存入数据库,前端发起请求携带相对路径,会被映射到后端地址。
比如前端发起profile/avatar/test.jpg,后端编写静态资源映射类
profile/avatar/最终会映射到 D:\\gooinfo\\security\\upload\\picture\\
第三步将文件保存到本地文件夹代码如下
getFileAbsolutePath(subDir, fileName)方法中指定保存到本地的位置
file_base_url 在此类中为
以下是工具类完整代码
package com.gooinfo.reconfiguration.upload;
import cn.hutool.core.date.DatePattern;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.file.FileNameUtil;
import cn.hutool.core.util.CharsetUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.URLUtil;
import com.gooinfo.reconfiguration.aop.error.ApiException;
import com.gooinfo.reconfiguration.aop.error.ErrorCode;
import com.gooinfo.reconfiguration.async.config.AgileBootConfig;
import com.gooinfo.reconfiguration.config.ApplicationConfig;
import org.apache.commons.io.FilenameUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpHeaders;
import org.springframework.stereotype.Component;
import org.springframework.util.MimeType;
import org.springframework.util.MimeTypeUtils;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import java.io.File;
import java.io.IOException;
import java.util.Objects;
/**
* 文件上传工具类
*
* @author valarchie
*/
public class FileUploadUtils {
/**
* 默认大小 50M
*/
public static final long MAX_FILE_SIZE = 50 * Constants.MB;
/**
* 默认的文件名最大长度 127
*/
public static final int MAX_FILE_NAME_LENGTH = 127;
public static final String file_base_url = "D:\\gooinfo\\security\\upload\\picture";
/**
* 允许上传和下载的文件类型
*/
private static final String[] ALLOWED_EXTENSIONS = {
// 图片
"bmp", "gif", "jpg", "jpeg", "png",
// word excel powerpoint
"doc", "docx", "xls", "xlsx", "ppt", "pptx", "html", "htm", "txt",
// 压缩文件
"rar", "zip", "gz", "bz2",
// 视频格式
"mp4", "avi", "rmvb",
"pdf"};
private FileUploadUtils() {
}
/**
* 根据文件路径上传
*
* @param subDir 相对应用的基目录
* @param file 上传的文件
* @return 文件名称
*/
public static String upload(String subDir, MultipartFile file) {
try {
return upload(subDir, file, ALLOWED_EXTENSIONS);
} catch (Exception e) {
throw new ApiException(ErrorCode.Business.UPLOAD_FILE_FAILED, e.getMessage());
}
}
/**
* 文件上传
*
* @param subDir 相对应用的子目录
* @param file 上传的文件
* @param allowedExtension 上传文件类型
* @return 返回上传成功的文件名
* @throws IOException 比如读写文件出错时
*/
public static String upload(String subDir, MultipartFile file, String[] allowedExtension)
throws IOException {
isAllowedUpload(file, allowedExtension);
String fileName = generateFilename(file);
saveFileToLocal(file, subDir, fileName);
return getRelativeFileUrl(subDir, fileName);
}
/**
* 将文件保存到服务器
*
* @param file 文件
* @param subDir 子目录
* @param fileName 文件名
*/
static void saveFileToLocal(MultipartFile file, String subDir, String fileName) throws IOException {
if (StrUtil.isEmpty(subDir) || StrUtil.isEmpty(fileName)) {
throw new ApiException(ErrorCode.Internal.INVALID_PARAMETER, "subDir or fileName");
}
File destination = new File(getFileAbsolutePath(subDir, fileName));
if (!destination.exists()) {
System.out.println(destination);
if (!destination.getParentFile().exists()) {
destination.getParentFile().mkdirs();
}
}
file.transferTo(destination);
}
/**
* 获取文件的相对地址
*
* @param subDir avatar
* @param fileName test.jpg
* @return /profile/avatar/test.jpg
*/
static String getRelativeFileUrl(String subDir, String fileName) {
// 相对地址用于网络请求 所以不使用File.separate
return StrUtil.format("/{}/{}/{}", Constants.RESOURCE_PREFIX, subDir, fileName);
}
/**
* 检测文件是否可以上传
*
* @param file 文件
* @param allowedExtension 允许的文件类型列表
*/
static void isAllowedUpload(MultipartFile file, String[] allowedExtension) {
int fileNameLength = Objects.requireNonNull(file.getOriginalFilename()).length();
if (fileNameLength > MAX_FILE_NAME_LENGTH) {
throw new ApiException(ErrorCode.Business.UPLOAD_FILE_NAME_EXCEED_MAX_LENGTH, MAX_FILE_NAME_LENGTH);
}
long size = file.getSize();
if (size > MAX_FILE_SIZE) {
throw new ApiException(ErrorCode.Business.UPLOAD_FILE_SIZE_EXCEED_MAX_SIZE, MAX_FILE_SIZE / Constants.MB);
}
String extension = getFileExtension(file);
if (!isExtensionAllowed(extension, allowedExtension)) {
throw new ApiException(ErrorCode.Business.UPLOAD_FILE_TYPE_NOT_ALLOWED,
StrUtil.join(",", (Object[]) allowedExtension));
}
}
/**
* 检查文件是否可下载
*
* @param resource 需要下载的文件
* @return true 正常 false 非法
*/
public static boolean isAllowDownload(String resource) {
// 禁止目录上跳级别
return !StrUtil.contains(resource, "..") &&
// 检查允许下载的文件规则
StrUtil.containsAnyIgnoreCase(FileNameUtil.getSuffix(resource), ALLOWED_EXTENSIONS);
}
/**
* 判断MIME类型是否是允许的MIME类型
*/
static boolean isExtensionAllowed(String extension, String[] allowedExtension) {
if (allowedExtension == null || allowedExtension.length == 0) {
return true;
}
return StrUtil.containsAnyIgnoreCase(extension, allowedExtension);
}
/**
* 获取文件名的后缀
*
* @param file 表单文件
* @return 后缀名
*/
static String getFileExtension(MultipartFile file) {
String extension = FilenameUtils.getExtension(file.getOriginalFilename());
if (StrUtil.isEmpty(extension)) {
MimeType mimeType = MimeTypeUtils.parseMimeType(Objects.requireNonNull(file.getContentType()));
extension = mimeType.getSubtype();
}
return extension;
}
/**
* 编码文件名
*/
static String generateFilename(MultipartFile file) {
return StrUtil.format("{}_{}_{}.{}",
DateUtil.format(DateUtil.date(), DatePattern.PURE_DATETIME_PATTERN),
FilenameUtils.getBaseName(file.getOriginalFilename()),
IdUtil.simpleUUID(),
getFileExtension(file));
}
/**
* 下载文件名重新编码
*
* @param fileName 真实文件名
*/
public static HttpHeaders getDownloadHeader(String fileName) {
String randomFileName = System.currentTimeMillis() + "_" + fileName;
String fileNameUrlEncoded = URLUtil.encode(randomFileName, CharsetUtil.CHARSET_UTF_8);
HttpHeaders headers = new HttpHeaders();
headers.set("Content-Disposition", String.format("attachment;filename=%s", fileNameUrlEncoded));
return headers;
}
public static String getFileAbsolutePath(String subDir, String fileName) {
// return AgileBootConfig.getFileBaseDir() + File.separator + subDir + File.separator + fileName;
return file_base_url+File.separator + fileName;
}
}
标签:java,String,fileName,return,头像,subDir,file,import,上传 From: https://www.cnblogs.com/wawaFish666/p/17958627