335 lines
14 KiB
Java
335 lines
14 KiB
Java
package com.gxwebsoft.common.system.controller;
|
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
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.web.ApiResult;
|
|
import com.gxwebsoft.common.core.web.BaseController;
|
|
import com.gxwebsoft.common.core.web.PageResult;
|
|
import com.gxwebsoft.common.system.entity.FileRecord;
|
|
import com.gxwebsoft.common.system.entity.Role;
|
|
import com.gxwebsoft.common.system.entity.User;
|
|
import com.gxwebsoft.common.system.param.FileRecordParam;
|
|
import com.gxwebsoft.common.system.service.FileRecordService;
|
|
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 java.io.File;
|
|
import java.util.ArrayList;
|
|
import java.util.Arrays;
|
|
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 {
|
|
@Resource
|
|
private ConfigProperties config;
|
|
@Resource
|
|
private FileRecordService fileRecordService;
|
|
|
|
@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);
|
|
result.setMerchantCode(getMerchantCode());
|
|
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);
|
|
result.setMerchantCode(getMerchantCode());
|
|
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(config.getFileServer() + path);
|
|
String contentType = FileServerUtil.getContentType(upload);
|
|
result.setContentType(contentType);
|
|
if (FileServerUtil.isImage(contentType)) {
|
|
result.setThumbnail(requestURL + "/thumbnail" + path);
|
|
}
|
|
result.setDownloadUrl(requestURL + "/download" + path);
|
|
result.setMerchantCode(getMerchantCode());
|
|
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: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);
|
|
if (fileRecordService.removeById(id)) {
|
|
if (StrUtil.isNotBlank(record.getPath())) {
|
|
fileRecordService.deleteFileAsync(Arrays.asList(
|
|
new File(getUploadDir(), record.getPath()),
|
|
new File(getUploadSmDir(), record.getPath())
|
|
));
|
|
}
|
|
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);
|
|
return success("删除成功");
|
|
}
|
|
return fail("删除失败");
|
|
}
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:list')")
|
|
@OperationLog
|
|
@ApiOperation("分页查询文件")
|
|
@GetMapping("/page")
|
|
public ApiResult<PageResult<FileRecord>> page(FileRecordParam param, HttpServletRequest request) {
|
|
// 搜索条件
|
|
if (getMerchantCode() != null) {
|
|
param.setMerchantCode(getMerchantCode());
|
|
}
|
|
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.setUrl(requestURL + record.getPath());
|
|
if (FileServerUtil.isImage(record.getContentType())) {
|
|
record.setThumbnail(requestURL + "/thumbnail" + record.getPath());
|
|
}
|
|
record.setDownloadUrl(requestURL + "/download" + record.getPath());
|
|
}
|
|
}
|
|
return success(result);
|
|
}
|
|
|
|
@PreAuthorize("hasAuthority('sys:file:list')")
|
|
@OperationLog
|
|
@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.setUrl(requestURL + record.getPath());
|
|
if (FileServerUtil.isImage(record.getContentType())) {
|
|
record.setThumbnail(requestURL + "/thumbnail" + record.getPath());
|
|
}
|
|
record.setDownloadUrl(requestURL + "/download" + record.getPath());
|
|
}
|
|
}
|
|
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("修改失败");
|
|
}
|
|
|
|
}
|