OSS业务存储适配器模式
流程
当我们配置了阿里云，腾讯云，minio等多个云存储厂商的业务代码时，如果我们要修改具体使用哪一种厂商的云存储，那么我们的controller层和service层就会需要改变业务代码；此时我们可以使用适配器模式来进行松耦合——>**1.**我们首先会定义一个关于文件存储的接口(非常丰富)，然后定义minio，阿里云等厂商的文件存储实现类，去实现文件存储的具体细节——>**2.**那么如何确定具体使用哪一个文件存储？——>**3.**我们利用nacos动态路由，得到storage.type——>**4.**然后再在我们的StorageConfig配置类中进行判断，如果是minio的，就返回minio的业务实现类，将其注入容器中，这样就实现了我们的动态路由，我们只需修改nacos上的配置文件进行发布即可
package com.wyh.oss.adapter;
import com.wyh.oss.entity.FileInfo;
import org.springframework.web.multipart.MultipartFile;
import java.io.InputStream;
import java.util.List;
/**
* 文件存储适配器
*
* @create: 2023/12/1 0:32
*/
public interface StorageAdapter {
/**
* 创建存储桶
*
* @param bucket 存储桶名称
*/
void createBucket(String bucket);
/**
* 上传文件
*
* @param multipartFile 文件流
* @param bucket 存储桶名称
* @param objectName 对象名称
*/
void uploadFile(MultipartFile multipartFile, String bucket, String objectName);
/**
* 获取所有存储桶
*
* @return 存储桶名称集合
*/
List<String> getAllBuckets();
/**
* 获取存储桶下所有文件
*
* @param bucket 存储桶名称
* @return 文件信息集合
*/
List<FileInfo> getAllFiles(String bucket);
/**
* 下载文件
*
* @param bucket 存储桶名称
* @param objectName 对象名称
* @return 文件流
*/
InputStream download(String bucket, String objectName);
/**
* 删除存储桶
*
* @param bucket 存储桶名称
*/
void deleteBucket(String bucket);
/**
* 删除文件
*
* @param bucket 存储桶名称
* @param objectName 对象名称
*/
void deleteObject(String bucket, String objectName);
String getUrl(String bucketName, String objectName);
}
minio的业务实现类：
package com.wyh.oss.adapter;
import com.wyh.oss.entity.FileInfo;
import com.wyh.oss.util.MinioUtil;
import lombok.SneakyThrows;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import java.io.InputStream;
import java.util.List;
/**
* minioIO存储适配器
*
* @create: 2023/12/1 0:38
*/
public class MinioStorageAdapter implements StorageAdapter {
@Resource
private MinioUtil minioUtil;
/**
* minioUrl
*/
@Value("${minio.url}")
private String url;
// 创建存储桶
@Override
@SneakyThrows
public void createBucket(String bucket) {
minioUtil.createBucket(bucket);
}
// 文件上传
@Override
@SneakyThrows
public void uploadFile(MultipartFile multipartFile, String bucket, String objectName) {
// 使用 minioUtil 对象创建存储桶
minioUtil.createBucket(bucket);
// 如果 objectName 不为空，则将文件名设置为 objectName + "/" + multipartFile.getName()
if (objectName != null) {
// 使用 minioUtil 对象将文件上传到 MinIO 存储服务中，并将其保存到指定的 bucket 以及 objectName + "/" + multipartFile.getName()
minioUtil.uploadFile(multipartFile.getInputStream(), bucket, objectName + "/" + multipartFile.getName());
} else {
// 使用 minioUtil 对象将文件上传到 MinIO 存储服务中，并将其保存到指定的 bucket 以及 multipartFile.getName()
minioUtil.uploadFile(multipartFile.getInputStream(), bucket, multipartFile.getName());
}
}
// 获取所有存储桶
@Override
@SneakyThrows
public List<String> getAllBuckets() {
return minioUtil.getAllBucket();
}
// 获取所有文件
@Override
@SneakyThrows
public List<FileInfo> getAllFiles(String bucket) {
return minioUtil.getAllFile(bucket);
}
// 文件下载
@Override
@SneakyThrows
public InputStream download(String bucket, String objectName) {
return minioUtil.downLoad(bucket, objectName);
}
// 桶删除
@Override
@SneakyThrows
public void deleteBucket(String bucket) {
minioUtil.deleteBucket(bucket);
}
// 文件删除
@Override
@SneakyThrows
public void deleteObject(String bucket, String objectName) {
minioUtil.deleteObject(bucket, objectName);
}
@Override
@SneakyThrows
public String getUrl(String bucket, String objectName) {
return url + "/" + bucket + "/" + objectName;
}
}
Util中封装了所有存储的具体细节：
package com.wyh.oss.util;
import com.wyh.oss.entity.FileInfo;
import io.minio.*;
import io.minio.errors.*;
import io.minio.http.Method;
import io.minio.messages.Bucket;
import io.minio.messages.Item;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.io.IOException;
import java.io.InputStream;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;
/**
* minio文件操作工具
*
* @author: ChickenWing
* @date: 2023/10/11
*/
@Component
public class MinioUtil {
@Resource
private MinioClient minioClient;
/**
* 创建bucket桶
*/
public void createBucket(String bucket) throws Exception {
boolean exists = minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucket).build());
if (!exists) {
minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucket).build());
}
}
/**
* 上传文件
*/
public void uploadFile(InputStream inputStream, String bucket, String objectName) throws Exception {
minioClient.putObject(PutObjectArgs.builder().bucket(bucket).object(objectName)
.stream(inputStream, -1, 5242889L).build());
}
/**
* 列出所有桶
*/
public List<String> getAllBucket() throws Exception {
List<Bucket> buckets = minioClient.listBuckets();
return buckets.stream().map(Bucket::name).collect(Collectors.toList());
}
/**
* 列出当前桶及文件
*/
public List<FileInfo> getAllFile(String bucket) throws Exception {
Iterable<Result<Item>> results = minioClient.listObjects(
ListObjectsArgs.builder().bucket(bucket).build());
List<FileInfo> fileInfoList = new LinkedList<>();
for (Result<Item> result : results) {
FileInfo fileInfo = new FileInfo();
Item item = result.get();
fileInfo.setFileName(item.objectName());
fileInfo.setDirectoryFlag(item.isDir());
fileInfo.setEtag(item.etag());
fileInfoList.add(fileInfo);
}
return fileInfoList;
}
/**
* 下载文件
*/
public InputStream downLoad(String bucket, String objectName) throws Exception {
return minioClient.getObject(
GetObjectArgs.builder().bucket(bucket).object(objectName).build()
);
}
/**
* 删除桶
*/
public void deleteBucket(String bucket) throws Exception {
minioClient.removeBucket(
RemoveBucketArgs.builder().bucket(bucket).build()
);
}
/**
* 删除文件
*/
public void deleteObject(String bucket, String objectName) throws Exception {
minioClient.removeObject(
RemoveObjectArgs.builder().bucket(bucket).object(objectName).build()
);
}
/**
* 获取文件url
*/
public String getPreviewFileUrl(String bucketName, String objectName) throws Exception{
GetPresignedObjectUrlArgs args = GetPresignedObjectUrlArgs.builder()
.method(Method.GET)
.bucket(bucketName).object(objectName).build();
return minioClient.getPresignedObjectUrl(args);
}
}
config的配置：
package com.wyh.oss.config;
import com.wyh.oss.adapter.AliStorageAdapter;
import com.wyh.oss.adapter.MinioStorageAdapter;
import com.wyh.oss.adapter.StorageAdapter;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cloud.context.config.annotation.RefreshScope;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
/**
* 文件存储config
*
* @create: 2023/12/1 10:35
*/
@Configuration
@RefreshScope
public class StorageConfig {
@Value("${storage.service.type}")
private String storageType;
/*@Resource
private StorageAdapter aliStorageAdapterImpl;
@Resource
private StorageAdapter minioStorageServiceImpl;*/
@Bean
@RefreshScope
public StorageAdapter storageService() {
if ("minio".equals(storageType)){
return new MinioStorageAdapter();
} else if("aliyun".equals(storageType)) {
return new AliStorageAdapter();
} else {
throw new IllegalArgumentException("未配置存储服务类型");
}
}
}