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

阿里云附件上传

pull/137/head
MyFaith 2019-04-05 23:07:22 +08:00
parent 2db1669e3f
commit 56e350b972
4 changed files with 174 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pom.xml
View File

@ -45,6 +45,7 @@
<hutool.version>4.5.0</hutool.version>
<upyun-java-sdk.version>4.0.1</upyun-java-sdk.version>
<qiniu-java-sdk.version>7.2.18</qiniu-java-sdk.version>
<aliyun-java-sdk.version>3.4.2</aliyun-java-sdk.version>
<thumbnailator.version>0.4.8</thumbnailator.version>
<commonmark.version>0.12.1</commonmark.version>
<commons-lang3.version>3.8.1</commons-lang3.version>
@ -171,6 +172,13 @@
</exclusions>
</dependency>
<!-- AliYun SDK -->
<dependency>
<groupId>com.aliyun.oss</groupId>
<artifactId>aliyun-sdk-oss</artifactId>
<version>${aliyun-java-sdk.version}</version>
</dependency>
<!-- Thumbnailator -->
<dependency>
<groupId>net.coobird</groupId>

125
src/main/java/run/halo/app/handler/file/AliYunFileHandler.java Normal file
View File

@ -0,0 +1,125 @@
package run.halo.app.handler.file;
import cn.hutool.core.lang.Assert;
import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.*;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Component;
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.AliYunProperties;
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.Date;
import java.util.Objects;
/**
* AliYun file handler.
* @author MyFaith
* @date 2019-04-04 00:06:13
*/
@Slf4j
@Component
public class AliYunFileHandler implements FileHandler {
private final OptionService optionService;
public AliYunFileHandler(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(AliYunProperties.OSS_ENDPOINT);
String ossAccessKey = optionService.getByPropertyOfNonNull(AliYunProperties.OSS_ACCESS_KEY);
String ossAccessSecret = optionService.getByPropertyOfNonNull(AliYunProperties.OSS_ACCESS_SECRET);
String ossBucketName = optionService.getByPropertyOfNonNull(AliYunProperties.OSS_BUCKET_NAME);
String ossSource = StringUtils.join("https://", ossBucketName, "." + ossEndPoint);
// Init OSS client
OSS ossClient = new OSSClientBuilder().build(ossEndPoint, ossAccessKey, ossAccessSecret);
try {
String basename = FilenameUtils.getBasename(file.getOriginalFilename());
String extension = FilenameUtils.getExtension(file.getOriginalFilename());
String timestamp = String.valueOf(new Date().getTime());
String upFilePath = StringUtils.join(basename, "_", timestamp, ".", extension);
String filePath = StringUtils.join(StringUtils.appendIfMissing(ossSource, "/"), upFilePath);
// Upload
PutObjectResult putObjectResult = ossClient.putObject(ossBucketName, upFilePath, file.getInputStream());
if (putObjectResult == null) {
throw new FileOperationException("Failed to upload file " + file.getOriginalFilename() + " to AliYun " + upFilePath);
}
// 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(filePath);
}
return uploadResult;
} catch (Exception e){
e.printStackTrace();
} finally {
ossClient.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(AliYunProperties.OSS_ENDPOINT);
String ossAccessKey = optionService.getByPropertyOfNonNull(AliYunProperties.OSS_ACCESS_KEY);
String ossAccessSecret = optionService.getByPropertyOfNonNull(AliYunProperties.OSS_ACCESS_SECRET);
String ossBucketName = optionService.getByPropertyOfNonNull(AliYunProperties.OSS_BUCKET_NAME);
String ossSource = StringUtils.join("https://", ossBucketName, "." + ossEndPoint);
// Init OSS client
OSS ossClient = new OSSClientBuilder().build(ossEndPoint, ossAccessKey, ossAccessSecret);
try {
ossClient.deleteObject(new DeleteObjectsRequest(ossBucketName).withKey(key));
} catch (Exception e) {
throw new FileOperationException("Failed to delete file " + key + " from AliYun", e);
} finally {
ossClient.shutdown();
}
}
@Override
public boolean supportType(AttachmentType type) {
return AttachmentType.ALIYUN.equals(type);
}
}

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

@ -26,7 +26,12 @@ public enum AttachmentType implements ValueEnum<Integer> {
/**
* sm.ms
*/
SMMS(3);
SMMS(3),
/**
* 阿里云
*/
ALIYUN(4);
private Integer value;

35
src/main/java/run/halo/app/model/properties/AliYunProperties.java Normal file
View File

@ -0,0 +1,35 @@
package run.halo.app.model.properties;
/**
* AliYun properties.
* @author MyFaith
* @date 2019-04-04 00:00:56
*/
public enum AliYunProperties implements PropertyEnum {
OSS_ENDPOINT("oss_aliyun_endpoint", String.class),
OSS_BUCKET_NAME("oss_aliyun_bucket_name", String.class),
OSS_ACCESS_KEY("oss_aliyun_access_key", String.class),
OSS_ACCESS_SECRET("oss_aliyun_access_secret", String.class);
private String value;
private Class<?> type;
AliYunProperties(String value, Class<?> type) {
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 getValue() {
return value;
}
}