FileObjectController.java
6.94 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
package com.diligrp.assistant.dfs.controller;
import com.diligrp.assistant.dfs.Constants;
import com.diligrp.assistant.dfs.DfsProperties;
import com.diligrp.assistant.dfs.domain.DfsAccessToken;
import com.diligrp.assistant.dfs.domain.DfsFile;
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.Message;
import com.diligrp.assistant.shared.util.ObjectUtils;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletResponse;
import org.apache.commons.io.IOUtils;
import org.springframework.http.MediaType;
import org.springframework.http.MediaTypeFactory;
import org.springframework.util.MimeType;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLDecoder;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Base64;
import java.util.List;
import java.util.Optional;
@RestController
@RequestMapping(value = "/dfs")
public class FileObjectController {
@Resource
private DfsProperties dfsProperties;
@Resource
private FileRepositoryService fileRepositoryService;
@RequestMapping(value = "/file/upload.do")
public Message<?> fileUpload(@RequestPart("file") MultipartFile file,
@RequestHeader(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);
return Message.success(fileId);
}
@RequestMapping(value = "/files/upload.do")
public Message<?> fileUploads(@RequestPart(value = "files") MultipartFile[] files,
@RequestHeader(Constants.HEADER_AUTHORIZATION) String authorization) throws IOException {
DfsAccessToken accessToken = accessAuthorization(authorization);
List<String> fileIds = new ArrayList<>();
for (MultipartFile file : files) {
Optional<MediaType> optional = MediaTypeFactory.getMediaType(file.getOriginalFilename());
MimeType mimeType = optional.orElse(MediaType.APPLICATION_OCTET_STREAM);
FileMetadata metadata = new FileMetadata(mimeType.toString());
DfsFile fileObject = new DfsFile(file.getOriginalFilename(), file.getInputStream(), metadata);
String fileId = fileRepositoryService.uploadFile(accessToken, fileObject);
fileIds.add(fileId);
}
return Message.success(fileIds);
}
@RequestMapping(value = "/file/uploadByBase64.do")
public Message<?> uploadByBase64(@RequestParam("file") String file, @RequestParam(value = "name") String name,
@RequestHeader(Constants.HEADER_AUTHORIZATION) String authorization) {
DfsAccessToken accessToken = accessAuthorization(authorization);
Optional<MediaType> optional = MediaTypeFactory.getMediaType(name);
MimeType mimeType = optional.orElse(MediaType.APPLICATION_OCTET_STREAM);
FileMetadata metadata = new FileMetadata(mimeType.toString());
ByteArrayInputStream inputStream = new ByteArrayInputStream(Base64.getDecoder().decode(file));
DfsFile fileObject = new DfsFile(name, inputStream, metadata);
String fileId = fileRepositoryService.uploadFile(accessToken, fileObject);
return Message.success(fileId);
}
@RequestMapping(value = "/file/preview.do")
public void preview(@RequestParam("fileId") String fileId, @RequestParam(name = "style", required = false) String style,
@RequestHeader(Constants.HEADER_AUTHORIZATION) String authorization, HttpServletResponse response) throws IOException {
// style参数为oss图片处理设置，比如：image/resize,m_fixed,h_300,w_300
DfsAccessToken accessToken = accessAuthorization(authorization);
DfsFile file = fileRepositoryService.downloadFile(accessToken, fileId, style);
InputStream fileStream = file.getStream();
response.setContentType(file.getMetadata().getMimeType());
// oss sdk存在bug, fileStream.available()和真实字节数存在差异
// response.setContentLength(fileStream.available());
try {
IOUtils.copy(fileStream, response.getOutputStream());
} finally {
IOUtils.closeQuietly(fileStream);
}
response.flushBuffer();
}
@RequestMapping(value = "/file/download.do")
public void download(@RequestParam("fileId") String fileId, @RequestHeader(Constants.HEADER_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);
fileRepositoryService.deleteFile(accessToken, fileId);
return Message.success();
}
private DfsAccessToken accessAuthorization(String authorization) {
if (authorization == null) {
throw new DfsServiceException(ErrorCode.UNAUTHORIZED_ACCESS_ERROR, ErrorCode.MESSAGE_ACCESS_DENIED);
}
return DfsAccessToken.of(authorization, dfsProperties.getPublicKey());
}
}