package com.mindskip.xzs.utility.minio;
|
|
import com.amazonaws.ClientConfiguration;
|
import com.amazonaws.Protocol;
|
import com.amazonaws.auth.AWSCredentials;
|
import com.amazonaws.auth.BasicAWSCredentials;
|
import com.amazonaws.services.s3.AmazonS3;
|
import com.amazonaws.services.s3.AmazonS3Client;
|
import com.amazonaws.services.s3.S3ClientOptions;
|
import com.amazonaws.services.s3.model.ObjectMetadata;
|
import com.amazonaws.services.s3.model.S3Object;
|
import org.apache.commons.io.FilenameUtils;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Component;
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
import java.io.InputStream;
|
import java.net.MalformedURLException;
|
import java.net.URL;
|
import java.util.Objects;
|
|
@Component
|
public class AwsMinIoUtil {
|
|
|
private static String endpoint;
|
|
private static final Logger logger = LoggerFactory.getLogger(AwsMinIoUtil.class);
|
|
private AmazonS3 s3client;
|
|
private static String accessKey;
|
|
private static String secretKey;
|
|
private static String bucket;
|
|
|
@Value("${aws.endpoint}")
|
public void setEndpoint(String endpoint) {
|
this.endpoint = endpoint;
|
}
|
|
@Value("${aws.accessKey}")
|
public void setAccessKey(String accessKey) {
|
this.accessKey = accessKey;
|
}
|
|
@Value("${aws.secretKey}")
|
public void setSecretKey(String secretKey) {
|
this.secretKey = secretKey;
|
}
|
|
@Value("${aws.bucket}")
|
public void setBucket(String bucket) {
|
this.bucket = bucket;
|
}
|
|
public static String getEndpoint() {
|
return endpoint;
|
}
|
|
public static String getAccessKey() {
|
return accessKey;
|
}
|
|
public static String getSecretKey() {
|
return secretKey;
|
}
|
|
public static String getBucket() {
|
return bucket;
|
}
|
|
public void init() throws MalformedURLException {
|
URL endpointUrl = new URL(endpoint);
|
String protocol = endpointUrl.getProtocol();
|
int port = endpointUrl.getPort() == -1 ? endpointUrl.getDefaultPort() : endpointUrl.getPort();
|
|
ClientConfiguration clientConfig = new ClientConfiguration();
|
clientConfig.setSignerOverride("S3SignerType");
|
clientConfig.setProtocol(Protocol.valueOf(protocol.toUpperCase()));
|
|
// 禁用证书检查，避免https自签证书校验失败
|
System.setProperty("com.amazonaws.sdk.disableCertChecking", "true");
|
// 屏蔽 AWS 的 MD5 校验，避免校验导致的下载抛出异常问题
|
System.setProperty("com.amazonaws.services.s3.disableGetObjectMD5Validation", "true");
|
AWSCredentials awsCredentials = new BasicAWSCredentials(accessKey, secretKey);
|
// 创建 S3Client 实例
|
AmazonS3 s3client = new AmazonS3Client(awsCredentials, clientConfig);
|
s3client.setEndpoint(endpointUrl.getHost() + ":" + port);
|
s3client.setS3ClientOptions(S3ClientOptions.builder().setPathStyleAccess(true).build());
|
this.s3client = s3client;
|
}
|
|
/**
|
* 创建桶
|
* @param bucket
|
* @return
|
*/
|
public boolean createBucket(String bucket) {
|
try {
|
s3client.createBucket(bucket);
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
return true;
|
}
|
|
/**
|
* 删除桶
|
* @param bucket
|
* @return
|
*/
|
public boolean deleteBucket(String bucket) {
|
try {
|
s3client.deleteBucket(bucket);
|
logger.info("删除bucket[{}]成功", bucket);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
return true;
|
}
|
|
/**
|
* 判断桶是否存在
|
* @param bucket
|
* @return
|
*/
|
public boolean bucketExists(String bucket) {
|
try {
|
return s3client.doesBucketExist(bucket);
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
return false;
|
}
|
|
/**
|
* 上传对象
|
* @param input
|
*/
|
public String upload( InputStream input, MultipartFile multipartFile) {
|
try {
|
init();
|
// 创建文件上传的元数据
|
ObjectMetadata meta = new ObjectMetadata();
|
// 设置文件上传长度
|
meta.setContentLength(input.available());
|
// 上传
|
String path = extractFilename(multipartFile);
|
s3client.putObject(bucket, path, input, meta);
|
return path;
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
return "";
|
}
|
|
/**
|
* 下载对象
|
* @param bucket
|
* @param objectId
|
* @return
|
*/
|
public InputStream download(String bucket, String objectId) {
|
try {
|
S3Object o = s3client.getObject(bucket, objectId);
|
return o.getObjectContent();
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
return null;
|
}
|
|
// public void download(String bucket, String objectId, OutputStream out) {
|
// S3Object o = s3client.getObject(bucket, objectId);
|
// try (InputStream in = o.getObjectContent()) {
|
// IOUtils.copyLarge(in, out);
|
// } catch (Exception e) {
|
// e.printStackTrace();
|
// }
|
// }
|
|
/**
|
* 判断对象是否存在
|
* @param bucket
|
* @param objectId
|
* @return
|
*/
|
public boolean existObject(String bucket, String objectId) {
|
try {
|
return s3client.doesObjectExist(bucket, objectId);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 删除对象
|
* @param bucket
|
* @param objectId
|
* @return
|
*/
|
public boolean deleteObject(String bucket, String objectId) {
|
try {
|
s3client.deleteObject(bucket, objectId);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
return true;
|
}
|
|
public void close() {
|
s3client = null;
|
}
|
|
/**
|
* 编码文件名
|
*/
|
public static final String extractFilename(MultipartFile file) {
|
return StringUtils.format("{}/{}_{}.{}", DateUtils.datePath(),
|
FilenameUtils.getBaseName(file.getOriginalFilename()), Seq.getId(Seq.uploadSeqType), getExtension(file));
|
}
|
|
/**
|
* 获取文件名的后缀
|
*
|
* @param file 表单文件
|
* @return 后缀名
|
*/
|
public static final String getExtension(MultipartFile file) {
|
String extension = FilenameUtils.getExtension(file.getOriginalFilename());
|
if (StringUtils.isEmpty(extension)) {
|
extension = MimeTypeUtils.getExtension(Objects.requireNonNull(file.getContentType()));
|
}
|
return extension;
|
}
|
|
|
|
}
|
