update 使用 策略+工厂 重写OSS模块

疯狂的狮子li

2021-07-18 089e288a6e55d2ae527ad733bbb8b8b5eaad6107

 ruoyi-oss/src/main/java/com/ruoyi/oss/service/impl/AliyunCloudStorageServiceImpl.java

@@ -1,28 +1,47 @@
package com.ruoyi.oss.service.impl;

import com.aliyun.oss.ClientConfiguration;
import com.aliyun.oss.OSSClient;
import com.ruoyi.oss.config.CloudStorageConfig;
import com.aliyun.oss.common.auth.DefaultCredentialProvider;
import com.ruoyi.oss.enumd.CloudServiceEnumd;
import com.ruoyi.oss.exception.OssException;
import com.ruoyi.oss.factory.OssFactory;
import com.ruoyi.oss.properties.AliyunProperties;
import com.ruoyi.oss.properties.CloudStorageProperties;
import com.ruoyi.oss.service.abstractd.AbstractCloudStorageService;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;

import java.io.ByteArrayInputStream;
import java.io.InputStream;

/**
 * 阿里云存储
 *
 * @author Lion Li
 */
public class AliyunCloudStorageServiceImpl extends AbstractCloudStorageService {
@Lazy
@Service
public class AliyunCloudStorageServiceImpl extends AbstractCloudStorageService implements InitializingBean {

   private OSSClient client;
   private final OSSClient client;
   private final AliyunProperties properties;

   public AliyunCloudStorageServiceImpl(CloudStorageConfig config) {
      this.config = config;
      // 初始化
      init();
   @Autowired
   public AliyunCloudStorageServiceImpl(CloudStorageProperties properties) {
      this.properties = properties.getAliyun();
      ClientConfiguration configuration = new ClientConfiguration();
      DefaultCredentialProvider credentialProvider = new DefaultCredentialProvider(
         this.properties.getAccessKeyId(),
         this.properties.getAccessKeySecret());
      client = new OSSClient(this.properties.getEndpoint(), credentialProvider, configuration);
   }

   private void init() {
      client = new OSSClient(config.getDomain(), config.getAccessKey(), config.getSecretKey());
   @Override
   public String getServiceType() {
      return CloudServiceEnumd.ALIYUN.getValue();
   }

   @Override
@@ -33,18 +52,18 @@
   @Override
   public String upload(InputStream inputStream, String path) {
      try {
         client.putObject(config.getBucketName(), path, inputStream);
         client.putObject(this.properties.getBucketName(), path, inputStream);
      } catch (Exception e) {
         throw new OssException("上传文件失败，请检查配置信息");
      }
      return config.getDomain() + "/" + path;
      return this.properties.getEndpoint() + "/" + path;
   }

   @Override
   public void delete(String path) {
      path = path.replace(config.getDomain() + "/", "");
      path = path.replace(this.properties.getEndpoint() + "/", "");
      try {
         client.deleteObject(config.getBucketName(), path);
         client.deleteObject(this.properties.getBucketName(), path);
      } catch (Exception e) {
         throw new OssException("上传文件失败，请检查配置信息");
      }
@@ -52,11 +71,16 @@

   @Override
   public String uploadSuffix(byte[] data, String suffix) {
      return upload(data, getPath(config.getPrefix(), suffix));
      return upload(data, getPath(this.properties.getPrefix(), suffix));
   }

   @Override
   public String uploadSuffix(InputStream inputStream, String suffix) {
      return upload(inputStream, getPath(config.getPrefix(), suffix));
      return upload(inputStream, getPath(this.properties.getPrefix(), suffix));
   }

   @Override
   public void afterPropertiesSet() throws Exception {
      OssFactory.register(getServiceType(),this);
   }
}