package org.sxkj.resource.endpoint;
|
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.AllArgsConstructor;
|
import lombok.SneakyThrows;
|
import org.jetbrains.annotations.NotNull;
|
import org.springblade.core.oss.model.BladeFile;
|
import org.springblade.core.oss.model.OssFile;
|
import org.springblade.core.secure.annotation.PreAuth;
|
import org.springblade.core.tenant.annotation.NonDS;
|
import org.springblade.core.tool.api.R;
|
import org.springblade.core.tool.constant.RoleConstant;
|
import org.springblade.core.tool.utils.FileUtil;
|
import org.springblade.core.tool.utils.Func;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.sxkj.resource.builder.OssBuilder;
|
import org.sxkj.resource.entity.Attach;
|
import org.sxkj.resource.service.IAttachService;
|
import org.sxkj.resource.util.ImageCompressorUtil;
|
import springfox.documentation.annotations.ApiIgnore;
|
|
import java.io.ByteArrayInputStream;
|
import java.io.IOException;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* 对象存储端点
|
*
|
* @author Chill
|
*/
|
@NonDS
|
@RestController
|
@AllArgsConstructor
|
@RequestMapping("/oss/endpoint")
|
@Api(value = "对象存储端点", tags = "文件上传接口")
|
public class OssEndpoint {
|
|
/**
|
* 对象存储构建类
|
*/
|
private final OssBuilder ossBuilder;
|
|
/**
|
* 附件表服务
|
*/
|
private final IAttachService attachService;
|
|
/**
|
* 创建存储桶
|
*
|
* @param bucketName 存储桶名称
|
* @return Bucket
|
*/
|
@SneakyThrows
|
@PostMapping("/make-bucket")
|
@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
|
@ApiIgnore
|
public R makeBucket(@RequestParam String bucketName) {
|
ossBuilder.template().makeBucket(bucketName);
|
return R.success("创建成功");
|
}
|
|
/**
|
* 创建存储桶
|
*
|
* @param bucketName 存储桶名称
|
* @return R
|
*/
|
@SneakyThrows
|
@PostMapping("/remove-bucket")
|
@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
|
@ApiIgnore
|
public R removeBucket(@RequestParam String bucketName) {
|
ossBuilder.template().removeBucket(bucketName);
|
return R.success("删除成功");
|
}
|
|
/**
|
* 拷贝文件
|
*
|
* @param fileName 存储桶对象名称
|
* @param destBucketName 目标存储桶名称
|
* @param destFileName 目标存储桶对象名称
|
* @return R
|
*/
|
@SneakyThrows
|
@PostMapping("/copy-file")
|
@ApiOperation(value = "拷贝文件", notes = "拷贝文件")
|
public R copyFile(@RequestParam String fileName, @RequestParam String destBucketName, String destFileName) {
|
ossBuilder.template().copyFile(fileName, destBucketName, destFileName);
|
return R.success("操作成功");
|
}
|
|
/**
|
* 获取文件信息
|
*
|
* @param fileName 存储桶对象名称
|
* @return InputStream
|
*/
|
@SneakyThrows
|
@GetMapping("/stat-file")
|
@ApiOperation(value = "获取文件信息", notes = "获取文件信息")
|
public R<OssFile> statFile(@RequestParam String fileName) {
|
return R.data(ossBuilder.template().statFile(fileName));
|
}
|
|
/**
|
* 获取文件相对路径
|
*
|
* @param fileName 存储桶对象名称
|
* @return String
|
*/
|
@SneakyThrows
|
@GetMapping("/file-path")
|
@ApiOperation(value = "获取文件相对路径", notes = "获取文件相对路径")
|
public R<String> filePath(@RequestParam String fileName) {
|
return R.data(ossBuilder.template().filePath(fileName));
|
}
|
|
|
/**
|
* 获取文件外链
|
*
|
* @param fileName 存储桶对象名称
|
* @return String
|
*/
|
@SneakyThrows
|
@GetMapping("/file-link")
|
@ApiOperation(value = "获取文件外链", notes = "获取文件外链")
|
public R<String> fileLink(@RequestParam String fileName) {
|
return R.data(ossBuilder.template().fileLink(fileName));
|
}
|
|
/**
|
* 上传文件
|
*
|
* @param file 文件
|
* @return ObjectStat
|
*/
|
@SneakyThrows
|
@PostMapping("/put-file")
|
@ApiOperation(value = "上传文件", notes = "上传文件")
|
public R<BladeFile> putFile(@RequestParam MultipartFile file) {
|
// 步骤1: 将文件转换为字节数组,避免流重置异常
|
byte[] bytes = file.getBytes();
|
// 步骤2: 使用字节数组输入流上传文件
|
BladeFile bladeFile = ossBuilder.template("estack").putFile(file.getOriginalFilename(), new ByteArrayInputStream(bytes));
|
return R.data(bladeFile);
|
}
|
|
/**
|
* 上传文件
|
*
|
* @param fileName 存储桶对象名称
|
* @param file 文件
|
* @return ObjectStat
|
*/
|
@SneakyThrows
|
@PostMapping("/put-file-by-name")
|
@ApiOperation(value = "上传文件", notes = "上传文件-存储桶对象名称")
|
public R<BladeFile> putFile(@RequestParam String fileName, @RequestParam MultipartFile file) {
|
// 步骤1: 将文件转换为字节数组,避免流重置异常
|
byte[] bytes = file.getBytes();
|
// 步骤2: 使用字节数组输入流上传文件
|
BladeFile bladeFile = ossBuilder.template().putFile(fileName, new ByteArrayInputStream(bytes));
|
return R.data(bladeFile);
|
}
|
|
/**
|
* 上传文件并保存至附件表
|
*
|
* @param file 文件
|
* @return ObjectStat
|
*/
|
@SneakyThrows
|
@PostMapping("/put-file-attach")
|
@ApiOperation(value = "上传文件并保存至附件表", notes = "上传文件并保存至附件表")
|
public R<BladeFile> putFileAttach(@RequestParam MultipartFile file) {
|
BladeFile bladeFile = getBladeFile(file);
|
return R.data(bladeFile);
|
}
|
|
/**
|
* 上传文件并保存至附件表
|
*
|
* @param fileName 存储桶对象名称
|
* @param resultType 结果类型
|
* @param file 上传文件
|
* @return BladeFile
|
*/
|
@SneakyThrows
|
@PostMapping("/put-file-attach-by-name")
|
@ApiOperation(value = "上传文件并保存至附件表", notes = "上传文件并保存至附件表-存储桶对象名称")
|
public R<BladeFile> putFileAttach(@RequestParam String fileName,
|
@RequestParam List<List<String>> resultType,
|
@RequestParam MultipartFile file) {
|
// 步骤1: 将文件转换为字节数组,避免流重置异常
|
byte[] bytes = file.getBytes();
|
// 步骤2: 使用字节数组输入流上传文件
|
BladeFile bladeFile = ossBuilder.template().putFile(fileName, new ByteArrayInputStream(bytes));
|
Long attachId = buildAttach(fileName, resultType, file.getSize(), bladeFile);
|
bladeFile.setAttachId(attachId);
|
return R.data(bladeFile);
|
}
|
|
|
/**
|
* 上传原图和压缩图并保存至附件表
|
*
|
* @param file 文件
|
* @return ObjectStat
|
*/
|
@SneakyThrows
|
@PostMapping("/put-file-cop-attach")
|
@ApiOperation(value = "上传原图和压缩图并保存至附件表", notes = "上传原图和压缩图并保存至附件表")
|
public R putFileCopAttach(@RequestParam MultipartFile file) {
|
// 上传之前先压缩
|
MultipartFile compressImage = ImageCompressorUtil.compressImage(file, null);
|
// 上传
|
BladeFile bladeFile = getBladeFile(file);
|
BladeFile compressFile = getBladeFile(compressImage);
|
//
|
Map<String, Object> map = new HashMap<>(2);
|
map.put("yt", bladeFile);
|
map.put("yst", compressFile);
|
// 返回
|
return R.data(bladeFile);
|
}
|
|
@NotNull
|
private BladeFile getBladeFile(@RequestParam MultipartFile file) throws IOException {
|
// 步骤1: 获取文件名
|
String fileName = file.getOriginalFilename();
|
// 步骤2: 将文件转换为字节数组,避免流重置异常
|
byte[] bytes = file.getBytes();
|
// 步骤3: 使用字节数组输入流上传文件
|
BladeFile bladeFile = ossBuilder.template("estack").putFile(fileName, new ByteArrayInputStream(bytes));
|
// 步骤4: 构建附件信息
|
Long attachId = buildAttach(fileName, null, file.getSize(), bladeFile);
|
bladeFile.setAttachId(attachId);
|
return bladeFile;
|
}
|
|
/**
|
* 构建附件表
|
*
|
* @param fileName 文件名
|
* @param fileSize 文件大小
|
* @param bladeFile 对象存储文件
|
* @return attachId
|
*/
|
private Long buildAttach(String fileName, List<List<String>> resultType, Long fileSize, BladeFile bladeFile) {
|
String fileExtension = FileUtil.getFileExtension(fileName);
|
Attach attach = new Attach();
|
attach.setDomainUrl(bladeFile.getDomain());
|
attach.setLink(bladeFile.getLink());
|
attach.setName(bladeFile.getName());
|
attach.setResultType(resultType);
|
attach.setOriginalName(bladeFile.getOriginalName());
|
attach.setAttachSize(fileSize);
|
attach.setExtension(fileExtension);
|
attachService.save(attach);
|
return attach.getId();
|
}
|
|
/**
|
* 删除文件
|
*
|
* @param fileName 存储桶对象名称
|
* @return R
|
*/
|
@SneakyThrows
|
@PostMapping("/remove-file")
|
@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
|
@ApiOperation(value = "删除文件", notes = "删除文件")
|
public R removeFile(@RequestParam String fileName) {
|
ossBuilder.template().removeFile(fileName);
|
return R.success("操作成功");
|
}
|
|
/**
|
* 批量删除文件
|
*
|
* @param fileNames 存储桶对象名称集合
|
* @return R
|
*/
|
@SneakyThrows
|
@PostMapping("/remove-files")
|
@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
|
@ApiOperation(value = "批量删除文件", notes = "批量删除文件")
|
public R removeFiles(@RequestParam String fileNames) {
|
ossBuilder.template().removeFiles(Func.toStrList(fileNames));
|
return R.success("操作成功");
|
}
|
|
}
|
