From c88f44a7d604bf91e22a5cb076018884a01a6261 Mon Sep 17 00:00:00 2001
From: Van333 <van666666@foxmail.com>
Date: 星期六, 19 十一月 2022 15:11:17 +0800
Subject: [PATCH] 替换阿里oss为本地minio，关闭对时存在定时任务时睡眠10min
---
sandu-common/src/main/java/com/sandu/common/file/impl/AliOssFileServiceImpl.java | 166 ++++++++++++++++++++++++++++++++++++++++--------------
1 files changed, 122 insertions(+), 44 deletions(-)
diff --git a/sandu-common/src/main/java/com/sandu/common/file/impl/AliOssFileServiceImpl.java b/sandu-common/src/main/java/com/sandu/common/file/impl/AliOssFileServiceImpl.java
index 6f927bc..41e8eea 100644
--- a/sandu-common/src/main/java/com/sandu/common/file/impl/AliOssFileServiceImpl.java
+++ b/sandu-common/src/main/java/com/sandu/common/file/impl/AliOssFileServiceImpl.java
@@ -2,41 +2,43 @@
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.lang.UUID;
-import cn.hutool.core.util.StrUtil;
-import com.aliyun.oss.OSS;
-import com.aliyun.oss.OSSClientBuilder;
-import com.aliyun.oss.model.CannedAccessControlList;
-import com.aliyun.oss.model.CreateBucketRequest;
-import com.aliyun.oss.model.PutObjectRequest;
-import com.aliyun.oss.model.PutObjectResult;
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 lombok.AllArgsConstructor;
+import com.sandu.common.file.config.minioConfig.MinIoPolicy;
+import io.minio.*;
+import io.minio.messages.Bucket;
+import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;
import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
-import java.io.ByteArrayInputStream;
import java.io.IOException;
+import java.io.InputStream;
+import java.math.BigInteger;
+import java.security.MessageDigest;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
+import java.util.List;
/**
* @author chenjiantian
* @date 2020/12/17 15:31
*/
@Slf4j
-@AllArgsConstructor
+@Component
public class AliOssFileServiceImpl implements IFileUpload {
- private final String endPoint;
- private final String keyId;
- private final String keySecret;
- private final String bucketName;
- private final String host;
+ @Autowired
+ private MinioClient minioClient;
+
+ private String bucketName = "remote-update";
+ public String MINIO_PREFIX = "vnnox.";
+
@Override
public FileUploadDto uploadFile(MultipartFile multipartFile) {
@@ -44,43 +46,33 @@
long size = multipartFile.getSize();
String suffix = FileUtil.getSuffix(originalFilename);
String dayStr = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMddHHmm"));
- String fileName = dayStr + UUID.fastUUID().toString() + "." + suffix;
+ String fileName = dayStr + UUID.fastUUID() + "." + suffix;
+ String md5 = getMD5(multipartFile);
FileUploadDto uploadDto = new FileUploadDto();
- OSS ossClient = new OSSClientBuilder().build(endPoint, keyId, keySecret);
- try {
- //瀹瑰櫒涓嶅瓨鍦紝灏卞垱寤�
- if (!ossClient.doesBucketExist(bucketName)) {
- ossClient.createBucket(bucketName);
- CreateBucketRequest createBucketRequest = new CreateBucketRequest(bucketName);
- createBucketRequest.setCannedACL(CannedAccessControlList.PublicRead);
- ossClient.createBucket(createBucketRequest);
- }
- //涓婁紶鏂囦欢
- PutObjectResult result = ossClient.putObject(new PutObjectRequest(bucketName, fileName, new ByteArrayInputStream(multipartFile.getBytes())));
- //璁剧疆鏉冮檺 杩欓噷鏄叕寮�璇�
-// ossClient.setBucketAcl(ossProperties.getBucketName(), CannedAccessControlList.PublicRead);
- if (result != null) {
- String url = "";
- // https://BucketName.Endpoint/ObjectName
- if (StrUtil.isBlank(host)) {
- url = "https://" + bucketName + "." + endPoint + "/" + fileName;
- } else {
- if (!host.endsWith("/")) {
- url = host + "/" + fileName;
- } else {
- url = host + fileName;
- }
+
+ try {
+ if (!bucketExists(bucketName)) {
+ createBucket(bucketName);
+ if (bucketName.contains(MINIO_PREFIX)) {
+ setBucketPolicy(bucketName);
}
- uploadDto.setFileUrl(url);
}
+
+ final InputStream is = multipartFile.getInputStream();
+ Long start = System.currentTimeMillis();
+ minioClient.putObject(bucketName, fileName, is, new PutObjectOptions(is.available(), -1));
+ is.close();
+ System.out.println(("涓婁紶鎬绘椂闂达細" + (System.currentTimeMillis() - start) + "ms"));
+
+// 浠巑inio鑾峰彇 url
+ String url = minioClient.presignedGetObject(bucketName, fileName);
+ System.out.println("鑾峰彇鍒扮殑璺緞锛�"+url);
+ uploadDto.setFileUrl(url);
} catch (Exception e) {
log.error(e.getMessage(), e);
throw new BusinessException(e.getMessage());
- } finally {
- //鍏抽棴
- ossClient.shutdown();
}
uploadDto.setOriginName(originalFilename);
@@ -88,6 +80,7 @@
uploadDto.setFileType(FileUtil.extName(originalFilename));
uploadDto.setStorageType(FileStorageEnums.ALIBABA.getCode());
uploadDto.setFileSize(size);
+ uploadDto.setMd5(md5);
if (IMAGE_SUFFIX_LIST.contains(uploadDto.getFileType())) {
BufferedImage img = null;
@@ -101,6 +94,91 @@
uploadDto.setHeight(img.getHeight());
}
}
+
return uploadDto;
}
+
+
+ /**
+ * 鑾峰彇涓婁紶鏂囦欢鐨凪D5鍊�
+ *
+ * @param file
+ * @return
+ */
+ public String getMD5(MultipartFile file) {
+
+ try {
+ byte[] uploadBytes = file.getBytes();
+ MessageDigest md5 = MessageDigest.getInstance("MD5");
+ byte[] digest = md5.digest(uploadBytes);
+ String hashString = new BigInteger(1, digest).toString(16);
+ if (hashString.length() % 2 != 0) {
+ hashString = "0" + hashString;
+ }
+ return hashString;
+ } catch (Exception e) {
+ }
+ return null;
+ }
+
+
+ /**
+ * 璁剧疆寮�鏀�*鐨勬《绛栫暐
+ *
+ * @param bucketName
+ * @return
+ */
+ public void setBucketPolicy(String bucketName) throws Exception {
+ minioClient.setBucketPolicy(
+ SetBucketPolicyArgs.builder().bucket(bucketName).config(MinIoPolicy.generateMinioPolicy(bucketName)).build()
+ );
+ }
+
+
+ /**
+ * 鍒ゆ柇 bucket鏄惁瀛樺湪
+ *
+ * @param bucketName: 妗跺悕
+ * @return: boolean
+ */
+ @SneakyThrows(Exception.class)
+ public boolean bucketExists(String bucketName) {
+ return minioClient.bucketExists(bucketName);
+ }
+
+ public boolean objectExists(String bucketName, String objectName) {
+ try {
+ getObjectInfo(bucketName, objectName);
+ } catch (Exception e) {
+ return false;
+ }
+ return true;
+ }
+
+ private void getObjectInfo(String bucketName, String objectName) {
+
+ }
+
+ /**
+ * 鍒涘缓 bucket
+ *
+ * @param bucketName: 妗跺悕
+ * @return: void
+ */
+ @SneakyThrows(Exception.class)
+ public void createBucket(String bucketName) {
+ boolean isExist = minioClient.bucketExists((BucketExistsArgs) ((io.minio.BucketExistsArgs.Builder) BucketExistsArgs.builder().bucket(bucketName)).build());
+ if (!isExist) {
+ minioClient.makeBucket((MakeBucketArgs) ((io.minio.MakeBucketArgs.Builder) MakeBucketArgs.builder().bucket(bucketName)).build());
+ }
+ }
+
+ /**
+ * 鑾峰彇鍏ㄩ儴bucket
+ */
+ @SneakyThrows(Exception.class)
+ public List<Bucket> getAllBuckets() {
+ return minioClient.listBuckets();
+ }
+
}
--
Gitblit v1.9.3