github
/
halo
mirror of https://github.com/halo-dev/halo
1
0
Fork 0
Code Issues Releases Wiki Activity

Merge pull request #251 from secondarycoder/dev-addBaiduYunOss 

新增百度云oss文件上传下载功能
pull/260/head
John Niang 2019-07-21 08:51:55 +08:00 committed by GitHub
parent b5294ed09c 9aa4532426
commit 7ad9f251dd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 210 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
build.gradle
View File

@ -54,6 +54,7 @@ dependencies {
implementation 'com.upyun:java-sdk:4.0.1'
implementation 'com.qiniu:qiniu-java-sdk:7.2.18'
implementation 'com.aliyun.oss:aliyun-sdk-oss:3.4.2'
implementation 'com.baidubce:bce-java-sdk:0.10.36'
implementation 'net.coobird:thumbnailator:0.4.8'
implementation 'io.springfox:springfox-swagger2:2.9.2'
implementation 'io.springfox:springfox-swagger-ui:2.9.2'

137
src/main/java/run/halo/app/handler/file/BaiDuYunFileHandler.java Normal file
View File

@ -0,0 +1,137 @@
package run.halo.app.handler.file;
import com.baidubce.auth.DefaultBceCredentials;
import com.baidubce.services.bos.BosClient;
import com.baidubce.services.bos.BosClientConfiguration;
import com.baidubce.services.bos.model.PutObjectResponse;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Component;
import org.springframework.util.Assert;
import org.springframework.web.multipart.MultipartFile;
import run.halo.app.exception.FileOperationException;
import run.halo.app.model.enums.AttachmentType;
import run.halo.app.model.properties.BaiDuYunProperties;
import run.halo.app.model.support.UploadResult;
import run.halo.app.service.OptionService;
import run.halo.app.utils.FilenameUtils;
import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.util.Objects;
/**
* BaiDuYun file handler.
*
* @author wangya
* @date 2019-07-20
*/
@Slf4j
@Component
public class BaiDuYunFileHandler implements FileHandler {
private final OptionService optionService;
public BaiDuYunFileHandler(OptionService optionService) {
this.optionService = optionService;
}
@Override
public UploadResult upload(MultipartFile file) {
Assert.notNull(file, "Multipart file must not be null");
// Get config
String ossEndPoint = optionService.getByPropertyOfNonNull(BaiDuYunProperties.OSS_ENDPOINT).toString();
String ossAccessKey = optionService.getByPropertyOfNonNull(BaiDuYunProperties.OSS_ACCESS_KEY).toString();
String ossAccessSecret = optionService.getByPropertyOfNonNull(BaiDuYunProperties.OSS_ACCESS_SECRET).toString();
String ossBucketName = optionService.getByPropertyOfNonNull(BaiDuYunProperties.OSS_BUCKET_NAME).toString();
String ossStyleRule = optionService.getByPropertyOfNonNull(BaiDuYunProperties.OSS_STYLE_RULE).toString();
String ossSource = StringUtils.join("https://", ossBucketName, "." + ossEndPoint);
BosClientConfiguration config = new BosClientConfiguration();
config.setCredentials(new DefaultBceCredentials(ossAccessKey,ossAccessSecret));
config.setEndpoint(ossEndPoint);
// Init OSS client
BosClient client = new BosClient(config);
try {
String basename = FilenameUtils.getBasename(file.getOriginalFilename());
String extension = FilenameUtils.getExtension(file.getOriginalFilename());
String timestamp = String.valueOf(System.currentTimeMillis());
String upFilePath = StringUtils.join(basename, "_", timestamp, ".", extension);
String filePath = StringUtils.join(StringUtils.appendIfMissing(ossSource, "/"), upFilePath);
// Upload
PutObjectResponse putObjectResponseFromInputStream = client.putObject(ossBucketName, upFilePath, file.getInputStream());
if (putObjectResponseFromInputStream == null) {
throw new FileOperationException("上传附件 " + file.getOriginalFilename() + " 到百度云失败 ");
}
// Response result
UploadResult uploadResult = new UploadResult();
uploadResult.setFilename(basename);
uploadResult.setFilePath(filePath);
uploadResult.setKey(upFilePath);
uploadResult.setMediaType(MediaType.valueOf(Objects.requireNonNull(file.getContentType())));
uploadResult.setSuffix(extension);
uploadResult.setSize(file.getSize());
// Handle thumbnail
if (FileHandler.isImageType(uploadResult.getMediaType())) {
BufferedImage image = ImageIO.read(file.getInputStream());
uploadResult.setWidth(image.getWidth());
uploadResult.setHeight(image.getHeight());
uploadResult.setThumbPath(StringUtils.isBlank(ossStyleRule) ? filePath : filePath + ossStyleRule);
}
return uploadResult;
} catch (Exception e) {
e.printStackTrace();
} finally {
client.shutdown();
}
// Build result
UploadResult result = new UploadResult();
log.info("File: [{}] uploaded successfully", file.getOriginalFilename());
return result;
}
@Override
public void delete(String key) {
Assert.notNull(key, "File key must not be blank");
// Get config
String ossEndPoint = optionService.getByPropertyOfNonNull(BaiDuYunProperties.OSS_ENDPOINT).toString();
String ossAccessKey = optionService.getByPropertyOfNonNull(BaiDuYunProperties.OSS_ACCESS_KEY).toString();
String ossAccessSecret = optionService.getByPropertyOfNonNull(BaiDuYunProperties.OSS_ACCESS_SECRET).toString();
String ossBucketName = optionService.getByPropertyOfNonNull(BaiDuYunProperties.OSS_BUCKET_NAME).toString();
String ossSource = StringUtils.join("https://", ossBucketName, "." + ossEndPoint);
BosClientConfiguration config = new BosClientConfiguration();
config.setCredentials(new DefaultBceCredentials(ossAccessKey,ossAccessSecret));
config.setEndpoint(ossEndPoint);
// Init OSS client
BosClient client = new BosClient(config);
try {
client.deleteObject(ossBucketName,key);
} catch (Exception e) {
throw new FileOperationException("附件 " + key + " 从百度云删除失败", e);
} finally {
client.shutdown();
}
}
@Override
public boolean supportType(AttachmentType type) {
return AttachmentType.BAIDUYUN.equals(type);
}
}

7
src/main/java/run/halo/app/model/enums/AttachmentType.java
View File

@ -31,7 +31,12 @@ public enum AttachmentType implements ValueEnum<Integer> {
/**
* 阿里云
*/
ALIYUN(4);
ALIYUN(4),
/**
* 百度云
*/
BAIDUYUN(5);
private Integer value;

66
src/main/java/run/halo/app/model/properties/BaiDuYunProperties.java Normal file
View File

@ -0,0 +1,66 @@
package run.halo.app.model.properties;
/**
* BaiDuYun properties.
*
* @author wangya
* @date 2019-07-19
*/
public enum BaiDuYunProperties implements PropertyEnum {
/**
* BaiDuyun oss endpoint.
*/
OSS_ENDPOINT("oss_baiduyun_endpoint", String.class, ""),
/**
* BaiDuyun oss bucket name.
*/
OSS_BUCKET_NAME("oss_baiduyun_bucket_name", String.class, ""),
/**
* BaiDuyun oss access key.
*/
OSS_ACCESS_KEY("oss_baiduyun_access_key", String.class, ""),
/**
* BaiDuyun oss access secret.
*/
OSS_ACCESS_SECRET("oss_baiduyun_access_secret", String.class, ""),
/**
* BaiDuyun oss style rule
*/
OSS_STYLE_RULE("oss_baiduyun_style_rule", String.class, "");
private final String value;
private final Class<?> type;
private final String defaultValue;
BaiDuYunProperties(String value, Class<?> type, String defaultValue) {
this.defaultValue = defaultValue;
if (!PropertyEnum.isSupportedType(type)) {
throw new IllegalArgumentException("Unsupported blog property type: " + type);
}
this.value = value;
this.type = type;
}
@Override
public Class<?> getType() {
return type;
}
@Override
public String defaultValue() {
return defaultValue;
}
@Override
public String getValue() {
return value;
}
}