疯狂的狮子li
2022-01-13 a0bed51d966ab5d161d3fdd5423ba84f59fb60ff
ruoyi-oss/src/main/java/com/ruoyi/oss/service/impl/QiniuOssStrategy.java
@@ -22,106 +22,106 @@
 */
public class QiniuOssStrategy extends AbstractOssStrategy {
   private UploadManager uploadManager;
   private BucketManager bucketManager;
   private Auth auth;
    private UploadManager uploadManager;
    private BucketManager bucketManager;
    private Auth auth;
   @Override
   public void init(OssProperties cloudStorageProperties) {
      properties = cloudStorageProperties;
      try {
         Configuration config = new Configuration(getRegion(properties.getRegion()));
         // https设置
         config.useHttpsDomains = false;
         config.useHttpsDomains = "Y".equals(properties.getIsHttps());
         uploadManager = new UploadManager(config);
         auth = Auth.create(properties.getAccessKey(), properties.getSecretKey());
         String bucketName = properties.getBucketName();
         bucketManager = new BucketManager(auth, config);
    @Override
    public void init(OssProperties cloudStorageProperties) {
        properties = cloudStorageProperties;
        try {
            Configuration config = new Configuration(getRegion(properties.getRegion()));
            // https设置
            config.useHttpsDomains = false;
            config.useHttpsDomains = "Y".equals(properties.getIsHttps());
            uploadManager = new UploadManager(config);
            auth = Auth.create(properties.getAccessKey(), properties.getSecretKey());
            String bucketName = properties.getBucketName();
            bucketManager = new BucketManager(auth, config);
         if (!ArrayUtil.contains(bucketManager.buckets(), bucketName)) {
            bucketManager.createBucket(bucketName, properties.getRegion());
         }
      } catch (Exception e) {
         throw new OssException("七牛云存储配置错误! 请检查系统配置:[" + e.getMessage() + "]");
      }
   }
            if (!ArrayUtil.contains(bucketManager.buckets(), bucketName)) {
                bucketManager.createBucket(bucketName, properties.getRegion());
            }
        } catch (Exception e) {
            throw new OssException("七牛云存储配置错误! 请检查系统配置:[" + e.getMessage() + "]");
        }
    }
   @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失败, 请核对七牛云配置信息:[" + e.getMessage() + "]");
      }
   }
    @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失败, 请核对七牛云配置信息:[" + e.getMessage() + "]");
        }
    }
   @Override
   public String getServiceType() {
      return OssEnumd.QINIU.getValue();
   }
    @Override
    public String getServiceType() {
        return OssEnumd.QINIU.getValue();
    }
   @Override
   public UploadResult upload(byte[] data, String path, String contentType) {
      try {
    @Override
    public UploadResult upload(byte[] data, String path, String contentType) {
        try {
            String token = auth.uploadToken(properties.getBucketName());
            Response res = uploadManager.put(data, path, token, null, contentType, false);
         if (!res.isOK()) {
            throw new RuntimeException("上传七牛出错:" + res.error);
         }
      } catch (Exception e) {
         throw new OssException("上传文件失败,请核对七牛配置信息:[" + e.getMessage() + "]");
      }
      return new UploadResult().setUrl(getEndpointLink() + "/" + path).setFilename(path);
   }
            if (!res.isOK()) {
                throw new RuntimeException("上传七牛出错:" + res.error);
            }
        } catch (Exception e) {
            throw new OssException("上传文件失败,请核对七牛配置信息:[" + e.getMessage() + "]");
        }
        return new UploadResult().setUrl(getEndpointLink() + "/" + path).setFilename(path);
    }
   @Override
   public void delete(String path) {
      try {
         path = path.replace(getEndpointLink() + "/", "");
         Response res = bucketManager.delete(properties.getBucketName(), path);
         if (!res.isOK()) {
            throw new RuntimeException("删除七牛文件出错:" + res.error);
         }
      } catch (Exception e) {
         throw new OssException(e.getMessage());
      }
   }
    @Override
    public void delete(String path) {
        try {
            path = path.replace(getEndpointLink() + "/", "");
            Response res = bucketManager.delete(properties.getBucketName(), path);
            if (!res.isOK()) {
                throw new RuntimeException("删除七牛文件出错:" + res.error);
            }
        } catch (Exception e) {
            throw new OssException(e.getMessage());
        }
    }
   @Override
   public UploadResult uploadSuffix(byte[] data, String suffix, String contentType) {
      return upload(data, getPath(properties.getPrefix(), suffix), contentType);
   }
    @Override
    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, String contentType) {
      return upload(inputStream, getPath(properties.getPrefix(), suffix), contentType);
   }
    @Override
    public UploadResult uploadSuffix(InputStream inputStream, String suffix, String contentType) {
        return upload(inputStream, getPath(properties.getPrefix(), suffix), contentType);
    }
   @Override
   public String getEndpointLink() {
      return properties.getEndpoint();
   }
    @Override
    public String getEndpointLink() {
        return properties.getEndpoint();
    }
   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();
      }
   }
    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();
        }
    }
}