package org.dromara.demo.util;
|
|
|
import cn.hutool.core.util.IdUtil;
|
import io.minio.MinioClient;
|
import io.minio.PutObjectOptions;
|
import jakarta.annotation.PostConstruct;
|
import lombok.Data;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.dromara.demo.domain.MinioResult;
|
import org.dromara.demo.exception.UploadFailException;
|
import org.springframework.boot.context.properties.ConfigurationProperties;
|
import org.springframework.stereotype.Component;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.ByteArrayInputStream;
|
|
/**
|
* @author gonghl
|
* @date 2024-3-6
|
*/
|
@Component
|
@ConfigurationProperties(prefix = "minio")
|
@Slf4j
|
@Data
|
public class MinioUtil {
|
|
|
private String url;
|
private String accessKey;
|
private String secretKey;
|
private MinioClient client;
|
|
|
public MinioResult uploadFile(MultipartFile file, String bucketName) {
|
try {
|
if (!client.bucketExists(bucketName)) {
|
client.makeBucket(bucketName);
|
client.setBucketPolicy(bucketName, "public");
|
}
|
|
String[] split = file.getOriginalFilename().split("\\.");
|
String suffix = split[split.length - 1];
|
|
if (split.length < 2) {
|
suffix = "";
|
}
|
|
String fileName = StringUtils.join(IdUtil.randomUUID(), ".", suffix);
|
|
client.putObject(bucketName, fileName, file.getInputStream(), new PutObjectOptions(file.getSize(), 5 * 1024 * 1024));
|
String fileUrl = getFilePath(bucketName, fileName);
|
if (StringUtils.isNotBlank(fileUrl)) {
|
return new MinioResult(bucketName, fileName, fileUrl, file.getOriginalFilename(), null);
|
}
|
} catch (Exception e) {
|
log.error("上传失败,失败详情信息:{}", e.getMessage());
|
throw new UploadFailException();
|
}
|
return null;
|
|
}
|
|
public MinioResult uploadFile(ByteArrayInputStream in, String name, String bucketName) {
|
|
String[] split = name.split("\\.");
|
String suffix = split[split.length - 1];
|
if (split.length < 2) {
|
suffix = "";
|
}
|
String fileName = StringUtils.join(IdUtil.randomUUID(), ".", suffix);
|
try {
|
client.putObject(bucketName, fileName, in, new PutObjectOptions(in.available(), 5 * 1024 * 1024));
|
String fileUrl = getFilePath(bucketName, fileName);
|
if (StringUtils.isNotBlank(fileUrl)) {
|
return new MinioResult(bucketName, fileName, fileUrl, name, null);
|
}
|
} catch (Exception e) {
|
log.error("上传失败,失败详情信息:{}", e.getMessage());
|
throw new UploadFailException();
|
}
|
|
return null;
|
|
}
|
|
public String uploadImage(ByteArrayInputStream in, String bucketName) {
|
try {
|
String fileName = IdUtil.randomUUID() + ".jpg";
|
client.putObject(bucketName, fileName, in, new PutObjectOptions(in.available(), 5 * 1024 * 1024));
|
String fileUrl = getFilePath(bucketName, fileName);
|
if (StringUtils.isNotBlank(fileUrl)) {
|
return fileUrl;
|
}
|
} catch (Exception e) {
|
log.error("上传失败,失败详情信息:{}", e.getMessage());
|
throw new UploadFailException();
|
}
|
|
return null;
|
|
}
|
|
public String getFilePath(String bucketName, String fileName) {
|
String wordUrl = null;
|
try {
|
if (client.getObject(bucketName, fileName) == null) {
|
return null;
|
}
|
|
return StringUtils.join("/", bucketName, "/", fileName);
|
} catch (Exception e) {
|
log.error("获取bucket为:{},文件名为{}时出错,原因:{}", bucketName, fileName, e.getMessage());
|
}
|
|
return null;
|
}
|
|
|
@PostConstruct
|
public void creatConnect() throws Exception {
|
this.client = new MinioClient(url, accessKey, secretKey);
|
|
|
}
|
|
}
|
