增加验证传参方式接口

jiangyong
1 parent e2c3879a
Showing 3 changed files with 143 additions and 1 deletions
assistant-dfs/src/main/java/com/diligrp/assistant/dfs/controller/FileObjectController.java
assistant-shared/src/main/java/com/diligrp/assistant/shared/domain/AmisMessage.java
assistant-shared/src/main/java/com/diligrp/assistant/shared/domain/AmisUploadResponse.java
@@ -8,6 +8,8 @@ import com.diligrp.assistant.dfs.domain.FileMetadata;
 import com.diligrp.assistant.dfs.exception.DfsServiceException;
 import com.diligrp.assistant.dfs.service.FileRepositoryService;
 import com.diligrp.assistant.shared.ErrorCode;
+import com.diligrp.assistant.shared.domain.AmisMessage;
+import com.diligrp.assistant.shared.domain.AmisUploadResponse;
 import com.diligrp.assistant.shared.domain.Message;
 import com.diligrp.assistant.shared.util.ObjectUtils;
 import jakarta.annotation.Resource;
@@ -55,6 +57,23 @@ public class FileObjectController {
         return Message.success(fileId);
     }
+    @RequestMapping(value = "/amis/file/upload.do")
+    public AmisMessage<AmisUploadResponse> amisFileUpload(@RequestPart("file") MultipartFile file,
+                                                          @RequestParam(Constants.HEADER_AUTHORIZATION) String authorization) throws IOException {
+        DfsAccessToken accessToken = accessAuthorization(authorization);
+
+        Optional<MediaType> optional = MediaTypeFactory.getMediaType(file.getOriginalFilename());
+        MimeType mimeType = optional.orElse(MediaType.APPLICATION_OCTET_STREAM);
+        FileMetadata metadata = new FileMetadata(mimeType.toString());
+
+        String fileName = ObjectUtils.isNotEmpty(file.getOriginalFilename()) ? file.getOriginalFilename() : "unknown";
+        DfsFile fileObject = new DfsFile(fileName, file.getInputStream(), metadata);
+        String fileId = fileRepositoryService.uploadFile(accessToken, fileObject);
+        AmisUploadResponse response = new AmisUploadResponse();
+        response.setUrl(fileId);
+        return AmisMessage.success(response);
+    }
+
     @RequestMapping(value = "/files/upload.do")
     public Message<?> fileUploads(@RequestPart(value = "files") MultipartFile[] files,
                                   @RequestHeader(Constants.HEADER_AUTHORIZATION) String authorization) throws IOException {
@@ -127,7 +146,26 @@ public class FileObjectController {
         }
         response.flushBuffer();
     }
-
+    @RequestMapping(value = "/amis/file/download.do")
+    public void amisDownload(@RequestParam("fileId") String fileId, @RequestParam("Dfs-Authorization") String authorization,
+                         HttpServletResponse response) throws IOException {
+        DfsAccessToken accessToken = accessAuthorization(authorization);
+        DfsFile file = fileRepositoryService.downloadFile(accessToken, fileId, null);
+        InputStream fileStream = file.getStream();
+        response.setContentType(file.getMetadata().getMimeType());
+        // oss sdk存在bug, fileStream.available()和真实字节数存在差异
+        // response.setContentLength(fileStream.available());
+        System.out.println(file.getName());
+        String fileName = URLEncoder.encode(file.getName(), StandardCharsets.UTF_8);
+        System.out.println(URLDecoder.decode(fileName, StandardCharsets.UTF_8));
+        response.addHeader("Content-Disposition", "attachment; filename*=UTF-8''" + fileName);
+        try {
+            IOUtils.copy(fileStream, response.getOutputStream());
+        } finally {
+            IOUtils.closeQuietly(fileStream);
+        }
+        response.flushBuffer();
+    }
     @RequestMapping(value = "/file/delete.do")
     public Message<?> deleteFile(@RequestParam("fileId") String fileId, @RequestHeader(Constants.HEADER_AUTHORIZATION) String authorization) {
         DfsAccessToken accessToken = accessAuthorization(authorization);
+package com.diligrp.assistant.shared.domain;
+
+import com.diligrp.assistant.shared.ErrorCode;
+
+/**
+ * @author jy
+ */
+public class AmisMessage<T> {
+    protected static final int CODE_SUCCESS = 0;
+    protected static final int CODE_FAILURE = ErrorCode.SYSTEM_UNKNOWN_ERROR;
+    protected static final String MSG_SUCCESS = "success";
+    private Integer status;
+    private String msg;
+    private T data;
+    public AmisMessage() {
+    }
+
+    public Integer getStatus() {
+        return status;
+    }
+
+    public void setStatus(Integer status) {
+        this.status = status;
+    }
+
+    public String getMsg() {
+        return msg;
+    }
+
+    public void setMsg(String msg) {
+        this.msg = msg;
+    }
+
+    public T getData() {
+        return this.data;
+    }
+
+    public void setData(T data) {
+        this.data = data;
+    }
+
+    public static AmisMessage<?> success() {
+        AmisMessage<?> result = new AmisMessage();
+        result.status = 0;
+        result.msg = "success";
+        return result;
+    }
+
+    public static <E> AmisMessage<E> success(E data) {
+        AmisMessage<E> result = new AmisMessage();
+        result.status = 0;
+        result.data = data;
+        result.msg = "success";
+        return result;
+    }
+
+    public static AmisMessage<?> failure(String message) {
+        AmisMessage<?> result = new AmisMessage();
+        result.status = 1000;
+        result.msg = message;
+        return result;
+    }
+
+    public static AmisMessage<?> failure(int code, String message) {
+        AmisMessage<?> result = new AmisMessage();
+        result.status = code;
+        result.msg = message;
+        return result;
+    }
+}
+package com.diligrp.assistant.shared.domain;
+
+/**
+ * @author jy
+ */
+public class AmisUploadResponse {
+    private String value;
+    private String filename;
+    private String url;
+
+    public String getValue() {
+        return value;
+    }
+
+    public void setValue(String value) {
+        this.value = value;
+    }
+
+    public String getFilename() {
+        return filename;
+    }
+
+    public void setFilename(String filename) {
+        this.filename = filename;
+    }
+
+    public String getUrl() {
+        return url;
+    }
+
+    public void setUrl(String url) {
+        this.url = url;
+    }
+}