458 lines
19 KiB
Java
458 lines
19 KiB
Java
package com.gxwebsoft.common.system.controller;
|
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
import com.alibaba.fastjson.JSONObject;
|
|
import com.aliyun.oss.OSS;
|
|
import com.aliyun.oss.OSSClientBuilder;
|
|
import com.aliyun.oss.common.auth.CredentialsProvider;
|
|
import com.aliyun.oss.common.auth.DefaultCredentialProvider;
|
|
import com.gxwebsoft.common.core.annotation.OperationLog;
|
|
import com.gxwebsoft.common.core.config.ConfigProperties;
|
|
import com.gxwebsoft.common.core.utils.FileServerUtil;
|
|
import com.gxwebsoft.common.core.utils.RedisUtil;
|
|
import com.gxwebsoft.common.core.web.ApiResult;
|
|
import com.gxwebsoft.common.core.web.BaseController;
|
|
import com.gxwebsoft.common.core.web.PageResult;
|
|
import com.gxwebsoft.common.system.entity.Company;
|
|
import com.gxwebsoft.common.system.entity.FileRecord;
|
|
import com.gxwebsoft.common.system.param.FileRecordParam;
|
|
import com.gxwebsoft.common.system.service.CompanyService;
|
|
import com.gxwebsoft.common.system.service.FileRecordService;
|
|
import com.gxwebsoft.common.system.service.SettingService;
|
|
import io.swagger.annotations.Api;
|
|
import io.swagger.annotations.ApiImplicitParam;
|
|
import io.swagger.annotations.ApiImplicitParams;
|
|
import io.swagger.annotations.ApiOperation;
|
|
import org.springframework.security.access.prepost.PreAuthorize;
|
|
import org.springframework.web.bind.annotation.*;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import javax.annotation.Resource;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import javax.servlet.http.HttpServletResponse;
|
|
import javax.imageio.ImageIO;
|
|
import java.awt.Color;
|
|
import java.awt.Graphics2D;
|
|
import java.awt.RenderingHints;
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.ByteArrayInputStream;
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.File;
|
|
import java.util.ArrayList;
|
|
import java.util.Arrays;
|
|
import java.util.Base64;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
|
|
/**
|
|
* 文件上传下载控制器
|
|
*
|
|
* @author WebSoft
|
|
* @since 2018-12-24 16:10:24
|
|
*/
|
|
@Api(tags = "文件上传下载")
|
|
@RestController
|
|
@RequestMapping("/api/file")
|
|
public class FileController extends BaseController {
|
|
private static final int FACE_MIN_WIDTH = 480;
|
|
private static final int FACE_MIN_HEIGHT = 640;
|
|
private static final int FACE_MAX_WIDTH = 1080;
|
|
private static final int FACE_MAX_HEIGHT = 1920;
|
|
|
|
@Resource
|
|
private ConfigProperties config;
|
|
@Resource
|
|
private RedisUtil redisUtil;
|
|
@Resource
|
|
private FileRecordService fileRecordService;
|
|
@Resource
|
|
private CompanyService companyService;
|
|
@Resource
|
|
private SettingService settingService;
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:upload')")
|
|
@OperationLog
|
|
@ApiOperation("上传文件")
|
|
@PostMapping("/upload")
|
|
public ApiResult<FileRecord> upload(@RequestParam MultipartFile file, HttpServletRequest request) {
|
|
FileRecord result = null;
|
|
try {
|
|
String dir = getUploadDir();
|
|
File upload = FileServerUtil.upload(file, dir, config.getUploadUuidName());
|
|
String path = upload.getAbsolutePath().replace("\\", "/").substring(dir.length() - 1);
|
|
// String requestURL = StrUtil.removeSuffix(request.getRequestURL(), "/upload");
|
|
String requestURL = config.getFileServer() + "/api/file";
|
|
String originalName = file.getOriginalFilename();
|
|
result = new FileRecord();
|
|
result.setCreateUserId(getLoginUserId());
|
|
result.setName(StrUtil.isBlank(originalName) ? upload.getName() : originalName);
|
|
result.setLength(upload.length());
|
|
result.setPath(path);
|
|
result.setUrl(requestURL + path);
|
|
String contentType = FileServerUtil.getContentType(upload);
|
|
result.setContentType(contentType);
|
|
if (FileServerUtil.isImage(contentType)) {
|
|
result.setThumbnail(requestURL + "/thumbnail" + path);
|
|
}
|
|
result.setDownloadUrl(config.getFileServer() + "/download" + path);
|
|
// 云存储配置
|
|
// final String s = redisUtil.get("setting:upload:" + getTenantId());
|
|
//
|
|
// final JSONObject jsonObject = JSONObject.parseObject(s);
|
|
// final String uploadMethod = jsonObject.getString("uploadMethod");
|
|
// final String bucketDomain = jsonObject.getString("bucketDomain");
|
|
// if(!uploadMethod.equals("file")){
|
|
// path = bucketDomain + path;
|
|
// }
|
|
result.setUrl(path);
|
|
fileRecordService.save(result);
|
|
return success(result);
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return fail("上传失败", result).setError(e.toString());
|
|
}
|
|
}
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:upload')")
|
|
@OperationLog
|
|
@ApiOperation("上传base64文件")
|
|
@ApiImplicitParams({
|
|
@ApiImplicitParam(name = "base64", value = "base64", required = true, dataType = "string", dataTypeClass = String.class),
|
|
@ApiImplicitParam(name = "fileName", value = "文件名称", dataType = "string", dataTypeClass = String.class)
|
|
})
|
|
@PostMapping("/upload/base64")
|
|
public ApiResult<FileRecord> uploadBase64(String base64, String fileName, HttpServletRequest request) {
|
|
FileRecord result = null;
|
|
try {
|
|
String dir = getUploadDir();
|
|
File upload = FileServerUtil.upload(base64, fileName, getUploadDir());
|
|
String path = upload.getAbsolutePath().substring(dir.length()).replace("\\", "/");
|
|
String requestURL = StrUtil.removeSuffix(request.getRequestURL(), "/upload/base64");
|
|
result = new FileRecord();
|
|
result.setCreateUserId(getLoginUserId());
|
|
result.setName(StrUtil.isBlank(fileName) ? upload.getName() : fileName);
|
|
result.setLength(upload.length());
|
|
result.setPath(path);
|
|
result.setUrl(requestURL + path);
|
|
result.setThumbnail(FileServerUtil.isImage(upload) ? (requestURL + "/thumbnail" + path) : null);
|
|
fileRecordService.save(result);
|
|
return success(result);
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return fail("上传失败", result).setError(e.toString());
|
|
}
|
|
}
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:upload')")
|
|
@OperationLog
|
|
@ApiOperation("上传图片")
|
|
@PostMapping("/image")
|
|
public HashMap<String, Object> image(@RequestParam MultipartFile file, HttpServletRequest request) {
|
|
FileRecord result = null;
|
|
try {
|
|
String dir = getUploadDir();
|
|
File upload = FileServerUtil.upload(file, dir, config.getUploadUuidName());
|
|
String path = upload.getAbsolutePath().replace("\\", "/").substring(dir.length() - 1);
|
|
// System.out.println("request.getRequestURL() = " + request.getRequestURL());
|
|
// String requestURL = StrUtil.removeSuffix(request.getRequestURL(), "/image");
|
|
String requestURL = config.getFileServer() + "/api/file";
|
|
// System.out.println("requestURL = " + requestURL);
|
|
// config.getServerUrl()
|
|
String originalName = file.getOriginalFilename();
|
|
result = new FileRecord();
|
|
result.setCreateUserId(getLoginUserId());
|
|
result.setName(StrUtil.isBlank(originalName) ? upload.getName() : originalName);
|
|
result.setLength(upload.length());
|
|
result.setPath(path);
|
|
result.setUrl(path);
|
|
String contentType = FileServerUtil.getContentType(upload);
|
|
result.setContentType(contentType);
|
|
if (FileServerUtil.isImage(contentType)) {
|
|
result.setThumbnail(requestURL + "/thumbnail" + path);
|
|
}
|
|
result.setDownloadUrl(requestURL + "/download" + path);
|
|
final HashMap<String, Object> map = new HashMap<>();
|
|
map.put("name",result.getName());
|
|
map.put("status","done");
|
|
map.put("thumbUrl",result.getThumbnail());
|
|
map.put("downloadUrl",result.getDownloadUrl());
|
|
map.put("url",result.getUrl());
|
|
fileRecordService.save(result);
|
|
return map;
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return null;
|
|
}
|
|
}
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:upload')")
|
|
@OperationLog
|
|
@ApiOperation("上传人脸图片并返回base64")
|
|
@PostMapping("/upload/face")
|
|
public ApiResult<String> uploadFace(@RequestParam MultipartFile file) {
|
|
try {
|
|
if (file == null || file.isEmpty()) {
|
|
return fail("上传失败，文件为空", null);
|
|
}
|
|
byte[] originalBytes = file.getBytes();
|
|
BufferedImage image;
|
|
try (ByteArrayInputStream inputStream = new ByteArrayInputStream(originalBytes)) {
|
|
image = ImageIO.read(inputStream);
|
|
}
|
|
if (image == null) {
|
|
return fail("上传失败，无法解析图片", null);
|
|
}
|
|
int width = image.getWidth();
|
|
int height = image.getHeight();
|
|
boolean inRange = width >= FACE_MIN_WIDTH && height >= FACE_MIN_HEIGHT
|
|
&& width <= FACE_MAX_WIDTH && height <= FACE_MAX_HEIGHT;
|
|
boolean isJpeg = isJpegImage(file.getContentType(), file.getOriginalFilename());
|
|
if (inRange && isJpeg) {
|
|
return success(Base64.getEncoder().encodeToString(originalBytes));
|
|
}
|
|
BufferedImage converted = resizeToRange(toRgbImage(image), width, height);
|
|
byte[] jpegBytes = toJpegBytes(converted);
|
|
return success(Base64.getEncoder().encodeToString(jpegBytes));
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return fail("上传失败", null).setError(e.toString());
|
|
}
|
|
}
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:list')")
|
|
@OperationLog
|
|
@ApiOperation("根据id查询文件")
|
|
@GetMapping("/{id}")
|
|
public ApiResult<FileRecord> get(@PathVariable("id") Integer id) {
|
|
return success(fileRecordService.getByIdRel(id));
|
|
}
|
|
|
|
@ApiOperation("查看原文件")
|
|
@GetMapping("/{dir}/{name:.+}")
|
|
public void preview(@PathVariable("dir") String dir, @PathVariable("name") String name,
|
|
HttpServletResponse response, HttpServletRequest request) {
|
|
File file = new File(getUploadDir(), dir + "/" + name);
|
|
FileServerUtil.preview(file, getPdfOutDir(), config.getOpenOfficeHome(), response, request);
|
|
}
|
|
|
|
@ApiOperation("下载原文件")
|
|
@GetMapping("/download/{dir}/{name:.+}")
|
|
public void download(@PathVariable("dir") String dir, @PathVariable("name") String name,
|
|
HttpServletResponse response, HttpServletRequest request) {
|
|
String path = dir + "/" + name;
|
|
FileRecord record = fileRecordService.getByIdPath(path);
|
|
File file = new File(getUploadDir(), path);
|
|
String fileName = record == null ? file.getName() : record.getName();
|
|
FileServerUtil.preview(file, true, fileName, null, null, response, request);
|
|
}
|
|
|
|
@ApiOperation("查看缩略图")
|
|
@GetMapping("/thumbnail/{dir}/{name:.+}")
|
|
public void thumbnail(@PathVariable("dir") String dir, @PathVariable("name") String name,
|
|
HttpServletResponse response, HttpServletRequest request) {
|
|
File file = new File(getUploadDir(), dir + "/" + name);
|
|
File thumbnail = new File(getUploadSmDir(), dir + "/" + name);
|
|
FileServerUtil.previewThumbnail(file, thumbnail, config.getThumbnailSize(), response, request);
|
|
}
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:remove')")
|
|
@OperationLog
|
|
@ApiOperation("删除文件")
|
|
@DeleteMapping("/remove/{id}")
|
|
public ApiResult<?> remove(@PathVariable("id") Integer id) {
|
|
FileRecord record = fileRecordService.getById(id);
|
|
List<FileRecord> fileRecords = new ArrayList<>();
|
|
fileRecords.add(record);
|
|
if (fileRecordService.removeById(id)) {
|
|
if (StrUtil.isNotBlank(record.getPath())) {
|
|
// 删除文件
|
|
fileRecordService.deleteFileAsync(Arrays.asList(
|
|
new File(getUploadDir(), record.getPath()),
|
|
new File(getUploadSmDir(), record.getPath())
|
|
));
|
|
}
|
|
fileRecordService.deleteOssFileAsync(fileRecords);
|
|
return success("删除成功");
|
|
}
|
|
return fail("删除失败");
|
|
}
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:remove')")
|
|
@OperationLog
|
|
@ApiOperation("批量删除文件")
|
|
@ApiImplicitParams({
|
|
@ApiImplicitParam(name = "ids", value = "id数组", required = true, dataType = "string", dataTypeClass = String.class)
|
|
})
|
|
@DeleteMapping("/remove/batch")
|
|
public ApiResult<?> deleteBatch(@RequestBody List<Integer> ids) {
|
|
List<FileRecord> fileRecords = fileRecordService.listByIds(ids);
|
|
if (fileRecordService.removeByIds(ids)) {
|
|
List<File> files = new ArrayList<>();
|
|
for (FileRecord record : fileRecords) {
|
|
if (StrUtil.isNotBlank(record.getPath())) {
|
|
files.add(new File(getUploadDir(), record.getPath()));
|
|
files.add(new File(getUploadSmDir(), record.getPath()));
|
|
}
|
|
}
|
|
fileRecordService.deleteFileAsync(files);
|
|
fileRecordService.deleteOssFileAsync(fileRecords);
|
|
return success("删除成功");
|
|
}
|
|
return fail("删除失败");
|
|
}
|
|
|
|
@ApiOperation("分页查询文件")
|
|
@GetMapping("/page")
|
|
public ApiResult<PageResult<FileRecord>> page(FileRecordParam param, HttpServletRequest request) {
|
|
PageResult<FileRecord> result = fileRecordService.pageRel(param);
|
|
// String requestURL = StrUtil.removeSuffix(request.getRequestURL(), "/page");
|
|
String requestURL = config.getFileServer();
|
|
for (FileRecord record : result.getList()) {
|
|
if (StrUtil.isNotBlank(record.getPath())) {
|
|
record.setDownloadUrl(record.getPath());
|
|
if (FileServerUtil.isImage(record.getContentType())) {
|
|
record.setThumbnail(record.getPath() + "?x-oss-process=image/resize,m_fixed,w_100,h_100/quality,Q_90");
|
|
record.setPath(record.getPath() + "?x-oss-process=image/resize,m_fixed,w_750/quality,Q_90");
|
|
record.setUrl(record.getDownloadUrl() + "?x-oss-process=image/resize,m_fixed,w_1680/quality,Q_90");
|
|
}
|
|
|
|
}
|
|
}
|
|
return success(result);
|
|
}
|
|
|
|
@ApiOperation("查询全部文件")
|
|
@GetMapping("/list")
|
|
public ApiResult<List<FileRecord>> list(FileRecordParam param, HttpServletRequest request) {
|
|
List<FileRecord> records = fileRecordService.listRel(param);
|
|
// String requestURL = StrUtil.removeSuffix(request.getRequestURL(), "/list");
|
|
String requestURL = config.getFileServer();
|
|
for (FileRecord record : records) {
|
|
if (StrUtil.isNotBlank(record.getPath())) {
|
|
record.setDownloadUrl(record.getPath());
|
|
if (FileServerUtil.isImage(record.getContentType())) {
|
|
record.setThumbnail(record.getPath() + "?x-oss-process=image/resize,m_fixed,w_100,h_100/quality,Q_90");
|
|
record.setPath(record.getPath() + "?x-oss-process=image/resize,m_fixed,w_750/quality,Q_90");
|
|
record.setUrl(record.getDownloadUrl() + "?x-oss-process=image/resize,m_fixed,w_1680/quality,Q_90");
|
|
}
|
|
}
|
|
}
|
|
return success(records);
|
|
}
|
|
|
|
/**
|
|
* 文件上传基目录
|
|
*/
|
|
private String getUploadBaseDir() {
|
|
return config.getUploadPath() + "file/";
|
|
}
|
|
|
|
/**
|
|
* 文件上传位置(服务器)
|
|
*/
|
|
private String getUploadDir() {
|
|
return config.getUploadPath() + "file/";
|
|
}
|
|
|
|
/**
|
|
* 文件上传位置(本地)
|
|
*/
|
|
// private String getUploadDir() {
|
|
// return "/Users/gxwebsoft/Documents/uploads/";
|
|
// }
|
|
|
|
/**
|
|
* 缩略图生成位置
|
|
*/
|
|
private String getUploadSmDir() {
|
|
return getUploadBaseDir() + "thumbnail/";
|
|
}
|
|
|
|
/**
|
|
* office转pdf输出位置
|
|
*/
|
|
private String getPdfOutDir() {
|
|
return getUploadBaseDir() + "pdf/";
|
|
}
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:upload')")
|
|
@OperationLog
|
|
@ApiOperation("添加文件")
|
|
@PostMapping()
|
|
public ApiResult<?> save(@RequestBody FileRecord fileRecord) {
|
|
if (fileRecordService.save(fileRecord)) {
|
|
return success("上传成功");
|
|
}
|
|
return fail("上传失败");
|
|
}
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:update')")
|
|
@OperationLog
|
|
@ApiOperation("修改文件")
|
|
@PutMapping()
|
|
public ApiResult<?> update(@RequestBody FileRecord fileRecord) {
|
|
if (fileRecordService.updateById(fileRecord)) {
|
|
return success("修改成功");
|
|
}
|
|
return fail("修改失败");
|
|
}
|
|
|
|
private static boolean isJpegImage(String contentType, String fileName) {
|
|
if (contentType != null) {
|
|
String ct = contentType.toLowerCase();
|
|
if (ct.contains("jpeg") || ct.contains("jpg")) {
|
|
return true;
|
|
}
|
|
}
|
|
if (fileName != null) {
|
|
String lower = fileName.toLowerCase();
|
|
return lower.endsWith(".jpg") || lower.endsWith(".jpeg");
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private static BufferedImage toRgbImage(BufferedImage source) {
|
|
if (source.getType() == BufferedImage.TYPE_INT_RGB) {
|
|
return source;
|
|
}
|
|
BufferedImage rgb = new BufferedImage(source.getWidth(), source.getHeight(), BufferedImage.TYPE_INT_RGB);
|
|
Graphics2D g2d = rgb.createGraphics();
|
|
g2d.setColor(Color.WHITE);
|
|
g2d.fillRect(0, 0, rgb.getWidth(), rgb.getHeight());
|
|
g2d.drawImage(source, 0, 0, null);
|
|
g2d.dispose();
|
|
return rgb;
|
|
}
|
|
|
|
private static BufferedImage resizeToRange(BufferedImage image, int width, int height) {
|
|
double scale = 1.0;
|
|
if (width < FACE_MIN_WIDTH || height < FACE_MIN_HEIGHT) {
|
|
scale = Math.max(FACE_MIN_WIDTH / (double) width, FACE_MIN_HEIGHT / (double) height);
|
|
} else if (width > FACE_MAX_WIDTH || height > FACE_MAX_HEIGHT) {
|
|
scale = Math.min(FACE_MAX_WIDTH / (double) width, FACE_MAX_HEIGHT / (double) height);
|
|
} else {
|
|
return image;
|
|
}
|
|
int targetWidth = Math.max(1, (int) Math.round(width * scale));
|
|
int targetHeight = Math.max(1, (int) Math.round(height * scale));
|
|
BufferedImage resized = new BufferedImage(targetWidth, targetHeight, BufferedImage.TYPE_INT_RGB);
|
|
Graphics2D g2d = resized.createGraphics();
|
|
g2d.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BICUBIC);
|
|
g2d.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_QUALITY);
|
|
g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
|
|
g2d.drawImage(image, 0, 0, targetWidth, targetHeight, null);
|
|
g2d.dispose();
|
|
return resized;
|
|
}
|
|
|
|
private static byte[] toJpegBytes(BufferedImage image) throws Exception {
|
|
try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
|
|
if (!ImageIO.write(image, "jpg", outputStream)) {
|
|
throw new IllegalStateException("jpeg encode fail");
|
|
}
|
|
return outputStream.toByteArray();
|
|
}
|
|
}
|
|
|
|
}
|