腾讯COS存储上传工具类

布满荆棘的人生 2023-10-10 14:15 46阅读 0赞

@Component
    public class TencentCOSUtil {
        
    
        @Value("${cos.accessKey}")
        private String accessKey;
    
        @Value("${cos.secretKey}")
        private String secretKey;
    
        @Value("${cos.regionName}")
        private String regionName;
    
        @Value("${cos.regionUrl}")
        private String regionUrl;
    
        @Value("${cos.bucketName}")
        private String bucketName;
    
        @Value("${cos.keyName}")
        private String keyName;
    
        /**
         * 腾讯COS文件上传统一处理
         * @param request
         * @param response
         * @return
         * @throws IOException
         */
        public String upload(HttpServletRequest request, HttpServletResponse response) throws IOException {
        
    
            MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
            MultipartFile file = multipartRequest.getFile("file");// 获取上传文件对象
            InputStream inputStream = file.getInputStream();
            String fileName = file.getOriginalFilename();
    
            String key = keyName + fileName;
    
            // 使用高级接口必须先保证本进程存在一个 TransferManager 实例，如果没有则创建
            TransferManager transferManager = createTransferManager();
    
            ObjectMetadata objectMetadata = new ObjectMetadata();
            // 上传的流如果能够获取准确的流长度，则推荐一定填写 content-length
            // 如果确实没办法获取到，则下面这行可以省略，但同时高级接口也没办法使用分块上传了
            objectMetadata.setContentLength(inputStream.available());
            PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, key, inputStream, objectMetadata);
    
            try {
        
                // 高级接口会返回一个异步结果Upload
                // 可同步地调用 waitForUploadResult 方法等待上传完成，成功返回UploadResult, 失败抛出异常
                Upload upload = transferManager.upload(putObjectRequest);
                UploadResult uploadResult = upload.waitForUploadResult();
                String savePath = regionUrl +"/"+ uploadResult.getKey();
    
                
            } catch (CosServiceException e) {
        
                
                e.printStackTrace();
            } catch (CosClientException e) {
        
                
                e.printStackTrace();
            } catch (InterruptedException e) {
        
                
                e.printStackTrace();
            }
            // 确定本进程不再使用 transferManager 实例之后，关闭之
            shutdownTransferManager(transferManager);
            return savePath;
        }
    
        /**
         * 创建 TransferManager 实例，这个实例用来后续调用高级接口
         * @return
         */
        public TransferManager createTransferManager() {
        
            // 创建一个 COSClient 实例，这是访问 COS 服务的基础实例。
            // 1 初始化用户身份信息（secretId, secretKey）。
            COSCredentials cred = new BasicCOSCredentials(accessKey, secretKey);
            // 2 设置 bucket 的地域, COS 地域的简称请参照 https://cloud.tencent.com/document/product/436/6224
            Region region = new Region(regionName);
            ClientConfig clientConfig = new ClientConfig(region);
            // 这里建议设置使用 https 协议
            // 从 5.6.54 版本开始，默认使用了 https
            clientConfig.setHttpProtocol(HttpProtocol.https);
            // 3 生成 cos 客户端。
            COSClient cosClient = new COSClient(cred, clientConfig);
    
            // 自定义线程池大小，建议在客户端与 COS 网络充足（例如使用腾讯云的 CVM，同地域上传 COS）的情况下，设置成16或32即可，可较充分的利用网络资源
            // 对于使用公网传输且网络带宽质量不高的情况，建议减小该值，避免因网速过慢，造成请求超时。
            ExecutorService threadPool = Executors.newFixedThreadPool(16);
    
            // 传入一个 threadpool, 若不传入线程池，默认 TransferManager 中会生成一个单线程的线程池。
            TransferManager transferManager = new TransferManager(cosClient, threadPool);
    
            // 设置高级接口的配置项
            // 分块上传阈值和分块大小分别为 5MB 和 1MB
            TransferManagerConfiguration transferManagerConfiguration = new TransferManagerConfiguration();
            transferManagerConfiguration.setMultipartUploadThreshold(5*1024*1024);
            transferManagerConfiguration.setMinimumUploadPartSize(1*1024*1024);
            transferManager.setConfiguration(transferManagerConfiguration);
    
            return transferManager;
        }
    
        /**
         * 在确定不再通过 TransferManager 的实例调用高级接口之后，一定要关闭这个实例，防止资源泄露。
         * @param transferManager
         */
        public void shutdownTransferManager(TransferManager transferManager) {
        
            // 指定参数为 true, 则同时会关闭 transferManager 内部的 COSClient 实例。
            // 指定参数为 false, 则不会关闭 transferManager 内部的 COSClient 实例。
            transferManager.shutdownNow(true);
        }
    
    }