|
@@ -1,12 +1,16 @@
|
|
|
package com.ruoyi.oss.service.impl;
|
|
|
|
|
|
+import cn.hutool.core.util.StrUtil;
|
|
|
import com.qcloud.cos.COSClient;
|
|
|
import com.qcloud.cos.ClientConfig;
|
|
|
import com.qcloud.cos.auth.BasicCOSCredentials;
|
|
|
import com.qcloud.cos.auth.COSCredentials;
|
|
|
+import com.qcloud.cos.http.HttpProtocol;
|
|
|
+import com.qcloud.cos.model.*;
|
|
|
import com.qcloud.cos.region.Region;
|
|
|
import com.ruoyi.oss.entity.UploadResult;
|
|
|
import com.ruoyi.oss.enumd.CloudServiceEnumd;
|
|
|
+import com.ruoyi.oss.exception.OssException;
|
|
|
import com.ruoyi.oss.factory.OssFactory;
|
|
|
import com.ruoyi.oss.properties.CloudStorageProperties;
|
|
|
import com.ruoyi.oss.properties.CloudStorageProperties.QcloudProperties;
|
|
@@ -16,6 +20,7 @@ import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.context.annotation.Lazy;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
+import java.io.ByteArrayInputStream;
|
|
|
import java.io.InputStream;
|
|
|
|
|
|
/**
|
|
@@ -41,12 +46,33 @@ public class QcloudCloudStorageServiceImpl extends AbstractCloudStorageService i
|
|
|
ClientConfig clientConfig = new ClientConfig();
|
|
|
// 设置bucket所在的区域,华南:gz 华北:tj 华东:sh
|
|
|
clientConfig.setRegion(new Region(this.properties.getRegion()));
|
|
|
+ if (this.properties.getIsHttps()) {
|
|
|
+ clientConfig.setHttpProtocol(HttpProtocol.https);
|
|
|
+ } else {
|
|
|
+ clientConfig.setHttpProtocol(HttpProtocol.http);
|
|
|
+ }
|
|
|
client = new COSClient(credentials, clientConfig);
|
|
|
+ createBucket();
|
|
|
} catch (Exception e) {
|
|
|
throw new IllegalArgumentException("腾讯云存储配置错误! 请检查系统配置!");
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+ @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失败, 请核对腾讯云配置信息");
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
@Override
|
|
|
public String getServiceType() {
|
|
|
return CloudServiceEnumd.QCLOUD.getValue();
|
|
@@ -54,43 +80,57 @@ public class QcloudCloudStorageServiceImpl extends AbstractCloudStorageService i
|
|
|
|
|
|
@Override
|
|
|
public UploadResult upload(byte[] data, String path, String contentType) {
|
|
|
- // 腾讯云必需要以"/"开头
|
|
|
- if (!path.startsWith("/")) {
|
|
|
- path = "/" + path;
|
|
|
+ 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("上传文件失败,请检查腾讯云配置信息");
|
|
|
}
|
|
|
- // 上传到腾讯云
|
|
|
-// UploadFileRequest request = new UploadFileRequest(config.getQcloudBucketName(), path, data);
|
|
|
-// String response = client.uploadFile(request);
|
|
|
-// Map<String, Object> jsonObject = JsonUtils.parseMap(response);
|
|
|
-// if (Convert.toInt(jsonObject.get("code")) != 0) {
|
|
|
-// throw new OssException("文件上传失败," + Convert.toStr(jsonObject.get("message")));
|
|
|
-// }
|
|
|
- return new UploadResult().setUrl(properties.getDomain() + "/" + path).setFilename(path);
|
|
|
+ return new UploadResult().setUrl(getEndpointLink() + "/" + path).setFilename(path);
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public void delete(String path) {
|
|
|
-// path = path.replace(config.getDomain(),"");
|
|
|
-// DelFileRequest request = new DelFileRequest(config.getBucketName(), path);
|
|
|
-// String response = client.delFile(request);
|
|
|
-// Map<String, Object> jsonObject = JsonUtils.parseMap(response);
|
|
|
-// if (Convert.toInt(jsonObject.get("code")) != 0) {
|
|
|
-// throw new OssException("文件删除失败," + Convert.toStr(jsonObject.get("message")));
|
|
|
-// }
|
|
|
+ path = path.replace(getEndpointLink() + "/", "");
|
|
|
+ try {
|
|
|
+ client.deleteObject(new DeleteObjectRequest(properties.getBucketName(), path));
|
|
|
+ } catch (Exception e) {
|
|
|
+ throw new OssException("上传文件失败,请检腾讯云查配置信息");
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public UploadResult uploadSuffix(byte[] data, String suffix, String contentType) {
|
|
|
- return upload(data, getPath(this.properties.getPrefix(), suffix), contentType);
|
|
|
+ return upload(data, getPath(properties.getPrefix(), suffix), contentType);
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public UploadResult uploadSuffix(InputStream inputStream, String suffix, String contentType) {
|
|
|
- return upload(inputStream, getPath(this.properties.getPrefix(), suffix), contentType);
|
|
|
+ return upload(inputStream, getPath(properties.getPrefix(), suffix), contentType);
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public void afterPropertiesSet() throws Exception {
|
|
|
OssFactory.register(getServiceType(),this);
|
|
|
}
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public String getEndpointLink() {
|
|
|
+ String endpoint = properties.getEndpoint();
|
|
|
+ StringBuilder sb = new StringBuilder(endpoint);
|
|
|
+ if (StrUtil.containsAnyIgnoreCase(endpoint, "http://")) {
|
|
|
+ sb.insert(7, properties.getBucketName() + ".");
|
|
|
+ } else if (StrUtil.containsAnyIgnoreCase(endpoint, "https://")) {
|
|
|
+ sb.insert(8, properties.getBucketName() + ".");
|
|
|
+ } else {
|
|
|
+ throw new OssException("Endpoint配置错误");
|
|
|
+ }
|
|
|
+ return sb.toString();
|
|
|
+ }
|
|
|
}
|