巡查任务

linwei

2026-01-19 077cedddebbfae2d740e259e9e16b0eb363f3a02

 drone-ops/drone-resource/src/main/java/org/sxkj/resource/endpoint/OssEndpoint.java

@@ -1,22 +1,7 @@
/*
 *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in the
 *  documentation and/or other materials provided with the distribution.
 *  Neither the name of the dreamlu.net developer nor the names of its
 *  contributors may be used to endorse or promote products derived from
 *  this software without specific prior written permission.
 *  Author: Chill 庄骞 (smallchill@163.com)
 */
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;
@@ -28,12 +13,13 @@
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.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.sxkj.resource.util.ImageCompressorUtil;
import springfox.documentation.annotations.ApiIgnore;

import java.io.IOException;
import java.util.HashMap;
@@ -48,7 +34,7 @@
@RestController
@AllArgsConstructor
@RequestMapping("/oss/endpoint")
@Api(value = "对象存储端点", tags = "对象存储端点")
@Api(value = "对象存储端点", tags = "文件上传接口")
public class OssEndpoint {

   /**
@@ -70,6 +56,7 @@
   @SneakyThrows
   @PostMapping("/make-bucket")
   @PreAuth(RoleConstant.HAS_ROLE_ADMIN)
    @ApiIgnore
   public R makeBucket(@RequestParam String bucketName) {
      ossBuilder.template().makeBucket(bucketName);
      return R.success("创建成功");
@@ -84,6 +71,7 @@
   @SneakyThrows
   @PostMapping("/remove-bucket")
   @PreAuth(RoleConstant.HAS_ROLE_ADMIN)
    @ApiIgnore
   public R removeBucket(@RequestParam String bucketName) {
      ossBuilder.template().removeBucket(bucketName);
      return R.success("删除成功");
@@ -99,6 +87,7 @@
    */
   @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("操作成功");
@@ -112,6 +101,7 @@
    */
   @SneakyThrows
   @GetMapping("/stat-file")
    @ApiOperation(value = "获取文件信息", notes = "获取文件信息")
   public R<OssFile> statFile(@RequestParam String fileName) {
      return R.data(ossBuilder.template().statFile(fileName));
   }
@@ -124,6 +114,7 @@
    */
   @SneakyThrows
   @GetMapping("/file-path")
    @ApiOperation(value = "获取文件相对路径", notes = "获取文件相对路径")
   public R<String> filePath(@RequestParam String fileName) {
      return R.data(ossBuilder.template().filePath(fileName));
   }
@@ -137,6 +128,7 @@
    */
   @SneakyThrows
   @GetMapping("/file-link")
    @ApiOperation(value = "获取文件外链", notes = "获取文件外链")
   public R<String> fileLink(@RequestParam String fileName) {
      return R.data(ossBuilder.template().fileLink(fileName));
   }
@@ -149,6 +141,7 @@
    */
   @SneakyThrows
   @PostMapping("/put-file")
    @ApiOperation(value = "上传文件", notes = "上传文件")
   public R<BladeFile> putFile(@RequestParam MultipartFile file) {
      BladeFile bladeFile = ossBuilder.template().putFile(file.getOriginalFilename(), file.getInputStream());
      return R.data(bladeFile);
@@ -163,6 +156,7 @@
    */
   @SneakyThrows
   @PostMapping("/put-file-by-name")
    @ApiOperation(value = "上传文件", notes = "上传文件-存储桶对象名称")
   public R<BladeFile> putFile(@RequestParam String fileName, @RequestParam MultipartFile file) {
      BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
      return R.data(bladeFile);
@@ -176,6 +170,7 @@
    */
   @SneakyThrows
   @PostMapping("/put-file-attach")
    @ApiOperation(value = "上传文件并保存至附件表", notes = "上传文件并保存至附件表")
   public R<BladeFile> putFileAttach(@RequestParam MultipartFile file) {
      BladeFile bladeFile = getBladeFile(file);
      return R.data(bladeFile);
@@ -185,14 +180,18 @@
    * 上传文件并保存至附件表
    *
    * @param fileName 存储桶对象名称
    * @param file     文件
    * @return ObjectStat
     * @param resultType 结果类型
     * @param file       上传文件
     * @return BladeFile
    */
   @SneakyThrows
   @PostMapping("/put-file-attach-by-name")
   public R<BladeFile> putFileAttach(@RequestParam String fileName, @RequestParam MultipartFile file) {
    @ApiOperation(value = "上传文件并保存至附件表", notes = "上传文件并保存至附件表-存储桶对象名称")
    public R<BladeFile> putFileAttach(@RequestParam String fileName,
                                      @RequestParam Integer resultType,
                                      @RequestParam MultipartFile file) {
      BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
      Long attachId = buildAttach(fileName, file.getSize(), bladeFile);
        Long attachId = buildAttach(fileName, resultType, file.getSize(), bladeFile);
      bladeFile.setAttachId(attachId);
      return R.data(bladeFile);
   }
@@ -206,6 +205,7 @@
    */
   @SneakyThrows
   @PostMapping("/put-file-cop-attach")
    @ApiOperation(value = "上传原图和压缩图并保存至附件表", notes = "上传原图和压缩图并保存至附件表")
   public R putFileCopAttach(@RequestParam MultipartFile file) {
      // 上传之前先压缩
      MultipartFile compressImage = ImageCompressorUtil.compressImage(file,null);
@@ -224,7 +224,7 @@
   private BladeFile getBladeFile(@RequestParam MultipartFile file) throws IOException {
      String fileName = file.getOriginalFilename();
      BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
      Long attachId = buildAttach(fileName, file.getSize(), bladeFile);
        Long attachId = buildAttach(fileName, null, file.getSize(), bladeFile);
      bladeFile.setAttachId(attachId);
      return bladeFile;
   }
@@ -237,12 +237,13 @@
    * @param bladeFile 对象存储文件
    * @return attachId
    */
   private Long buildAttach(String fileName, Long fileSize, BladeFile bladeFile) {
    private Long buildAttach(String fileName, Integer 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);
@@ -259,6 +260,7 @@
   @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("操作成功");
@@ -273,6 +275,7 @@
   @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("操作成功");