|
|
package com.yeqifu.sys.controller;
|
|
|
|
|
|
import cn.hutool.core.date.DateUtil;
|
|
|
import com.yeqifu.sys.common.AppFileUtils;
|
|
|
import org.springframework.http.MediaType;
|
|
|
import org.springframework.http.ResponseEntity;
|
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
import org.springframework.http.HttpStatus;
|
|
|
|
|
|
import java.io.File;
|
|
|
import java.io.IOException;
|
|
|
import java.nio.file.Files;
|
|
|
import java.nio.file.Path;
|
|
|
import java.nio.file.Paths;
|
|
|
import java.util.Date;
|
|
|
import java.util.HashMap;
|
|
|
import java.util.Map;
|
|
|
|
|
|
/**
|
|
|
* @Author: 落亦-
|
|
|
* @Date: 2019/12/15 23:46
|
|
|
*/
|
|
|
@RestController
|
|
|
@RequestMapping("file")
|
|
|
public class FileController {
|
|
|
|
|
|
/**
|
|
|
* 图片上传
|
|
|
* @param mf
|
|
|
* @return
|
|
|
*/
|
|
|
@RequestMapping("uploadImage")
|
|
|
public Map<String, Object> uploadImage(MultipartFile mf) {
|
|
|
// 1. 得到文件名
|
|
|
String oldName = mf.getOriginalFilename();
|
|
|
// 2. 根据旧的文件名生成新的文件名
|
|
|
String newName = AppFileUtils.createNewFileName(oldName);
|
|
|
// 3. 得到当前日期的字符串
|
|
|
String dirName = DateUtil.format(new Date(), "yyyy-MM-dd");
|
|
|
// 4. 构造文件夹
|
|
|
File dirFile = new File(AppFileUtils.UPLOAD_PATH, dirName);
|
|
|
// 5. 判断当前文件夹是否存在
|
|
|
if (!dirFile.exists()) {
|
|
|
// 如果不存在则创建新文件夹
|
|
|
dirFile.mkdirs();
|
|
|
}
|
|
|
// 6. 构造文件对象
|
|
|
File file = new File(dirFile, newName + "_temp");
|
|
|
// 7. 把 mf 里面的图片信息写入 file
|
|
|
try {
|
|
|
mf.transferTo(file);
|
|
|
} catch (IllegalStateException | IOException e) {
|
|
|
e.printStackTrace();
|
|
|
}
|
|
|
Map<String, Object> map = new HashMap<>();
|
|
|
map.put("path", dirName + "/" + newName + "_temp");
|
|
|
return map;
|
|
|
}
|
|
|
|
|
|
|
|
|
/**
|
|
|
* 图片下载
|
|
|
* @param path 图片路径
|
|
|
* return
|
|
|
*/
|
|
|
@RequestMapping("showImageByPath")
|
|
|
public ResponseEntity<byte[]> downloadImage(String path) {
|
|
|
// 规范路径，防止路径穿越
|
|
|
Path normalizedPath = Paths.get(AppFileUtils.UPLOAD_PATH, path).normalize();
|
|
|
|
|
|
// 防止路径越界，确保文件在允许的上传目录下
|
|
|
if (!normalizedPath.startsWith(AppFileUtils.UPLOAD_PATH)) {
|
|
|
return ResponseEntity.status(HttpStatus.FORBIDDEN).build(); // 如果路径越界，返回403 Forbidden
|
|
|
}
|
|
|
|
|
|
// 检查文件是否存在
|
|
|
File file = normalizedPath.toFile();
|
|
|
if (!file.exists() || !file.isFile()) {
|
|
|
return ResponseEntity.status(HttpStatus.NOT_FOUND).build(); // 文件不存在，返回404 Not Found
|
|
|
}
|
|
|
|
|
|
try {
|
|
|
// 读取文件内容
|
|
|
byte[] fileContent = Files.readAllBytes(normalizedPath);
|
|
|
|
|
|
// 根据文件扩展名推测文件类型并返回适当的 Content-Type
|
|
|
String contentType = Files.probeContentType(normalizedPath);
|
|
|
if (contentType == null) {
|
|
|
contentType = "application/octet-stream"; // 默认二进制流类型
|
|
|
}
|
|
|
|
|
|
// 返回文件内容，设置响应头和状态码
|
|
|
return ResponseEntity.ok()
|
|
|
.contentType(MediaType.valueOf(contentType))
|
|
|
.body(fileContent);
|
|
|
|
|
|
} catch (IOException e) {
|
|
|
// 异常捕获并返回500服务器错误
|
|
|
e.printStackTrace();
|
|
|
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build(); // 服务器错误
|
|
|
}
|
|
|
}
|
|
|
|
|
|
}
|
