springboot+cos文件存储

package com.example.cos.services.impl; import cn.hutool.core.util.StrUtil; import com.example.cos.config.CosConfig; import com.example.cos.services.ICosFileService; import com.example.cos.vo.RestApiResponse; import com.qcloud.cos.COSClient; import com.qcloud.cos.model.CannedAccessControlList; import com.qcloud.cos.model.ObjectMetadata; import com.qcloud.cos.model.PutObjectRequest; import lombok.extern.slf4j.Slf4j; import org.joda.time.DateTime; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import org.springframework.web.multipart.MultipartFile; import javax.annotation.Resource; import java.io.IOException; import java.io.InputStream; /** * @author phf */ @Slf4j @Service public class ICosFileServiceImpl implements ICosFileService { @Resource private COSClient cosClient; @Resource private CosConfig cosConfig; @Override @Transactional(rollbackFor = Exception.class) public RestApiResponse<String> upload(MultipartFile[] files) { RestApiResponse<String> response = RestApiResponse.success(); String res = ""; try { for (MultipartFile file : files) { String originalFileName = file.getOriginalFilename(); // 获得文件流 InputStream inputStream = null; inputStream = file.getInputStream(); // 设置文件路径 String filePath = getFilePath(originalFileName, "你的桶内文件路径abc/def/test/"); // 上传文件 String bucketName = cosConfig.getBucketName(); ObjectMetadata objectMetadata = new ObjectMetadata(); objectMetadata.setContentLength(file.getSize()); objectMetadata.setContentType(file.getContentType()); PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, filePath, inputStream, objectMetadata); cosClient.putObject(putObjectRequest); cosClient.setBucketAcl(bucketName, CannedAccessControlList.PublicRead); String url = cosConfig.getPath() + "/" + filePath; res += url + ","; } String paths = res.substring(0, res.length() - 1); response.setData(paths); return response; } catch (IOException e) { e.printStackTrace(); } finally { cosClient.shutdown(); } return RestApiResponse.fail(); } @Override public RestApiResponse<Void> delete(String fileName) { cosConfig.cosClient(); // 文件桶内路径 String filePath = getDelFilePath(fileName, "你的桶内文件路径abc/def/test/"); cosClient.deleteObject(cosConfig.getBucketName(), filePath); return RestApiResponse.success(); } /** * 生成文件路径 * @param originalFileName 原始文件名称 * @param folder 存储路径 * @return */ private String getFilePath(String originalFileName, String folder) { // 获取后缀名 String fileType = originalFileName.substring(originalFileName.lastIndexOf(".")); // 以文件后缀来存储在存储桶中生成文件夹方便管理 String filePath = folder + "/"; // 去除文件后缀 替换所有特殊字符 String fileStr = StrUtil.removeSuffix(originalFileName, fileType).replaceAll("[^0-9a-zA-Z\\u4e00-\\u9fa5]", "_"); filePath += new DateTime().toString("yyyyMMddHHmmss") + "_" + fileStr + fileType; log.info("filePath：" + filePath); return filePath; } /** * 生成文件路径 * @param originalFileName 原始文件名称 * @param folder 存储路径 * @return */ private String getDelFilePath(String originalFileName, String folder) { // 获取后缀名 String fileType = originalFileName.substring(originalFileName.lastIndexOf(".")); // 以文件后缀来存储在存储桶中生成文件夹方便管理 String filePath = folder + "/"; // 去除文件后缀 替换所有特殊字符 String fileStr = StrUtil.removeSuffix(originalFileName, fileType).replaceAll("[^0-9a-zA-Z\\u4e00-\\u9fa5]", "_"); filePath += fileStr + fileType; log.info("filePath：" + filePath); return filePath; } }