devbox
盐析白兔
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

SpringBoot集成腾讯COS流程

作者：盐析白兔 | 2024-06-01 22:09:43

赞

踩

SpringBoot集成腾讯COS流程

1.pom.xml中添加cos配置

  1. <!--腾讯cos -->
  2. <dependency>
  3. 	<groupId>com.qcloud</groupId>
  4. 	<artifactId>cos_api</artifactId>
  5. 	<version>5.6.28</version>
  6. </dependency>

2.application.yaml中添加cos配置

  1. # 腾讯云存储cos相关公共配置
  2. tencent:
  3.   cos:
  4.     secretId: ABIDFG5gz36gMp2xbyHvYw3u
  5.     secretKey: 93ima8OcaHhDUUDjEdmfYEd
  6.     bucketName: haha-18888888
  7.     folder: video
  8.     region: ap-shanghai

3.创建属性映射类CosProperties

  1. /**
  2.  * Cos配置
  3.  */
  4. @Data
  5. @Component
  6. @RefreshScope
  7. @ConfigurationProperties(prefix = "tencent.cos")
  8. public class CosProperties {
  9.  
  10.     private String secretId;
  11.     private String secretKey;
  12.     private String bucketName;
  13.     private String folder;
  14.     private String region;
  15.  
  16. }

