|
|
@@ -0,0 +1,392 @@
|
|
|
+package com.jzg.organization.utils;
|
|
|
+
|
|
|
+import io.minio.*;
|
|
|
+import io.minio.http.Method;
|
|
|
+import io.minio.messages.Bucket;
|
|
|
+import io.minio.messages.DeleteObject;
|
|
|
+import io.minio.messages.Item;
|
|
|
+import jakarta.annotation.Resource;
|
|
|
+import lombok.SneakyThrows;
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
+import org.apache.commons.lang3.StringUtils;
|
|
|
+import org.springframework.context.annotation.Bean;
|
|
|
+import org.springframework.stereotype.Component;
|
|
|
+import org.springframework.web.multipart.MultipartFile;
|
|
|
+
|
|
|
+import java.beans.BeanProperty;
|
|
|
+import java.io.ByteArrayInputStream;
|
|
|
+import java.io.InputStream;
|
|
|
+import java.io.UnsupportedEncodingException;
|
|
|
+import java.net.URLDecoder;
|
|
|
+import java.util.*;
|
|
|
+
|
|
|
+/**
|
|
|
+ * @program: xxkfz-minio
|
|
|
+ * @ClassName MinioUtils.java
|
|
|
+ * @author: 公众号：小小开发者
|
|
|
+ * @create: 2024-03-13 10:55
|
|
|
+ * @description: MinIO操作工具类
|
|
|
+ **/
|
|
|
+@Slf4j
|
|
|
+@Component
|
|
|
+public class MinioUtils {
|
|
|
+
|
|
|
+
|
|
|
+ @Resource
|
|
|
+ private MinioClient minioClient;
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 启动SpringBoot容器的时候初始化Bucket
|
|
|
+ * 如果没有Bucket则创建
|
|
|
+ *
|
|
|
+ * @param bucketName
|
|
|
+ */
|
|
|
+ public void createBucket(String bucketName) {
|
|
|
+ try {
|
|
|
+ if (!bucketExists(bucketName)) {
|
|
|
+ minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucketName).build());
|
|
|
+ log.info("创建bucketName = {}完成!", bucketName);
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ log.info("bucketName = {}已存在！策略为：{}", bucketName, getBucketPolicy(bucketName));
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("创建bucketName = {}异常!e = {}", bucketName, e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 判断Bucket是否存在，true：存在，false：不存在
|
|
|
+ *
|
|
|
+ * @param bucketName
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @SneakyThrows
|
|
|
+ public boolean bucketExists(String bucketName) {
|
|
|
+ return minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获得Bucket的策略
|
|
|
+ *
|
|
|
+ * @param bucketName
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @SneakyThrows
|
|
|
+ public String getBucketPolicy(String bucketName) {
|
|
|
+ return minioClient.getBucketPolicy(GetBucketPolicyArgs.builder().bucket(bucketName).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获得所有Bucket列表
|
|
|
+ *
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @SneakyThrows
|
|
|
+ public List<Bucket> getAllBuckets() {
|
|
|
+ return minioClient.listBuckets();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 根据bucketName获取其相关信息
|
|
|
+ *
|
|
|
+ * @param bucketName
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public Optional<Bucket> getBucket(String bucketName) {
|
|
|
+ return getAllBuckets().stream().filter(b -> b.name().equals(bucketName)).findFirst();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 根据bucketName删除Bucket，true：删除成功； false：删除失败，文件或已不存在
|
|
|
+ *
|
|
|
+ * @param bucketName
|
|
|
+ * @throws Exception
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public void removeBucket(String bucketName) {
|
|
|
+ minioClient.removeBucket(RemoveBucketArgs.builder().bucket(bucketName).build());
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 判断文件是否存在
|
|
|
+ *
|
|
|
+ * @param bucketName
|
|
|
+ * @param objectName
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ public boolean isObjectExist(String bucketName, String objectName) {
|
|
|
+ boolean exist = true;
|
|
|
+ try {
|
|
|
+ minioClient.statObject(StatObjectArgs.builder().bucket(bucketName).object(objectName).build());
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("[Minio工具类]>>>> 判断文件是否存在, 异常：", e);
|
|
|
+ exist = false;
|
|
|
+ }
|
|
|
+ return exist;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 判断文件夹是否存在
|
|
|
+ *
|
|
|
+ * @param bucketName
|
|
|
+ * @param objectName
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ public boolean isFolderExist(String bucketName, String objectName) {
|
|
|
+ boolean exist = false;
|
|
|
+ try {
|
|
|
+ Iterable<Result<Item>> results = minioClient.listObjects(ListObjectsArgs.builder().bucket(bucketName).prefix(objectName).recursive(false).build());
|
|
|
+ for (Result<Item> result : results) {
|
|
|
+ Item item = result.get();
|
|
|
+ if (item.isDir() && objectName.equals(item.objectName())) {
|
|
|
+ exist = true;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("[Minio工具类]>>>> 判断文件夹是否存在，异常：", e);
|
|
|
+ exist = false;
|
|
|
+ }
|
|
|
+ return exist;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 根据文件前置查询文件
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param prefix 前缀
|
|
|
+ * @param recursive 是否使用递归查询
|
|
|
+ * @return MinioItem 列表
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public List<Item> getAllObjectsByPrefix(String bucketName, String prefix, boolean recursive) {
|
|
|
+ List<Item> list = new ArrayList<>();
|
|
|
+ Iterable<Result<Item>> objectsIterator = minioClient.listObjects(ListObjectsArgs.builder().bucket(bucketName).prefix(prefix).recursive(recursive).build());
|
|
|
+ if (objectsIterator != null) {
|
|
|
+ for (Result<Item> o : objectsIterator) {
|
|
|
+ Item item = o.get();
|
|
|
+ list.add(item);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return list;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取文件流
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param objectName 文件名
|
|
|
+ * @return 二进制流
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public InputStream getObject(String bucketName, String objectName) {
|
|
|
+ return minioClient.getObject(GetObjectArgs.builder().bucket(bucketName).object(objectName).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 断点下载
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param objectName 文件名称
|
|
|
+ * @param offset 起始字节的位置
|
|
|
+ * @param length 要读取的长度
|
|
|
+ * @return 二进制流
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public InputStream getObject(String bucketName, String objectName, long offset, long length) {
|
|
|
+ return minioClient.getObject(GetObjectArgs.builder().bucket(bucketName).object(objectName).offset(offset).length(length).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取路径下文件列表
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param prefix 文件名称
|
|
|
+ * @param recursive 是否递归查找，false：模拟文件夹结构查找
|
|
|
+ * @return 二进制流
|
|
|
+ */
|
|
|
+ public Iterable<Result<Item>> listObjects(String bucketName, String prefix, boolean recursive) {
|
|
|
+ return minioClient.listObjects(ListObjectsArgs.builder().bucket(bucketName).prefix(prefix).recursive(recursive).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 使用MultipartFile进行文件上传
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param file 文件名
|
|
|
+ * @param objectName 对象名
|
|
|
+ * @param contentType 类型
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public ObjectWriteResponse uploadFile(String bucketName, MultipartFile file, String objectName, String contentType) {
|
|
|
+ InputStream inputStream = file.getInputStream();
|
|
|
+ return minioClient.putObject(PutObjectArgs.builder().bucket(bucketName).object(objectName).contentType(contentType).stream(inputStream, inputStream.available(), -1).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 图片上传
|
|
|
+ *
|
|
|
+ * @param bucketName
|
|
|
+ * @param imageBase64
|
|
|
+ * @param imageName
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ public ObjectWriteResponse uploadImage(String bucketName, String imageBase64, String imageName) {
|
|
|
+ if (!StringUtils.isEmpty(imageBase64)) {
|
|
|
+ InputStream in = base64ToInputStream(imageBase64);
|
|
|
+ String newName = System.currentTimeMillis() + "_" + imageName + ".jpg";
|
|
|
+ String year = String.valueOf(new Date().getYear());
|
|
|
+ String month = String.valueOf(new Date().getMonth());
|
|
|
+ return uploadFile(bucketName, year + "/" + month + "/" + newName, in);
|
|
|
+
|
|
|
+ }
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+
|
|
|
+ public static InputStream base64ToInputStream(String base64) {
|
|
|
+ ByteArrayInputStream stream = null;
|
|
|
+ try {
|
|
|
+ byte[] bytes = Base64.getEncoder().encode(base64.trim().getBytes());
|
|
|
+ stream = new ByteArrayInputStream(bytes);
|
|
|
+ } catch (Exception e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+ return stream;
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 上传本地文件
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param objectName 对象名称
|
|
|
+ * @param fileName 本地文件路径
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public ObjectWriteResponse uploadFile(String bucketName, String objectName, String fileName) {
|
|
|
+ return minioClient.uploadObject(UploadObjectArgs.builder().bucket(bucketName).object(objectName).filename(fileName).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 通过流上传文件
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param objectName 文件对象
|
|
|
+ * @param inputStream 文件流
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public ObjectWriteResponse uploadFile(String bucketName, String objectName, InputStream inputStream) {
|
|
|
+ return minioClient.putObject(PutObjectArgs.builder().bucket(bucketName).object(objectName).stream(inputStream, inputStream.available(), -1).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 创建文件夹或目录
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param objectName 目录路径
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public ObjectWriteResponse createDir(String bucketName, String objectName) {
|
|
|
+ return minioClient.putObject(PutObjectArgs.builder().bucket(bucketName).object(objectName).stream(new ByteArrayInputStream(new byte[]{}), 0, -1).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取文件信息, 如果抛出异常则说明文件不存在
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param objectName 文件名称
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public String getFileStatusInfo(String bucketName, String objectName) {
|
|
|
+ return minioClient.statObject(StatObjectArgs.builder().bucket(bucketName).object(objectName).build()).toString();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 拷贝文件
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param objectName 文件名
|
|
|
+ * @param srcBucketName 目标存储桶
|
|
|
+ * @param srcObjectName 目标文件名
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public ObjectWriteResponse copyFile(String bucketName, String objectName, String srcBucketName, String srcObjectName) {
|
|
|
+ return minioClient.copyObject(CopyObjectArgs.builder().source(CopySource.builder().bucket(bucketName).object(objectName).build()).bucket(srcBucketName).object(srcObjectName).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 删除文件
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param objectName 文件名称
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public void removeFile(String bucketName, String objectName) {
|
|
|
+ minioClient.removeObject(RemoveObjectArgs.builder().bucket(bucketName).object(objectName).build());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 批量删除文件
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param keys 需要删除的文件列表
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ public void removeFiles(String bucketName, List<String> keys) {
|
|
|
+ List<DeleteObject> objects = new LinkedList<>();
|
|
|
+ keys.forEach(s -> {
|
|
|
+ objects.add(new DeleteObject(s));
|
|
|
+ try {
|
|
|
+ removeFile(bucketName, s);
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("[Minio工具类]>>>> 批量删除文件，异常：", e);
|
|
|
+ }
|
|
|
+ });
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取文件外链
|
|
|
+ *
|
|
|
+ * @param bucketName 存储桶
|
|
|
+ * @param objectName 文件名
|
|
|
+ * @param expires 过期时间 <=7 秒 （外链有效时间（单位：秒））
|
|
|
+ * @return url
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public String getPresignedObjectUrl(String bucketName, String objectName, Integer expires) {
|
|
|
+ GetPresignedObjectUrlArgs args = GetPresignedObjectUrlArgs.builder().expiry(expires).bucket(bucketName).object(objectName).build();
|
|
|
+ return minioClient.getPresignedObjectUrl(args);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获得文件外链
|
|
|
+ *
|
|
|
+ * @param bucketName
|
|
|
+ * @param objectName
|
|
|
+ * @return url
|
|
|
+ */
|
|
|
+ @SneakyThrows(Exception.class)
|
|
|
+ public String getPresignedObjectUrl(String bucketName, String objectName) {
|
|
|
+ GetPresignedObjectUrlArgs args = GetPresignedObjectUrlArgs.builder().bucket(bucketName).object(objectName).method(Method.GET).build();
|
|
|
+ return minioClient.getPresignedObjectUrl(args);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 将URLDecoder编码转成UTF8
|
|
|
+ *
|
|
|
+ * @param str
|
|
|
+ * @return
|
|
|
+ * @throws UnsupportedEncodingException
|
|
|
+ */
|
|
|
+ public String getUtf8ByURLDecoder(String str) throws UnsupportedEncodingException {
|
|
|
+ String url = str.replaceAll("%(?![0-9a-fA-F]{2})", "%25");
|
|
|
+ return URLDecoder.decode(url, "UTF-8");
|
|
|
+ }
|
|
|
+}
|
