|
@@ -0,0 +1,68 @@
|
|
1
|
+package com.yunzhi.niucai.common;
|
|
2
|
+
|
|
3
|
+import com.aliyun.oss.OSSClient;
|
|
4
|
+import com.yunzhi.niucai.config.AliyunConfig;
|
|
5
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
6
|
+import org.springframework.stereotype.Component;
|
|
7
|
+import org.springframework.web.multipart.MultipartFile;
|
|
8
|
+
|
|
9
|
+import java.io.*;
|
|
10
|
+
|
|
11
|
+@Component
|
|
12
|
+public class OSSUtils {
|
|
13
|
+
|
|
14
|
+ @Autowired
|
|
15
|
+ AliyunConfig aliyunConfig;
|
|
16
|
+
|
|
17
|
+ /**
|
|
18
|
+ * 上传文件
|
|
19
|
+ * @param mf 上传的文件
|
|
20
|
+ * @param toDirs 上传到的目录
|
|
21
|
+ * @return
|
|
22
|
+ */
|
|
23
|
+ public String putObject(MultipartFile mf, String ... toDirs) throws IOException {
|
|
24
|
+ String fName = formatFileName(mf.getOriginalFilename());
|
|
25
|
+ String preFix = String.valueOf(System.currentTimeMillis());
|
|
26
|
+ String toDir = "/";
|
|
27
|
+ if (null != toDirs && toDirs.length > 0) {
|
|
28
|
+ toDir = toDirs[0];
|
|
29
|
+ }
|
|
30
|
+ String nwFName = toDir + preFix + "-" + fName;
|
|
31
|
+ nwFName = StringUtils.trim(nwFName,"/");
|
|
32
|
+
|
|
33
|
+ return putFile(nwFName, new ByteArrayInputStream(mf.getBytes()));
|
|
34
|
+ }
|
|
35
|
+
|
|
36
|
+ public String putObject(File f, String ... toDirs) throws FileNotFoundException {
|
|
37
|
+ String fName = formatFileName(f.getName());
|
|
38
|
+ String preFix = String.valueOf(System.currentTimeMillis());
|
|
39
|
+ String toDir = "/";
|
|
40
|
+ if (null != toDirs && toDirs.length > 0) {
|
|
41
|
+ toDir = toDirs[0];
|
|
42
|
+ }
|
|
43
|
+ String nwFName = toDir + preFix + "-" + fName;
|
|
44
|
+ nwFName = StringUtils.trim(nwFName,"/");
|
|
45
|
+
|
|
46
|
+ InputStream inputStream = new FileInputStream(f);
|
|
47
|
+ return putFile(nwFName, inputStream);
|
|
48
|
+ }
|
|
49
|
+
|
|
50
|
+ private String formatFileName(String fName) {
|
|
51
|
+ return StringUtils.ifNull(fName, StringUtils.random(16) + ".png");
|
|
52
|
+ }
|
|
53
|
+
|
|
54
|
+ private String putFile(String fname, InputStream input) {
|
|
55
|
+ OSSClient ossClient = new OSSClient(aliyunConfig.getOss().getEndpoint(), aliyunConfig.getAccessKeyId(), aliyunConfig.getAccessKeySecret());
|
|
56
|
+ ossClient.putObject(aliyunConfig.getOss().getBucketName(), fname, input);
|
|
57
|
+ ossClient.shutdown();
|
|
58
|
+
|
|
59
|
+ // String url = ossClient.generatePresignedUrl(AliOSSUtils.oss.getBucketName(), fname, expiration).toString();
|
|
60
|
+
|
|
61
|
+ String preURL = aliyunConfig.getOss().getBucketURL();
|
|
62
|
+ if (!preURL.endsWith("/")) {
|
|
63
|
+ preURL += "/";
|
|
64
|
+ }
|
|
65
|
+
|
|
66
|
+ return preURL + fname.replaceAll("\\ ", "%20");
|
|
67
|
+ }
|
|
68
|
+}
|