4.封装Cos工具类

  1. /**
  2.  * cos 工具类
  3.  */
  4. @Slf4j
  5. public class CosUtils {
  6.     private static CosUtils cosUtils = new CosUtils();
  7.  
  8.     private COSClient cosClient;
  9.  
  10.  
  11.     public static CosUtils getInstance() {
  12.         return cosUtils;
  13.     }
  14.  
  15.     private CosProperties cosProperties;
  16.  
  17.     public CosUtils setCosProperties(CosProperties cosProperties) {
  18.         this.cosProperties = cosProperties;
  19.         this.cosClient = createCOSClient(cosProperties);
  20.         return this;
  21.     }
  22.  
  23.     public String getUploadTemporaryToken(String key) {
  24.         if (StrUtil.hasBlank(cosProperties.getSecretId(), cosProperties.getSecretKey())) {
  25.             return null;
  26.         }
  27.         COSCredentials cred = new BasicCOSCredentials(cosProperties.getSecretId(), cosProperties.getSecretKey());
  28.         COSSigner signer = new COSSigner();
  29.         // 设置过期时间为1个小时
  30.         LocalDateTime now = LocalDateTime.now();
  31.         Date expiredTime = new Date(now.toInstant(ZoneOffset.of("+8")).toEpochMilli() + 3600L * 1000L);
  32.         // 要签名的 key, 生成的签名只能用于对应此 key 的上传
  33.         log.info("待签名key[{}], now[{}]", key, now);
  34.         String signStr = signer.buildAuthorizationStr(HttpMethodName.PUT, key, cred, expiredTime);
  35.         log.info("签名成功, key[{}], now[{}], signStr[{}]", key, now, signStr);
  36.         return signStr;
  37.     }
  38.  
  39.  
  40.     /**
  41.      * 上传文件
  42.      * 1.创建本地文件 2.上传
  43.      *
  44.      * @param fileName    文件名(带后缀)
  45.      * @param fileContent 文件内容
  46.      * @return
  47.      */
  48.     public String upload(String fileName, String fileContent, String customizeFolder) {
  49.         try {
  50.             String cosSecretId = cosProperties.getSecretId();
  51.             String cosSecretKey = cosProperties.getSecretKey();
  52.             String folder = StringUtils.isEmpty(customizeFolder) ? cosProperties.getFolder() : customizeFolder;
  53.             String bucketName = cosProperties.getBucketName();
  54.  
  55.             if (StringUtils.isEmpty(cosSecretId) ||
  56.                     StringUtils.isEmpty(cosSecretKey) ||
  57.                     StringUtils.isEmpty(bucketName) ||
  58.                     StringUtils.isEmpty(folder)) {
  59.                 log.error("cos upload params Incomplete");
  60.                 return "";
  61.             }
  62.  
  63.             String root = Objects.requireNonNull(CosUtils.class.getResource("/")).getPath();
  64.             String s = root + "/temp/upload/";
  65.             File localFile = getLocalFile(fileContent, s, fileName);
  66.             if (localFile == null) {
  67.                 return "";
  68.             }
  69.             DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
  70.             String fromatDate = LocalDateTime.now().format(formatter);
  71.             String fileUrl = folder + "/" + fromatDate + "/" + fileName;
  72.  
  73.             PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, fileUrl, localFile);
  74.             // 设置存储类型, 默认是标准(Standard), 低频(standard_ia)
  75.             putObjectRequest.setStorageClass(StorageClass.Standard);
  76.             try {
  77.                 this.cosClient.putObject(putObjectRequest);
  78.             } catch (CosClientException e) {
  79.                 log.error("An exception occurs during execution cos upload,error message:{}", e.getMessage());
  80.             }
  81.             //删除本地缓存文件
  82.             if (localFile.exists()) {
  83.                 localFile.delete();
  84.             }
  85.             return fileUrl.startsWith("/") ? fileUrl : "/" + fileUrl;
  86.         }catch (Exception e){
  87.             log.error("文件上传失败,{}",e);
  88.         }
  89.         return StrUtil.EMPTY;
  90.     }
  91.  
  92.     /**
  93.      * 删除文件
  94.      *
  95.      * @param bucketName
  96.      * @param key
  97.      * @return
  98.      */
  99.     public Boolean deleteCosFile(String bucketName, String key) {
  100.         String cosSecretId = cosProperties.getSecretId();
  101.         String cosSecretKey = cosProperties.getSecretKey();
  102.         Boolean executeFlag = true;
  103.  
  104.         if (StringUtils.isEmpty(cosSecretId) ||
  105.                 StringUtils.isEmpty(cosSecretKey) ||
  106.                 StringUtils.isEmpty(bucketName) ||
  107.                 StringUtils.isEmpty(key)) {
  108.  
  109.             log.error("cos delete file  params Incomplete");
  110.  
  111.             return false;
  112.         }
  113.  
  114.         COSCredentials cred = new BasicCOSCredentials(cosSecretId, cosSecretKey);
  115.         // 2 设置bucket的区域, COS地域的简称请参照 https://www.qcloud.com/document/product/436/6224
  116.         ClientConfig clientConfig = new ClientConfig(new Region("ap-nanjing"));
  117.         // 3 生成cos客户端
  118.         COSClient cosclient = new COSClient(cred, clientConfig);
  119.         try {
  120.             cosclient.deleteObject(bucketName, key);
  121.  
  122.         } catch (CosClientException e) {
  123.             log.error("An exception occurs during execution cos delete,error message:{}", e.getMessage());
  124.             executeFlag = false;
  125.         }
  126.  
  127.         // 关闭客户端
  128.         cosclient.shutdown();
  129.  
  130.         return executeFlag;
  131.     }
  132.  
  133.     private void getDir(String path) {
  134.         File localFile = new File(path);
  135.         if (!localFile.exists()) {
  136.             localFile.mkdirs();
  137.         }
  138.     }
  139.  
  140.  
  141.     private File getLocalFile(String instructionSet, String dir, String fileName) {
  142.         File localFile = null;
  143.         try {
  144.  
  145.             getDir(dir);
  146.  
  147.             localFile = new File(dir, fileName);
  148.             if (!localFile.exists()) {
  149.                 localFile.createNewFile();
  150.             }
  151.             FileOutputStream fos = new FileOutputStream(localFile, true);
  152.             OutputStreamWriter osw = new OutputStreamWriter(fos);
  153.             BufferedWriter bw = new BufferedWriter(osw);
  154.             bw.write(instructionSet);
  155.             bw.newLine();
  156.             bw.flush();
  157.             bw.close();
  158.             osw.close();
  159.             fos.close();
  160.  
  161.             return localFile;
  162.         } catch (IOException e2) {
  163.             log.error("An exception occurs during execution create local file,error message:{} ", e2.getMessage());
  164.             return null;
  165.         }
  166.     }
  167.  
  168.     /**
  169.      * 获取二进制文件
  170.      */
  171.     public static byte[] downLoadBinary(String urlStr) throws IOException {
  172.         HttpURLConnection conn = null;
  173.         InputStream inputStream = null;
  174.         ByteArrayOutputStream bos = null;
  175.         try {
  176.             URL url = new URL(urlStr);
  177.             conn = (HttpURLConnection) url.openConnection();
  178.             //设置超时间为10秒
  179.             conn.setConnectTimeout(10 * 1000);
  180.             conn.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.135 Safari/537.36");
  181.             //得到输入流
  182.             inputStream = conn.getInputStream();
  183.             bos = new ByteArrayOutputStream();
  184.             //获取数据数组
  185.             return readInputStream(inputStream, bos);
  186.         } finally {
  187.             if (bos != null) {
  188.                 bos.close();
  189.             }
  190.             if (inputStream != null) {
  191.                 inputStream.close();
  192.             }
  193.             if (conn != null) {
  194.                 conn.disconnect();
  195.             }
  196.         }
  197.     }
  198.  
  199.  
  200.     /**
  201.      * 获取字符串列表
  202.      */
  203.     public static List<String> downLoadList(String urlStr) throws IOException {
  204.         HttpURLConnection conn = null;
  205.         InputStream inputStream = null;
  206.         try {
  207.             URL url = new URL(urlStr);
  208.             conn = (HttpURLConnection) url.openConnection();
  209.             //设置超时间为10秒
  210.             conn.setConnectTimeout(10 * 1000);
  211.             conn.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.135 Safari/537.36");
  212.             //得到输入流
  213.             inputStream = conn.getInputStream();
  214.             //获取数据数组 windows操作系统默认编码：GB18030
  215.             return IoUtil.readLines(new InputStreamReader(inputStream, Charset.forName("GB18030")), new ArrayList<>());
  216.         } catch (IOException e){
  217.             log.error("An exception occurs during execution download file,error message:{} ", e.getMessage());
  218.             return Collections.EMPTY_LIST;
  219.         }finally {
  220.             if (inputStream != null) {
  221.                 inputStream.close();
  222.             }
  223.             if (conn != null) {
  224.                 conn.disconnect();
  225.             }
  226.         }
  227.     }
  228.     /**
  229.      * 输入流转二进制
  230.      */
  231.     public static byte[] readInputStream(InputStream inputStream, ByteArrayOutputStream bos) throws IOException {
  232.         byte[] buffer = new byte[1024];
  233.         int len = 0;
  234.         while ((len = inputStream.read(buffer)) != -1) {
  235.             bos.write(buffer, 0, len);
  236.         }
  237.         return bos.toByteArray();
  238.     }
  239.  
  240.     /**
  241.      * 创建cosClient
  242.      *
  243.      * @param cosProperties
  244.      * @return
  245.      */
  246.     public static COSClient createCOSClient(CosProperties cosProperties) {
  247.         // 1 初始化用户身份信息（secretId, secretKey）。
  248.         // SECRETID和SECRETKEY请登录访问管理控制台 https://console.cloud.tencent.com/cam/capi
  249.         // 进行查看和管理
  250.         String secretId = cosProperties.getSecretId();
  251.         String secretKey = cosProperties.getSecretKey();
  252.         COSCredentials cred = new BasicCOSCredentials(secretId, secretKey);
  253.         // 2 设置 bucket 的地域, COS 地域的简称请参照
  254.         // https://cloud.tencent.com/document/product/436/6224
  255.         // clientConfig 中包含了设置 region, https(默认 http), 超时, 代理等 set 方法, 使用可参见源码或者常见问题
  256.         // Java SDK 部分。
  257.         Region region = new Region(cosProperties.getRegion());
  258.         ClientConfig clientConfig = new ClientConfig(region);
  259.         // 这里建议设置使用 https 协议
  260.         // 从 5.6.54 版本开始，默认使用了 https
  261.         clientConfig.setHttpProtocol(HttpProtocol.https);
  262.  
  263.         // 3 生成 cos 客户端。
  264.         return new COSClient(cred, clientConfig);
  265.     }
  266.  
  267.     /**
  268.      * 获取视频属性: (宽度、高度、时长)
  269.      * 获取方式:从oss获取
  270.      *
  271.      * @param ossUrl
  272.      * @return
  273.      */
  274.     public VideoProperties getVideoPropertiesFromCos(String ossUrl) {
  275.         try {
  276.             // 此处的key为对象键，对象键是对象在存储桶内的唯一标识
  277.             String key = ossUrl;
  278.             GeneratePresignedUrlRequest req = new GeneratePresignedUrlRequest("greatpan-1300159541", key, HttpMethodName.GET);
  279.             // 设置签名过期时间(可选), 若未进行设置, 则默认使用 ClientConfig 中的签名过期时间(1小时)
  280.             // 可以设置任意一个未来的时间，推荐是设置 10 分钟到 3 天的过期时间
  281.             // 这里设置签名在半个小时后过期
  282.             Date expirationDate = new Date(System.currentTimeMillis() + 30L * 60L * 1000L);
  283.             req.setExpiration(expirationDate);
  284.             req.addRequestParameter("ci-process", "videoinfo");
  285.             URL url = this.cosClient.generatePresignedUrl(req);
  286.             String mediaInfoXml = HttpClientUtil.doGet(url.toString());
  287.             if (mediaInfoXml != null) {
  288.                 Document document = XmlUtil.readXML(mediaInfoXml);
  289.                 Node error = document.getElementsByTagName("Error").item(0);
  290.                 if (!ObjectUtils.isEmpty(error)) {
  291.                     Node message = document.getElementsByTagName("Message").item(0);
  292.                     log.error("获取视频基础信息出错.ossurl:{}, e:{}", ossUrl, Optional.ofNullable(message).map(Node::getTextContent).orElse(""));
  293.                     return null;
  294.                 }
  295.                 String width = document.getElementsByTagName("Width").item(0).getTextContent();
  296.                 String height = document.getElementsByTagName("Height").item(0).getTextContent();
  297.                 String rotation = document.getElementsByTagName("Rotation").item(0).getTextContent();
  298.                 if (StringUtils.isEmpty(width) || StringUtils.isEmpty(height) || StringUtils.isEmpty(rotation)) {
  299.                     return null;
  300.                 }
  301.                 VideoProperties videoProperties = new VideoProperties();
  302.                 int w = Integer.parseInt(width);
  303.                 int h = Integer.parseInt(height);
  304.                 int r = (int) Double.parseDouble(rotation);
  305.                 // 如果r是90或者270, 说明视频有旋转操作, 并宽高比有变化，需要把w 和 h 调换
  306.                 if (r % 90 == 0 && (r / 90 % 2) == 1) {
  307.                     videoProperties.setHeight(w);
  308.                     videoProperties.setWidth(h);
  309.                     return videoProperties;
  310.                 }
  311.                 videoProperties.setHeight(h);
  312.                 videoProperties.setWidth(w);
  313.                 return videoProperties;
  314.             }
  315.             return null;
  316.         } catch (Exception e) {
  317.             log.error("获取视频基础信息出错.ossurl:{}, e:", ossUrl, e);
  318.             return null;
  319.         }
  320.     }
  321.  
  322.     public static void main(String[] args) throws Exception {
  323. //        VideoProperties videoPropertiesFromCos = getVideoPropertiesFromCos("/video/2024-05-30/723d5de3-f874-4744-819f-0a31e6e8e507.mp4");
  324. //        System.out.println(videoPropertiesFromCos);
  325.         byte[] bytes = CosUtils.downLoadBinary("http://fs.haha.com/video/2024-05-30/1484098659191754752.txt");
  326.         System.out.println(new String(bytes));
  327.     }
  328. }

5.前端获取cos上传token

  1. @Resource
  2. private CosUtils cosUtils;
  3.  
  4. @GetMapping("/token")
  5. @ApiOperation("获取文件存储token")
  6. public ApiResponse<?> getFileUploadToken(@RequestParam String fileFullPath) {
  7. 	if(StringUtils.isEmpty(fileFullPath)){
  8. 		return ApiResponse.error("参数错误");
  9. 	}
  10. 	return ApiResponse.ok(cosUtils.getUploadTemporaryToken(fileFullPath));
  11. }

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/659607
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号