package com.sandu.common.file.impl;
|
|
import cn.hutool.core.io.FileUtil;
|
import cn.hutool.core.lang.UUID;
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
import com.qiniu.http.Response;
|
import com.qiniu.storage.Configuration;
|
import com.qiniu.storage.UploadManager;
|
import com.qiniu.util.Auth;
|
import com.sandu.common.enums.FileStorageEnums;
|
import com.sandu.common.execption.BusinessException;
|
import com.sandu.common.file.FileUploadDto;
|
import com.sandu.common.file.IFileUpload;
|
import com.sandu.common.file.config.QiNiuOssProperties;
|
import lombok.AllArgsConstructor;
|
import lombok.SneakyThrows;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.imageio.ImageIO;
|
import java.awt.image.BufferedImage;
|
import java.io.File;
|
import java.io.IOException;
|
|
/**
|
* @author chenjiantian
|
* @date 2021/9/1 18:38
|
* 七牛上传
|
*/
|
@AllArgsConstructor
|
public class QiNiuFileUploadImpl implements IFileUpload {
|
|
private QiNiuOssProperties qiNiuOssProperties;
|
|
|
@SneakyThrows
|
@Override
|
public FileUploadDto uploadFile(MultipartFile multipartFile) {
|
String originalFilename = multipartFile.getOriginalFilename();
|
FileUploadDto uploadDto = new FileUploadDto();
|
String suffix = FileUtil.getSuffix(originalFilename);
|
String fileName = UUID.fastUUID().toString() + "." + suffix;
|
|
Configuration cfg = new Configuration();
|
UploadManager uploadManager = new UploadManager(cfg);
|
Auth auth = Auth.create("accessKey", qiNiuOssProperties.getAccessKey());
|
String upToken = auth.uploadToken(qiNiuOssProperties.getBucket());
|
Response response = uploadManager.put(multipartFile.getBytes(), null, upToken);
|
if (!response.isOK()) {
|
throw new BusinessException("七牛上传出错" + response.statusCode);
|
}
|
JSONObject jsonObject = JSON.parseObject(response.bodyString());
|
String key = jsonObject.getString("key");
|
String url = new File(qiNiuOssProperties.getHost(),key).getAbsolutePath();
|
uploadDto.setFileUrl(url);
|
|
uploadDto.setOriginName(originalFilename);
|
uploadDto.setFileName(fileName);
|
uploadDto.setFileType(FileUtil.extName(originalFilename));
|
uploadDto.setStorageType(FileStorageEnums.QINIU.getCode());
|
uploadDto.setFileSize(multipartFile.getSize());
|
|
if (IMAGE_SUFFIX_LIST.contains(uploadDto.getFileType())) {
|
BufferedImage img = null;
|
try {
|
img = ImageIO.read(multipartFile.getInputStream());
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
if (img != null && img.getWidth() >= 0 && img.getHeight() >= 0) {
|
uploadDto.setWidth(img.getWidth());
|
uploadDto.setHeight(img.getHeight());
|
}
|
}
|
return uploadDto;
|
}
|
}
|
