package com.qianwen.core.oss;
|
|
import com.obs.services.ObsClient;
|
import com.obs.services.model.GetObjectRequest;
|
import com.obs.services.model.ObjectMetadata;
|
import com.obs.services.model.ObsObject;
|
import com.obs.services.model.PutObjectResult;
|
import java.io.InputStream;
|
import java.math.BigDecimal;
|
import java.util.List;
|
import com.qianwen.core.oss.model.BladeFile;
|
import com.qianwen.core.oss.model.OssFile;
|
import com.qianwen.core.oss.props.OssProperties;
|
import com.qianwen.core.oss.rule.OssRule;
|
import org.springframework.util.StringUtils;
|
import org.springframework.web.multipart.MultipartFile;
|
|
public class HuaweiObsTemplate implements OssTemplate {
|
private final ObsClient obsClient;
|
private final OssProperties ossProperties;
|
private final OssRule ossRule;
|
|
public HuaweiObsTemplate(final ObsClient obsClient, final OssProperties ossProperties, final OssRule ossRule) {
|
this.obsClient = obsClient;
|
this.ossProperties = ossProperties;
|
this.ossRule = ossRule;
|
}
|
|
@Override
|
public void makeBucket(String bucketName) {
|
if (!bucketExists(bucketName)) {
|
this.obsClient.createBucket(getBucketName(bucketName));
|
}
|
}
|
|
@Override
|
public void removeBucket(String bucketName) {
|
this.obsClient.deleteBucket(getBucketName(bucketName));
|
}
|
|
@Override
|
public boolean bucketExists(String bucketName) {
|
return this.obsClient.headBucket(getBucketName(bucketName));
|
}
|
|
@Override
|
public void copyFile(String bucketName, String fileName, String destBucketName) {
|
this.obsClient.copyObject(getBucketName(bucketName), fileName, getBucketName(destBucketName), fileName);
|
}
|
|
@Override
|
public void copyFile(String bucketName, String fileName, String destBucketName, String destFileName) {
|
this.obsClient.copyObject(getBucketName(bucketName), fileName, getBucketName(destBucketName), destFileName);
|
}
|
|
@Override
|
public OssFile statFile(String fileName) {
|
return statFile(this.ossProperties.getBucketName(), fileName);
|
}
|
|
@Override
|
public OssFile statFile(String bucketName, String fileName) {
|
ObjectMetadata stat = this.obsClient.getObjectMetadata(getBucketName(bucketName), fileName);
|
OssFile ossFile = new OssFile();
|
ossFile.setName(fileName);
|
ossFile.setLink(fileLink(ossFile.getName()));
|
ossFile.setHash(stat.getContentMd5());
|
ossFile.setLength(stat.getContentLength().longValue());
|
ossFile.setPutTime(stat.getLastModified());
|
ossFile.setContentType(stat.getContentType());
|
return ossFile;
|
}
|
|
@Override
|
public String filePath(String fileName) {
|
return getOssHost(getBucketName()).concat("/").concat(fileName);
|
}
|
|
@Override
|
public String filePath(String bucketName, String fileName) {
|
return getOssHost(getBucketName(bucketName)).concat("/").concat(fileName);
|
}
|
|
@Override
|
public String fileLink(String fileName) {
|
return getOssHost().concat("/").concat(fileName);
|
}
|
|
@Override
|
public String fileLink(String bucketName, String fileName) {
|
return getOssHost(getBucketName(bucketName)).concat("/").concat(fileName);
|
}
|
|
@Override
|
public InputStream getObject(String bucketName, String fileName, Long offset, Long length) {
|
return get(bucketName, fileName, offset, length);
|
}
|
|
@Override
|
public InputStream getObject(String fileName, Long offset, Long length) {
|
return get(this.ossProperties.getBucketName(), fileName, offset, length);
|
}
|
|
@Override
|
public InputStream getObject(String bucketName, String fileName) {
|
return get(bucketName, fileName);
|
}
|
|
@Override
|
public InputStream getObject(String fileName) {
|
return getObject(this.ossProperties.getBucketName(), fileName);
|
}
|
|
public InputStream get(String bucketName, String key) {
|
ObsObject ossObject = this.obsClient.getObject(getBucketName(bucketName), key);
|
return ossObject.getObjectContent();
|
}
|
|
public InputStream get(String bucketName, String key, Long offset, Long length) {
|
GetObjectRequest getObjectRequest = new GetObjectRequest(getBucketName(bucketName), key);
|
getObjectRequest.setRangeStart(offset);
|
getObjectRequest.setRangeEnd(Long.valueOf(new BigDecimal(offset.longValue()).add(new BigDecimal(length.longValue())).longValue()));
|
ObsObject ossObject = this.obsClient.getObject(getObjectRequest);
|
return ossObject.getObjectContent();
|
}
|
|
@Override
|
public BladeFile putFile(MultipartFile file) {
|
return putFile(this.ossProperties.getBucketName(), file.getOriginalFilename(), file);
|
}
|
|
@Override
|
public BladeFile putFile(String fileName, MultipartFile file) {
|
return putFile(this.ossProperties.getBucketName(), fileName, file);
|
}
|
|
@Override
|
public BladeFile putFile(String bucketName, String fileName, MultipartFile file) {
|
|
try {
|
return putFile(bucketName, fileName, file.getInputStream());
|
} catch (Exception ex) {
|
throw new RuntimeException(ex);
|
}
|
}
|
|
@Override
|
public BladeFile putFile(String fileName, InputStream stream) {
|
return putFile(this.ossProperties.getBucketName(), fileName, stream);
|
}
|
|
@Override
|
public BladeFile putFile(String bucketName, String fileName, InputStream stream) {
|
return put(bucketName, stream, fileName, false);
|
}
|
|
@Override
|
public void removeFile(String fileName) {
|
this.obsClient.deleteObject(getBucketName(), fileName);
|
}
|
|
@Override
|
public void removeFile(String bucketName, String fileName) {
|
this.obsClient.deleteObject(getBucketName(bucketName), fileName);
|
}
|
|
@Override
|
public void removeFiles(List<String> fileNames) {
|
fileNames.forEach(this::removeFile);
|
}
|
|
@Override
|
public void removeFiles(String bucketName, List<String> fileNames) {
|
fileNames.forEach(fileName -> {
|
removeFile(getBucketName(bucketName), fileName);
|
});
|
}
|
|
public BladeFile put(String bucketName, InputStream stream, String key, boolean cover) {
|
makeBucket(bucketName);
|
String key2 = getFileName(key);
|
if (cover) {
|
this.obsClient.putObject(getBucketName(bucketName), key2, stream);
|
} else {
|
PutObjectResult response = this.obsClient.putObject(getBucketName(bucketName), key2, stream);
|
for (int retry = 0; StringUtils.isEmpty(response.getEtag()) && retry < 5; retry++) {
|
response = this.obsClient.putObject(getBucketName(bucketName), key2, stream);
|
}
|
}
|
BladeFile file = new BladeFile();
|
file.setOriginalName(key);
|
file.setName(key2);
|
file.setLink(fileLink(bucketName, key2));
|
return file;
|
}
|
|
private String getFileName(String originalFilename) {
|
return this.ossRule.fileName(originalFilename);
|
}
|
|
private String getBucketName() {
|
return getBucketName(this.ossProperties.getBucketName());
|
}
|
|
private String getBucketName(String bucketName) {
|
return this.ossRule.bucketName(bucketName);
|
}
|
|
public String getOssHost(String bucketName) {
|
String prefix = this.ossProperties.getEndpoint().contains("https://") ? "https://" : "http://";
|
return prefix + getBucketName(bucketName) + "." + this.ossProperties.getEndpoint().replaceFirst(prefix, "");
|
}
|
|
public String getOssHost() {
|
return getOssHost(this.ossProperties.getBucketName());
|
}
|
}
|
