疯狂的狮子li
2021-08-02 f8ab5663ef7220f8adc0fd8fcbd7d0cf84c6d77d
ruoyi-oss/src/main/java/com/ruoyi/oss/service/impl/QiniuCloudStorageServiceImpl.java
@@ -1,5 +1,6 @@
package com.ruoyi.oss.service.impl;
import cn.hutool.core.util.ArrayUtil;
import com.qiniu.http.Response;
import com.qiniu.storage.BucketManager;
import com.qiniu.storage.Configuration;
@@ -38,21 +39,40 @@
   public QiniuCloudStorageServiceImpl(CloudStorageProperties properties) {
      this.properties = properties.getQiniu();
      try {
         // z0 z1 z2
         Configuration config = new Configuration(Region.autoRegion());
         // 默认不使用https
         Configuration config = new Configuration(getRegion(this.properties.getRegion()));
         // https设置
         config.useHttpsDomains = false;
         if (this.properties.getIsHttps() != null) {
            config.useHttpsDomains = this.properties.getIsHttps();
         }
         uploadManager = new UploadManager(config);
         Auth auth = Auth.create(
            this.properties.getAccessKey(),
            this.properties.getSecretKey());
         token = auth.uploadToken(this.properties.getBucketName());
         String bucketName = this.properties.getBucketName();
         token = auth.uploadToken(bucketName);
         bucketManager = new BucketManager(auth, config);
         if (!ArrayUtil.contains(bucketManager.buckets(), bucketName)) {
            bucketManager.createBucket(bucketName, this.properties.getRegion());
         }
      } catch (Exception e) {
         throw new IllegalArgumentException("七牛云存储配置错误! 请检查系统配置!");
      }
   }
   @Override
   public void createBucket() {
      try {
         String bucketName = properties.getBucketName();
         if (ArrayUtil.contains(bucketManager.buckets(), bucketName)) {
            return;
         }
         bucketManager.createBucket(bucketName, properties.getRegion());
      } catch (Exception e) {
         throw new OssException("创建Bucket失败, 请核对七牛云配置信息");
      }
   }
   @Override
   public String getServiceType() {
@@ -60,23 +80,23 @@
   }
   @Override
   public UploadResult upload(byte[] data, String path) {
   public UploadResult upload(byte[] data, String path, String contentType) {
      try {
         Response res = uploadManager.put(data, path, token);
         Response res = uploadManager.put(data, path, token, null, contentType, false);
         if (!res.isOK()) {
            throw new RuntimeException("上传七牛出错:" + res.toString());
         }
      } catch (Exception e) {
         throw new OssException("上传文件失败,请核对七牛配置信息");
      }
      return new UploadResult().setUrl(properties.getDomain() + "/" + path).setFilename(path);
      return new UploadResult().setUrl(getEndpointLink() + "/" + path).setFilename(path);
   }
   @Override
   public void delete(String path) {
      try {
         path = path.replace(this.properties.getDomain() + "/", "");
         Response res = bucketManager.delete(this.properties.getBucketName(), path);
         path = path.replace(getEndpointLink() + "/", "");
         Response res = bucketManager.delete(properties.getBucketName(), path);
         if (!res.isOK()) {
            throw new RuntimeException("删除七牛文件出错:" + res.toString());
         }
@@ -86,13 +106,13 @@
   }
   @Override
   public UploadResult uploadSuffix(byte[] data, String suffix) {
      return upload(data, getPath(this.properties.getPrefix(), suffix));
   public UploadResult uploadSuffix(byte[] data, String suffix, String contentType) {
      return upload(data, getPath(properties.getPrefix(), suffix), contentType);
   }
   @Override
   public UploadResult uploadSuffix(InputStream inputStream, String suffix) {
      return upload(inputStream, getPath(this.properties.getPrefix(), suffix));
   public UploadResult uploadSuffix(InputStream inputStream, String suffix, String contentType) {
      return upload(inputStream, getPath(properties.getPrefix(), suffix), contentType);
   }
   @Override
@@ -100,4 +120,26 @@
      OssFactory.register(getServiceType(),this);
   }
   @Override
   public String getEndpointLink() {
      return properties.getDomain();
   }
   private Region getRegion(String region) {
      switch (region) {
         case "z0":
            return Region.region0();
         case "z1":
            return Region.region1();
         case "z2":
            return Region.region2();
         case "na0":
            return Region.regionNa0();
         case "as0":
            return Region.regionAs0();
         default:
            return Region.autoRegion();
      }
   }
}