package com.yuanchu.limslaboratory.utils;
|
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.tomcat.util.http.fileupload.IOUtils;
|
import org.springframework.stereotype.Component;
|
|
import java.io.ByteArrayInputStream;
|
import java.io.FileOutputStream;
|
import java.io.IOException;
|
|
/**
|
* 保存文件工具
|
*/
|
@Slf4j
|
@Component
|
public class FileSaveUtil {
|
|
// 取yml中的路径 + /
|
// @Value("${file.path}")
|
private String FILE_PATH;
|
|
/**
|
* 存储文件主函数
|
* @param content 文件二进制流
|
* @param originalFilename 文件名称
|
* @return 返回文件名称用于存储数据库
|
*/
|
public String StoreFile(byte[] content, String originalFilename) {
|
// 生成随机名称：时间_随机6位数字
|
String FileName = System.currentTimeMillis() + "_" + MyUtil.getNumber(6);
|
String suffix = originalFilename.substring(originalFilename.lastIndexOf("."));
|
// 名称拼接
|
String fileName = FileName + suffix;
|
// 进行存储
|
storeFileWithFileName(content, fileName);
|
return fileName;
|
}
|
|
/**
|
* 存储文件函数
|
* @param content 文件二进制流
|
* @param fileName 文件名称
|
*/
|
private void storeFileWithFileName(byte[] content, String fileName) {
|
// 存储路径
|
String path = FILE_PATH + java.io.File.separatorChar;
|
// 目录不存在则创建
|
java.io.File file = new java.io.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);
|
}
|
}
|
}
|
