add 增加 OSS 对象存储模块 相关代码(未完成)

疯狂的狮子li

2021-07-17 7e90d845987127718d34f27b4ebfdcafaa8e4e4e

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
package com.ruoyi.oss.service.impl;
 
import cn.hutool.core.io.IoUtil;
import com.qiniu.http.Response;
import com.qiniu.storage.BucketManager;
import com.qiniu.storage.Configuration;
import com.qiniu.storage.Region;
import com.qiniu.storage.UploadManager;
import com.qiniu.util.Auth;
import com.ruoyi.oss.config.CloudStorageConfig;
import com.ruoyi.oss.exception.OssException;
import com.ruoyi.oss.service.abstractd.AbstractCloudStorageService;
 
import java.io.InputStream;
 
/**
 * 七牛云存储
 */
public class QiniuCloudStorageServiceImpl extends AbstractCloudStorageService {
 
    private UploadManager uploadManager;
    private BucketManager bucketManager;
    private String token;
 
    public QiniuCloudStorageServiceImpl(CloudStorageConfig config) {
        this.config = config;
        // 初始化
        init();
    }
 
    private void init() {
        // z0 z1 z2
        Configuration config = new Configuration(Region.autoRegion());
        // 默认不使用https
        config.useHttpsDomains = false;
        uploadManager = new UploadManager(config);
        Auth auth = Auth.create(this.config.getAccessKey(), this.config.getSecretKey());
        token = auth.uploadToken(this.config.getBucketName());
        bucketManager = new BucketManager(auth, config);
    }
 
    @Override
    public String upload(byte[] data, String path) {
        try {
            Response res = uploadManager.put(data, path, token);
            if (!res.isOK()) {
                throw new RuntimeException("上传七牛出错：" + res.toString());
            }
        } catch (Exception e) {
            throw new OssException("上传文件失败，请核对七牛配置信息");
        }
        return config.getDomain() + "/" + path;
    }
 
    @Override
    public void delete(String path) {
        try {
            path = path.replace(config.getDomain() + "/", "");
            Response res = bucketManager.delete(config.getBucketName(), path);
            if (!res.isOK()) {
                throw new RuntimeException("删除七牛文件出错：" + res.toString());
            }
        } catch (Exception e) {
            throw new OssException(e.getMessage());
        }
    }
 
    @Override
    public String upload(InputStream inputStream, String path) {
        byte[] data = IoUtil.readBytes(inputStream);
        return this.upload(data, path);
    }
 
    @Override
    public String uploadSuffix(byte[] data, String suffix) {
        return upload(data, getPath(config.getPrefix(), suffix));
    }
 
    @Override
    public String uploadSuffix(InputStream inputStream, String suffix) {
        return upload(inputStream, getPath(config.getPrefix(), suffix));
    }
 
}