update 基于S3协议重新实现 OSS模块 支持自定义域名
疯狂的狮子li
4 years ago
parent
6d9a89a65f
commit
3430b090e6
@ -1,52 +0,0 @@
|
|||||||
package com.ruoyi.common.oss.enumd;
|
|
||||||
|
|
||||||
import com.ruoyi.common.oss.service.impl.AliyunOssStrategy;
|
|
||||||
import com.ruoyi.common.oss.service.impl.MinioOssStrategy;
|
|
||||||
import com.ruoyi.common.oss.service.impl.QcloudOssStrategy;
|
|
||||||
import com.ruoyi.common.oss.service.impl.QiniuOssStrategy;
|
|
||||||
import lombok.AllArgsConstructor;
|
|
||||||
import lombok.Getter;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 对象存储服务商枚举
|
|
||||||
*
|
|
||||||
* @author Lion Li
|
|
||||||
*/
|
|
||||||
@Getter
|
|
||||||
@AllArgsConstructor
|
|
||||||
public enum OssEnumd {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 七牛云
|
|
||||||
*/
|
|
||||||
QINIU("qiniu", QiniuOssStrategy.class),
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 阿里云
|
|
||||||
*/
|
|
||||||
ALIYUN("aliyun", AliyunOssStrategy.class),
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 腾讯云
|
|
||||||
*/
|
|
||||||
QCLOUD("qcloud", QcloudOssStrategy.class),
|
|
||||||
|
|
||||||
/**
|
|
||||||
* minio
|
|
||||||
*/
|
|
||||||
MINIO("minio", MinioOssStrategy.class);
|
|
||||||
|
|
||||||
private final String value;
|
|
||||||
|
|
||||||
private final Class<?> beanClass;
|
|
||||||
|
|
||||||
public static OssEnumd find(String value) {
|
|
||||||
for (OssEnumd enumd : values()) {
|
|
||||||
if (enumd.getValue().equals(value)) {
|
|
||||||
return enumd;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,69 +0,0 @@
|
|||||||
package com.ruoyi.common.oss.service.abstractd;
|
|
||||||
|
|
||||||
import cn.hutool.core.io.IoUtil;
|
|
||||||
import cn.hutool.core.util.IdUtil;
|
|
||||||
import com.ruoyi.common.core.utils.DateUtils;
|
|
||||||
import com.ruoyi.common.core.utils.StringUtils;
|
|
||||||
import com.ruoyi.common.oss.entity.UploadResult;
|
|
||||||
import com.ruoyi.common.oss.enumd.OssEnumd;
|
|
||||||
import com.ruoyi.common.oss.properties.OssProperties;
|
|
||||||
import com.ruoyi.common.oss.service.IOssStrategy;
|
|
||||||
|
|
||||||
import java.io.InputStream;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 对象存储策略(支持七牛、阿里云、腾讯云、minio)
|
|
||||||
*
|
|
||||||
* @author Lion Li
|
|
||||||
*/
|
|
||||||
public abstract class AbstractOssStrategy implements IOssStrategy {
|
|
||||||
|
|
||||||
protected OssProperties properties;
|
|
||||||
public boolean isInit = false;
|
|
||||||
|
|
||||||
public void init(OssProperties properties) {
|
|
||||||
this.properties = properties;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public abstract void createBucket();
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public abstract OssEnumd getServiceType();
|
|
||||||
|
|
||||||
public String getPath(String prefix, String suffix) {
|
|
||||||
// 生成uuid
|
|
||||||
String uuid = IdUtil.fastSimpleUUID();
|
|
||||||
// 文件路径
|
|
||||||
String path = DateUtils.datePath() + "/" + uuid;
|
|
||||||
if (StringUtils.isNotBlank(prefix)) {
|
|
||||||
path = prefix + "/" + path;
|
|
||||||
}
|
|
||||||
return path + suffix;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public abstract UploadResult upload(byte[] data, String path, String contentType);
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public abstract void delete(String path);
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public UploadResult upload(InputStream inputStream, String path, String contentType) {
|
|
||||||
byte[] data = IoUtil.readBytes(inputStream);
|
|
||||||
return this.upload(data, path, contentType);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public abstract UploadResult uploadSuffix(byte[] data, String suffix, String contentType);
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public abstract UploadResult uploadSuffix(InputStream inputStream, String suffix, String contentType);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 获取域名访问链接
|
|
||||||
*
|
|
||||||
* @return 域名访问链接
|
|
||||||
*/
|
|
||||||
public abstract String getEndpointLink();
|
|
||||||
}
|
|
||||||
@ -1,115 +0,0 @@
|
|||||||
package com.ruoyi.common.oss.service.impl;
|
|
||||||
|
|
||||||
import com.aliyun.oss.ClientConfiguration;
|
|
||||||
import com.aliyun.oss.OSSClient;
|
|
||||||
import com.aliyun.oss.common.auth.DefaultCredentialProvider;
|
|
||||||
import com.aliyun.oss.model.CannedAccessControlList;
|
|
||||||
import com.aliyun.oss.model.CreateBucketRequest;
|
|
||||||
import com.aliyun.oss.model.ObjectMetadata;
|
|
||||||
import com.aliyun.oss.model.PutObjectRequest;
|
|
||||||
import com.ruoyi.common.core.utils.StringUtils;
|
|
||||||
import com.ruoyi.common.oss.entity.UploadResult;
|
|
||||||
import com.ruoyi.common.oss.enumd.OssEnumd;
|
|
||||||
import com.ruoyi.common.oss.exception.OssException;
|
|
||||||
import com.ruoyi.common.oss.properties.OssProperties;
|
|
||||||
import com.ruoyi.common.oss.service.abstractd.AbstractOssStrategy;
|
|
||||||
import org.springframework.stereotype.Component;
|
|
||||||
|
|
||||||
import java.io.ByteArrayInputStream;
|
|
||||||
import java.io.InputStream;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 阿里云存储策略
|
|
||||||
*
|
|
||||||
* @author Lion Li
|
|
||||||
*/
|
|
||||||
@Component
|
|
||||||
public class AliyunOssStrategy extends AbstractOssStrategy {
|
|
||||||
|
|
||||||
private OSSClient client;
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void init(OssProperties ossProperties) {
|
|
||||||
super.init(ossProperties);
|
|
||||||
try {
|
|
||||||
ClientConfiguration configuration = new ClientConfiguration();
|
|
||||||
DefaultCredentialProvider credentialProvider = new DefaultCredentialProvider(
|
|
||||||
properties.getAccessKey(), properties.getSecretKey());
|
|
||||||
client = new OSSClient(properties.getEndpoint(), credentialProvider, configuration);
|
|
||||||
createBucket();
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new OssException("阿里云存储配置错误! 请检查系统配置:[" + e.getMessage() + "]");
|
|
||||||
}
|
|
||||||
isInit = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void createBucket() {
|
|
||||||
try {
|
|
||||||
String bucketName = properties.getBucketName();
|
|
||||||
if (client.doesBucketExist(bucketName)) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
CreateBucketRequest createBucketRequest = new CreateBucketRequest(bucketName);
|
|
||||||
createBucketRequest.setCannedACL(CannedAccessControlList.PublicRead);
|
|
||||||
client.createBucket(createBucketRequest);
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new OssException("创建Bucket失败, 请核对阿里云配置信息:[" + e.getMessage() + "]");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public OssEnumd getServiceType() {
|
|
||||||
return OssEnumd.ALIYUN;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public UploadResult upload(byte[] data, String path, String contentType) {
|
|
||||||
return upload(new ByteArrayInputStream(data), path, contentType);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public UploadResult upload(InputStream inputStream, String path, String contentType) {
|
|
||||||
try {
|
|
||||||
ObjectMetadata metadata = new ObjectMetadata();
|
|
||||||
metadata.setContentType(contentType);
|
|
||||||
client.putObject(new PutObjectRequest(properties.getBucketName(), path, inputStream, metadata));
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new OssException("上传文件失败，请检查阿里云配置信息:[" + e.getMessage() + "]");
|
|
||||||
}
|
|
||||||
return UploadResult.builder().url(getEndpointLink() + "/" + path).filename(path).build(); }
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void delete(String path) {
|
|
||||||
path = path.replace(getEndpointLink() + "/", "");
|
|
||||||
try {
|
|
||||||
client.deleteObject(properties.getBucketName(), path);
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new OssException("上传文件失败，请检查阿里云配置信息:[" + e.getMessage() + "]");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public UploadResult uploadSuffix(byte[] data, String suffix, String contentType) {
|
|
||||||
return upload(data, getPath(properties.getPrefix(), suffix), contentType);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public UploadResult uploadSuffix(InputStream inputStream, String suffix, String contentType) {
|
|
||||||
return upload(inputStream, getPath(properties.getPrefix(), suffix), contentType);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String getEndpointLink() {
|
|
||||||
String endpoint = properties.getEndpoint();
|
|
||||||
StringBuilder sb = new StringBuilder(endpoint);
|
|
||||||
if (StringUtils.containsAnyIgnoreCase(endpoint, "http://")) {
|
|
||||||
sb.insert(7, properties.getBucketName() + ".");
|
|
||||||
} else if (StringUtils.containsAnyIgnoreCase(endpoint, "https://")) {
|
|
||||||
sb.insert(8, properties.getBucketName() + ".");
|
|
||||||
} else {
|
|
||||||
throw new OssException("Endpoint配置错误");
|
|
||||||
}
|
|
||||||
return sb.toString();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,127 +0,0 @@
|
|||||||
package com.ruoyi.common.oss.service.impl;
|
|
||||||
|
|
||||||
import cn.hutool.core.util.ArrayUtil;
|
|
||||||
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.common.oss.entity.UploadResult;
|
|
||||||
import com.ruoyi.common.oss.enumd.OssEnumd;
|
|
||||||
import com.ruoyi.common.oss.exception.OssException;
|
|
||||||
import com.ruoyi.common.oss.properties.OssProperties;
|
|
||||||
import com.ruoyi.common.oss.service.abstractd.AbstractOssStrategy;
|
|
||||||
import org.springframework.stereotype.Component;
|
|
||||||
|
|
||||||
import java.io.InputStream;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 七牛云存储策略
|
|
||||||
*
|
|
||||||
* @author Lion Li
|
|
||||||
*/
|
|
||||||
@Component
|
|
||||||
public class QiniuOssStrategy extends AbstractOssStrategy {
|
|
||||||
|
|
||||||
private UploadManager uploadManager;
|
|
||||||
private BucketManager bucketManager;
|
|
||||||
private Auth auth;
|
|
||||||
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void init(OssProperties ossProperties) {
|
|
||||||
super.init(ossProperties);
|
|
||||||
try {
|
|
||||||
Configuration config = new Configuration(getRegion(properties.getRegion()));
|
|
||||||
// https设置
|
|
||||||
config.useHttpsDomains = false;
|
|
||||||
config.useHttpsDomains = "Y".equals(properties.getIsHttps());
|
|
||||||
uploadManager = new UploadManager(config);
|
|
||||||
auth = Auth.create(properties.getAccessKey(), properties.getSecretKey());
|
|
||||||
bucketManager = new BucketManager(auth, config);
|
|
||||||
createBucket();
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new OssException("七牛云存储配置错误! 请检查系统配置:[" + e.getMessage() + "]");
|
|
||||||
}
|
|
||||||
isInit = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void createBucket() {
|
|
||||||
try {
|
|
||||||
String bucketName = properties.getBucketName();
|
|
||||||
if (ArrayUtil.contains(bucketManager.buckets(), bucketName)) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
bucketManager.createBucket(bucketName, properties.getRegion());
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new OssException("创建Bucket失败, 请核对七牛云配置信息:[" + e.getMessage() + "]");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public OssEnumd getServiceType() {
|
|
||||||
return OssEnumd.QINIU;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public UploadResult upload(byte[] data, String path, String contentType) {
|
|
||||||
try {
|
|
||||||
String token = auth.uploadToken(properties.getBucketName());
|
|
||||||
Response res = uploadManager.put(data, path, token, null, contentType, false);
|
|
||||||
if (!res.isOK()) {
|
|
||||||
throw new RuntimeException("上传七牛出错：" + res.error);
|
|
||||||
}
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new OssException("上传文件失败，请核对七牛配置信息:[" + e.getMessage() + "]");
|
|
||||||
}
|
|
||||||
return UploadResult.builder().url(getEndpointLink() + "/" + path).filename(path).build();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void delete(String path) {
|
|
||||||
try {
|
|
||||||
path = path.replace(getEndpointLink() + "/", "");
|
|
||||||
Response res = bucketManager.delete(properties.getBucketName(), path);
|
|
||||||
if (!res.isOK()) {
|
|
||||||
throw new RuntimeException("删除七牛文件出错：" + res.error);
|
|
||||||
}
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new OssException(e.getMessage());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public UploadResult uploadSuffix(byte[] data, String suffix, String contentType) {
|
|
||||||
return upload(data, getPath(properties.getPrefix(), suffix), contentType);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public UploadResult uploadSuffix(InputStream inputStream, String suffix, String contentType) {
|
|
||||||
return upload(inputStream, getPath(properties.getPrefix(), suffix), contentType);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String getEndpointLink() {
|
|
||||||
return properties.getEndpoint();
|
|
||||||
}
|
|
||||||
|
|
||||||
private Region getRegion(String region) {
|
|
||||||
switch (region) {
|
|
||||||
case "z0":
|
|
||||||
return Region.region0();
|
|
||||||
case "z1":
|
|
||||||
return Region.region1();
|
|
||||||
case "z2":
|
|
||||||
return Region.region2();
|
|
||||||
case "na0":
|
|
||||||
return Region.regionNa0();
|
|
||||||
case "as0":
|
|
||||||
return Region.regionAs0();
|
|
||||||
default:
|
|
||||||
return Region.autoRegion();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,5 +1 @@
|
|||||||
org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
|
org.springframework.boot.autoconfigure.EnableAutoConfiguration=
|
||||||
com.ruoyi.common.oss.service.impl.AliyunOssStrategy,\
|
|
||||||
com.ruoyi.common.oss.service.impl.MinioOssStrategy,\
|
|
||||||
com.ruoyi.common.oss.service.impl.QcloudOssStrategy,\
|
|
||||||
com.ruoyi.common.oss.service.impl.QiniuOssStrategy
|
|
||||||
|
|||||||
Loading…
Reference in New Issue