package com.yuanchu.mom.utils;
|
|
import com.yuanchu.mom.exception.MyFileException;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.tomcat.util.http.fileupload.IOUtils;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.FileSystemUtils;
|
import org.springframework.web.multipart.MultipartFile;
|
import javax.annotation.PostConstruct;
|
import java.io.ByteArrayInputStream;
|
import java.io.File;
|
import java.io.FileOutputStream;
|
import java.io.IOException;
|
|
/**
|
* 保存文件工具
|
*/
|
@Slf4j
|
@Component
|
public class FileSaveUtil {
|
|
// 取yml中的路径 + /
|
private static String FILE_PATH;
|
|
private static String[] ALLOWED;
|
|
@Value("${file.path}")
|
private String file;
|
|
@Value("${file.allowed}")
|
private String[] allowed;
|
|
@PostConstruct
|
public void getFile() {
|
FILE_PATH = this.file;
|
}
|
|
@PostConstruct
|
public void getAllowed(){
|
ALLOWED = this.allowed;
|
}
|
/**
|
* 存储文件主函数
|
* @param file 文件二进制流
|
* @return 返回文件名称用于存储数据库
|
*/
|
public static String StoreFile(MultipartFile file) {
|
String originalFilename = file.getOriginalFilename();
|
// 生成随机名称：时间_随机6位数字
|
String FileName = System.currentTimeMillis() + "_" + MyUtil.getNumber(6);
|
String suffix = null;
|
if (originalFilename != null) {
|
suffix = originalFilename.substring(originalFilename.lastIndexOf("."));
|
// 如果后缀名不通过抛出异常
|
if (!isFileAllowed(suffix)){
|
throw new MyFileException(suffix);
|
}
|
}
|
// 名称拼接
|
String fileName = FileName + suffix;
|
// 进行存储
|
try {
|
storeFileWithFileName(file.getBytes(), fileName);
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
return fileName;
|
}
|
|
public static Boolean DeleteFile(String fileName) {
|
return FileSystemUtils.deleteRecursively(new File(FILE_PATH + "/" + fileName));
|
}
|
/**
|
* 存储文件函数
|
* @param content 文件二进制流
|
* @param fileName 文件名称
|
*/
|
private static void storeFileWithFileName(byte[] content, String fileName) {
|
// 存储路径
|
String path = FILE_PATH + File.separatorChar;
|
// 目录不存在则创建
|
File file = new File(path);
|
if (!file.exists()) {
|
file.mkdirs();
|
}
|
// 开始存储
|
try (FileOutputStream os = new FileOutputStream(path + fileName);
|
ByteArrayInputStream is = new ByteArrayInputStream(content)) {
|
IOUtils.copy(is, os);
|
} catch (IOException e) {
|
MyUtil.PrintLog("存储文件异常：" + e);
|
}
|
}
|
|
/**
|
* 判断文件是否被允许上传
|
*
|
* @param fileName 文件名
|
* @return 允许true, 否则false
|
*/
|
private static boolean isFileAllowed(String fileName) {
|
// 获取后缀名
|
String suffixName = fileName.substring(fileName.lastIndexOf(".") + 1).toLowerCase();
|
for (String allow : ALLOWED) {
|
if (allow.equals(suffixName)) {
|
return true;
|
}
|
}
|
return false;
|
}
|
}
|
